tjh.dev
forked from
tangled.org/core
Mirror of @tangled.org/core. Running on a Raspberry Pi Zero 2
spindle/models: interface-ify WorkflowLogger
Signed-off-by: oppiliappan <me@oppi.li>
authored by
+14
-10
spindle/models/logger.go
+14
-10
spindle/models/logger.go
···
9
9
"strings"
10
10
)
11
11
12
-
type WorkflowLogger struct {
12
+
type WorkflowLogger interface {
13
+
Close() error
14
+
DataWriter(idx int, stream string) io.Writer
15
+
ControlWriter(idx int, step Step, stepStatus StepStatus) io.Writer
16
+
}
17
+
18
+
type FileWorkflowLogger struct {
13
19
file *os.File
14
20
encoder *json.Encoder
15
21
mask *SecretMask
16
22
}
17
23
18
-
func NewWorkflowLogger(baseDir string, wid WorkflowId, secretValues []string) (*WorkflowLogger, error) {
24
+
func NewFileWorkflowLogger(baseDir string, wid WorkflowId, secretValues []string) (*FileWorkflowLogger, error) {
19
25
path := LogFilePath(baseDir, wid)
20
-
21
26
file, err := os.OpenFile(path, os.O_CREATE|os.O_WRONLY|os.O_APPEND, 0644)
22
27
if err != nil {
23
28
return nil, fmt.Errorf("creating log file: %w", err)
24
29
}
25
-
26
-
return &WorkflowLogger{
30
+
return &FileWorkflowLogger{
27
31
file: file,
28
32
encoder: json.NewEncoder(file),
29
33
mask: NewSecretMask(secretValues),
···
39
35
return logFilePath
40
36
}
41
37
42
-
func (l *WorkflowLogger) Close() error {
38
+
func (l *FileWorkflowLogger) Close() error {
43
39
return l.file.Close()
44
40
}
45
41
46
-
func (l *WorkflowLogger) DataWriter(idx int, stream string) io.Writer {
42
+
func (l *FileWorkflowLogger) DataWriter(idx int, stream string) io.Writer {
47
43
return &dataWriter{
48
44
logger: l,
49
45
idx: idx,
···
51
47
}
52
48
}
53
49
54
-
func (l *WorkflowLogger) ControlWriter(idx int, step Step, stepStatus StepStatus) io.Writer {
50
+
func (l *FileWorkflowLogger) ControlWriter(idx int, step Step, stepStatus StepStatus) io.Writer {
55
51
return &controlWriter{
56
52
logger: l,
57
53
idx: idx,
···
61
57
}
62
58
63
59
type dataWriter struct {
64
-
logger *WorkflowLogger
60
+
logger *FileWorkflowLogger
65
61
idx int
66
62
stream string
67
63
}
···
79
75
}
80
76
81
77
type controlWriter struct {
82
-
logger *WorkflowLogger
78
+
logger *FileWorkflowLogger
83
79
idx int
84
80
step Step
85
81
stepStatus StepStatus