-
Notifications
You must be signed in to change notification settings - Fork 0
/
main.go
111 lines (89 loc) · 2.42 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
package main
import (
"context"
"flag"
"fmt"
"os"
"os/signal"
"syscall"
"time"
"github.com/vrischmann/envconfig"
yaml "gopkg.in/yaml.v3"
"github.com/ngalaiko/miniboard/backend"
"github.com/ngalaiko/miniboard/backend/logger"
)
func main() {
configPath := flag.String("config", "", "Path to the configuration file, required")
verbose := flag.Bool("verbose", false, "Enable verbose logging")
flag.Parse()
log := logger.New(logger.Info)
if *verbose {
log = logger.New(logger.Debug)
}
cfg, err := parseConfiguration(configPath)
if err != nil {
log.Fatal("failed to parse configuration: %s", err)
}
log.Info("application is starting")
srv, err := backend.New(log, cfg)
if err != nil {
log.Fatal("failed to initialize server: %s", err)
}
// Wait for shut down in a separate goroutine.
errCh := make(chan error)
go func() {
shutdownCh := make(chan os.Signal, 1)
signal.Notify(shutdownCh, os.Interrupt, syscall.SIGTERM)
sig := <-shutdownCh
log.Info("received %s, shutting down", sig)
shutdownTimeout := 15 * time.Second
shutdownCtx, cancel := context.WithTimeout(context.Background(), shutdownTimeout)
defer cancel()
errCh <- srv.Shutdown(shutdownCtx)
}()
mainCtx := context.Background()
if err := srv.Start(mainCtx); err != nil {
log.Fatal("failed to start the server: %s", err)
}
// Handle shutdown errors.
if err := <-errCh; err != nil {
log.Error("error during shutdown: %s", err)
}
log.Info("application stopped")
}
func parseConfiguration(path *string) (*backend.Config, error) {
cfg := &backend.Config{}
if path != nil && *path != "" {
var err error
cfg, err = parseConfigurationFromYaml(*path)
if err != nil {
return nil, err
}
}
if err := parseConfigurationFromEnvironment(cfg); err != nil {
return nil, err
}
return cfg, nil
}
func parseConfigurationFromEnvironment(cfg *backend.Config) error {
if err := envconfig.InitWithOptions(cfg, envconfig.Options{
Prefix: "MINIBOARD",
AllOptional: true,
}); err != nil {
return fmt.Errorf("failed to parse config from env: %w", err)
}
return nil
}
func parseConfigurationFromYaml(path string) (*backend.Config, error) {
file, err := os.Open(path)
if err != nil {
return nil, fmt.Errorf("failed to open file: %w", err)
}
cfg := &backend.Config{}
d := yaml.NewDecoder(file)
d.KnownFields(true)
if err := d.Decode(cfg); err != nil {
return nil, fmt.Errorf("failed to parse file: %w", err)
}
return cfg, nil
}