fix: Weighted votes

allinbits · Mar 8, 2024 · f5997f6 · f5997f6
1 parent dda0c20
commit f5997f6
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 3 deletions.
diff --git a/database/gov.go b/database/gov.go
@@ -288,16 +288,16 @@ WHERE proposal_deposit.height <= excluded.height`
 // SaveVote allows to save for the given height and the message vote
 func (db *Db) SaveVote(vote types.Vote) error {
 	query := `
-INSERT INTO proposal_vote (proposal_id, voter_address, option, timestamp, height) 
-VALUES ($1, $2, $3, $4, $5)`
+INSERT INTO proposal_vote (proposal_id, voter_address, option, weight, timestamp, height) 
+VALUES ($1, $2, $3, $4, $5, $6)`
 
 	// Store the voter account
 	err := db.SaveAccounts([]types.Account{types.NewAccount(vote.Voter)})
 	if err != nil {
 		return fmt.Errorf("error while storing voter account: %s", err)
 	}
 
-	_, err = db.SQL.Exec(query, vote.ProposalID, vote.Voter, vote.Option.String(), vote.Timestamp, vote.Height)
+	_, err = db.SQL.Exec(query, vote.ProposalID, vote.Voter, vote.Option.String(), vote.Weight, vote.Timestamp, vote.Height)
 	if err != nil {
 		return fmt.Errorf("error while storing vote: %s", err)
 	}

diff --git a/database/schema/08-gov.sql b/database/schema/08-gov.sql
@@ -43,6 +43,7 @@ CREATE TABLE proposal_vote
     proposal_id   INTEGER NOT NULL REFERENCES proposal (id),
     voter_address TEXT    NOT NULL REFERENCES account (address),
     option        TEXT    NOT NULL,
+    weight        TEXT    NOT NULL,
     timestamp     TIMESTAMP,
     height        BIGINT  NOT NULL
 );

diff --git a/modules/gov/handle_msg.go b/modules/gov/handle_msg.go
@@ -32,6 +32,8 @@ func (m *Module) HandleMsg(index int, msg sdk.Msg, tx *juno.Tx) error {
 		return m.handleMsgDeposit(tx, cosmosMsg)
 	case *govtypes.MsgVote:
 		return m.handleMsgVote(tx, cosmosMsg)
+	case *govtypes.MsgVoteWeighted:
+		return m.handleMsgVoteWeighted(tx, cosmosMsg)
 	}
 
 	return nil