logy/console.go at main · codnect/logy · GitHub

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
73
package logy

import (
	"os"
)

type ConsoleHandler struct {
	commonHandler
}

func newConsoleHandler() *ConsoleHandler {
	handler := &ConsoleHandler{}
	handler.initializeHandler()

	handler.setTarget(TargetStderr)

	handler.SetEnabled(true)
	handler.SetLevel(LevelDebug)

	handler.SetColorEnabled(true)
	return handler
}

func (h *ConsoleHandler) SetColorEnabled(enabled bool) {
	h.color.Store(enabled)
}

func (h *ConsoleHandler) IsColorEnabled() bool {
	return h.color.Load().(bool)
}

func (h *ConsoleHandler) setTarget(target Target) {
	var consoleWriter *syncWriter
	if h.writer != nil {
		consoleWriter = h.writer.(*syncWriter)
	} else {
		consoleWriter = newSyncWriter(nil, true)
		h.writer = consoleWriter
	}

	defer consoleWriter.mu.Unlock()
	consoleWriter.mu.Lock()

	switch target {
	case TargetStdout:
		h.target.Store(target)
		consoleWriter.writer = os.Stdout
		consoleWriter.setDiscarded(false)
	case TargetStderr:
		h.target.Store(target)
		consoleWriter.writer = os.Stderr
		consoleWriter.setDiscarded(false)
	default:
		h.target.Store(target)
		consoleWriter.setDiscarded(true)
	}
}

func (h *ConsoleHandler) Target() Target {
	return h.target.Load().(Target)
}

func (h *ConsoleHandler) OnConfigure(config Config) error {
	h.SetEnabled(config.Console.Enabled)
	h.SetLevel(config.Console.Level)
	h.SetFormat(config.Console.Format)

	h.setTarget(config.Console.Target)
	h.SetColorEnabled(config.Console.Color)

	h.applyJsonConfig(config.Console.Json)
	return nil
}