Add debugging option to the default logger

mikelsr · Sep 9, 2023 · 8a1de29 · 8a1de29
1 parent 49fec14
commit 8a1de29
Showing 1 changed file with 9 additions and 1 deletion.
diff --git a/raft/defaults.go b/raft/defaults.go
@@ -10,10 +10,18 @@ import (
 	"go.etcd.io/raft/v3/raftpb"
 )
 
-var DefaultLogger = &raft.DefaultLogger{Logger: log.New(os.Stderr, "raft", log.LstdFlags)}
+var defaultLogger = &raft.DefaultLogger{Logger: log.New(os.Stderr, "raft", log.LstdFlags)}
+
+func DefaultLogger(debug bool) raft.Logger {
+	if debug {
+		defaultLogger.EnableDebug()
+	}
+	return defaultLogger
+}
 
 func DefaultConfig() *raft.Config {
 	return &raft.Config{
+		ID:              DefaultID(),
 		HeartbeatTick:   HeartbeatTick,
 		ElectionTick:    ElectionTick,
 		MaxSizePerMsg:   MaxSizePerMsg,