fix: get or create delegate when creating vote

deployer has voted without delegating for example
VenusProtocol · Feb 16, 2024 · f3259de · f3259de
1 parent 1bed22c
commit f3259de
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/subgraphs/venus-governance/src/operations/create.ts b/subgraphs/venus-governance/src/operations/create.ts
@@ -5,7 +5,8 @@ import { VoteCast as VoteCastBravo } from '../../generated/GovernorBravoDelegate
 import { Proposal, Vote } from '../../generated/schema';
 import { ABSTAIN, AGAINST, BIGINT_ONE, FOR, NORMAL } from '../constants';
 import { getVoteId } from '../utilities/ids';
-import { getDelegate, getGovernanceEntity, getProposal } from './get';
+import { getGovernanceEntity, getProposal } from './get';
+import { getOrCreateDelegate } from './getOrCreate';
 
 export function createProposal<E>(event: E): Proposal {
   const id = event.params.id.toString();
@@ -51,7 +52,7 @@ export function createVoteAlpha(event: VoteCastAlpha): Vote {
 
 export function createVoteBravo(event: VoteCastBravo): Vote {
   const proposal = getProposal(event.params.proposalId.toString());
-  const voter = getDelegate(event.params.voter);
+  const voter = getOrCreateDelegate(event.params.voter).entity;
   const id = getVoteId(event.params.voter, event.params.proposalId);
   const vote = new Vote(id);
   vote.proposal = proposal.id;