feat(@142vip/egg): 完善SequelizeORM基础类

packages/egg/src/orm/sequelize.ts

@@ -1,4 +1,5 @@
-import { DataTypes } from 'sequelize'
+import type { Options } from 'sequelize'
+import { DataTypes, Sequelize } from 'sequelize'
 
 export const BaseSequelizeOptions = {
   // 全局默认配置
@@ -58,3 +59,47 @@ export const VipSequelize = {
   BaseEntity: BaseSequelizeEntity,
   createEntity: createSequelizeEntity,
 }
+
+export interface SequelizeOptions extends Options {
+  connectUri?: string
+  Sequelize: typeof Sequelize
+}
+
+/**
+ * Sequelize基础类
+ */
+export class SequelizeORM {
+  private readonly options: SequelizeOptions
+  private sequelize: Sequelize
+  constructor(options: SequelizeOptions) {
+    this.options = options
+    this.sequelize = this.getConnect()
+  }
+
+  /**
+   * 连接
+   */
+  getConnect() {
+    if (this.sequelize == null) {
+      const _Sequelize = this.options.Sequelize || Sequelize
+      this.sequelize = this.options.connectUri != null
+        ? new _Sequelize(this.options.connectUri, this.options)
+        : new _Sequelize(this.options)
+    }
+    return this.sequelize
+  }
+
+  /**
+   * 连接重试
+   */
+  public async retry() {
+    await this.sequelize.authenticate()
+  }
+
+  /**
+   * 关闭连接
+   */
+  public async disconnect() {
+    await this.sequelize.close()
+  }
+}