1629606507543-migrate-orders.js

/* eslint-disable no-await-in-loop */
/* eslint-disable guard-for-in */
/* eslint-disable no-restricted-syntax */
const _ = require('lodash');
const path = require('path');
const { logger: rootLogger, mongo, cache } = require('../app/helpers');

module.exports.up = async () => {
  const logger = rootLogger.child({
    gitHash: process.env.GIT_HASH || 'unspecified',
    migration: path.basename(__filename)
  });

  const database = await mongo.connect(logger);

  logger.info('Start migration');

  let keys;
  let result;

  // Create index
  const gridTradeOrderscollection = database.collection(
    'trailing-trade-grid-trade-orders'
  );
  try {
    result = await gridTradeOrderscollection.dropIndex(
      `trailing-trade-grid-trade-orders-key-idx`
    );
  } catch (e) {
    logger.warn(
      `Index 'trailing-trade-grid-trade-orders-key-idx' is not found. It's ok.`
    );
  }

  result = await gridTradeOrderscollection.createIndex(
    { key: 1 },
    { name: `trailing-trade-grid-trade-orders-key-idx` }
  );

  logger.info({ result }, 'Created index for orders');

  // Migreate cache to mongo
  keys = await cache.keys('*grid-trade-last*');
  if (keys && keys.length > 0) {
    for (const key of keys) {
      // Get cache
      const order = JSON.parse(await cache.get(key)) || {};

      await mongo.insertOne(logger, 'trailing-trade-grid-trade-orders', {
        key,
        order
      });

      await cache.del(key);
    }
  }

  // Create index
  const manualOrdersCollection = database.collection(
    'trailing-trade-manual-orders'
  );
  try {
    result = await manualOrdersCollection.dropIndex(
      `trailing-trade-manual-orders-key-idx`
    );
  } catch (e) {
    logger.warn(
      `Index 'trailing-trade-manual-orders-key-idx' is not found. It's ok.`
    );
  }

  result = await manualOrdersCollection.createIndex(
    { symbol: 1, orderId: 1 },
    { name: `trailing-trade-manual-orders-key-idx` }
  );

  logger.info({ result }, 'Created index for manual orders');

  keys = await cache.keys('trailing-trade-manual-order*');
  if (keys && keys.length > 0) {
    for (const key of keys) {
      const manualOrders = await cache.hgetall(key);

      if (_.isEmpty(manualOrders) === false) {
        for (const rawOrder of Object.values(manualOrders)) {
          const order = JSON.parse(rawOrder);
          await mongo.insertOne(logger, 'trailing-trade-manual-orders', {
            symbol: order.symbol,
            orderId: order.orderId,
            order
          });
        }
      }

      await cache.del(key);
    }
  }

  logger.info({ result }, 'Finish migration');
};

module.exports.down = next => {
  next();
};