Move saveOnlyGossipData and stopGossipTxData flags to the config

shardeum · Jun 24, 2024 · b80f544 · b80f544
1 parent c241ead
commit b80f544
Show file tree

Hide file tree

Showing 4 changed files with 9 additions and 10 deletions.
diff --git a/src/Config.ts b/src/Config.ts
@@ -61,6 +61,10 @@ export interface Config {
   randomGossipArchiversCount: 2 // Number of random archivers to gossip to
   subscribeToMoreConsensors: boolean // To subscribe to more consensors when the number of active archivers is less than 4
   extraConsensorsToSubscribe: 1 // Number of extra consensors to subscribe to
+  // For debugging gossip data, set this to true. This will save only the gossip data received from the gossip archivers.
+  saveOnlyGossipData: boolean
+  // For debugging purpose, set this to true to stop gossiping tx data
+  stopGossipTxData: boolean
 }
 
 let config: Config = {
@@ -120,6 +124,8 @@ let config: Config = {
   randomGossipArchiversCount: 2,
   subscribeToMoreConsensors: true,
   extraConsensorsToSubscribe: 1,
+  saveOnlyGossipData: false,
+  stopGossipTxData: false,
 }
 // Override default config params from config file, env vars, and cli args
 export async function overrideDefaultConfig(file: string): Promise<void> {

diff --git a/src/Data/Collector.ts b/src/Data/Collector.ts
@@ -47,9 +47,6 @@ interface MissingTx {
   senders: string[]
 }
 
-// For debugging gossip data, set this to true. This will save only the gossip data received from the gossip archivers.
-export const saveOnlyGossipData = false
-
 type GET_TX_RECEIPT_RESPONSE = {
   success: boolean
   receipt?: Receipt.ArchiverReceipt | Receipt.AppliedReceipt2

diff --git a/src/Data/Data.ts b/src/Data/Data.ts
@@ -27,7 +27,6 @@ import {
   storeAccountData,
   storingAccountData,
   storeOriginalTxData,
-  saveOnlyGossipData,
 } from './Collector'
 import * as CycleDB from '../dbstore/cycles'
 import * as ReceiptDB from '../dbstore/receipts'
@@ -258,7 +257,7 @@ export function initSocketClient(node: NodeList.ConsensusNodeInfo): void {
         storeOriginalTxData(
           newData.responses.ORIGINAL_TX_DATA,
           sender.nodeInfo.ip + ':' + sender.nodeInfo.port,
-          saveOnlyGossipData
+          config.saveOnlyGossipData
         )
       }
       if (newData.responses && newData.responses.RECEIPT) {
@@ -274,7 +273,7 @@ export function initSocketClient(node: NodeList.ConsensusNodeInfo): void {
           newData.responses.RECEIPT,
           sender.nodeInfo.ip + ':' + sender.nodeInfo.port,
           true,
-          saveOnlyGossipData
+          config.saveOnlyGossipData
         )
       }
       if (newData.responses && newData.responses.CYCLE) {

diff --git a/src/Data/GossipData.ts b/src/Data/GossipData.ts
@@ -24,9 +24,6 @@ export interface GossipData {
   sign: Signature
 }
 
-// For debugging purpose, set this to true to stop gossiping tx data
-const stopGossipTxData = false
-
 // List of archivers that are not adjacent to the current archiver
 const remainingArchivers = []
 
@@ -69,7 +66,7 @@ export async function sendDataToAdjacentArchivers(
   dataType: DataType,
   data: GossipData['data']
 ): Promise<void> {
-  if (stopGossipTxData) return
+  if (config.stopGossipTxData) return
   if (State.otherArchivers.length === 0) return
   const gossipPayload = {
     dataType,