fix loans count (#16)

impactMarket · Jun 1, 2023 · 9179464 · 9179464
1 parent 73fac4d
commit 9179464
Show file tree

Hide file tree

Showing 3 changed files with 14 additions and 1 deletion.
diff --git a/generated/schema.ts b/generated/schema.ts
@@ -100,6 +100,15 @@ export class Borrower extends Entity {
   set loans(value: Array<string>) {
     this.set("loans", Value.fromStringArray(value));
   }
+
+  get loansCount(): i32 {
+    let value = this.get("loansCount");
+    return value!.toI32();
+  }
+
+  set loansCount(value: i32) {
+    this.set("loansCount", Value.fromI32(value));
+  }
 }
 
 export class Loan extends Entity {

diff --git a/schema.graphql b/schema.graphql
@@ -19,6 +19,8 @@ type Borrower @entity {
   id: ID!
   # borrower loans
   loans: [Loan!]! @derivedFrom(field: "borrower")
+  # number of loans
+  loansCount: Int!
 }
 
 """

diff --git a/src/mappings/micro-credit.ts b/src/mappings/micro-credit.ts
@@ -121,6 +121,7 @@ export function handleLoanAdded(event: LoanAdded): void {
     if (!borrower) {
         // create borrower entity
         borrower = new Borrower(event.params.userAddress.toHex());
+        borrower.loansCount = 0;
     }
 
     loan.borrower = event.params.userAddress.toHex();
@@ -216,7 +217,8 @@ export function handleLoanClaimed(event: LoanClaimed): void {
     // update loan entity data
     loan.claimed = event.block.timestamp.toI32();
 
-    if (borrower.loans.length === 1) {
+    borrower.loansCount += 1;
+    if (borrower.loansCount === 1) {
         loanManager.borrowers += 1;
     }
     loanManager.loans += 1;