完成 MysqlConnection、RedisConnection

src/config.ts

@@ -4,6 +4,17 @@ const config = {
         allowedHeaders: ['Content-Type'],
         methods: ['GET', 'POST']
     },
+    mysql: {
+        host: 'localhost',
+        database: '',
+        user: 'root',
+        password: ''
+    },
+    redis: {
+        host: 'localhost',
+        port: 6379,
+        password: '' // localhost
+    },
     API_Port: 8080
 };
 export default config;

src/index.ts

@@ -1,12 +1,18 @@
 import { APILoader } from "@lib/API/APILoader";
 import Logger from '@lib/Logger/Logger'
 import config from "./config";
+import MySQLConnection from "@lib/Database/MySQLConnection";
+import RedisConnection from "@lib/Database/RedisConnection";
+MySQLConnection
+RedisConnection
+// import API
 import GetTest from "./api/GetTest";
 const logger = new Logger('Server')
+
 async function main(): Promise<void> {
     logger.info('Starting...');
     const apiLoader = new APILoader(config.cors);
-    // loadAPI
+    // addAPI
     apiLoader.add(GetTest);
 
     await apiLoader.start(config.API_Port);

src/lib/Database/MySQLConnection.ts

@@ -0,0 +1,84 @@
+/**
+ * @file MySQLConnection.ts
+ * @version 1.0.0
+ * @description MySQL数据库连接池
+ */
+import mysql from "mysql2/promise";
+import Logger from "@lib/Logger/Logger";
+import config from "../../config";
+
+class MySQLConnectPool {
+    private pool: any;
+    private logger = new Logger('MySQLConnection');
+
+    constructor() {
+        this.pool = this.createConnectPool();
+        this.logger.info("Database connection pool created")
+        setTimeout(async () => {
+            let res = await this.testConnection();
+            if (res)
+                this.logger.info("Database test successful")
+            else
+                this.logger.error("Database test failed")
+        }, 10);
+    }
+
+    private createConnectPool() {
+        return mysql.createPool({
+            host: config.mysql.host,
+            database: config.mysql.database,
+            user: config.mysql.user,
+            password: config.mysql.password,
+            waitForConnections: true,
+            connectionLimit: 10,
+            queueLimit: 0
+        })
+    }
+
+    private 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) {
+            this.logger.error(`An error occurred during the database test: ` + error);
+            return 0;
+        }
+
+    }
+
+    /**
+     * 执行SQL查询
+     * @param sql SQL语句
+     * @param values 可选的查询参数列表
+     * @param database 可选的数据库
+     * @returns Promise<any | undefined> 查询结果
+     */
+    public async execute(sql: string, values?: any[], database?: string): Promise<any | undefined> {
+        let connection: any;
+        try {
+            connection = await this.pool.getConnection();
+
+            // 如果指定了数据库，则更改当前连接的数据库
+            if (database) {
+                await connection.changeUser({ database });
+            }
+
+            let [rows, fields] = await connection.execute(sql, values);
+            return rows;
+        } catch (error) {
+            this.logger.error("An error occurred in the database: " + error, '\n##', sql, '\n##', JSON.stringify(values));
+            return undefined;
+        } finally {
+            if (database)
+                await connection.changeUser({ database: config.mysql.database });// 恢复默认数据库
+            if (connection)
+                connection.release();
+        }
+    }
+}
+
+const MySQLConnection = new MySQLConnectPool();
+export default MySQLConnection;

src/lib/Database/RedisConnection.ts

@@ -0,0 +1,43 @@
+import Redis from 'ioredis';
+import config from '../../config';
+import Logger from '@lib/Logger/Logger';
+
+class RedisConnection {
+    private pool?: Redis
+    private logger = new Logger('Redis')
+
+    constructor() {
+        try {
+            this.pool = new Redis({
+                port: config.redis.port,
+                host: config.redis.host,
+                password: config.redis.password,
+                maxRetriesPerRequest: 10,
+            });
+            this.logger.info('Database connection pool created')
+        } catch (error) {
+            this.logger.error('Failed to create database connection pool: ' + error)
+        }
+        setTimeout(async () => {
+            if (this.pool == undefined)
+                return;
+            try {
+                let res = await this.pool.set('redis_test', '1');
+                if (res)
+                    this.logger.info('Database test successful')
+                else
+                    throw new Error('Unexpected return value')
+            } catch (error) {
+                this.logger.error('Database test failed: ' + error)
+            }
+
+        }, 10);
+    }
+
+    public getPool(): Redis {
+        return <Redis>this.pool;
+    }
+}
+
+const redisConnection = new RedisConnection();
+export default redisConnection.getPool();