node: SupplyManager restared by SetHead API call

kaiachain · Aug 13, 2024 · 2ce37f0 · 2ce37f0
1 parent e077455
commit 2ce37f0
Show file tree

Hide file tree

Showing 2 changed files with 15 additions and 0 deletions.
diff --git a/node/cn/api_backend.go b/node/cn/api_backend.go
@@ -100,6 +100,8 @@ func (b *CNAPIBackend) SetHead(number uint64) error {
 	b.cn.protocolManager.Downloader().Cancel()
 	b.cn.protocolManager.SetSyncStop(true)
 	defer b.cn.protocolManager.SetSyncStop(false)
+	b.cn.supplyManager.Stop()
+	defer b.cn.supplyManager.Start()
 	return doSetHead(b.cn.blockchain, b.cn.engine, b.cn.governance, b.gpo, number)
 }
 

diff --git a/node/cn/api_backend_test.go b/node/cn/api_backend_test.go
@@ -173,6 +173,18 @@ func testGov() *governance.MixedEngine {
 	return governance.NewMixedEngine(config, db)
 }
 
+type testSupplyManager struct{}
+
+func (sm *testSupplyManager) Start() {
+}
+
+func (sm *testSupplyManager) Stop() {
+}
+
+func (sm *testSupplyManager) GetTotalSupply(num uint64) (*reward.TotalSupply, error) {
+	return &reward.TotalSupply{}, nil
+}
+
 func TestCNAPIBackend_SetHead(t *testing.T) {
 	mockCtrl, mockBlockChain, _, api := newCNAPIBackend(t)
 	defer mockCtrl.Finish()
@@ -183,6 +195,7 @@ func TestCNAPIBackend_SetHead(t *testing.T) {
 	api.cn.protocolManager = pm
 	api.cn.engine = gxhash.NewFullFaker()
 	api.cn.governance = testGov()
+	api.cn.supplyManager = &testSupplyManager{}
 	api.gpo = gasprice.NewOracle(api, gasprice.Config{}, nil, api.cn.governance)
 
 	number := uint64(123)