/
main.go
35 lines (30 loc) · 684 Bytes
/
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
package main
import (
"fmt"
"os"
"github.com/docker/go-plugins-helpers/sdk"
"github.com/sirupsen/logrus"
)
var logLevels = map[string]logrus.Level{
"debug": logrus.DebugLevel,
"info": logrus.InfoLevel,
"warn": logrus.WarnLevel,
"error": logrus.ErrorLevel,
}
func main() {
levelVal := os.Getenv("LOG_LEVEL")
if levelVal == "" {
levelVal = "info"
}
if level, exists := logLevels[levelVal]; exists {
logrus.SetLevel(level)
} else {
fmt.Fprintln(os.Stderr, "invalid log level: ", levelVal)
os.Exit(1)
}
h := sdk.NewHandler(`{"Implements": ["LogDriver"]}`)
handlers(&h, newDriver())
if err := h.ServeUnix("elasticlogger", 0); err != nil {
panic(err)
}
}