feat: Cherry-Pick (0.56): Define throttle for GetBalance queries (#16338

) Signed-off-by: Valentin Tronkov <[email protected]>
hashgraph · Oct 31, 2024 · 8978fa6 · 8978fa6
1 parent 76c54fc
commit 8978fa6
Show file tree

Hide file tree

Showing 5 changed files with 61 additions and 7 deletions.
diff --git a/hedera-node/configuration/mainnet/upgrade/throttles.json b/hedera-node/configuration/mainnet/upgrade/throttles.json
@@ -181,11 +181,24 @@
           "opsPerSec": 0,
           "milliOpsPerSec": 1000000000,
           "operations": [
-            "CryptoGetAccountBalance",
             "TransactionGetReceipt"
           ]
         }
       ]
+    },
+    {
+      "burstPeriod": 0,
+      "burstPeriodMs": 1000,
+      "name": "BalanceQueryLimits",
+      "throttleGroups": [
+        {
+          "opsPerSec": 0,
+          "milliOpsPerSec": 1000000,
+          "operations": [
+            "CryptoGetAccountBalance"
+          ]
+        }
+      ]
     }
   ]
 }
diff --git a/hedera-node/configuration/previewnet/upgrade/throttles.json b/hedera-node/configuration/previewnet/upgrade/throttles.json
@@ -181,11 +181,24 @@
           "opsPerSec": 0,
           "milliOpsPerSec": 1000000000,
           "operations": [
-            "CryptoGetAccountBalance",
             "TransactionGetReceipt"
           ]
         }
       ]
+    },
+    {
+      "burstPeriod": 0,
+      "burstPeriodMs": 1000,
+      "name": "BalanceQueryLimits",
+      "throttleGroups": [
+        {
+          "opsPerSec": 0,
+          "milliOpsPerSec": 1000000,
+          "operations": [
+            "CryptoGetAccountBalance"
+          ]
+        }
+      ]
     }
   ]
-}
+}
diff --git a/hedera-node/configuration/testnet/upgrade/throttles.json b/hedera-node/configuration/testnet/upgrade/throttles.json
@@ -181,11 +181,24 @@
           "opsPerSec": 0,
           "milliOpsPerSec": 1000000000,
           "operations": [
-            "CryptoGetAccountBalance",
             "TransactionGetReceipt"
           ]
         }
       ]
+    },
+    {
+      "burstPeriod": 0,
+      "burstPeriodMs": 1000,
+      "name": "BalanceQueryLimits",
+      "throttleGroups": [
+        {
+          "opsPerSec": 0,
+          "milliOpsPerSec": 1000000,
+          "operations": [
+            "CryptoGetAccountBalance"
+          ]
+        }
+      ]
     }
   ]
 }
diff --git a/hedera-node/hedera-config/src/main/java/com/hedera/node/config/data/StatsConfig.java b/hedera-node/hedera-config/src/main/java/com/hedera/node/config/data/StatsConfig.java
@@ -25,7 +25,9 @@
 public record StatsConfig(
         @ConfigProperty(defaultValue = "<GAS>,ThroughputLimits,CreationLimits") @NodeProperty
                 List<String> consThrottlesToSample,
-        @ConfigProperty(defaultValue = "<GAS>,ThroughputLimits,OffHeapQueryLimits,CreationLimits,FreeQueryLimits")
+        @ConfigProperty(
+                        defaultValue =
+                                "<GAS>,ThroughputLimits,OffHeapQueryLimits,CreationLimits,FreeQueryLimits,BalanceQueryLimits")
                 @NodeProperty
                 List<String> hapiThrottlesToSample,
         @ConfigProperty(defaultValue = "0") @NodeProperty int executionTimesToTrack,

diff --git a/hedera-node/hedera-file-service-impl/src/main/resources/genesis/throttles.json b/hedera-node/hedera-file-service-impl/src/main/resources/genesis/throttles.json
@@ -181,11 +181,24 @@
           "opsPerSec": 0,
           "milliOpsPerSec": 1000000000,
           "operations": [
-            "CryptoGetAccountBalance",
             "TransactionGetReceipt"
           ]
         }
       ]
+    },
+    {
+      "burstPeriod": 0,
+      "burstPeriodMs": 1000,
+      "name": "BalanceQueryLimits",
+      "throttleGroups": [
+        {
+          "opsPerSec": 0,
+          "milliOpsPerSec": 1000000,
+          "operations": [
+            "CryptoGetAccountBalance"
+          ]
+        }
+      ]
     }
   ]
-}
+}