indexer: update maxCensusSize when SET_PROCESS_CENSUS

Signed-off-by: p4u <[email protected]>
vocdoni · Jun 11, 2024 · c12e60a · c12e60a
1 parent dc27947
commit c12e60a
Show file tree

Hide file tree

Showing 9 changed files with 28 additions and 23 deletions.
diff --git a/vochain/indexer/db/processes.sql.go b/vochain/indexer/db/processes.sql.go
diff --git a/vochain/indexer/indexer.go b/vochain/indexer/indexer.go
@@ -697,7 +697,7 @@ func (idx *Indexer) OnTransferTokens(tx *vochaintx.TokenTransfer) {
 
 // OnCensusUpdate adds the process to blockUpdateProcs in order to update the census.
 // This function call is triggered by the SET_PROCESS_CENSUS tx.
-func (idx *Indexer) OnCensusUpdate(pid, _ []byte, _ string) {
+func (idx *Indexer) OnCensusUpdate(pid, _ []byte, _ string, _ uint64) {
 	idx.blockMu.Lock()
 	defer idx.blockMu.Unlock()
 	idx.blockUpdateProcs[string(pid)] = true

diff --git a/vochain/indexer/process.go b/vochain/indexer/process.go
@@ -202,13 +202,14 @@ func (idx *Indexer) updateProcess(ctx context.Context, queries *indexerdb.Querie
 
 	// Update the process in the indexer database
 	if _, err := queries.UpdateProcessFromState(ctx, indexerdb.UpdateProcessFromStateParams{
-		ID:          pid,
-		CensusRoot:  nonNullBytes(p.CensusRoot),
-		CensusUri:   p.GetCensusURI(),
-		PrivateKeys: indexertypes.EncodeJSON(p.EncryptionPrivateKeys),
-		PublicKeys:  indexertypes.EncodeJSON(p.EncryptionPublicKeys),
-		Metadata:    p.GetMetadata(),
-		Status:      int64(p.Status),
+		ID:            pid,
+		CensusRoot:    nonNullBytes(p.CensusRoot),
+		CensusUri:     p.GetCensusURI(),
+		PrivateKeys:   indexertypes.EncodeJSON(p.EncryptionPrivateKeys),
+		PublicKeys:    indexertypes.EncodeJSON(p.EncryptionPublicKeys),
+		Metadata:      p.GetMetadata(),
+		Status:        int64(p.Status),
+		MaxCensusSize: int64(p.GetMaxCensusSize()),
 	}); err != nil {
 		return err
 	}

diff --git a/vochain/indexer/queries/processes.sql b/vochain/indexer/queries/processes.sql
@@ -51,7 +51,8 @@ SET census_root         = sqlc.arg(census_root),
 	private_keys        = sqlc.arg(private_keys),
 	public_keys         = sqlc.arg(public_keys),
 	metadata            = sqlc.arg(metadata),
-	status              = sqlc.arg(status)
+	status              = sqlc.arg(status),
+	max_census_size	 	= sqlc.arg(max_census_size)
 WHERE id = sqlc.arg(id);
 
 -- name: GetProcessStatus :one

diff --git a/vochain/keykeeper/keykeeper.go b/vochain/keykeeper/keykeeper.go
@@ -272,7 +272,7 @@ func (*KeyKeeper) OnNewTx(_ *vochaintx.Tx, _ uint32, _ int32) {}
 func (*KeyKeeper) OnBeginBlock(_ state.BeginBlock) {}
 
 // OnCensusUpdate is not used by the KeyKeeper
-func (*KeyKeeper) OnCensusUpdate(_, _ []byte, _ string) {}
+func (*KeyKeeper) OnCensusUpdate(_, _ []byte, _ string, _ uint64) {}
 
 // OnCancel does nothing
 func (k *KeyKeeper) OnCancel(_ []byte, _ int32) {}
diff --git a/vochain/offchaindatahandler/offchaindatahandler.go b/vochain/offchaindatahandler/offchaindatahandler.go
@@ -122,7 +122,7 @@ func (d *OffChainDataHandler) OnProcess(p *models.Process, _ int32) {
 }
 
 // OnCensusUpdate is triggered when the census is updated during an election.
-func (d *OffChainDataHandler) OnCensusUpdate(pid, censusRoot []byte, censusURI string) {
+func (d *OffChainDataHandler) OnCensusUpdate(pid, censusRoot []byte, censusURI string, _ uint64) {
 	if d.importOnlyNew && !d.isSynced {
 		return
 	}

diff --git a/vochain/state/eventlistener.go b/vochain/state/eventlistener.go
@@ -29,7 +29,7 @@ type EventListener interface {
 	OnSetAccount(addr []byte, account *Account)
 	OnTransferTokens(tx *vochaintx.TokenTransfer)
 	OnSpendTokens(addr []byte, txType models.TxType, cost uint64, reference string)
-	OnCensusUpdate(pid, censusRoot []byte, censusURI string)
+	OnCensusUpdate(pid, censusRoot []byte, censusURI string, censusSize uint64)
 	Commit(height uint32) (err error)
 	OnBeginBlock(BeginBlock)
 	Rollback()

diff --git a/vochain/state/process.go b/vochain/state/process.go
@@ -363,7 +363,7 @@ func (v *State) SetProcessCensus(pid, censusRoot []byte, censusURI string, censu
 			return err
 		}
 		for _, l := range v.eventListeners {
-			l.OnCensusUpdate(process.ProcessId, process.CensusRoot, censusURI)
+			l.OnCensusUpdate(process.ProcessId, process.CensusRoot, censusURI, censusSize)
 		}
 	}
 

diff --git a/vochain/state/state_test.go b/vochain/state/state_test.go
@@ -190,7 +190,7 @@ func (*Listener) OnCancel(_ []byte, _ int32)
 func (*Listener) OnProcessKeys(_ []byte, _ string, _ int32)                       {}
 func (*Listener) OnRevealKeys(_ []byte, _ string, _ int32)                        {}
 func (*Listener) OnProcessResults(_ []byte, _ *models.ProcessResult, _ int32)     {}
-func (*Listener) OnCensusUpdate(_, _ []byte, _ string)                            {}
+func (*Listener) OnCensusUpdate(_, _ []byte, _ string, _ uint64)                  {}
 func (*Listener) OnSetAccount(_ []byte, _ *Account)                               {}
 func (*Listener) OnTransferTokens(_ *vochaintx.TokenTransfer)                     {}
 func (*Listener) OnSpendTokens(_ []byte, _ models.TxType, _ uint64, _ string)     {}