import { PrismaClient } from '@prisma/client/edge'
import { withOptimize } from '@prisma/extension-optimize'
import { withAccelerate } from '@prisma/extension-accelerate'

const prisma = new PrismaClient({ log: ['query', 'info', 'warn', 'error'] })
  .$extends(
    withOptimize({
      apiKey: process.env.OPTIMIZE_API_KEY!,
      enable: true,
    })
  )
  .$extends(withAccelerate())
  .$extends({
    query: {
      $allModels: {
        async $allOperations({ operation, model, args, query }) {
          const start = performance.now()
          const result = await query(args)
          const end = performance.now()
          const duration = end - start
          console.log(`Query ${model}.${operation} took ${duration.toFixed(2)}ms`)
          return result
        },
      },
    },
  })

export default prisma

import { PrismaClient } from '@prisma/client/edge'
import { withOptimize } from '@prisma/extension-optimize'
import { withAccelerate } from '@prisma/extension-accelerate'

const prisma = new PrismaClient({ log: ['query', 'info', 'warn', 'error'] })
  .$extends(
    withOptimize({
      apiKey: process.env.OPTIMIZE_API_KEY!,
      enable: true,
    })
  )
  .$extends(withAccelerate())
  .$extends({
    query: {
      $allModels: {
        async $allOperations({ operation, model, args, query }) {
          const start = performance.now()
          const result = await query(args)
          const end = performance.now()
          const duration = end - start
          console.log(`Query ${model}.${operation} took ${duration.toFixed(2)}ms`)
          return result
        },
      },
    },
  })

export default prisma