Prisma+fakerでサクッとデモデータを作る

MYJLab Advent Calendar 2022、14日目の記事、担当は3年の@takumiakasaka1231です！

昨日の記事は、@mahirooooonのWord2Vecについてでした。

言葉の足し算引き算が楽しいらしく、授業後に目をキラキラさせながらお話していたのを思い出しました。（笑）

さて、本日は、モダンなORMであるprismaとfakerを使って、サクッとデモデータをDB内に入れていってみようと思います！

では、僕の初qiita記事、スタートです！

公式ducument

背景

backend開発の時、DBにデモデータをいれるのが面倒くさいときってありますよね？

そんなときに劇的に開発効率を上げてくれる方法があります。

(seedで、チーム全員でデモデータをそろえたいときは、この手法は向いていません)

環境構築(prisma,faker,mysql)

prismaのinstall(node 14.17.0以上が必要らしいです)(そこでめっちゃ怒られた)

npm init -y 
npm install typescript ts-node @types/node --save-dev

npx tsc --init

npm install prisma --save-dev

npx prisma init --datasource-provider sqlite

fakerのinstall

npm install @faker-js/faker --save-dev

localでmysqlを立ち上げてください

docker pull mysql

docker run --name your-container-name -e MYSQL_ROOT_PASSWORD=your-password -d -p 3306:3306 mysql:tag

これで環境が整いました。

ファイルの中身

これらのコマンドを実行した後、ファイルがいろいろできたと思います。
今回は、関連する部分だけ

prisma/schema.prisma

DBの中身を実際に書いていく部分になります。

早速、中身を変更していきましょう。

schemaの変更

schema.prismaをこんな感じに変更してみましょう

// This is your Prisma schema file,
// learn more about it in the docs: https://pris.ly/d/prisma-schema

generator client {
  provider = "prisma-client-js"
}

datasource db {
  provider = "mysql"
  url      = env("DATABASE_URL")
}

model User {
  id    Int     @id @default(autoincrement())
  email String  @unique
  name  String?
  posts Post[]
}

model Post {
  id        Int     @id @default(autoincrement())
  title     String
  content   String?
  published Boolean @default(false)
  author    User    @relation(fields: [authorId], references: [id])
  authorId  Int
}

.envにDATABASE_URLの追加をお忘れなく

そしたら、migrateをしていきます

npx prisma migrate dev --name init

次は、実際にデータを入れるseed.tsを用意していきましょう！

seed.ts

まずは、package.jsonに以下を追加します。

"prisma": {
    "seed": "ts-node prisma/seed.ts"
  }

これで、seed.tsを実行する準備が完了しました。
次に、実際のファイルを作成、編集していきましょう。

prismaフォルダー下にseed.tsを作成します

seed.tsを以下のように記述します。

import { PrismaClient } from "@prisma/client";
import { faker } from "@faker-js/faker";

const prisma = new PrismaClient();
async function main() {
  for (let index = 0; index < 100; index++) {
    await prisma.user.create({
      data: {
        name: faker.name.fullName(),
        email: faker.internet.email(),
        posts: {
          createMany: {
            data: [
              {
                title: faker.name.jobTitle(),
              },
              {
                title: faker.name.jobTitle(),
              },
              {
                title: faker.name.jobTitle(),
              },
              {
                title: faker.name.jobTitle(),
              },
              {
                title: faker.name.jobTitle(),
              },
            ],
          },
        },
      },
    });
  }
}
main()
  .then(async () => {
    await prisma.$disconnect();
  })
  .catch(async (e) => {
    console.error(e);
    await prisma.$disconnect();
    process.exit(1);
  });

その後、seed.tsを動かしていきます

npx prisma migrate reset

🌱  The seed command has been executed.

と表示されれば成功です！！

prisma studioを覗きに行きましょう

npx prisma studio

完璧です。

100人と、1人につき5つ投稿が紐づいたデモデータを作成することができました！！！

終わりに

どうでしたか？

初投稿の記事となりますが、皆さまの知識になったり、参考になれば幸いです。

これを機に、たくさん投稿したいです！

ではまたっ！！！

参考文献

prisma 公式ドキュメント

faker.js 公式ドキュメント

Seeding Database with Prisma & Faker -by Leigh Halliday