forked from rancher/go-machine-service
/
main.go
91 lines (79 loc) · 2.34 KB
/
main.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
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
package main
import (
"flag"
"fmt"
"os"
log "github.com/Sirupsen/logrus"
"github.com/rancherio/go-machine-service/events"
"github.com/rancherio/go-machine-service/handlers"
)
var (
GITCOMMIT = "HEAD"
)
func main() {
processCmdLineFlags()
log.WithFields(log.Fields{
"gitcommit": GITCOMMIT,
}).Info("Starting go-machine-service...")
eventHandlers := map[string]events.EventHandler{
"physicalhost.create": handlers.CreateMachine,
"physicalhost.bootstrap": handlers.ActivateMachine,
"physicalhost.remove": handlers.PurgeMachine,
"ping": handlers.PingNoOp,
}
apiUrl := os.Getenv("CATTLE_URL")
accessKey := os.Getenv("CATTLE_ACCESS_KEY")
secretKey := os.Getenv("CATTLE_SECRET_KEY")
router, err := events.NewEventRouter("goMachineService", 2000, apiUrl, accessKey, secretKey,
nil, eventHandlers, 10)
if err != nil {
log.WithFields(log.Fields{
"Err": err,
}).Error("Unable to create EventRouter")
} else {
err := router.Start(nil)
if err != nil {
log.WithFields(log.Fields{
"Err": err,
}).Error("Unable to start EventRouter")
}
}
log.Info("Exiting go-machine-service...")
}
func processCmdLineFlags() {
// Define command line flags
logLevel := flag.String("loglevel", "info", "Set the default loglevel (default:info) [debug|info|warn|error]")
version := flag.Bool("v", false, "read the version of the go-machine-service")
output := flag.String("o", "", "set the output file to write logs into, default is stdout")
flag.Parse()
if *output != "" {
var f *os.File
if _, err := os.Stat(*output); os.IsNotExist(err) {
f, err = os.Create(*output)
if err != nil {
fmt.Printf("could not create file=%s for logging, err=%v\n", *output, err)
os.Exit(1)
}
} else {
var err error
f, err = os.OpenFile(*output, os.O_RDWR|os.O_APPEND, 0)
if err != nil {
fmt.Printf("could not open file=%s for writing, err=%v\n", *output, err)
os.Exit(1)
}
}
log.SetFormatter(&log.JSONFormatter{})
log.SetOutput(f)
}
if *version {
fmt.Printf("go-machine-service\t gitcommit=%s\n", GITCOMMIT)
os.Exit(0)
}
// Process log level. If an invalid level is passed in, we simply default to info.
if parsedLogLevel, err := log.ParseLevel(*logLevel); err == nil {
log.WithFields(log.Fields{
"logLevel": *logLevel,
}).Info("Setting log level")
log.SetLevel(parsedLogLevel)
}
}