152 lines
3.9 KiB
TypeScript
152 lines
3.9 KiB
TypeScript
/* eslint-disable @typescript-eslint/no-explicit-any */
|
|
|
|
/**
|
|
* Vercel Postgres (Neon/Postgres) 适配器
|
|
*
|
|
* 将 Vercel Postgres API 转换为与 D1 兼容的接口
|
|
*
|
|
* 注意：此模块仅在服务端使用，通过 webpack 配置排除客户端打包
|
|
*/
|
|
|
|
import { sql } from '@vercel/postgres';
|
|
import { DatabaseAdapter, D1PreparedStatement, D1Result } from './d1-adapter';
|
|
|
|
/**
|
|
* Vercel Postgres 适配器
|
|
*
|
|
* 使用 @vercel/postgres 包装为 D1 兼容接口
|
|
*/
|
|
export class PostgresAdapter implements DatabaseAdapter {
|
|
private queryParams: { query: string; values: any[] } | null = null;
|
|
|
|
prepare(query: string): D1PreparedStatement {
|
|
return new PostgresPreparedStatement(query);
|
|
}
|
|
|
|
batch(statements: D1PreparedStatement[]): Promise<D1Result[]> {
|
|
// Postgres 使用事务模拟 batch
|
|
return new Promise((resolve, reject) => {
|
|
Promise.all(statements.map((stmt) => (stmt as PostgresPreparedStatement).execute()))
|
|
.then((results) => resolve(results))
|
|
.catch((err) => reject(err));
|
|
});
|
|
}
|
|
|
|
exec(query: string): void {
|
|
// Vercel Postgres 不支持直接 exec，需要使用 sql 模板
|
|
throw new Error('exec() is not supported for Vercel Postgres. Use prepare() instead.');
|
|
}
|
|
}
|
|
|
|
/**
|
|
* Vercel Postgres PreparedStatement 包装器
|
|
* 将 Vercel Postgres API 转换为 D1 兼容 API
|
|
*/
|
|
class PostgresPreparedStatement implements D1PreparedStatement {
|
|
private params: any[] = [];
|
|
private paramIndex = 1;
|
|
|
|
constructor(private query: string) {}
|
|
|
|
bind(...values: any[]): D1PreparedStatement {
|
|
this.params = values;
|
|
return this;
|
|
}
|
|
|
|
/**
|
|
* 将 SQLite 风格的 ? 占位符替换为 Postgres 风格的 $1, $2, ...
|
|
*/
|
|
private convertQuery(query: string): string {
|
|
let index = 1;
|
|
return query.replace(/\?/g, () => `$${index++}`);
|
|
}
|
|
|
|
/**
|
|
* 将 SQL 查询中的表名和列名转换为双引号包裹（Postgres 要求）
|
|
* 注意：需要排除已经有引号的内容
|
|
*/
|
|
private quoteIdentifiers(query: string): string {
|
|
// 这个方法主要用于处理列值，表名在 schema 中已经创建好
|
|
return query;
|
|
}
|
|
|
|
/**
|
|
* 执行查询并返回第一行
|
|
*/
|
|
async first<T = any>(colName?: string): Promise<T | null> {
|
|
try {
|
|
const convertedQuery = this.convertQuery(this.query);
|
|
|
|
// 使用 Vercel Postgres 的 query 方法执行参数化查询
|
|
const result = await sql.query(convertedQuery, this.params);
|
|
|
|
if (!result || result.rows.length === 0) return null;
|
|
|
|
const row = result.rows[0];
|
|
|
|
if (colName) return row[colName] ?? null;
|
|
|
|
return row as T;
|
|
} catch (err) {
|
|
console.error('Postgres first() error:', err);
|
|
return null;
|
|
}
|
|
}
|
|
|
|
/**
|
|
* 执行查询并返回结果
|
|
*/
|
|
async run<T = any>(): Promise<D1Result<T>> {
|
|
try {
|
|
const convertedQuery = this.convertQuery(this.query);
|
|
|
|
const result = await sql.query(convertedQuery, this.params);
|
|
|
|
return {
|
|
success: true,
|
|
meta: {
|
|
changes: result.rowCount || 0,
|
|
last_row_id: null, // Postgres 不直接返回 lastInsertId
|
|
},
|
|
results: result.rows,
|
|
};
|
|
} catch (err: any) {
|
|
console.error('Postgres run() error:', err);
|
|
return {
|
|
success: false,
|
|
error: err.message,
|
|
};
|
|
}
|
|
}
|
|
|
|
/**
|
|
* 执行查询并返回所有行
|
|
*/
|
|
async all<T = any>(): Promise<D1Result<T>> {
|
|
try {
|
|
const convertedQuery = this.convertQuery(this.query);
|
|
|
|
const result = await sql.query(convertedQuery, this.params);
|
|
|
|
return {
|
|
success: true,
|
|
results: result.rows || [],
|
|
};
|
|
} catch (err: any) {
|
|
console.error('Postgres all() error:', err);
|
|
return {
|
|
success: false,
|
|
error: err.message,
|
|
results: [],
|
|
};
|
|
}
|
|
}
|
|
|
|
/**
|
|
* 内部执行方法（用于 batch 操作）
|
|
*/
|
|
async execute(): Promise<D1Result> {
|
|
return this.run();
|
|
}
|
|
}
|