@ocap/statedb-memory 1.28.9 → 1.29.0

package/lib/table/base.js

@@ -1,135 +0,0 @@
-const { StateDBTable } = require('@ocap/statedb');
-const omit = require('lodash/omit');
-
-const debug = require('debug')(require('../../package.json').name);
-
-class MemoryTable extends StateDBTable {
-  /**
-   * @param {Object} param
-   * @param {string} param.name db name
-   * @param {string|string[]} param.uniqIndex primary keys
-   * @param {boolean} param.syncBalance sync balance table when create / update / get
-   * @param {BalanceTable} param.balanceTable balance table
-   * @param {import('lokijs')} param.db LokiJS database instance
-   */
-  constructor({ name, uniqIndex, balanceTable, syncBalance = false, db }) {
-    super(uniqIndex);
-
-    this.name = name;
-    this.uniqIndex = uniqIndex;
-    this.balanceTable = balanceTable;
-    this.syncBalance = syncBalance;
-
-    if (this.syncBalance && !this.balanceTable) {
-      throw new Error('balanceTable is required when syncBalance is true');
-    }
-
-    if (!db) {
-      throw new Error('db is required for MemoryTable');
-    }
-
-    this.collection = db.addCollection(name, { unique: [this.primaryKey], clone: true });
-    this.markReady();
-  }
-
-  async _create(key, attrs) {
-    const id = this.generatePrimaryKey(key);
-    // Don't call this._get here cause _get may be overwritten
-    const doc = await MemoryTable.prototype._get.call(this, id);
-    if (doc) {
-      throw new Error(`${this.name} already exists: ${key}`);
-    }
-
-    debug(`insert ${this.name}`, attrs);
-
-    const insertAttrs = { ...attrs };
-    // insert without tokens cause we've migrate token to balance table
-    if (this.syncBalance) {
-      delete insertAttrs.tokens;
-    }
-
-    const result = await this.collection.insert({ [this.primaryKey]: id, ...insertAttrs });
-
-    if (this.syncBalance && attrs.tokens) {
-      debug(`update balance for ${this.name}`, attrs);
-
-      result.tokens = await this.balanceTable.updateBalance({
-        address: attrs.address,
-        tokens: attrs.tokens,
-        context: attrs.context,
-      });
-    }
-
-    return result;
-  }
-
-  async _get(key) {
-    if (!key) return null;
-
-    const id = this.generatePrimaryKey(key);
-    const result = await this.collection.by(this.primaryKey, id);
-
-    if (result && this.syncBalance) {
-      const balance = await this.balanceTable.getBalance(result.address);
-      result.tokens = { ...(result.tokens || {}), ...balance };
-    }
-
-    return result;
-  }
-
-  _history() {
-    return [];
-  }
-
-  async _update(key, updates) {
-    const id = this.generatePrimaryKey(key);
-    // Don't call this._get here cause _get may be overwritten
-    const doc = await MemoryTable.prototype._get.call(this, id);
-    if (!doc) {
-      throw new Error(`${this.name} does not exists: ${key}`);
-    }
-
-    Object.assign(doc, updates);
-
-    const updateAttrs = { ...doc };
-    // insert without tokens cause we've migrate token to balance table
-    if (this.syncBalance) {
-      delete updateAttrs.tokens;
-    }
-
-    await this.collection.update(updateAttrs);
-
-    if (this.syncBalance && doc.tokens) {
-      debug(`update balance for ${this.name}`, doc);
-
-      doc.tokens = await this.balanceTable.updateBalance({
-        address: doc.address,
-        tokens: doc.tokens,
-        context: doc.context,
-      });
-    }
-
-    return doc;
-  }
-
-  updateOrCreate(exist, state, ctx) {
-    const id = this.generatePrimaryKey(state);
-    const attrs = omit(state, this.uniqIndex);
-
-    if (!id) {
-      throw new Error('Cannot update or create without uniq index');
-    }
-
-    if (exist) {
-      return this.update(id, attrs, ctx);
-    }
-
-    return this.create(id, attrs, ctx);
-  }
-
-  _reset() {
-    this.collection.removeWhere({});
-  }
-}
-
-module.exports = MemoryTable;

package/lib/table/rollup.js

@@ -1,13 +0,0 @@
-const MemoryTable = require('./base');
-
-class RollupTable extends MemoryTable {
-  async existByToken(token) {
-    if (!token) {
-      return false;
-    }
-
-    return this.collection.count({ tokenAddress: token, paused: false }) > 0;
-  }
-}
-
-module.exports = RollupTable;

package/lib/table/token.js

@@ -1,13 +0,0 @@
-const MemoryTable = require('./base');
-
-class TokenTable extends MemoryTable {
-  async existBySymbol(symbol) {
-    if (!symbol) {
-      throw new Error('param symbol is required');
-    }
-
-    return this.collection.count({ symbol }) > 0;
-  }
-}
-
-module.exports = TokenTable;