1174 set log level to debug in golang

go/core/logger/level_filter.go

@@ -0,0 +1,50 @@
+// Copyright 2024 Google LLC
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+//     http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+
+// Package logger provides a context-scoped slog.Logger.
+package logger
+
+import (
+	"context"
+	"log/slog"
+)
+
+// LevelFilterHandler is a custom handler that only logs DEBUG messages.
+type LevelFilterHandler struct {
+	level slog.Level
+	h     slog.Handler
+}
+
+func (h *LevelFilterHandler) Enabled(ctx context.Context, level slog.Level) bool {
+	// Display the message if its level is greater than or equal to the configured level
+	return level >= h.level
+}
+
+func (h *LevelFilterHandler) Handle(ctx context.Context, r slog.Record) error {
+	return h.h.Handle(ctx, r)
+}
+
+func (h *LevelFilterHandler) WithAttrs(attrs []slog.Attr) slog.Handler {
+	return &LevelFilterHandler{
+		level: h.level,
+		h:     h.h.WithAttrs(attrs),
+	}
+}
+
+func (h *LevelFilterHandler) WithGroup(name string) slog.Handler {
+	return &LevelFilterHandler{
+		level: h.level,
+		h:     h.h.WithGroup(name),
+	}
+}

go/core/logger/logger.go

@@ -26,10 +26,12 @@ import (
 func init() {
 	// TODO: Remove this. The main program should be responsible for configuring logging.
 	// This is just a convenience during development.
-	h := slog.New(slog.NewTextHandler(os.Stderr, &slog.HandlerOptions{
-		Level: slog.LevelDebug,
-	}))
-	slog.SetDefault(h)
+	baseHandler := slog.NewTextHandler(os.Stderr, &slog.HandlerOptions{})
+	debugHandler := &LevelFilterHandler{
+		h:     baseHandler,
+		level: slog.LevelDebug,
+	}
+	slog.SetDefault(slog.New(debugHandler))
 }
 
 var loggerKey = base.NewContextKey[*slog.Logger]()

go/genkit/servers.go

@@ -158,7 +158,7 @@ func findProjectRoot() (string, error) {
 //
 // To construct a server with additional routes, use [NewFlowServeMux].
 func startFlowServer(addr string, flows []string, errCh chan<- error) *http.Server {
-	slog.Info("starting flow server")
+	slog.Debug("starting flow server")
 	addr = serverAddress(addr, "PORT", "127.0.0.1:3400")
 	mux := NewFlowServeMux(flows)
 	return startServer(addr, mux, errCh)
@@ -182,7 +182,7 @@ func startServer(addr string, handler http.Handler, errCh chan<- error) *http.Se
 	}
 
 	go func() {
-		slog.Info("server listening", "addr", addr)
+		slog.Debug("server listening", "addr", addr)
 		if err := server.ListenAndServe(); err != nil && err != http.ErrServerClosed {
 			errCh <- fmt.Errorf("server error on %s: %w", addr, err)
 		}
@@ -205,7 +205,7 @@ func shutdownServers(servers []*http.Server) error {
 			if err := srv.Shutdown(ctx); err != nil {
 				slog.Error("server shutdown failed", "addr", srv.Addr, "err", err)
 			} else {
-				slog.Info("server shutdown successfully", "addr", srv.Addr)
+				slog.Debug("server shutdown successfully", "addr", srv.Addr)
 			}
 		}(server)
 	}

go/internal/registry/registry.go

@@ -101,7 +101,7 @@ func (r *Registry) RegisterAction(typ atype.ActionType, a action.Action) {
 	}
 	a.SetTracingState(r.tstate)
 	r.actions[key] = a
-	slog.Info("RegisterAction",
+	slog.Debug("RegisterAction",
 		"type", typ,
 		"name", a.Name())
 }

go/plugins/googlecloud/googlecloud_test.go

@@ -74,7 +74,7 @@ func TestGCP(t *testing.T) {
 		if err := setLogHandler(*projectID, slog.LevelInfo); err != nil {
 			t.Fatal(err)
 		}
-		slog.Info("testing GCP logging",
+		slog.Debug("testing GCP logging",
 			"binaryName", os.Args[0],
 			"goVersion", runtime.Version())
 		// Allow time to export.