demo db

electron/controller/framework.js

@@ -32,31 +32,31 @@ class FrameworkController extends Controller {
   /**
    * json数据库操作
    */   
-  async dbOperation(args) {
-    const paramsObj = args;
-    //Log.info('eeeee paramsObj:', paramsObj);
+  async jsondbOperation(args) {
+    const { action, info, delete_name, update_name, update_age, search_age } = args;
+
     const data = {
-      action: paramsObj.action,
+      action,
       result: null,
       all_list: []
     };
     
-    switch (paramsObj.action) {
+    switch (action) {
       case 'add' :
-        data.result = await Services.get('storage').addTestData(paramsObj.info);
+        data.result = await Services.get('database.jsondb').addTestData(info);
         break;
       case 'del' :
-        data.result = await Services.get('storage').delTestData(paramsObj.delete_name);
+        data.result = await Services.get('database.jsondb').delTestData(delete_name);
         break;
       case 'update' :
-        data.result = await Services.get('storage').updateTestData(paramsObj.update_name, paramsObj.update_age);
+        data.result = await Services.get('database.jsondb').updateTestData(update_name, update_age);
         break;
       case 'get' :
-        data.result = await Services.get('storage').getTestData(paramsObj.search_age);
+        data.result = await Services.get('database.jsondb').getTestData(search_age);
         break;
     }
 
-    data.all_list = await Services.get('storage').getAllTestData();
+    data.all_list = await Services.get('database.jsondb').getAllTestData();
 
     return data;
   }
@@ -65,36 +65,46 @@ class FrameworkController extends Controller {
    * sqlite数据库操作
    */   
   async sqlitedbOperation(args) {
-    const paramsObj = args;
-    //Log.info('eeeee paramsObj:', paramsObj);
+    const { action, info, delete_name, update_name, update_age, search_age } = args;
+
     const data = {
-      action: paramsObj.action,
+      action,
       result: null,
-      all_list: []
+      all_list: [],
+      code: 0
     };
-    
-    switch (paramsObj.action) {
+
+    try {
+      // test
+      Services.get('database.sqlitedb').getDataDir();
+    } catch (err) {
+      console.log(err);
+      data.code = -1;
+      return data;
+    }
+
+    switch (action) {
       case 'add' :
-        data.result = await Services.get('storage').addTestDataSqlite(paramsObj.info);;
+        data.result = await Services.get('database.sqlitedb').addTestDataSqlite(info);;
         break;
       case 'del' :
-        data.result = await Services.get('storage').delTestDataSqlite(paramsObj.delete_name);;
+        data.result = await Services.get('database.sqlitedb').delTestDataSqlite(delete_name);;
         break;
       case 'update' :
-        data.result = await Services.get('storage').updateTestDataSqlite(paramsObj.update_name, paramsObj.update_age);
+        data.result = await Services.get('database.sqlitedb').updateTestDataSqlite(update_name, update_age);
         break;
       case 'get' :
-        data.result = await Services.get('storage').getTestDataSqlite(paramsObj.search_age);
+        data.result = await Services.get('database.sqlitedb').getTestDataSqlite(search_age);
         break;
       case 'getDataDir' :
-        data.result = await Services.get('storage').getDataDir();
+        data.result = await Services.get('database.sqlitedb').getDataDir();
         break;
       case 'setDataDir' :
-        data.result = await Services.get('storage').setCustomDataDir(paramsObj.data_dir);
+        data.result = await Services.get('database.sqlitedb').setCustomDataDir(data_dir);
         break;            
     }
 
-    data.all_list = await Services.get('storage').getAllTestDataSqlite();
+    data.all_list = await Services.get('database.sqlitedb').getAllTestDataSqlite();
 
     return data;
   }  

electron/service/database/jsondb.js

@@ -0,0 +1,112 @@
+'use strict';
+
+const { Service } = require('ee-core');
+const Storage = require('ee-core/storage');
+const _ = require('lodash');
+
+/**
+ * json数据存储
+ * @class
+ */
+class JsondbService extends Service {
+
+  constructor (ctx) {
+    super(ctx);
+
+    // jsondb数据库
+    this.demoDB = Storage.connection('demo');  
+    this.demoDBKey = {
+      test_data: 'test_data'
+    };
+  }
+
+  /*
+   * 增 Test data
+   */
+  async addTestData(user) {
+    const key = this.demoDBKey.test_data;
+    if (!this.demoDB.db.has(key).value()) {
+      this.demoDB.db.set(key, []).write();
+    }
+    
+    const data = this.demoDB.db
+    .get(key)
+    .push(user)
+    .write();
+
+    return data;
+  }
+
+  /*
+   * 删 Test data
+   */
+  async delTestData(name = '') {
+    const key = this.demoDBKey.test_data;
+    const data = this.demoDB.db
+    .get(key)
+    .remove({name: name})
+    .write();
+
+    return data;
+  }
+
+  /*
+   * 改 Test data
+   */
+  async updateTestData(name= '', age = 0) {
+    const key = this.demoDBKey.test_data;
+    const data = this.demoDB.db
+    .get(key)
+    .find({name: name}) // 修改找到的第一个数据，貌似无法批量修改 todo
+    .assign({age: age})
+    .write();
+
+    return data;
+  }
+
+  /*
+   * 查 Test data
+   */
+  async getTestData(age = 0) {
+    const key = this.demoDBKey.test_data;
+    let data = this.demoDB.db
+    .get(key)
+    //.find({age: age}) 查找单个
+    .filter(function(o) {
+      let isHas = true;
+      isHas = age === o.age ? true : false;
+      return isHas;
+    })
+    //.orderBy(['age'], ['name']) 排序
+    //.slice(0, 10) 分页
+    .value();
+
+    if (_.isEmpty(data)) {
+      data = []
+    }
+
+    return data;
+  }
+
+  /*
+   * all Test data
+   */
+  async getAllTestData() {
+    const key = this.demoDBKey.test_data;
+    if (!this.demoDB.db.has(key).value()) {
+      this.demoDB.db.set(key, []).write();
+    }
+    let data = this.demoDB.db
+    .get(key)
+    .value();
+
+    if (_.isEmpty(data)) {
+      data = []
+    }
+
+    return data;
+  }
+}
+
+JsondbService.toString = () => '[class JsondbService]';
+module.exports = JsondbService;

electron/service/database/sqlitedb.js

@@ -0,0 +1,163 @@
+'use strict';
+
+const { Service } = require('ee-core');
+const Storage = require('ee-core/storage');
+const _ = require('lodash');
+const path = require('path');
+
+/**
+ * sqlite数据存储
+ * @class
+ */
+class SqlitedbService extends Service {
+
+  constructor (ctx) {
+    super(ctx);
+
+    this.sqliteFile = 'sqlite-demo.db';
+    let sqliteOptions = {
+      driver: 'sqlite',
+      default: {
+        timeout: 6000,
+        verbose: console.log // 打印sql语法
+      }
+    }
+    this.demoSqliteDB = Storage.connection(this.sqliteFile, sqliteOptions);
+  }
+
+  /*
+   * 检查并创建表 (sqlite)
+   */
+  async checkAndCreateTableSqlite(tableName = '') {
+    if (_.isEmpty(tableName)) {
+      throw new Error(`table name is required`);
+    }
+    // 检查表是否存在
+    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 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);
+
+  }
+
+  /*
+   * 增 Test data (sqlite)
+   */
+  async addTestDataSqlite(data) {
+    //console.log("add data:", data);
+
+    let table = 'user';
+    await this.checkAndCreateTableSqlite(table);
+
+    const insert = this.demoSqliteDB.db.prepare(`INSERT INTO ${table} (name, age) VALUES (@name, @age)`);
+    insert.run(data);
+
+    return true;
+  }
+
+  /*
+   * 删 Test data (sqlite)
+   */
+  async delTestDataSqlite(name = '') {
+    //console.log("delete name:", name);
+
+    let table = 'user';
+    await this.checkAndCreateTableSqlite(table);
+
+    const delUser = this.demoSqliteDB.db.prepare(`DELETE FROM ${table} WHERE name = ?`);
+    delUser.run(name);
+
+    return true;
+  }
+
+  /*
+   * 改 Test data (sqlite)
+   */
+  async updateTestDataSqlite(name= '', age = 0) {
+    //console.log("update :", {name, age});
+
+    let table = 'user';
+    await this.checkAndCreateTableSqlite(table);
+
+    const updateUser = this.demoSqliteDB.db.prepare(`UPDATE ${table} SET age = ? WHERE name = ?`);
+    updateUser.run(age, name);
+
+    return true;
+  }  
+
+  /*
+   * 查 Test data (sqlite)
+   */
+  async getTestDataSqlite(age = 0) {
+    //console.log("select :", {age});
+
+    let table = 'user';
+    await this.checkAndCreateTableSqlite(table);
+
+    const selectUser = this.demoSqliteDB.db.prepare(`SELECT * FROM ${table} WHERE age = @age`);
+    const users = selectUser.all({age: age});
+    //console.log("select users:", users);
+    return users;
+  }  
+  
+  /*
+   * all Test data (sqlite)
+   */
+  async getAllTestDataSqlite() {
+    //console.log("select all user");
+
+    let table = 'user';
+    await this.checkAndCreateTableSqlite(table);
+
+    const selectAllUser = this.demoSqliteDB.db.prepare(`SELECT * FROM ${table} `);
+    const allUser =  selectAllUser.all();
+    //console.log("select allUser:", allUser);
+    return allUser;
+  }
+  
+  /*
+   * get data dir (sqlite)
+   */
+  async getDataDir() {
+    const dir = this.demoSqliteDB.getStorageDir();    
+
+    return dir;
+  } 
+
+  /*
+   * set custom data dir (sqlite)
+   */
+  async setCustomDataDir(dir) {
+    if (_.isEmpty(dir)) {
+      return;
+    }
+
+    // the absolute path of the db file
+    const dbFile = path.join(dir, this.sqliteFile);
+    const sqliteOptions = {
+      driver: 'sqlite',
+      default: {
+        timeout: 6000,
+        verbose: console.log
+      }
+    }
+    this.demoSqliteDB = Storage.connection(dbFile, sqliteOptions);    
+
+    return;
+  }
+}
+
+SqlitedbService.toString = () => '[class SqlitedbService]';
+module.exports = SqlitedbService;

frontend/src/api/main.js

@@ -1,5 +1,5 @@
-import storage from 'store2'
 import request from '@/utils/request'
+import storage from 'store2'
 
 /**
  * 路由定义（主进程与渲染进程通信频道定义）
@@ -9,7 +9,7 @@ const ipcApiRoute = {
   test: 'controller.example.test',
   checkForUpdater: 'controller.framework.checkForUpdater',
   downloadApp: 'controller.framework.downloadApp',
-  dbOperation: 'controller.framework.dbOperation',
+  jsondbOperation: 'controller.framework.jsondbOperation',
   sqlitedbOperation: 'controller.framework.sqlitedbOperation',
   uploadFile: 'controller.framework.uploadFile',
   checkHttpServer: 'controller.framework.checkHttpServer',
@@ -82,7 +82,6 @@ const requestHttp = (uri, parameter) => {
 }
 
 export {
-  ipcApiRoute,
-  specialIpcRoute,
-  requestHttp,
+  ipcApiRoute, requestHttp, specialIpcRoute
 }
+

frontend/src/router/routerMap.js

@@ -37,9 +37,9 @@ const constantRouterMap = [
             component: () => import('@/views/framework/socket/SocketServer.vue')
           }, 
           {
-            path: '/framework/db/index',
-            name: 'FrameworkDBIndex',
-            component: () => import('@/views/framework/db/Index.vue')
+            path: '/framework/jsondb/index',
+            name: 'FrameworkJsonDBIndex',
+            component: () => import('@/views/framework/jsondb/Index.vue')
           },
           {
             path: '/framework/sqlitedb/index',

frontend/src/router/subMenu.js

@@ -24,7 +24,7 @@ export default {
 		'menu_103' : {
 			icon: 'profile',
 			title: 'json数据库',
-			pageName: 'FrameworkDBIndex',
+			pageName: 'FrameworkJsonDBIndex',
 			params: {}
 		},
 		'menu_104' : {

frontend/src/views/framework/jsondb/Index.vue

@@ -156,7 +156,7 @@ export default {
       const params = {
         action: 'all',
       }
-      ipc.invoke(ipcApiRoute.dbOperation, params).then(res => {
+      ipc.invoke(ipcApiRoute.jsondbOperation, params).then(res => {
         console.log('res:', res);
         if (res.all_list.length == 0) {
           return false;
@@ -179,7 +179,7 @@ export default {
       if (ac == 'add' && this.name.length == 0) {
         this.$message.error(`请填写数据`);
       }
-      ipc.invoke(ipcApiRoute.dbOperation, params).then(res => {
+      ipc.invoke(ipcApiRoute.jsondbOperation, params).then(res => {
         console.log('res:', res);
         if (ac == 'get') {
           if (res.result.length == 0) {

frontend/src/views/framework/sqlitedb/Index.vue

@@ -175,7 +175,7 @@ export default {
   },
   mounted () {
     this.init();
-    this.getAllTestData();
+    
   },
   methods: {
     init() {
@@ -183,7 +183,13 @@ export default {
         action: 'getDataDir',
       }
       ipc.invoke(ipcApiRoute.sqlitedbOperation, params).then(res => {
+        if (res.code == -1) {
+          this.$message.error('请检查sqlite数据库是否');
+          return
+        }
+
         this.data_dir = res.result;
+        this.getAllTestData();
       }) 
     },
     getAllTestData () {