Merge pull request #53 from zkLinkProtocol/feat-lppufpoint

fixed display lp puffer points.
zkLinkProtocol · May 17, 2024 · fae976c · fae976c
2 parents af07620 + e48f3cf
commit fae976c
Show file tree

Hide file tree

Showing 4 changed files with 14 additions and 5 deletions.
diff --git a/src/entities/blockAddressPointOfLp.entity.ts b/src/entities/blockAddressPointOfLp.entity.ts
@@ -4,7 +4,7 @@ import { bigIntNumberTransformer } from "../transformers/bigIntNumber.transforme
 import { hexTransformer } from "../transformers/hex.transformer";
 
 @Entity({ name: "blockAddressPointOfLp" })
-export class BlockAddressPointOfLp extends BaseEntity {
+export class BlockAddressPointOfLp {
   @PrimaryColumn({ type: "bigint", transformer: bigIntNumberTransformer })
   public readonly blockNumber: number;
 

diff --git a/src/nova/nova.controller.ts b/src/nova/nova.controller.ts
@@ -186,11 +186,13 @@ export class NovaController {
       return NOT_FOUND_EXCEPTION;
     }
     // Get real puffer points in layerbank.
-    let points: number;
+    let points: number = 0;
     try {
       const tokenData =
-        await this.pufferPointsSercie.getPointsData(tokenAddress);
-      points = tokenData.items[0]?.realPoints ?? 0;
+        this.pufferPointsSercie.getPoolPufferPoints(tokenAddress);
+      if (tokenData) {
+        points = tokenData.realPoints ?? 0;
+      }
     } catch (err) {
       this.logger.error(
         "Get puffer real points in layerbank failed",

diff --git a/src/nova/nova.service.ts b/src/nova/nova.service.ts
@@ -51,7 +51,6 @@ export class NovaService {
         addresses,
         project,
       );
-
     if (Array.isArray(points) && totalPoints) {
       return this.getPointData(points, totalPoints);
     } else {

diff --git a/src/puffer/puffPoints.service.ts b/src/puffer/puffPoints.service.ts
@@ -240,6 +240,14 @@ export class PuffPointsService extends Worker {
     return await this.projectGraphService.getPoints(this.projectName, address);
   }
 
+  public getPoolPufferPoints(poolAddress: string): PufferPointItem {
+    const _poolAddress = poolAddress.toLowerCase();
+    const result = this.localPoints.filter(
+      (item) => item.address === _poolAddress,
+    );
+    return result[0];
+  }
+
   // return real totalPoints
   public async getRealPointsData(): Promise<number> {
     const realData = await fetch(