feat: compress assetIds

shapeshift · Dec 20, 2024 · 1a79883 · 1a79883
1 parent 4fbd84f
commit 1a79883
Show file tree

Hide file tree

Showing 3 changed files with 51 additions and 4 deletions.
diff --git a/packages/utils/src/assetData/decodeAssetData.ts b/packages/utils/src/assetData/decodeAssetData.ts
@@ -8,8 +8,22 @@ import { FIELDS } from './constants'
 import { getBaseAsset } from './getBaseAsset'
 import type { EncodedAssetData, FieldToType } from './types'
 
+const decodeAssetId = (encodedAssetId: string, assetIdPrefixes: string[]) => {
+  // Performance sensitive. `lastIndexOf` + `substring` faster than `split`
+  const colonIndex = encodedAssetId.lastIndexOf(':')
+  const prefixIdx = Number(encodedAssetId.substring(0, colonIndex))
+  const assetReference = encodedAssetId.substring(colonIndex + 1)
+
+  // Performance sensitive. String concatenation faster than template literal
+  return assetIdPrefixes[prefixIdx] + ':' + assetReference
+}
+
 export const decodeAssetData = (encodedAssetData: EncodedAssetData) => {
-  const { sortedAssetIds, encodedAssets } = encodedAssetData
+  const { assetIdPrefixes, encodedAssetIds, encodedAssets } = encodedAssetData
+
+  const sortedAssetIds = encodedAssetIds.map(encodedAssetId =>
+    decodeAssetId(encodedAssetId, assetIdPrefixes),
+  )
 
   const assetData = encodedAssets.reduce<Record<AssetId, Asset>>((acc, encodedAsset, idx) => {
     const assetId = sortedAssetIds[idx]

diff --git a/packages/utils/src/assetData/encodeAssetData.ts b/packages/utils/src/assetData/encodeAssetData.ts
@@ -5,6 +5,35 @@ import { assertUnreachable } from '../assertUnreachable'
 import { FIELDS } from './constants'
 import type { EncodedAsset, EncodedAssetData, Field, FieldToType } from './types'
 
+const encodeAssetIds = (sortedAssetIds: AssetId[]) => {
+  const assetIdPrefixes: string[] = []
+  const encodedAssetIds = []
+  for (const assetId of sortedAssetIds) {
+    // `indexOf` + `substring` faster than `split`
+    const colonIndex = assetId.lastIndexOf(':')
+    const prefix = assetId.substring(0, colonIndex)
+    const assetReference = assetId.substring(colonIndex + 1)
+
+    const prefixIdx = (() => {
+      const prefixIdx = assetIdPrefixes.indexOf(prefix)
+
+      if (prefixIdx !== -1) {
+        return prefixIdx
+      } else {
+        assetIdPrefixes.push(prefix)
+        return assetIdPrefixes.length - 1
+      }
+    })()
+
+    encodedAssetIds.push(`${prefixIdx}:${assetReference}`)
+  }
+
+  return {
+    assetIdPrefixes,
+    encodedAssetIds,
+  }
+}
+
 const encodeField = <F extends Field>(
   field: F,
   assetIdToAssetIdx: Record<AssetId, number>,
@@ -38,21 +67,24 @@ export const encodeAssetData = (
   sortedAssetIds: AssetId[],
   assetsById: AssetsById,
 ): EncodedAssetData => {
+  const { assetIdPrefixes, encodedAssetIds } = encodeAssetIds(sortedAssetIds)
+
   const assetIdToAssetIdx = sortedAssetIds.reduce<Record<AssetId, number>>((acc, val, idx) => {
     acc[val] = idx
     return acc
   }, {})
 
   const encodedAssets: EncodedAsset[] = []
-  for (let i = 0; i < sortedAssetIds.length; i++) {
+  for (let i = 0; i < encodedAssetIds.length; i++) {
     const assetId = sortedAssetIds[i]
     const asset = assetsById[assetId]
     const record = FIELDS.map(field => encodeField(field, assetIdToAssetIdx, asset)) as EncodedAsset
     encodedAssets.push(record)
   }
 
   return {
-    sortedAssetIds,
+    assetIdPrefixes,
+    encodedAssetIds,
     encodedAssets,
   }
 }
diff --git a/packages/utils/src/assetData/types.ts b/packages/utils/src/assetData/types.ts
@@ -27,6 +27,7 @@ export type EncodedAsset = [
 ]
 
 export type EncodedAssetData = {
-  sortedAssetIds: AssetId[]
+  assetIdPrefixes: string[]
+  encodedAssetIds: string[]
   encodedAssets: EncodedAsset[]
 }