aboutsummaryrefslogtreecommitdiffstats
path: root/client/go/internal/admin/vespa-wrapper/logfmt/levelflags.go
blob: 4e6c128475328d9363b6f88cdfc501b83c52b8ce (plain) (blame)
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
65
66
67
68
69
70
71
72
// Copyright Yahoo. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root.
// vespa logfmt command
// Author: arnej

package logfmt

import (
	"strings"
)

// handle CLI flags for log level filtering

type flagValueForLevel struct {
	levels  map[string]bool
	changed bool
}

func defaultLevelFlags() map[string]bool {
	return map[string]bool{
		"fatal":   true,
		"error":   true,
		"warning": true,
		"info":    true,
		"config":  false,
		"event":   false,
		"debug":   false,
		"spam":    false,
	}
}

func (v *flagValueForLevel) Type() string {
	return "level flags"
}

func (v *flagValueForLevel) String() string {
	var buf strings.Builder
	flagNames := []string{
		"fatal",
		"error",
		"warning",
		"info",
		"config",
		"event",
		"debug",
		"spam",
	}
	for _, flag := range flagNames {
		if v.levels[flag] {
			buf.WriteString(" +")
		} else {
			buf.WriteString(" -")
		}
		buf.WriteString(flag)
	}
	return buf.String()
}

func (v *flagValueForLevel) flags() map[string]bool {
	return v.levels
}

func (v *flagValueForLevel) name() string {
	return "level"
}

func (v *flagValueForLevel) unchanged() bool {
	return !v.changed
}

func (v *flagValueForLevel) Set(val string) error {
	return applyPlusMinus(val, v)
}