lnd: Create the callback function to update the estimator info on main cfg

Create the callback function (UpdateEstimatorValue) to update the
estimator info on main cfg. The callback function is a parameter
on NewMissionControl function.

Author: MPins

server.go

@@ -923,8 +923,11 @@ func newServer(cfg *Config, listenAddrs []net.Addr,
 				routingConfig.ProbabilityEstimatorType)
 		}
 	}
-
+	// UpdateEstimatorValue is a function that will be called by the
+	// mission control instance every time a new estimator value 
+	// is setted.
 	mcCfg := &routing.MissionControlConfig{
+		UpdateEstimatorCB:       s.UpdateEstimatorValue,
 		Estimator:               estimator,
 		MaxMcHistory:            routingConfig.MaxMcHistory,
 		McFlushInterval:         routingConfig.McFlushInterval,
@@ -1680,6 +1683,29 @@ func newServer(cfg *Config, listenAddrs []net.Addr,
 	return s, nil
 }
 
+// UpdateEstimatorValue is a callback function to change estimator value in the  
+// main cfg when the lncli setmccfg is called to change the estimator value. 
+ func (s *server) UpdateEstimatorValue(estimator routing.Estimator) {
+	switch c := estimator.Config().(type) {
+	case routing.AprioriConfig:
+		s.cfg.SubRPCServers.RouterRPC.RoutingConfig.ProbabilityEstimatorType =
+		routing.AprioriEstimatorName
+		targetCfg := s.cfg.SubRPCServers.RouterRPC.AprioriConfig
+		targetCfg.PenaltyHalfLife = c.PenaltyHalfLife
+		targetCfg.Weight = c.AprioriWeight
+		targetCfg.CapacityFraction = c.CapacityFraction
+		targetCfg.HopProbability = c.AprioriHopProbability
+
+	case routing.BimodalConfig:
+		s.cfg.SubRPCServers.RouterRPC.RoutingConfig.ProbabilityEstimatorType =
+		routing.BimodalEstimatorName
+		targetCfg := s.cfg.SubRPCServers.RouterRPC.BimodalConfig
+		targetCfg.Scale = int64(c.BimodalScaleMsat)
+		targetCfg.NodeWeight = c.BimodalNodeWeight
+		targetCfg.DecayTime = c.BimodalDecayTime 
+	}
+}
+
 // signAliasUpdate takes a ChannelUpdate and returns the signature. This is
 // used for option_scid_alias channels where the ChannelUpdate to be sent back
 // may differ from what is on disk.