Function make

repos/effect/packages/sql/d1/src/D1Client.ts:99–231

(
  options: D1ClientConfig
)

 * context.
 *
 * @category services
 * @since 4.0.0
 */
export const D1Client = Context.Service<D1Client>("@effect/sql-d1/D1Client")

/**
 * Configuration for a Cloudflare D1 client, including the `D1Database`, prepared statement cache settings, span attributes, and query/result name transforms.
 *
 * @category models
 * @since 4.0.0
 */
export interface D1ClientConfig {
  readonly db: D1Database
  readonly prepareCacheSize?: number | undefined
  readonly prepareCacheTTL?: Duration.Input | undefined
  readonly spanAttributes?: Record<string, unknown> | undefined

  readonly transformResultNames?: ((str: string) => string) | undefined
  readonly transformQueryNames?: ((str: string) => string) | undefined
}

type TransformRows = <A extends object>(rows: ReadonlyArray<A>) => ReadonlyArray<A>

type BatchResults<Statements extends ReadonlyArray<Statement.Statement<any>>> = {
  readonly [K in keyof Statements]: Effect.Success<Statements[K]>
}

interface StatementWithTransformRows extends Statement.Statement<any> {
  readonly transformRows: TransformRows | undefined
}

const makeBatch = (options: {
  readonly db: D1Database
  readonly prepareCache: Cache.Cache<string, D1PreparedStatement, SqlError>
  readonly spanAttributes: ReadonlyArray<readonly [string, unknown]>
  readonly getClient: () => D1Client
}): D1Client["batch"] =>
<const Statements extends ReadonlyArray<Statement.Statement<any>>>(
  statements: Statements
) => {
  if (statements.length === 0) {
    return Effect.succeed([] as unknown as BatchResults<Statements>)
  }
  return Effect.useSpan(
    "sql.execute",
    { kind: "client" },
    (span) =>
      Effect.withFiber(Effect.fnUntraced(function*(fiber) {
        const transformer = fiber.getRef(Statement.CurrentTransformer)
        const prepared: Array<D1PreparedStatement> = []
        const transforms: Array<TransformRows | undefined> = []
        const queryTexts: Array<string> = []

        for (const original of statements) {
          const statement = transformer === undefined
            ? original
            : yield* transformer(original, options.getClient(), fiber, span)
          const [sql, params] = statement.compile()