@abtnode/connect-storage 1.16.19-beta-340de95d → 1.16.19-beta-e6aac665

package/lib/index.js

@@ -8,19 +8,25 @@ class SequelizeStorage extends EventEmitter {
     if (!options.dbPath) {
       throw new Error('SequelizeStorage requires dbPath to be set');
     }
+    if (!options.model) {
+      throw new Error('SequelizeStorage requires model to be set');
+    }
+    if (!options.primaryKey) {
+      throw new Error('SequelizeStorage requires primaryKey to be set');
+    }
 
     super(options);
 
     this.sequelize = createSequelize(options.dbPath);
     this.models = getConnectModels();
+
     setupModels(this.models, this.sequelize);
 
-    this.state = new ConnectionState(this.models.Connection);
+    this.state = new ConnectionState(this.models[options.model], options);
+    this.primaryKey = options.primaryKey;
 
     doSchemaMigration(options.dbPath, 'connect')
       .then(() => {
-        // eslint-disable-next-line no-console
-        console.info(`Connection storage schema migration succeed: ${options.dbPath}`);
         if (typeof options.onload === 'function') {
           options.onload();
         }
@@ -51,13 +57,13 @@ class SequelizeStorage extends EventEmitter {
   }
 
   async delete(token) {
-    const num = await this.state.remove({ token });
+    const num = await this.state.remove({ [this.primaryKey]: token });
     this.emit('destroy', token);
     return num;
   }
 
   async exist(token, did) {
-    const data = await this.state.count({ token, did });
+    const data = await this.state.count({ [this.primaryKey]: token, did });
     return data;
   }
 

package/lib/state.js

@@ -7,12 +7,16 @@ const { BaseState } = require('@abtnode/models');
  */
 class Connection extends BaseState {
   async start(token, status = 'created') {
-    const doc = await this.insert({ token, status });
+    const attrs =
+      status && typeof status === 'object'
+        ? { [this.config.primaryKey]: token, ...status }
+        : { [this.config.primaryKey]: token, status };
+    const doc = await this.insert(attrs);
     return this._format(doc);
   }
 
   async update(token, updates) {
-    const doc = await this.findOne({ token });
+    const doc = await this.findOne({ [this.config.primaryKey]: token });
     if (!doc) {
       throw new Error(`Connect session does not exist: ${token}`);
     }
@@ -20,12 +24,15 @@ class Connection extends BaseState {
     const knownKeys = Object.keys(this.model.getAttributes());
     const extra = omit(updates, knownKeys);
 
-    await super.update({ token }, { ...pick(updates, knownKeys), __extra: { ...doc.__extra, ...extra } });
+    await super.update(
+      { [this.config.primaryKey]: token },
+      { ...pick(updates, knownKeys), __extra: { ...doc.__extra, ...extra } }
+    );
     return this._format(doc, updates);
   }
 
   async read(token) {
-    const doc = await this.findOne({ token });
+    const doc = await this.findOne({ [this.config.primaryKey]: token });
     return this._format(doc);
   }
 

package/package.json

@@ -3,7 +3,7 @@
   "publishConfig": {
     "access": "public"
   },
-  "version": "1.16.19-beta-340de95d",
+  "version": "1.16.19-beta-e6aac665",
   "description": "Sequelize storage for @arcblock/did-auth",
   "main": "lib/index.js",
   "files": [
@@ -19,11 +19,11 @@
   "author": "wangshijun <wangshijun2010@gmail.com> (http://github.com/wangshijun)",
   "license": "Apache-2.0",
   "dependencies": {
-    "@abtnode/models": "1.16.19-beta-340de95d",
+    "@abtnode/models": "1.16.19-beta-e6aac665",
     "lodash": "^4.17.21"
   },
   "devDependencies": {
     "jest": "^27.5.1"
   },
-  "gitHead": "cc7afa5fe28ef4b0304f97b7e0b9ad8194427ae7"
+  "gitHead": "09b0172831521ef73f7879940b977c2696b5b247"
 }