les/vflux/server: set inactiveFlag in priorityPool

ethereum · Mar 29, 2021 · f1000d6 · f1000d6
1 parent 50ddf0e
commit f1000d6
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 6 deletions.
diff --git a/les/vflux/server/clientpool.go b/les/vflux/server/clientpool.go
@@ -132,12 +132,6 @@ func NewClientPool(balanceDb ethdb.KeyValueStore, minCap uint64, connectedBias t
 		}
 	})
 
-	ns.SubscribeField(setup.balanceField, func(node *enode.Node, state nodestate.Flags, oldValue, newValue interface{}) {
-		if newValue != nil {
-			ns.SetStateSub(node, setup.inactiveFlag, nodestate.Flags{}, 0)
-		}
-	})
-
 	ns.SubscribeField(setup.capacityField, func(node *enode.Node, state nodestate.Flags, oldValue, newValue interface{}) {
 		if c, ok := ns.GetField(node, setup.clientField).(clientPeer); ok {
 			newCap, _ := newValue.(uint64)

diff --git a/les/vflux/server/prioritypool.go b/les/vflux/server/prioritypool.go
@@ -116,6 +116,7 @@ func newPriorityPool(ns *nodestate.NodeStateMachine, setup *serverSetup, clock m
 				stepDiv:       pp.capacityStepDiv,
 			}
 			ns.SetFieldSub(node, pp.setup.queueField, c)
+			ns.SetStateSub(node, setup.inactiveFlag, nodestate.Flags{}, 0)
 		} else {
 			ns.SetStateSub(node, nodestate.Flags{}, pp.setup.activeFlag.Or(pp.setup.inactiveFlag), 0)
 			if n, _ := pp.ns.GetField(node, pp.setup.queueField).(*ppNodeInfo); n != nil {