This commit is contained in:
xiaojunnuo
2024-10-31 22:35:05 +08:00
parent 584378a32b
commit b817cb4a1b
13 changed files with 125 additions and 41 deletions
@@ -0,0 +1,3 @@
export interface SqlAdapter {
date(columnName: string): string;
}
@@ -0,0 +1,34 @@
import { SqliteAdapter } from './sqlite.js';
import { PostgresqlAdapter } from './postgresql.js';
import { Config, Init, Provide, Scope, ScopeEnum } from '@midwayjs/core';
import { SqlAdapter } from './d.js';
@Provide()
@Scope(ScopeEnum.Singleton)
export class DbAdapter implements SqlAdapter {
adapter: SqlAdapter;
@Config('typeorm.dataSource.default.type')
dbType: string;
@Init()
async init() {
if (this.isSqlite()) {
this.adapter = new SqliteAdapter();
} else if (this.isPostgresql()) {
this.adapter = new PostgresqlAdapter();
} else {
throw new Error(`dbType ${this.dbType} not support`);
}
}
isSqlite() {
return this.dbType === 'better-sqlite3';
}
isPostgresql() {
return this.dbType === 'postgres';
}
date(columnName: string) {
return this.adapter.date(columnName);
}
}
@@ -0,0 +1,7 @@
import { SqlAdapter } from './d.js';
export class PostgresqlAdapter implements SqlAdapter {
date(columnName: string) {
return `to_char(${columnName}, 'YYYY-MM-DD')`;
}
}
@@ -0,0 +1,7 @@
import { SqlAdapter } from './d.js';
export class SqliteAdapter implements SqlAdapter {
date(columnName: string) {
return `date(${columnName}, 'localtime')`;
}
}