96 lines
2.6 KiB
TypeScript
96 lines
2.6 KiB
TypeScript
import { JSClass } from '@ccms/container'
|
|
import { Model } from './model'
|
|
|
|
/**
|
|
* 数据库配置
|
|
*/
|
|
export interface DataBaseConfig {
|
|
/**
|
|
* 数据库连接串
|
|
*/
|
|
url: string | javax.sql.DataSource
|
|
/**
|
|
* 数据库驱动
|
|
*/
|
|
driverClassName?: string
|
|
/**
|
|
* 用户名
|
|
*/
|
|
username?: string
|
|
/**
|
|
* 密码
|
|
*/
|
|
password?: string
|
|
}
|
|
|
|
/**
|
|
* 数据库封装类
|
|
*/
|
|
export class DataBase {
|
|
private dataSource: javax.sql.DataSource
|
|
private jdbcTemplate: org.springframework.jdbc.core.JdbcTemplate
|
|
|
|
@JSClass('com.zaxxer.hikari.HikariDataSource')
|
|
private HikariDataSource: any
|
|
@JSClass('com.zaxxer.hikari.HikariConfig')
|
|
private HikariConfig: any
|
|
@JSClass('org.springframework.jdbc.core.JdbcTemplate')
|
|
private JdbcTemplate: typeof org.springframework.jdbc.core.JdbcTemplate
|
|
|
|
constructor(dbConfig: DataBaseConfig) {
|
|
if (!dbConfig.url) { throw new Error('DataBase url can\'t be null!') }
|
|
this.createDataSource(dbConfig)
|
|
this.initialize()
|
|
}
|
|
|
|
private createDataSource(dbConfig: DataBaseConfig) {
|
|
if (typeof dbConfig.url === "string") {
|
|
let config = new this.HikariConfig()
|
|
if (dbConfig.driverClassName) {
|
|
config.setDriverClassName(dbConfig.driverClassName)
|
|
}
|
|
if (dbConfig.username) {
|
|
config.setUsername(dbConfig.username)
|
|
}
|
|
if (dbConfig.password) {
|
|
config.setPassword(dbConfig.password)
|
|
}
|
|
config.setJdbcUrl(dbConfig.url)
|
|
this.dataSource = new this.HikariDataSource(config)
|
|
} else {
|
|
this.dataSource = dbConfig.url
|
|
}
|
|
}
|
|
|
|
private initialize() {
|
|
this.jdbcTemplate = new this.JdbcTemplate(this.dataSource)
|
|
}
|
|
|
|
/**
|
|
* 执行SQL查询
|
|
* @param sql SQL语句
|
|
*/
|
|
query<T>(sql: string, ...args: any[]): Array<T> {
|
|
let startTime = Date.now()
|
|
let result = Java.from<any>(this.jdbcTemplate.queryForList(sql, args))
|
|
console.debug(java.lang.String.format(`\n[DB] query \nSQL : ${sql.replace(/\?/ig, '%s')} \nCOST : ${Date.now() - startTime}ms`, args))
|
|
return result
|
|
}
|
|
|
|
/**
|
|
* 执行SQL更新
|
|
* @param sql SQL语句
|
|
*/
|
|
update(sql: string, ...args: any[]): number {
|
|
let startTime = Date.now()
|
|
let result = this.jdbcTemplate.update(sql, args)
|
|
console.debug(java.lang.String.format(`\n[DB] update \nSQL : ${sql.replace(/\?/ig, '%s')} \nCOST : ${Date.now() - startTime}ms`, args))
|
|
return result
|
|
}
|
|
|
|
close() {
|
|
//@ts-ignore
|
|
this.dataSource.close()
|
|
}
|
|
}
|