-
Notifications
You must be signed in to change notification settings - Fork 0
/
palworld_exporter.go
115 lines (100 loc) · 3.11 KB
/
palworld_exporter.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
package main
import (
"context"
"log/slog"
"net/http"
"os"
"os/signal"
"syscall"
"time"
"github.com/alecthomas/kingpin/v2"
"github.com/jimmysharp/palworld_exporter/collector"
"github.com/jimmysharp/palworld_exporter/config"
"github.com/jimmysharp/palworld_exporter/log"
"github.com/prometheus/client_golang/prometheus"
"github.com/prometheus/client_golang/prometheus/promhttp"
"github.com/prometheus/common/version"
)
const (
shutdownTimeout = 5 * time.Second
)
var (
app = kingpin.New("palworld_exporter", "Prometheus exporter for Palworld")
listenAddress = app.
Flag("web.listen-address", "Address to expose metrics.").
Default(":18212").
Envar("LISTEN_ADDRESS").
String()
scrapeURI = app.
Flag("scrape_uri", "URI to Palworld REST API metrics endpoint.").
Default("http://localhost:8212/v1/api/metrics").
Envar("SCRAPE_URI").
String()
httpUsername = app.
Flag("http_user", "Username for Palworld REST API basic authentication.").
Envar("HTTP_USER").
Default("admin").
String()
httpPassword = app.
Flag("http_password", "Password for Palworld REST API basic authentication.").
Envar("HTTP_PASSWORD").
Required().
String()
logLevel = app.
Flag("log.level", "Only log messages with the given severity or above. Valid levels: [debug, info, warn, error]").
Default("info").
Envar("LOG_LEVEL").
HintOptions("debug", "info", "warn", "error").
String()
logFormat = app.
Flag("log.format", "Output format of log messages. Valid formats: [default, text, json]").
Default("default").
Envar("LOG_FORMAT").
HintOptions("default", "text", "json").
String()
)
func createServer(config *config.Config, logger *slog.Logger) *http.Server {
exporter := collector.NewExporter(config, logger)
registry := prometheus.NewRegistry()
registry.MustRegister(exporter)
mux := http.NewServeMux()
mux.Handle("/metrics", promhttp.HandlerFor(registry, promhttp.HandlerOpts{}))
server := &http.Server{
Addr: config.ListenAddress,
Handler: mux,
}
return server
}
func main() {
ctx, stop := signal.NotifyContext(context.Background(), syscall.SIGINT, syscall.SIGTERM)
defer stop()
app.HelpFlag.Short('h')
app.Version(version.Print("palworld_exporter"))
kingpin.MustParse(app.Parse(os.Args[1:]))
config := &config.Config{
ListenAddress: *listenAddress,
ScrapeURI: *scrapeURI,
HttpUsername: *httpUsername,
HttpPassword: *httpPassword,
LogLevel: *logLevel,
LogFormat: *logFormat,
}
logger := log.NewLogger(config)
server := createServer(config, logger)
go func() {
logger.Info("Starting palworld_exporter", slog.String("version", version.Info()))
if err := server.ListenAndServe(); err != nil && err != http.ErrServerClosed {
logger.Error("Error starting server", slog.String("err", err.Error()))
os.Exit(1)
}
}()
<-ctx.Done()
stop()
logger.Info("Caught signal, Shutting down server...")
ctx, cancel := context.WithTimeout(context.Background(), shutdownTimeout)
defer cancel()
if err := server.Shutdown(ctx); err != nil {
logger.Error("Error safely shutting down server", slog.String("err", err.Error()))
os.Exit(1)
}
}