@ocap/indexdb-elasticsearch/lib/table/account.js

OCAP indexdb adapter that uses elasticsearch as backend

/* eslint-disable no-undef */
/* eslint-disable no-underscore-dangle */
const ESIndex = require('./base');
const getParams = require('../model/account');
const { getTableName } = require('../util');

class Account extends ESIndex {
  constructor(db) {
    super({
      name: getTableName(db.config.prefix, 'account'),
      docId: 'address',
      client: db.client,
      http: db.http,
      indexParams: getParams(),
    });

    this.tokenLength = db.config.tokenLength;
  }

  static formatBeforeUpdate(data, tokenLength) {
    data.balance = super.padBalance(data.balance, tokenLength);

    if (Array.isArray(data.tokens)) {
      data.tokens = data.tokens.map((x) => {
        x.balance = super.padBalance(x.balance, tokenLength);
        return x;
      });
    }

    return data;
  }

  static formatAfterRead(data) {
    if (!data) return data;

    data.balance = super.trimBalance(data.balance);

    if (Array.isArray(data.tokens)) {
      data.tokens = data.tokens.map((x) => {
        x.balance = super.trimBalance(x.balance);
        return x;
      });
    }

    return data;
  }

  batchInsert(rows) {
    const formatted = rows.map((x) => Account.formatBeforeUpdate(x, this.tokenLength));
    return super.batchInsert(formatted);
  }

  _insert(row) {
    return super._insert(Account.formatBeforeUpdate(row, this.tokenLength));
  }

  _update(key, updates) {
    return super._update(key, Account.formatBeforeUpdate(updates, this.tokenLength));
  }

  async _get(key) {
    const item = await super._get(key);

    return Account.formatAfterRead(item);
  }
}

module.exports = Account;