tweaks

AccumulateNetwork · Sep 4, 2024 · 5c01ca2 · 5c01ca2
1 parent 75e0f3e
commit 5c01ca2
Show file tree

Hide file tree

Showing 13 changed files with 25 additions and 26 deletions.
diff --git a/cmd/accumulated-bootstrap/main.go b/cmd/accumulated-bootstrap/main.go
@@ -70,7 +70,6 @@ func run(*cobra.Command, []string) {
 	ctx := ContextForMainProcess(context.Background())
 	inst, err := Start(ctx, cfg)
 	Check(err)
-
-	<-ctx.Done()
+	<-inst.Done()
 	inst.Stop()
 }
diff --git a/cmd/accumulated-faucet/main.go b/cmd/accumulated-faucet/main.go
@@ -58,15 +58,12 @@ func init() {
 	cmd.Flags().Var(&flag.Key, "key", "The key used to sign faucet transactions")
 	cmd.Flags().Var(&flag.Account, "account", "The faucet account")
 	cmd.Flags().Var((*MultiaddrSliceFlag)(&flag.Listen), "listen", "P2P listening address(es)")
-	cmd.Flags().Var((*MultiaddrSliceFlag)(&flag.PromListen), "prom-listen", "Prometheus listening address(es) (default /ip4/0.0.0.0/tcp/8081/http)")
+	cmd.Flags().Var((*MultiaddrSliceFlag)(&flag.PromListen), "prom-listen", "Prometheus listening address(es)")
 	cmd.Flags().VarP((*MultiaddrSliceFlag)(&flag.Peers), "peer", "p", "Peers to connect to")
 	cmd.Flags().StringVar(&flag.PeerDatabase, "peer-db", flag.PeerDatabase, "Track peers using a persistent database.")
 	cmd.Flags().StringVar(&flag.LogLevel, "log-level", "error", "Log level")
 
 	cmd.PersistentPreRun = func(cmd *cobra.Command, args []string) {
-		if !cmd.Flag("prom-listen").Changed {
-			flag.PromListen = []multiaddr.Multiaddr{multiaddr.StringCast("/ip4/0.0.0.0/tcp/8081/http")}
-		}
 		if !cmd.Flag("peer").Changed {
 			flag.Peers = accumulate.BootstrapServers
 		}
@@ -107,6 +104,6 @@ func run(_ *cobra.Command, args []string) {
 	ctx := cmdutil.ContextForMainProcess(context.Background())
 	inst, err := Start(ctx, cfg)
 	Check(err)
-	<-ctx.Done()
+	<-inst.Done()
 	inst.Stop()
 }
diff --git a/cmd/accumulated-http/main.go b/cmd/accumulated-http/main.go
@@ -175,7 +175,7 @@ func run(cmd *cobra.Command, args []string) {
 	ctx := cmdutil.ContextForMainProcess(context.Background())
 	inst, err := Start(ctx, cfg)
 	Check(err)
-	<-ctx.Done()
+	<-inst.Done()
 	inst.Stop()
 }
 

diff --git a/cmd/accumulated/main.go b/cmd/accumulated/main.go
@@ -103,7 +103,7 @@ func runCfg(c *run.Config, predicate func(run.Service) bool) {
 
 	check(i.StartFiltered(predicate))
 	color.HiBlue("\n----- Running -----\n\n")
-	<-ctx.Done()
+	<-i.Done()
 	color.HiBlack("\n----- Stopping -----\n\n")
 	i.Stop()
 }

diff --git a/cmd/accumulated/run/consensus.go b/cmd/accumulated/run/consensus.go
@@ -235,7 +235,7 @@ func (c *ConsensusService) start(inst *Instance) error {
 		return errors.UnknownError.WithFormat("start consensus: %w", err)
 	}
 
-	inst.cleanup(func(context.Context) error {
+	inst.cleanup("consensus node", func(context.Context) error {
 		err := node.Stop()
 		node.Wait()
 		return err

diff --git a/cmd/accumulated/run/faucet.go b/cmd/accumulated/run/faucet.go
@@ -103,7 +103,7 @@ func (f *FaucetService) start(inst *Instance) error {
 			return
 		}
 
-		inst.cleanup(func(context.Context) error { impl.Stop(); return nil })
+		inst.cleanup("faucet", func(context.Context) error { impl.Stop(); return nil })
 		registerRpcService(inst, impl.ServiceAddress(), message.Faucet{Faucet: impl})
 		inst.logger.Info("Ready", "module", "run", "service", "faucet")
 	}()

diff --git a/cmd/accumulated/run/http.go b/cmd/accumulated/run/http.go
@@ -154,7 +154,7 @@ func (h *HttpListener) startHTTP(inst *Instance, handler http.Handler) (*http.Se
 		ReadHeaderTimeout: h.ReadHeaderTimeout.Get(),
 	}
 
-	inst.cleanup(server.Shutdown)
+	inst.cleanup("http listener", server.Shutdown)
 
 	for _, l := range h.Listen {
 		l, secure, err := httpListen(l)

diff --git a/cmd/accumulated/run/instance.go b/cmd/accumulated/run/instance.go
@@ -199,7 +199,7 @@ func (inst *Instance) StartFiltered(predicate func(Service) bool) (err error) {
 			serviceUp.Add(inst.context, 1, metric.WithAttributes(
 				attribute.String("type", svc.Type().String())))
 
-			inst.cleanup(func(ctx context.Context) error {
+			inst.cleanup("service metrics", func(ctx context.Context) error {
 				serviceUp.Add(inst.context, -1, metric.WithAttributes(
 					attribute.String("type", svc.Type().String())))
 				return nil
@@ -256,17 +256,20 @@ func (i *Instance) run(fn func()) {
 	}()
 }
 
-func (i *Instance) cleanup(fn func(context.Context) error) {
+func (i *Instance) cleanup(name string, fn func(context.Context) error) {
 	i.running.Add(1)
 	go func() {
 		defer i.running.Done()
 		<-i.context.Done()
 
+		slog.Debug("Stopping", "process", name)
 		ctx, cancel := context.WithTimeout(context.Background(), 10*time.Second)
 		defer cancel()
 		err := fn(ctx)
 		if err != nil {
-			slog.Error("Error during shutdown", "error", err)
+			slog.Error("Error during shutdown", "error", err, "process", name)
+		} else {
+			slog.Debug("Stopped", "process", name)
 		}
 	}()
 }

diff --git a/cmd/accumulated/run/instrumentation.go b/cmd/accumulated/run/instrumentation.go
@@ -82,7 +82,7 @@ func (i *Instrumentation) startPprof(inst *Instance) error {
 		slog.Error("Server stopped (pprof)", "error", err)
 	})
 
-	inst.cleanup(s.Shutdown)
+	inst.cleanup("pprof http", s.Shutdown)
 	return nil
 }
 
@@ -110,7 +110,7 @@ func (m *Monitor) start(inst *Instance) error {
 
 func (m *Monitor) pollMemory(inst *Instance) {
 	tick := time.NewTicker(m.MemoryPollingRate.Get())
-	inst.cleanup(func(context.Context) error { tick.Stop(); return nil })
+	inst.cleanup("memory tracking", func(context.Context) error { tick.Stop(); return nil })
 
 	var s1, s2 runtime.MemStats
 	runtime.ReadMemStats(&s1)

diff --git a/cmd/accumulated/run/p2p.go b/cmd/accumulated/run/p2p.go
@@ -38,7 +38,7 @@ func (p *P2P) start(inst *Instance) error {
 
 	slog.InfoContext(inst.context, "We are", "node-id", node.ID(), "instance-id", inst.id, "module", "run")
 
-	inst.cleanup(func(context.Context) error {
+	inst.cleanup("p2p node", func(context.Context) error {
 		err := node.Close()
 		if err != nil {
 			return err

diff --git a/cmd/accumulated/run/storage.go b/cmd/accumulated/run/storage.go
@@ -107,7 +107,7 @@ func (s *BadgerStorage) open(inst *Instance) (keyvalue.Beginner, error) {
 		return nil, err
 	}
 
-	inst.cleanup(func(context.Context) error { return db.Close() })
+	inst.cleanup("storage", func(context.Context) error { return db.Close() })
 	return db, nil
 }
 
@@ -117,7 +117,7 @@ func (s *BoltStorage) open(inst *Instance) (keyvalue.Beginner, error) {
 		return nil, err
 	}
 
-	inst.cleanup(func(context.Context) error { return db.Close() })
+	inst.cleanup("storage", func(context.Context) error { return db.Close() })
 	return db, nil
 }
 
@@ -127,7 +127,7 @@ func (s *LevelDBStorage) open(inst *Instance) (keyvalue.Beginner, error) {
 		return nil, err
 	}
 
-	inst.cleanup(func(context.Context) error { return db.Close() })
+	inst.cleanup("storage", func(context.Context) error { return db.Close() })
 	return db, nil
 }
 
@@ -137,6 +137,6 @@ func (s *ExpBlockDBStorage) open(inst *Instance) (keyvalue.Beginner, error) {
 		return nil, err
 	}
 
-	inst.cleanup(func(context.Context) error { return db.Close() })
+	inst.cleanup("storage", func(context.Context) error { return db.Close() })
 	return db, nil
 }
diff --git a/cmd/accumulated/run/telemetry.go b/cmd/accumulated/run/telemetry.go
@@ -102,7 +102,7 @@ func (t *Telemetry) setupTraceProvider(res *resource.Resource, inst *Instance) e
 			trace.WithBatchTimeout(time.Second)),
 	)
 
-	inst.cleanup(provider.Shutdown)
+	inst.cleanup("otel (traces)", provider.Shutdown)
 	otel.SetTracerProvider(provider)
 	return nil
 }
@@ -252,7 +252,7 @@ func (t *Telemetry) setupMeterProvider(res *resource.Resource, inst *Instance) e
 	}
 
 	provider := metric.NewMeterProvider(options...)
-	inst.cleanup(provider.Shutdown)
+	inst.cleanup("otel (metrics)", provider.Shutdown)
 	otel.SetMeterProvider(provider)
 	return nil
 }
@@ -271,7 +271,7 @@ func (t *Telemetry) setupLoggerProvider(res *resource.Resource, inst *Instance)
 		log.WithProcessor(log.NewBatchProcessor(exporter)),
 	)
 
-	inst.cleanup(provider.Shutdown)
+	inst.cleanup("otel (logs)", provider.Shutdown)
 	global.SetLoggerProvider(provider)
 	return nil
 }

diff --git a/internal/api/routing/router.go b/internal/api/routing/router.go
@@ -88,7 +88,7 @@ func NewRouter(opts RouterOptions) *RouterInstance {
 			}
 
 			r.tree = tree
-			markReady(true)
+			markReady(false)
 			return nil
 		})
 	}