Merge pull request #180 from helix-bridge/xiaoch05-optimise-lnv3

optimise lnv3

apollo/prisma/migrations/20241014083428_init/migration.sql

@@ -0,0 +1,7 @@
+-- CreateTable
+CREATE TABLE "ScanCursor" (
+    "id" TEXT NOT NULL,
+    "cursor" BIGINT NOT NULL,
+
+    CONSTRAINT "ScanCursor_pkey" PRIMARY KEY ("id")
+);

apollo/prisma/schema.prisma

@@ -46,6 +46,11 @@ model HistoryRecord {
   extData String?
 }
 
+model ScanCursor {
+  id String @id
+  cursor BigInt
+}
+
 model LnBridgeRelayInfo {
   id String @id
   version String

apollo/src/aggregation/aggregation.service.ts

@@ -28,6 +28,25 @@ export class AggregationService extends PrismaClient implements OnModuleInit {
     });
   }
 
+  async writeCursor(id: string, cursor: bigint) {
+    return await this.scanCursor.upsert({
+      where: { id: id },
+      update: { cursor: cursor },
+      create: { id: id, cursor: cursor },
+    });
+  }
+
+  async readCursor(id: string): Promise<bigint> {
+    const cursor = await this.scanCursor.findUnique({
+      where: { id: id },
+    });
+    if (cursor) {
+      return cursor.cursor;
+    } else {
+      return BigInt(0);
+    }
+  }
+
   async createHistoryRecord(data: Prisma.HistoryRecordCreateInput): Promise<HistoryRecord> {
     return this.historyRecord.create({
       data,
@@ -51,6 +70,19 @@ export class AggregationService extends PrismaClient implements OnModuleInit {
     });
   }
 
+  async saveHistoryRecord(params: {
+    where: Prisma.HistoryRecordWhereUniqueInput;
+    dataCreate: Prisma.HistoryRecordCreateInput;
+    dataUpdate: Prisma.HistoryRecordUpdateInput;
+  }): Promise<HistoryRecord> {
+    const { where, dataCreate, dataUpdate } = params;
+    return await this.historyRecord.upsert({
+      where: where,
+      update: dataUpdate,
+      create: dataCreate,
+    });
+  }
+
   async queryLnBridgeRelayInfoFirst(
     lnBridgeRelayInfoWhereInput: Prisma.LnBridgeRelayInfoWhereInput,
     orderBy?: Prisma.Enumerable<Prisma.LnBridgeRelayInfoOrderByWithRelationAndSearchRelevanceInput>