forked from Loyalsoldier/geoip
-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathinstance.go
73 lines (61 loc) · 1.35 KB
/
instance.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
package lib
import (
"encoding/json"
"errors"
"os"
"strings"
)
type Instance struct {
config *config
input []InputConverter
output []OutputConverter
}
func NewInstance() (*Instance, error) {
return &Instance{
config: new(config),
input: make([]InputConverter, 0),
output: make([]OutputConverter, 0),
}, nil
}
func (i *Instance) Init(configFile string) error {
var content []byte
var err error
configFile = strings.TrimSpace(configFile)
if strings.HasPrefix(configFile, "http://") || strings.HasPrefix(configFile, "https://") {
content, err = getRemoteURLContent(configFile)
} else {
content, err = os.ReadFile(configFile)
}
if err != nil {
return err
}
if err := json.Unmarshal(content, &i.config); err != nil {
return err
}
for _, input := range i.config.Input {
i.input = append(i.input, input.converter)
}
for _, output := range i.config.Output {
i.output = append(i.output, output.converter)
}
return nil
}
func (i *Instance) Run() error {
if len(i.input) == 0 || len(i.output) == 0 {
return errors.New("input type and output type must be specified")
}
var err error
container := NewContainer()
for _, ic := range i.input {
container, err = ic.Input(container)
if err != nil {
return err
}
}
for _, oc := range i.output {
if err := oc.Output(container); err != nil {
return err
}
}
return nil
}