feat: build migrated to Bazel, project refactored

feat: build migrated to Bazel, project refactored

Author: MrDan4es

.env.example

@@ -1,3 +1,2 @@
-REGCLOUD_TOKEN=ace6ahvaezuqueh3Eec9eikaeth2Oonaiwo3ohzoongu2oom1aigae9phahgae2oi4Oof6oda2naiqueh6eshe0die0Zaibaufoo
-REGCLOUD_HTTP_PORT=322
-TZ=Europe/Moscow
+COLLECTORS_CLOUD_REG_RU_AUTH_TOKEN={{https://help.reg.ru/support/servery-vps/oblachnyye-servery/rabota-s-serverom/api-dlya-oblachnykh-serverov}}
+COLLECTORS_SELECTEL_AUTH_TOKEN={{https://developers.selectel.ru/docs/control-panel/authorization/}}

.github/workflows/on_push_main.yml

@@ -0,0 +1,27 @@
+name: on_push_main_stage
+
+on:
+  push:
+    branches: [main]
+
+jobs:
+  update_stage:
+    runs-on: ubuntu-latest
+
+    steps:
+      - uses: actions/checkout@v4
+      - name: Login to GitHub Container Registry
+        uses: docker/login-action@v3
+        with:
+          registry: ghcr.io
+          username: ${{ github.actor }}
+          password: ${{ secrets.GITHUB_TOKEN }}
+      - uses: bazelbuild/setup-bazelisk@v3
+      - name: Mount bazel cache
+        uses: actions/cache@v4
+        with:
+          path: "~/.cache/bazel"
+          key: bazel-${{ hashFiles('.bazelversion', '.bazelrc', 'WORKSPACE', 'WORKSPACE.bazel', 'MODULE.bazel', '**/BUILD.bazel') }}
+          restore-keys: |
+            bazel-
+      - run: bazel run push

.gitignore

@@ -1,5 +1,14 @@
-# Environment
-.env
-
-# Golang
-go.sum
+#----Environment----
+/**/.env
+#-------------------
+#-----Intellij------
+.idea/
+.idea_modules/
+#-------------------
+#-------Bazel-------
+bazel-*
+.ijwb
+#-------------------
+#-------VSCode------
+.vscode
+#-------------------

BUILD.bazel

@@ -0,0 +1,17 @@
+load("@gazelle//:def.bzl", "gazelle")
+load("@rules_oci//oci:defs.bzl", "oci_push", "oci_tarball")
+
+gazelle(name = "gazelle")
+
+oci_tarball(
+    name = "bundle",
+    image = "//cmd/http_exporter:oci",
+    repo_tags = ["local/mrdan4es/http_exporter:latest"],
+)
+
+oci_push(
+    name = "push",
+    image = "//cmd/http_exporter:oci",
+    remote_tags = ["latest"],
+    repository = "ghcr.io/mrdan4es/http_exporter",
+)

MODULE.bazel

@@ -0,0 +1,31 @@
+module(name = "http_exporter")
+
+bazel_dep(name = "rules_go", version = "0.48.0")
+bazel_dep(name = "gazelle", version = "0.37.0")
+bazel_dep(name = "rules_oci", version = "2.0.0-alpha5")
+bazel_dep(name = "rules_pkg", version = "0.9.1")
+
+oci = use_extension("@rules_oci//oci:extensions.bzl", "oci")
+oci.pull(
+    name = "distroless_base_debian12",
+    digest = "sha256:2102ce121ff1448316b93c5f347118a8e604f5ba7ec9dd7a5c2d8b0eac2941fe",
+    image = "gcr.io/distroless/base-debian12",
+    platforms = [
+        "linux/amd64",
+    ],
+)
+use_repo(oci, "distroless_base_debian12", "distroless_base_debian12_linux_amd64")
+
+go_sdk = use_extension("@rules_go//go:extensions.bzl", "go_sdk")
+go_sdk.download(version = "1.22.5")
+
+go_deps = use_extension("@gazelle//:extensions.bzl", "go_deps")
+go_deps.from_file(go_mod = "//:go.mod")
+use_repo(
+    go_deps,
+    "com_github_jmespath_go_jmespath",
+    "com_github_prometheus_client_golang",
+    "com_github_rs_zerolog",
+    "com_github_spf13_pflag",
+    "com_github_spf13_viper",
+)

MODULE.bazel.lock

@@ -0,0 +1,38 @@
+load("@rules_go//go:def.bzl", "go_binary", "go_library")
+load("@rules_oci//oci:defs.bzl", "oci_image")
+load("@rules_pkg//:pkg.bzl", "pkg_tar")
+
+go_library(
+    name = "http_exporter_lib",
+    srcs = ["http_exporter.go"],
+    importpath = "github.com/mrdan4es/http_exporter/cmd/http_exporter",
+    visibility = ["//visibility:private"],
+    deps = [
+        "//pkg/collector",
+        "//pkg/config",
+        "@com_github_prometheus_client_golang//prometheus",
+        "@com_github_prometheus_client_golang//prometheus/promhttp",
+        "@com_github_rs_zerolog//:zerolog",
+        "@com_github_rs_zerolog//log",
+        "@com_github_spf13_pflag//:pflag",
+    ],
+)
+
+go_binary(
+    name = "http_exporter",
+    embed = [":http_exporter_lib"],
+    visibility = ["//visibility:public"],
+)
+
+pkg_tar(
+    name = "layer",
+    srcs = ["http_exporter"],
+)
+
+oci_image(
+    name = "oci",
+    base = "@distroless_base_debian12",
+    entrypoint = ["/http_exporter"],
+    tars = ["layer"],
+    visibility = ["//visibility:public"],
+)

README.md

@@ -0,0 +1,105 @@
+package main
+
+import (
+	"context"
+	"encoding/json"
+	"errors"
+	"flag"
+	"fmt"
+	"net/http"
+	"os/signal"
+	"syscall"
+	"time"
+
+	"github.com/prometheus/client_golang/prometheus"
+	"github.com/prometheus/client_golang/prometheus/promhttp"
+	"github.com/rs/zerolog"
+	zlog "github.com/rs/zerolog/log"
+	"github.com/spf13/pflag"
+
+	"github.com/mrdan4es/http_exporter/pkg/collector"
+	"github.com/mrdan4es/http_exporter/pkg/config"
+)
+
+func main() {
+	ctx, cancel := signal.NotifyContext(context.Background(), syscall.SIGINT, syscall.SIGTERM)
+	defer cancel()
+
+	configPath := pflag.StringP("config", "c", "", "path to config file")
+	pflag.CommandLine.AddGoFlagSet(flag.CommandLine)
+	pflag.Parse()
+
+	cfg, err := config.Load(*configPath)
+	if err != nil {
+		panic(err)
+	}
+
+	log := zlog.Output(
+		zerolog.NewConsoleWriter(
+			func(w *zerolog.ConsoleWriter) {
+				w.TimeFormat = time.RFC3339Nano
+			},
+		),
+	)
+	ctx = log.WithContext(ctx)
+
+	msg, err := json.Marshal(cfg)
+	if err != nil {
+		log.Fatal().
+			Err(err).
+			Msg("marshal config")
+	}
+
+	log.Info().
+		RawJSON("config", msg).
+		Send()
+
+	if err := run(ctx, cfg); err != nil {
+		log.Fatal().
+			Err(err).
+			Send()
+	}
+}
+
+func run(ctx context.Context, cfg *config.Config) error {
+	log := zerolog.Ctx(ctx)
+
+	for _, collectorCfg := range cfg.Collectors {
+		c, err := collector.New(ctx, collectorCfg)
+		if err != nil {
+			return fmt.Errorf("create %s collector: %w", collectorCfg.Name, err)
+		}
+
+		prometheus.MustRegister(c)
+	}
+
+	mux := http.NewServeMux()
+	mux.Handle("/metrics", promhttp.Handler())
+
+	httpServer := &http.Server{
+		Addr:    cfg.HTTP.ListenAddr,
+		Handler: mux,
+	}
+
+	go func() {
+		<-ctx.Done()
+
+		log.Info().Msg("Shutting down HTTP server...")
+
+		shutdownCtx := context.Background()
+		shutdownCtx, cancel := context.WithTimeout(ctx, 10*time.Second)
+		defer cancel()
+
+		if err := httpServer.Shutdown(shutdownCtx); err != nil {
+			log.Err(err).Send()
+		}
+	}()
+
+	log.Info().Msgf("Serving /metrics on addr %s...", cfg.HTTP.ListenAddr)
+
+	if err := httpServer.ListenAndServe(); err != nil && !errors.Is(err, http.ErrServerClosed) {
+		return err
+	}
+
+	return nil
+}

cmd/http_exporter/BUILD.bazel

@@ -0,0 +1,25 @@
+HTTP:
+  listenAddr: ":322"
+
+collectors:
+  - name: cloud_reg_ru
+    url: https://api.cloudvps.reg.ru/v1/balance_data
+    auth:
+      method: bearer
+      token: ${COLLECTORS_CLOUD_REG_RU_AUTH_TOKEN}
+    fields:
+      - name: balance
+        description: Balance
+        query:  balance_data.balance
+      - name: hours_left
+        description: Hours left
+        query: balance_data.hours_left
+  - name: selectel
+    url: https://api.selectel.ru/v3/balances
+    auth:
+      method: x-token
+      token: ${COLLECTORS_SELECTEL_AUTH_TOKEN}
+    fields:
+      - name: balance
+        description: Balance
+        query: data.billings[0].final_sum

cmd/http_exporter/http_exporter.go

@@ -1,8 +1,10 @@
-version: "3.8"
 services:
-  regcloud-exporter:
-    build: src/
+  http_exporter:
+    image: local/mrdan4es/http_exporter
     ports:
-      - "${REGCLOUD_HTTP_PORT}:${REGCLOUD_HTTP_PORT}"
+      - "322:322"
+    volumes:
+      - ./config.example.yml:/tmp/config.yml
+    command: ["--config", "/tmp/config.yml"]
     env_file:
-      - .env
+      - .env.example

config.example.yml

@@ -0,0 +1,39 @@
+module github.com/mrdan4es/http_exporter
+
+go 1.22
+
+require (
+	github.com/jmespath/go-jmespath v0.4.0
+	github.com/prometheus/client_golang v1.19.1
+	github.com/rs/zerolog v1.33.0
+	github.com/spf13/pflag v1.0.5
+	github.com/spf13/viper v1.19.0
+)
+
+require (
+	github.com/beorn7/perks v1.0.1 // indirect
+	github.com/cespare/xxhash/v2 v2.2.0 // indirect
+	github.com/fsnotify/fsnotify v1.7.0 // indirect
+	github.com/hashicorp/hcl v1.0.0 // indirect
+	github.com/magiconair/properties v1.8.7 // indirect
+	github.com/mattn/go-colorable v0.1.13 // indirect
+	github.com/mattn/go-isatty v0.0.20 // indirect
+	github.com/mitchellh/mapstructure v1.5.0 // indirect
+	github.com/pelletier/go-toml/v2 v2.2.2 // indirect
+	github.com/prometheus/client_model v0.5.0 // indirect
+	github.com/prometheus/common v0.48.0 // indirect
+	github.com/prometheus/procfs v0.12.0 // indirect
+	github.com/sagikazarmark/locafero v0.4.0 // indirect
+	github.com/sagikazarmark/slog-shim v0.1.0 // indirect
+	github.com/sourcegraph/conc v0.3.0 // indirect
+	github.com/spf13/afero v1.11.0 // indirect
+	github.com/spf13/cast v1.6.0 // indirect
+	github.com/subosito/gotenv v1.6.0 // indirect
+	go.uber.org/multierr v1.11.0 // indirect
+	golang.org/x/exp v0.0.0-20230905200255-921286631fa9 // indirect
+	golang.org/x/sys v0.23.0 // indirect
+	golang.org/x/text v0.14.0 // indirect
+	google.golang.org/protobuf v1.33.0 // indirect
+	gopkg.in/ini.v1 v1.67.0 // indirect
+	gopkg.in/yaml.v3 v3.0.1 // indirect
+)