fix governance vote inserting

Author: Fraccaman

chain/src/repository/gov.rs

@@ -1,4 +1,4 @@
-use std::collections::HashSet;
+use std::collections::{HashSet};
 
 use anyhow::Context;
 use diesel::upsert::excluded;

chain/src/services/namada.rs

@@ -605,30 +605,55 @@ pub async fn query_all_votes(
     client: &HttpClient,
     proposals_ids: Vec<u64>,
 ) -> anyhow::Result<HashSet<GovernanceVote>> {
-    let votes: Vec<HashSet<GovernanceVote>> =
-        futures::stream::iter(proposals_ids)
-            .filter_map(|proposal_id| async move {
-                let votes = rpc::query_proposal_votes(client, proposal_id)
-                    .await
-                    .ok()?;
+    let votes = futures::stream::iter(proposals_ids)
+        .filter_map(|proposal_id| async move {
+            let votes =
+                rpc::query_proposal_votes(client, proposal_id).await.ok()?;
+
+            let votes = votes
+                .into_iter()
+                .map(|vote| GovernanceVote {
+                    proposal_id,
+                    vote: ProposalVoteKind::from(vote.data),
+                    address: Id::from(vote.delegator),
+                })
+                .collect::<HashSet<_>>();
+
+            Some(votes)
+        })
+        .map(futures::future::ready)
+        .buffer_unordered(32)
+        .collect::<Vec<_>>()
+        .await;
 
-                let votes = votes
-                    .into_iter()
-                    .map(|vote| GovernanceVote {
-                        proposal_id,
-                        vote: ProposalVoteKind::from(vote.data),
-                        address: Id::from(vote.delegator),
-                    })
-                    .collect::<HashSet<_>>();
+    let mut voter_count: HashMap<(Id, u64), u64> = HashMap::new();
+    for vote in votes.iter().flatten() {
+        *voter_count
+            .entry((vote.address.clone(), vote.proposal_id))
+            .or_insert(0) += 1;
+    }
 
-                Some(votes)
+    let mut seen_voters = HashSet::new();
+    anyhow::Ok(
+        votes
+            .iter()
+            .flatten().filter(|&vote| {
+                seen_voters.insert((vote.address.clone(), vote.proposal_id))
+            }).cloned()
+            .map(|mut vote| {
+                if let Some(count) =
+                    voter_count.get(&(vote.address.clone(), vote.proposal_id))
+                {
+                    if *count > 1_u64 {
+                        vote.vote = ProposalVoteKind::Unknown;
+                    }
+                    vote
+                } else {
+                    vote
+                }
             })
-            .map(futures::future::ready)
-            .buffer_unordered(32)
-            .collect::<Vec<_>>()
-            .await;
-
-    anyhow::Ok(votes.iter().flatten().cloned().collect())
+            .collect(),
+    )
 }
 
 pub async fn get_validator_set_at_epoch(

orm/migrations/2025-01-13-115313_vote_unknown/down.sql

@@ -0,0 +1 @@
+-- This file should undo anything in `up.sql`

orm/migrations/2025-01-13-115313_vote_unknown/up.sql

@@ -0,0 +1,2 @@
+-- Your SQL goes here
+ALTER TYPE VOTE_KIND ADD VALUE 'unknown';

orm/src/governance_votes.rs

@@ -10,6 +10,7 @@ pub enum GovernanceVoteKindDb {
     Nay,
     Yay,
     Abstain,
+    Unknown,
 }
 
 impl From<ProposalVoteKind> for GovernanceVoteKindDb {
@@ -18,6 +19,7 @@ impl From<ProposalVoteKind> for GovernanceVoteKindDb {
             ProposalVoteKind::Nay => Self::Nay,
             ProposalVoteKind::Yay => Self::Yay,
             ProposalVoteKind::Abstain => Self::Abstain,
+            ProposalVoteKind::Unknown => Self::Unknown,
         }
     }
 }

shared/src/vote.rs

@@ -9,6 +9,7 @@ pub enum ProposalVoteKind {
     Nay,
     Yay,
     Abstain,
+    Unknown,
 }
 
 impl From<ProposalVote> for ProposalVoteKind {

webserver/src/response/governance.rs

@@ -58,6 +58,7 @@ pub enum VoteType {
     Yay,
     Nay,
     Abstain,
+    Unknown,
 }
 
 #[derive(Clone, Debug, Serialize, Deserialize)]
@@ -219,6 +220,7 @@ impl From<GovernanceProposalVoteDb> for ProposalVote {
                 GovernanceVoteKindDb::Nay => VoteType::Nay,
                 GovernanceVoteKindDb::Yay => VoteType::Yay,
                 GovernanceVoteKindDb::Abstain => VoteType::Abstain,
+                GovernanceVoteKindDb::Unknown => VoteType::Unknown,
             },
             voter_address: value.voter_address,
         }