Merge pull request #1138 from multiversx/MEX-351

MEX-351: governance contract shard field
multiversx · Aug 10, 2023 · c261ddf · c261ddf
2 parents b0f0160 + be84fad
commit c261ddf
Show file tree

Hide file tree

Showing 3 changed files with 17 additions and 0 deletions.
diff --git a/src/modules/governance/models/governance.contract.model.ts b/src/modules/governance/models/governance.contract.model.ts
@@ -6,6 +6,8 @@ import { EsdtToken } from '../../tokens/models/esdtToken.model';
 export class GovernanceTokenSnapshotContract {
     @Field()
     address: string;
+    @Field(() => Int)
+    shard: number;
     @Field()
     minFeeForPropose: string;
     @Field()

diff --git a/src/modules/governance/resolvers/governance.contract.resolver.ts b/src/modules/governance/resolvers/governance.contract.resolver.ts
@@ -13,6 +13,11 @@ export class GovernanceTokenSnapshotContractResolver {
     ) {
     }
 
+    @ResolveField()
+    async shard(@Parent() contract: GovernanceTokenSnapshotContract): Promise<number> {
+        return this.governanceAbi.getAddressShardID(contract.address);
+    }
+
     @ResolveField()
     async minFeeForPropose(@Parent() contract: GovernanceTokenSnapshotContract): Promise<string> {
         return this.governanceAbi.minFeeForPropose(contract.address);

diff --git a/src/modules/governance/services/governance.abi.service.ts b/src/modules/governance/services/governance.abi.service.ts
@@ -29,6 +29,16 @@ export class GovernanceTokenSnapshotAbiService
         super(mxProxy);
     }
 
+    @ErrorLoggerAsync({ className: GovernanceTokenSnapshotAbiService.name })
+    @GetOrSetCache({
+        baseKey: 'governance',
+        remoteTtl: CacheTtlInfo.ContractState.remoteTtl,
+        localTtl: CacheTtlInfo.ContractState.localTtl,
+    })
+    async getAddressShardID(scAddress: string): Promise<number> {
+        return await this.mxProxy.getAddressShardID(scAddress);
+    }
+
     @ErrorLoggerAsync({ className: GovernanceTokenSnapshotAbiService.name })
     @GetOrSetCache({
         baseKey: 'governance',