update proposal contract.

Sin7Y · Jan 30, 2024 · 31cce04 · 31cce04
1 parent 4d1bf5d
commit 31cce04
Showing 1 changed file with 7 additions and 2 deletions.
diff --git a/examples/source/quickstart/proposal.ola b/examples/source/quickstart/proposal.ola
@@ -45,13 +45,18 @@ contract ProposalContract {
             assert(proposals[i] != _contentHash);
         }
 
+        u32 weight = 1;
+        if (oldProposal.votingType == VotingType.Weighted) {
+            weight = _weight;
+        }
+
         // check if deadline is in the future
         Proposal proposal = proposalByHash[_contentHash];
         assert(block_timestamp() < proposal.deadline, "Voting period has ended");
         if (_support) {
-            proposal.totalSupport += _weight;
+            proposal.totalSupport += weight;
         } else {
-            proposal.totalAgainst += _weight;
+            proposal.totalAgainst += weight;
         }
         proposalByHash[_contentHash] = proposal;
         proposalsByVoter[origin_address()].push(_contentHash);