git init
This commit is contained in:
69
Server/plugs/database/MySQLConnection.ts
Normal file
69
Server/plugs/database/MySQLConnection.ts
Normal file
@@ -0,0 +1,69 @@
|
||||
// MYSQL数据库连接池
|
||||
// 版本:v0.1
|
||||
const mysql = require('mysql2/promise');
|
||||
import Logger from "../Logger";
|
||||
import SettingConfig from "../Settingconfig";
|
||||
|
||||
class MySQLConnectPool{
|
||||
pool: any;
|
||||
|
||||
constructor()
|
||||
{
|
||||
this.pool = this.createConnectPool();
|
||||
Logger.info("[MySQL] 数据库连接池已创建")
|
||||
setTimeout(async () => {
|
||||
let res = await this.testConnection();
|
||||
if(res)
|
||||
Logger.info("[MySQL] 数据库测试成功")
|
||||
else
|
||||
Logger.error("[MySQL] 数据库测试失败")
|
||||
}, 10);
|
||||
}
|
||||
|
||||
// 内部函数,无需手动调用
|
||||
createConnectPool(){
|
||||
return mysql.createPool({
|
||||
host: SettingConfig.mysql.host,
|
||||
database: SettingConfig.mysql.database,
|
||||
user: SettingConfig.mysql.user,
|
||||
password: SettingConfig.mysql.password,
|
||||
waitForConnections:true,
|
||||
connectionLimit:10,
|
||||
queueLimit:0
|
||||
})
|
||||
}
|
||||
|
||||
// 内部函数,无需手动调用
|
||||
async testConnection(){
|
||||
try {
|
||||
let res = await this.execute("SELECT 1 + 1 As result");
|
||||
if(res[0].result == 2)
|
||||
return 1;
|
||||
else
|
||||
return 0;
|
||||
} catch (error) {
|
||||
Logger.error(`[MYSQL] 数据库测试发生了错误:`+error);
|
||||
return 0;
|
||||
}
|
||||
|
||||
}
|
||||
|
||||
// 执行SQL语句
|
||||
async execute(sql: string,values = undefined){
|
||||
let connection;
|
||||
try {
|
||||
connection = await this.pool.getConnection();
|
||||
let [rows, fields] = await connection.execute(sql,values);
|
||||
return rows;
|
||||
} catch (error) {
|
||||
Logger.error("[MYSQL] 数据库发生错误:"+error);
|
||||
return undefined;
|
||||
} finally{
|
||||
if(connection)
|
||||
connection.release();
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
let MySQLConnection = new MySQLConnectPool();
|
||||
export default MySQLConnection;
|
||||
98
Server/plugs/database/RedisConnection.ts
Normal file
98
Server/plugs/database/RedisConnection.ts
Normal file
@@ -0,0 +1,98 @@
|
||||
const Redis = require('ioredis');
|
||||
import Logger from "../Logger";
|
||||
|
||||
class _RedisConnection{
|
||||
pool: any;
|
||||
|
||||
constructor(){
|
||||
this.pool = new Redis({
|
||||
host: 'server.tonesc.cn',
|
||||
password: '2Pj4Ss9al3mS1',
|
||||
connectionPoolSize: 10,
|
||||
});
|
||||
Logger.info('[Redis] 数据库连接池已创建')
|
||||
setTimeout(async () => {
|
||||
let res = await this.set('redis_test', '1');
|
||||
if(res)
|
||||
Logger.info('[Redis] 数据库测试成功')
|
||||
else
|
||||
Logger.error('[Redis] 数据库测试失败')
|
||||
}, 10);
|
||||
}
|
||||
|
||||
/**
|
||||
* 设置键值对
|
||||
* @param {*} key
|
||||
* @param {*} value
|
||||
* @returns { Promise<Boolean> } 成功返回true,失败返回false
|
||||
*/
|
||||
set(key: string, value: string){
|
||||
return new Promise((resolve) => {
|
||||
this.pool.set(key, value, (err: any, value: any) => {
|
||||
if (err) {
|
||||
Logger.error('[Redis] 设置键时发生错误:' + err);
|
||||
resolve(false);
|
||||
} else {
|
||||
resolve(true);
|
||||
}
|
||||
})
|
||||
})
|
||||
}
|
||||
|
||||
/**
|
||||
* 获取键值对,成功返回值,失败返回undefined
|
||||
* @param {string} key
|
||||
* @returns { Promise<String> } 成功返回值,失败返回undefined
|
||||
*/
|
||||
get(key: string): Promise<string> {
|
||||
return new Promise((resolve) => {
|
||||
this.pool.get(key, (err: any, value: any) => {
|
||||
if (err) {
|
||||
Logger.error('[Redis] 获取键时发生错误:' + err);
|
||||
throw new Error('Redis连接错误')
|
||||
} else {
|
||||
resolve(value);
|
||||
}
|
||||
})
|
||||
})
|
||||
}
|
||||
|
||||
/**
|
||||
* 删除键值对,成功返回true,失败返回false
|
||||
* @param {string} key
|
||||
* @returns { Promise<Boolean> } 成功返回true,失败返回false
|
||||
*/
|
||||
del(key: string){
|
||||
return new Promise((resolve) => {
|
||||
this.pool.del(key, (err: any, value: any) => {
|
||||
if (err) {
|
||||
Logger.error('[Redis] 删除键时发生错误:' + err);
|
||||
resolve(false);
|
||||
} else {
|
||||
resolve(true);
|
||||
}
|
||||
})
|
||||
})
|
||||
}
|
||||
|
||||
/**
|
||||
* 设置键的过期时间
|
||||
* @param {*} key
|
||||
* @param {*} seconds
|
||||
* @returns { Promise<Boolean> } 成功返回true,失败返回false
|
||||
*/
|
||||
expire(key: any, seconds: number){
|
||||
return new Promise((resolve) => {
|
||||
this.pool.expire(key, seconds, (err: number, value: number) => {
|
||||
if (err) {
|
||||
Logger.error('[Redis] 设置键过期时发生错误:' + err);
|
||||
resolve(false);
|
||||
} else {
|
||||
resolve(true);
|
||||
}
|
||||
})
|
||||
})
|
||||
}
|
||||
}
|
||||
const RedisConnection = new _RedisConnection();
|
||||
export default RedisConnection;
|
||||
Reference in New Issue
Block a user