1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
|
// Package slog is my set of wrappers around package slog.
package slog
import (
"flag"
"fmt"
"io"
"log/slog"
"net/http"
"os"
)
var (
slogLevel = flag.String("slog-level", "INFO", "log level")
// The current slog handler.
Handler slog.Handler
leveler *slog.LevelVar
)
func Init() {
var programLevel slog.Level
if err := (&programLevel).UnmarshalText([]byte(*slogLevel)); err != nil {
fmt.Fprintf(os.Stderr, "invalid log level %s: %v, using info\n", *slogLevel, err)
programLevel = slog.LevelInfo
}
leveler = &slog.LevelVar{}
leveler.Set(programLevel)
h := slog.NewJSONHandler(os.Stderr, &slog.HandlerOptions{
AddSource: true,
Level: leveler,
})
slog.SetDefault(slog.New(h))
Handler = h
http.HandleFunc("/.within/debug/slog-level", func(w http.ResponseWriter, r *http.Request) {
var level, old slog.Level
old = leveler.Level()
if r.Method == http.MethodPost {
data, err := io.ReadAll(http.MaxBytesReader(w, r.Body, 64))
defer r.Body.Close()
if err != nil {
http.Error(w, err.Error(), http.StatusBadRequest)
return
}
if err := (&level).UnmarshalText(data); err != nil {
http.Error(w, err.Error(), http.StatusBadRequest)
return
}
leveler.Set(level)
slog.Info("changed level", "from", old, "to", level)
fmt.Fprintln(w, level)
} else {
fmt.Fprintln(w, old)
}
})
}
|
