-
Notifications
You must be signed in to change notification settings - Fork 2
/
main.go
154 lines (128 loc) · 3.57 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
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
package main
import (
"fmt"
flags "github.com/jessevdk/go-flags"
yaml "gopkg.in/yaml.v2"
"io/ioutil"
"log"
"os"
"strings"
"sync"
)
const (
Name = "go-syslogd"
Author = "webdevops.io"
Version = "0.2.1"
LogPrefix = ""
)
var (
argparser *flags.Parser
configuration ConfigurationDefinition
LoggerStdout SyslogLogger
LoggerStderr SyslogLogger
)
type ConfigurationDefinition struct {
Syslog Syslog
Pipes []Pipe
}
var opts struct {
Configuration string `short:"c" long:"configuration" description:"Configuration file (yml)" default:"/etc/go-syslog.yml"`
Verbose bool `short:"v" long:"verbose" description:"verbose mode"`
ShowVersion bool `short:"V" long:"version" description:"show version and exit"`
ShowOnlyVersion bool ` long:"dumpversion" description:"show only version number and exit"`
}
// handle special cli options
// eg. --help
// --version
// --path
// --mode=...
func handleSpecialCliOptions(args []string) {
// --dumpversion
if opts.ShowOnlyVersion {
fmt.Println(Version)
os.Exit(0)
}
// --version
if opts.ShowVersion {
fmt.Println(fmt.Sprintf("%s version %s", Name, Version))
fmt.Println(fmt.Sprintf("Copyright (C) 2017 %s", Author))
os.Exit(0)
}
}
// Parse configuration from yml file
func parseConfiguration() {
confData, err := ioutil.ReadFile(opts.Configuration)
if err != nil {
panic(fmt.Sprintf("Failed opening configuration file %s: %v", opts.Configuration, err.Error()))
}
err = yaml.Unmarshal(confData, &configuration)
if err != nil {
panic(fmt.Sprintf("Unable to parse configuration file %s: %v", opts.Configuration, err.Error()))
}
// set internal defaults
configuration.Syslog.Filter.facility = 2147483647
configuration.Syslog.Filter.severity = 2147483647
// init syslog configuration
if configuration.Syslog.Path != "" {
// Facility filter
if configuration.Syslog.Filter.Facility != "" {
for _, facility := range strings.Split(configuration.Syslog.Filter.Facility, ",") {
if facilityId, ok := SyslogFacilityMap[facility]; ok {
configuration.Syslog.Filter.facility = clearBit(configuration.Syslog.Filter.facility, uint(facilityId))
}
}
}
// Severity filter
if configuration.Syslog.Filter.Severity != "" {
for _, severity := range strings.Split(configuration.Syslog.Filter.Severity, ",") {
if severityId, ok := SyslogPriorityMap[severity]; ok {
configuration.Syslog.Filter.severity = clearBit(configuration.Syslog.Filter.severity, uint(severityId))
}
}
}
}
}
// Prints help
func printHelp() {
argparser.WriteHelp(os.Stdout)
os.Exit(1)
}
// Init system loggers
func initLogger() {
LoggerStdout = SyslogLogger{log.New(os.Stdout, LogPrefix, 0)}
LoggerStderr = SyslogLogger{log.New(os.Stderr, LogPrefix, 0)}
}
// Main function
func main() {
var wg sync.WaitGroup
// init logger
initLogger()
// init argument parser
argparser = flags.NewParser(&opts, flags.Default)
args, err := argparser.Parse()
handleSpecialCliOptions(args)
// check if there is an parse error
if err != nil {
if flagsErr, ok := err.(*flags.Error); ok && flagsErr.Type == flags.ErrHelp {
os.Exit(0)
} else {
os.Exit(1)
}
}
// parse yml configuration
parseConfiguration()
LoggerStdout.Println(fmt.Sprintf("Starting %s version %s", Name, Version))
// init pipes
for _, pipe := range configuration.Pipes {
wg.Add(1)
go func(pipe Pipe) {
handlePipe(pipe)
wg.Done()
}(pipe)
}
// init syslog
if configuration.Syslog.Path != "" {
handleSyslog()
}
wg.Wait()
}