🍷

Prismaでtruncateする

2022/01/29に公開1件

Prisma

tech

今までデータをリセットする際はMySQLサーバーのコンテナを作り直して再度データを入れていました。

これだと約40秒ほどかかってしまい結構時間を無駄にしていたので、コンテナを壊さずにデータを初期化するように対応しました。

これで約10秒ほどでDB内のデータをリセットすることができるようになりました。

truncate.ts

import { PrismaClient } from ".primsa/client"
 
export async function truncate(prisma: PrismaClient) {
  const allProperties = Object.keys(prisma);

  const modelNames = allProperties.filter(
    (x) => !(typeof x === "string" && (x.startsWith("$") || x.startsWith("_")))
  );

  // 外部キーを無効化
  await prisma.$queryRaw`SET FOREIGN_KEY_CHECKS=0`;

  for (const modelName of modelNames) {
    const query = `TRUNCATE TABLE ${modelName}`;
    await prisma.$queryRawUnsafe(query);
  };

  await prisma.$queryRaw`SET FOREIGN_KEY_CHECKS=1`;
}

補足

rawQueryのテンプレートリテラルには、列名とテーブル名に変数を入れることができません。

await prisma.$queryRawUnsafe`TRUNCATE TABLE ${modelName}`; // シンタックスエラーになる

なので、上述のように一旦変数に格納して引数に入れるようにすれば動作させることができます。

参考

Discussion

TakashiAihara 2023/07/08に更新

有用な記事助かりました。

ちょっとPrismaClientのindex.d.ts見てましたが、ModelNameというのがあるはずなので、そちら使ったほうが見やすいかもしれません。

import { PrismaClient } from ".primsa/client"
...
const allProperties = Object.keys(prisma);

const modelNames = allProperties.filter(
  (x) => !(typeof x === "string" && (x.startsWith("$") || x.startsWith("_")))
);

↓こう

import { PrismaClient, Prisma } from ".primsa/client"
...
const modelNames = Object.values(Prisma.ModelName).map((m) => m.toLowerCase())

環境は Prisma 4.15 です。

    "@prisma/client": "^4.15.0",
    "prisma": "^4.15.0",