study-univ3-on-hardhat/scripts/path/level2Generate.ts

import logger from "../../utils/logger";
import history from "../interface/history";
import SwapPath from "../interface/swapPath";

class Level2Generate {
  maxMemoryOfByte: number = 0
  maxMemoryChanged: boolean = true
  lpSumObj: any = {}

  buildLpObj (lpList: any) {
    for (const lp of lpList) {
      lp.dataObj = JSON.parse(lp.data)
    }
  }

  showMemory (mainInfo: string) {
    const memoryUsage = process.memoryUsage()
    if (this.maxMemoryOfByte < memoryUsage.rss) {
      this.maxMemoryOfByte = memoryUsage.rss
      this.maxMemoryChanged = true
    }

    if (this.maxMemoryChanged) {
      logger.debug(`${mainInfo}  ${this.format(memoryUsage.rss)}/${this.format(this.maxMemoryOfByte)}`)

      this.maxMemoryChanged = false
    }
  }

  format (bytes: any) {
    return (bytes / 1024 / 1024).toFixed(2) + ' MB';
  }

  buildSumObjGroupBySum2(allTypeLpList: any) {
    // 根据sum值进行池子分类
    logger.debug(`generate sum group by sum...${allTypeLpList.length}`)
    for (const lpDbObj of allTypeLpList) {
      if (this.lpSumObj[lpDbObj.dataObj.sum2]) {
        this.lpSumObj[lpDbObj.dataObj.sum2].push(lpDbObj.dataObj)
      } else {
        this.lpSumObj[lpDbObj.dataObj.sum2] = [lpDbObj.dataObj]
      }
    }
  }

  async generatePathAndUpdate() {
    // 组合所有路径并更新
    for (const sum2 of Object.keys(this.lpSumObj)) {
      const lpList = this.lpSumObj[sum2]

      // 组成路径
      const pathList: any = []
      for (let i = 0; i < lpList.length; i++) {
        const iLp = lpList[i]
        for (let j = i + 1; j < lpList.length; j++) {
          const jLp = lpList[j]

          const path = [iLp, jLp]
          pathList.push(path)
        }
      }

      await SwapPath.appendOrUpdate(sum2, '2', pathList)
    }
  }

  async run() {
    logger.debug('Level2 maintenance start.')
    while (true) {
      if (this.maxMemoryChanged) {
        this.showMemory('Level2 a loop...')
      }
      this.maxMemoryChanged = false

      this.lpSumObj = {}

      const topLpPullRst = await history.findByBlock('topLp')
      if (!topLpPullRst.state) continue
      const topLpList = topLpPullRst.data
      const normalLpPullRst = await history.findByBlock('normalLp')
      if (!normalLpPullRst.state) continue
      const normalLpList = normalLpPullRst.data

      const allTypeLpList: any = topLpList.concat(normalLpList)

      // 将原始data构造成dataObj
      this.buildLpObj(allTypeLpList)
      // 根据Sum构造Lp索引
      this.buildSumObjGroupBySum2(allTypeLpList)
      // 生成路径并更新到库
      await this.generatePathAndUpdate()
    }
  }
}

async function main() {
  await new Level2Generate().run()
}

main().catch((error) => {
  console.error(error);
  process.exitCode = 1;
})