Cloudflare d1のbatchをPrismaで使う

const { PrismaClient } = require('@prisma/client');
const prisma = new PrismaClient();

async function updateUserRanks() {
  const userUpdates = [
    { user_id: 17, rank: 1 },
    { user_id: 19, rank: 2 },
  ];

  const promises = userUpdates.map(async (update) => {
    return prisma.users.update({
      where: { user_id: update.user_id },
      data: { rank: update.rank },
    });
  });

  await prisma.$transaction(promises);
}

// 使用例
updateUserRanks()
  .then(() => {
    console.log('ユーザーのランクが更新されました');
  })
  .catch((error) => {
    console.error('エラーが発生しました:', error);
  })
  .finally(async () => {
    await prisma.$disconnect();
  });

const { PrismaClient } = require('@prisma/client');
const prisma = new PrismaClient();

async function batchUpdateUserRanks(db) {
  const updateQueries = [
    { rank: 1, user_id: 17 },
    { rank: 2, user_id: 19 },
  ];

  const batch = updateQueries.map((update) => {
    const sql = `
      UPDATE users
      SET rank = ${update.rank}
      WHERE user_id = ${update.user_id};
    `;
    return db.prepare(sql);
  });

  await db.batch(batch);
}

// 使用例
(async () => {
  const rawDb = env.DB; // 環境変数からDB接続を取得
  await batchUpdateUserRanks(rawDb);
})();

説明:

1. SQLクエリの生成:

   • updateQueries配列に、更新したいデータを格納します。
   • map関数を使って、それぞれの更新クエリを生成し、SQL文字列を作成します。

2. DBバッチ処理:

   • db.prepare(sql)メソッドを用いて、各SQLを準備します。
   • 最後に、db.batch(batch)を実行して、複数のSQLを一度に送信します。

SQL文字列を取得する方法

1. 環境変数の設定:
   PrismaのDEBUG環境変数を使って、クエリが実行される際のロギングを設定します。ターミナルで以下のコマンドを実行します。

   export DEBUG="prisma:query"

2. 実際のクエリを実行:
   Prismaクライアントを使用してクエリを実行します。以下は、例としてデータをクエリするコードです。

   const { PrismaClient } = require('@prisma/client');
   const prisma = new PrismaClient();
   
   async function main() {
       const users = await prisma.user.findMany();
       console.log(users);
   }
   
   main()
       .catch(e => console.error(e))
       .finally(async () => {
           await prisma.$disconnect();
   });

3. コンソール出力を確認:
   上記のスクリプトを実行すると、生成されたSQLクエリがコンソールに出力されます。これにより、Prismaが生成したSQL文が確認できます。

注意事項

• 上記の方法は開発環境でのデバッグ用です。本番環境でこのロギングを有効にすることは、パフォーマンスに影響を及ぼす可能性があるため、注意が必要です。

• Prismaは、SQL文を明示的に返すためのAPIを提供していないため、クエリごとにどのようなSQLが生成されるかを確認する際に、デバッグ環境でのロギングが最も簡単で一般的な手法となります。