blob: 4791b93deb0da222c80c349971b44f2da3fab65e (
plain)
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
|
package main
import (
"flag"
"fmt"
"os"
"github.com/prometheus/client_golang/prometheus"
"github.com/sirupsen/logrus"
"gitlab.com/gitlab-org/gitaly/internal/blackbox"
"gitlab.com/gitlab-org/gitaly/internal/log"
"gitlab.com/gitlab-org/gitaly/internal/version"
)
var flagVersion = flag.Bool("version", false, "Print version and exit")
func flagUsage() {
fmt.Println(version.GetVersionString())
fmt.Printf("Usage: %v [OPTIONS] configfile\n", os.Args[0])
flag.PrintDefaults()
}
func main() {
flag.Usage = flagUsage
flag.Parse()
// If invoked with -version
if *flagVersion {
fmt.Println(version.GetVersionString())
os.Exit(0)
}
if flag.NArg() != 1 || flag.Arg(0) == "" {
flag.Usage()
os.Exit(1)
}
if err := run(flag.Arg(0)); err != nil {
logrus.WithError(err).Fatal()
}
}
func run(configPath string) error {
configRaw, err := os.ReadFile(configPath)
if err != nil {
return err
}
config, err := blackbox.ParseConfig(string(configRaw))
if err != nil {
return err
}
bb := blackbox.New(config)
prometheus.MustRegister(bb)
log.Configure(log.Loggers, config.Logging.Format, config.Logging.Level)
return bb.Run()
}
|
