From 83f6cc33b17442c46cae45fa486b770409ebe41b Mon Sep 17 00:00:00 2001
From: gaoshuaixing <gaoshuaixing@bilibili.com>
Date: Fri, 27 Dec 2024 19:33:19 +0800
Subject: [PATCH] feat rewrite sqlite

---
 electron/controller/framework.js              |  2 +-
 electron/service/database/basedb.js           | 43 ++++++++++++++++
 electron/service/database/sqlitedb.js         | 51 ++++++++-----------
 .../src/views/framework/sqlitedb/Index.vue    |  4 +-
 package.json                                  |  5 +-
 5 files changed, 70 insertions(+), 35 deletions(-)
 create mode 100644 electron/service/database/basedb.js

diff --git a/electron/controller/framework.js b/electron/controller/framework.js
index 3433449..e39e864 100644
--- a/electron/controller/framework.js
+++ b/electron/controller/framework.js
@@ -9,7 +9,7 @@ const { getExtraResourcesDir } = require('ee-core/ps');
 const { logger } = require('ee-core/log');
 const { getConfig } = require('ee-core/config');
 const { frameworkService } = require('../service/framework');
-const { getHttpServer } = require('ee-core/socket');
+const { frameworkService } = require('../service/database/sqlitedb');
 
 /**
  * framework - demo
diff --git a/electron/service/database/basedb.js b/electron/service/database/basedb.js
new file mode 100644
index 0000000..c0a3a48
--- /dev/null
+++ b/electron/service/database/basedb.js
@@ -0,0 +1,43 @@
+'use strict';
+
+const { SqliteStorage } = require('ee-core/storage');
+const { getStorageDir } = require('ee-core/ps');
+const path = require('path');
+
+/**
+ * sqlite数据存储
+ * @class
+ */
+class BasedbService {
+
+  constructor(options) {
+    super();
+
+    const { dbname } = options;
+    this.dbname = dbname;
+    this.db = undefined;
+    this._init();
+  }
+
+  /*
+   * 初始化
+   */
+  _init() {
+    // 定义数据文件
+    const dbFile = path.join(getStorageDir(), "db", this.dbname);
+    const sqliteOptions = {
+      driver: 'sqlite',
+      default: {
+        timeout: 6000,
+        verbose: null // 打印sql语法 console.log
+      }
+    }
+    const storage = new SqliteStorage(dbFile, sqliteOptions);
+    this.db = storage.db;
+  }
+}  
+
+BasedbService.toString = () => '[class BasedbService]';
+module.exports = {
+  BasedbService,
+};
\ No newline at end of file
diff --git a/electron/service/database/sqlitedb.js b/electron/service/database/sqlitedb.js
index 7971e03..63a6528 100644
--- a/electron/service/database/sqlitedb.js
+++ b/electron/service/database/sqlitedb.js
@@ -1,5 +1,6 @@
 'use strict';
 
+const { BasedbService } = require('./basedb');
 const Storage = require('ee-core/storage');
 const _ = require('lodash');
 const path = require('path');
@@ -8,45 +9,35 @@ const path = require('path');
  * sqlite数据存储
  * @class
  */
-class SqlitedbService {
+class SqlitedbService extends BasedbService {
 
   constructor () {
-    this.sqliteFile = 'sqlite-demo.db';
-    let sqliteOptions = {
-      driver: 'sqlite',
-      default: {
-        timeout: 6000,
-        verbose: console.log // 打印sql语法
-      }
+    const options = {
+      dbname:'sqlite-demo.db',
     }
-    this.demoSqliteDB = Storage.connection(this.sqliteFile, sqliteOptions);
+    super(options);
+    this._initTable();
   }
 
   /*
-   * 检查并创建表 (sqlite)
+   * 初始化表
    */
-  async checkAndCreateTableSqlite(tableName = '') {
-    if (_.isEmpty(tableName)) {
-      throw new Error(`table name is required`);
-    }
+  _initTable() {
     // 检查表是否存在
-    const userTable = this.demoSqliteDB.db.prepare('SELECT * FROM sqlite_master WHERE type=? AND name = ?');
-    const result = userTable.get('table', tableName);
-    //console.log('result:', result);
-    if (result) {
-      return;
+    const userTableName = 'user';
+    const masterStmt = this.db.prepare('SELECT * FROM sqlite_master WHERE type=? AND name = ?');
+    let tableExists = masterStmt.get('table', userTableName);
+    if (!tableExists) {
+      // 创建表
+      const create_user_table_sql =
+      `CREATE TABLE ${userTableName}
+      (
+         id INTEGER PRIMARY KEY AUTOINCREMENT,
+         name CHAR(50) NOT NULL,
+         age INT
+      );`
+      this.db.exec(create_user_table_sql);
     }
-
-    // 创建表
-    const create_table_user =
-    `CREATE TABLE ${tableName}
-     (
-        id INTEGER PRIMARY KEY AUTOINCREMENT,
-        name CHAR(50) NOT NULL,
-        age INT
-     );`
-    this.demoSqliteDB.db.exec(create_table_user);
-
   }
 
   /*
diff --git a/frontend/src/views/framework/sqlitedb/Index.vue b/frontend/src/views/framework/sqlitedb/Index.vue
index 5a8871c..83d8771 100644
--- a/frontend/src/views/framework/sqlitedb/Index.vue
+++ b/frontend/src/views/framework/sqlitedb/Index.vue
@@ -1,5 +1,5 @@
 <template>
-  <div id="app-base-db">
+  <div id="app-sqlite-db">
     <div class="one-block-1">
       <span>
         1. sqlite本地数据库
@@ -260,7 +260,7 @@ export default {
 };
 </script>
 <style lang="less" scoped>
-#app-base-db {
+#app-sqlite-db {
   padding: 0px 10px;
   text-align: left;
   width: 100%;
diff --git a/package.json b/package.json
index 6ec612a..8c3adcf 100644
--- a/package.json
+++ b/package.json
@@ -34,7 +34,7 @@
     "build-l-armv7l": "electron-builder --config=./electron/config/builder.json -l=deb --armv7l",
     "build-lr-64": "electron-builder --config=./electron/config/builder.json -l=rpm --x64",
     "build-lp-64": "electron-builder --config=./electron/config/builder.json -l=pacman --x64",
-    "test": "cross-env DEBUG=* ee-bin dev --serve=electron"
+    "test": "cross-env DEBUG=ee-core* ee-bin dev --serve=electron"
   },
   "repository": "https://github.com/dromara/electron-egg.git",
   "keywords": [
@@ -45,15 +45,16 @@
   "author": "哆啦好梦, Inc <530353222@qq.com>",
   "license": "Apache",
   "devDependencies": {
+    "@electron/rebuild": "^3.7.1",
     "cross-env": "^7.0.3",
     "debug": "^4.4.0",
     "ee-bin": "file:.yalc/ee-bin",
-    "ee-bin-ts": "file:.yalc/ee-bin-ts",
     "electron": "^29.4.0",
     "electron-builder": "^25.1.8"
   },
   "dependencies": {
     "axios": "^1.7.9",
+    "better-sqlite3": "^11.7.0",
     "dayjs": "^1.11.13",
     "ee-core": "file:.yalc/ee-core",
     "electron-updater": "^6.3.8"