/
json.go
74 lines (61 loc) · 1.43 KB
/
json.go
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
74
package sink
import (
"encoding/json"
"os"
"path/filepath"
"github.com/devopsext/discovery/common"
sreCommon "github.com/devopsext/sre/common"
"github.com/devopsext/utils"
)
type JsonOptions struct {
Dir string
Providers []string
}
type Json struct {
options JsonOptions
logger sreCommon.Logger
observability *common.Observability
}
func (j *Json) Name() string {
return "Json"
}
func (j *Json) Providers() []string {
return j.options.Providers
}
func (j *Json) Process(d common.Discovery, so common.SinkObject) {
m := so.Map()
j.logger.Debug("Json has to process %d objects from %s...", len(m), d.Name())
data, err := json.Marshal(m)
if err != nil {
j.logger.Error("Json Sink: %v", err)
return
}
f, err := os.Create(filepath.Join(j.options.Dir, d.Name()+".json"))
if err != nil {
j.logger.Error("Json Sink: %v", err)
return
}
defer func(f *os.File) {
err := f.Close()
if err != nil {
j.logger.Error("Json Sink: %v", err)
}
}(f)
_, err = f.Write(data)
if err != nil {
j.logger.Error("Json Sink: %v", err)
}
}
func NewJson(options JsonOptions, observability *common.Observability) *Json {
logger := observability.Logs()
if utils.IsEmpty(options.Dir) {
logger.Debug("Json has no directory. Skipped")
return nil
}
options.Providers = common.RemoveEmptyStrings(options.Providers)
return &Json{
options: options,
logger: logger,
observability: observability,
}
}