Welcome to mirror list, hosted at ThFree Co, Russian Federation.

check.go « gitaly « cmd - gitlab.com/gitlab-org/gitaly.git - Unnamed repository; edit this file 'description' to name the repository.
summaryrefslogtreecommitdiff
blob: 00dcb7603fac07cbf02b0d454f85caa3a22485a9 (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
61
62
63
64
65
66
67
68
69
70
71
72
package main

import (
	"context"
	"flag"
	"fmt"
	"os"

	"github.com/sirupsen/logrus"
	"gitlab.com/gitlab-org/gitaly/v15/internal/git"
	"gitlab.com/gitlab-org/gitaly/v15/internal/gitaly/config"
	"gitlab.com/gitlab-org/gitaly/v15/internal/gitaly/config/prometheus"
	"gitlab.com/gitlab-org/gitaly/v15/internal/gitaly/hook"
	"gitlab.com/gitlab-org/gitaly/v15/internal/gitlab"
)

func execCheck() {
	logrus.SetLevel(logrus.ErrorLevel)

	checkCmd := flag.NewFlagSet("check", flag.ExitOnError)
	checkCmd.Usage = func() {
		fmt.Fprintf(os.Stderr, "Usage: %v check <configfile>\n", os.Args[0])
		checkCmd.PrintDefaults()
	}

	_ = checkCmd.Parse(os.Args[2:])

	if checkCmd.NArg() != 1 || checkCmd.Arg(0) == "" {
		fmt.Fprintf(os.Stderr, "error: invalid argument(s)")
		checkCmd.Usage()
		os.Exit(2)
	}

	configPath := checkCmd.Arg(0)

	fmt.Print("Checking GitLab API access: ")
	info, err := checkAPI(configPath)
	if err != nil {
		fmt.Println("FAILED")
		fmt.Fprintln(os.Stderr, err)
		os.Exit(1)
	}

	fmt.Println("OK")
	fmt.Printf("GitLab version: %s\n", info.Version)
	fmt.Printf("GitLab revision: %s\n", info.Revision)
	fmt.Printf("GitLab Api version: %s\n", info.APIVersion)
	fmt.Printf("Redis reachable for GitLab: %t\n", info.RedisReachable)
	fmt.Println("OK")

	os.Exit(0)
}

func checkAPI(configPath string) (*gitlab.CheckInfo, error) {
	cfg, err := loadConfig(configPath)
	if err != nil {
		return nil, fmt.Errorf("load config: config_path %q: %w", configPath, err)
	}

	gitlabAPI, err := gitlab.NewHTTPClient(logrus.New(), cfg.Gitlab, cfg.TLS, prometheus.Config{})
	if err != nil {
		return nil, err
	}

	gitCmdFactory, cleanup, err := git.NewExecCommandFactory(cfg)
	if err != nil {
		return nil, err
	}
	defer cleanup()

	return hook.NewManager(cfg, config.NewLocator(cfg), gitCmdFactory, nil, gitlabAPI).Check(context.Background())
}