Compare commits
2 Commits
| Author | SHA1 | Date |
|---|---|---|
|
|
391f703dec | |
|
|
f836ae401b |
|
|
@ -1,7 +0,0 @@
|
||||||
root = "."
|
|
||||||
tmp_dir = "tmp"
|
|
||||||
|
|
||||||
[build]
|
|
||||||
cmd = "go build -o ./tmp/app ."
|
|
||||||
bin = "./tmp/app"
|
|
||||||
include_ext = ["go"]
|
|
||||||
|
|
@ -0,0 +1,7 @@
|
||||||
|
.devcontainer/
|
||||||
|
tests/
|
||||||
|
*.md
|
||||||
|
Dockerfile
|
||||||
|
.dockerignore
|
||||||
|
.gitignore
|
||||||
|
.git
|
||||||
|
|
@ -1,2 +1 @@
|
||||||
.devcontainer/
|
.devcontainer/
|
||||||
tmp/
|
|
||||||
|
|
@ -0,0 +1,21 @@
|
||||||
|
FROM golang:1.23.5 as builder
|
||||||
|
|
||||||
|
WORKDIR /app
|
||||||
|
|
||||||
|
COPY go.mod go.sum ./
|
||||||
|
|
||||||
|
RUN go mod download
|
||||||
|
|
||||||
|
COPY . .
|
||||||
|
|
||||||
|
RUN go build -o /app/main ./cmd
|
||||||
|
|
||||||
|
FROM golang:1.23.5-alpine
|
||||||
|
|
||||||
|
WORKDIR /app
|
||||||
|
|
||||||
|
COPY --from=builder /app/main .
|
||||||
|
|
||||||
|
EXPOSE 9101
|
||||||
|
|
||||||
|
CMD ["./main"]
|
||||||
|
|
@ -0,0 +1,29 @@
|
||||||
|
package main
|
||||||
|
|
||||||
|
import (
|
||||||
|
"exporter/internal/app"
|
||||||
|
"log"
|
||||||
|
"net/http"
|
||||||
|
"os"
|
||||||
|
|
||||||
|
"github.com/prometheus/client_golang/prometheus/promhttp"
|
||||||
|
)
|
||||||
|
|
||||||
|
func main() {
|
||||||
|
log.SetOutput(os.Stdout) // Логируем в стандартный вывод
|
||||||
|
log.SetFlags(log.Ldate | log.Ltime | log.Lshortfile)
|
||||||
|
|
||||||
|
exporter := app.NewMetricsExporter()
|
||||||
|
|
||||||
|
// Используем кастомный реестр в обработчике /metrics
|
||||||
|
http.Handle("/metrics", promhttp.HandlerFor(app.CustomRegistry, promhttp.HandlerOpts{}))
|
||||||
|
http.HandleFunc("/update", exporter.JSONHandler) // Обработчик для приёма JSON
|
||||||
|
|
||||||
|
port := ":9101"
|
||||||
|
log.Printf("Starting server on port %s\n", port)
|
||||||
|
if err := http.ListenAndServe(port, nil); err != nil {
|
||||||
|
log.Fatalf("Error starting server: %s\n", err)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
//TODO: сделать переменные окружения, настроить канал, дописать юнит тесты, добавить интеграционные тесты
|
||||||
|
|
@ -1,16 +0,0 @@
|
||||||
services:
|
|
||||||
exporter:
|
|
||||||
image: cosmtrek/air:v1.61.7
|
|
||||||
volumes:
|
|
||||||
- .:/app
|
|
||||||
working_dir: /app
|
|
||||||
ports:
|
|
||||||
- "9101:9101"
|
|
||||||
command: ["air", "-c", ".air.toml"]
|
|
||||||
|
|
||||||
prometheus:
|
|
||||||
image: prom/prometheus:v3.1.0
|
|
||||||
volumes:
|
|
||||||
- ./prometheus.yml:/etc/prometheus/prometheus.yml
|
|
||||||
ports:
|
|
||||||
- "9090:9090"
|
|
||||||
|
|
@ -1,4 +1,4 @@
|
||||||
package main
|
package app
|
||||||
|
|
||||||
import (
|
import (
|
||||||
"encoding/json"
|
"encoding/json"
|
||||||
|
|
@ -6,15 +6,13 @@ import (
|
||||||
"io"
|
"io"
|
||||||
"log"
|
"log"
|
||||||
"net/http"
|
"net/http"
|
||||||
"os"
|
|
||||||
"sync"
|
"sync"
|
||||||
|
|
||||||
"github.com/prometheus/client_golang/prometheus"
|
"github.com/prometheus/client_golang/prometheus"
|
||||||
"github.com/prometheus/client_golang/prometheus/promhttp"
|
|
||||||
)
|
)
|
||||||
|
|
||||||
// Создаём кастомный реестр
|
// Создаём кастомный реестр
|
||||||
var customRegistry = prometheus.NewRegistry()
|
var CustomRegistry = prometheus.NewRegistry()
|
||||||
|
|
||||||
// Структура JSON
|
// Структура JSON
|
||||||
type MetricRequest struct {
|
type MetricRequest struct {
|
||||||
|
|
@ -28,20 +26,20 @@ type MetricRequest struct {
|
||||||
|
|
||||||
// Экспортёр метрик
|
// Экспортёр метрик
|
||||||
type MetricsExporter struct {
|
type MetricsExporter struct {
|
||||||
gaugeMetrics map[string]*prometheus.GaugeVec
|
GaugeMetrics map[string]*prometheus.GaugeVec
|
||||||
counterMetrics map[string]*prometheus.CounterVec
|
CounterMetrics map[string]*prometheus.CounterVec
|
||||||
histogramMetrics map[string]*prometheus.HistogramVec
|
HistogramMetrics map[string]*prometheus.HistogramVec
|
||||||
summaryMetrics map[string]*prometheus.SummaryVec
|
SummaryMetrics map[string]*prometheus.SummaryVec
|
||||||
mu sync.Mutex // Защита от одновременного доступа
|
mu sync.Mutex // Защита от одновременного доступа
|
||||||
}
|
}
|
||||||
|
|
||||||
// Создаём новый экспортёр
|
// Создаём новый экспортёр
|
||||||
func NewMetricsExporter() *MetricsExporter {
|
func NewMetricsExporter() *MetricsExporter {
|
||||||
return &MetricsExporter{
|
return &MetricsExporter{
|
||||||
gaugeMetrics: make(map[string]*prometheus.GaugeVec),
|
GaugeMetrics: make(map[string]*prometheus.GaugeVec),
|
||||||
counterMetrics: make(map[string]*prometheus.CounterVec),
|
CounterMetrics: make(map[string]*prometheus.CounterVec),
|
||||||
histogramMetrics: make(map[string]*prometheus.HistogramVec),
|
HistogramMetrics: make(map[string]*prometheus.HistogramVec),
|
||||||
summaryMetrics: make(map[string]*prometheus.SummaryVec),
|
SummaryMetrics: make(map[string]*prometheus.SummaryVec),
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
@ -76,7 +74,7 @@ func (me *MetricsExporter) UpdateMetric(request MetricRequest) {
|
||||||
|
|
||||||
// Обновление `Gauge` метрик
|
// Обновление `Gauge` метрик
|
||||||
func (me *MetricsExporter) updateGauge(name string, labels []string, labelValues []string, value float64) {
|
func (me *MetricsExporter) updateGauge(name string, labels []string, labelValues []string, value float64) {
|
||||||
if _, exists := me.gaugeMetrics[name]; !exists {
|
if _, exists := me.GaugeMetrics[name]; !exists {
|
||||||
gaugeVec := prometheus.NewGaugeVec(
|
gaugeVec := prometheus.NewGaugeVec(
|
||||||
prometheus.GaugeOpts{
|
prometheus.GaugeOpts{
|
||||||
Name: name,
|
Name: name,
|
||||||
|
|
@ -84,10 +82,10 @@ func (me *MetricsExporter) updateGauge(name string, labels []string, labelValues
|
||||||
},
|
},
|
||||||
labels,
|
labels,
|
||||||
)
|
)
|
||||||
customRegistry.MustRegister(gaugeVec)
|
CustomRegistry.MustRegister(gaugeVec)
|
||||||
me.gaugeMetrics[name] = gaugeVec
|
me.GaugeMetrics[name] = gaugeVec
|
||||||
}
|
}
|
||||||
me.gaugeMetrics[name].WithLabelValues(labelValues...).Set(value)
|
me.GaugeMetrics[name].WithLabelValues(labelValues...).Set(value)
|
||||||
log.Printf("Gauge обновлён: %s = %f\n", name, value)
|
log.Printf("Gauge обновлён: %s = %f\n", name, value)
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
@ -97,7 +95,7 @@ func (me *MetricsExporter) updateCounter(name string, labels []string, labelValu
|
||||||
log.Printf("Ошибка: Counter %s не может быть отрицательным\n", name)
|
log.Printf("Ошибка: Counter %s не может быть отрицательным\n", name)
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
if _, exists := me.counterMetrics[name]; !exists {
|
if _, exists := me.CounterMetrics[name]; !exists {
|
||||||
counterVec := prometheus.NewCounterVec(
|
counterVec := prometheus.NewCounterVec(
|
||||||
prometheus.CounterOpts{
|
prometheus.CounterOpts{
|
||||||
Name: name,
|
Name: name,
|
||||||
|
|
@ -105,16 +103,16 @@ func (me *MetricsExporter) updateCounter(name string, labels []string, labelValu
|
||||||
},
|
},
|
||||||
labels,
|
labels,
|
||||||
)
|
)
|
||||||
customRegistry.MustRegister(counterVec)
|
CustomRegistry.MustRegister(counterVec)
|
||||||
me.counterMetrics[name] = counterVec
|
me.CounterMetrics[name] = counterVec
|
||||||
}
|
}
|
||||||
me.counterMetrics[name].WithLabelValues(labelValues...).Add(value)
|
me.CounterMetrics[name].WithLabelValues(labelValues...).Add(value)
|
||||||
log.Printf("Counter обновлён: %s += %f\n", name, value)
|
log.Printf("Counter обновлён: %s += %f\n", name, value)
|
||||||
}
|
}
|
||||||
|
|
||||||
// Обновление `Histogram` метрик
|
// Обновление `Histogram` метрик
|
||||||
func (me *MetricsExporter) updateHistogram(name string, labels []string, labelValues []string, value float64) {
|
func (me *MetricsExporter) updateHistogram(name string, labels []string, labelValues []string, value float64) {
|
||||||
if _, exists := me.histogramMetrics[name]; !exists {
|
if _, exists := me.HistogramMetrics[name]; !exists {
|
||||||
histogramVec := prometheus.NewHistogramVec(
|
histogramVec := prometheus.NewHistogramVec(
|
||||||
prometheus.HistogramOpts{
|
prometheus.HistogramOpts{
|
||||||
Name: name,
|
Name: name,
|
||||||
|
|
@ -123,16 +121,16 @@ func (me *MetricsExporter) updateHistogram(name string, labels []string, labelVa
|
||||||
},
|
},
|
||||||
labels,
|
labels,
|
||||||
)
|
)
|
||||||
customRegistry.MustRegister(histogramVec)
|
CustomRegistry.MustRegister(histogramVec)
|
||||||
me.histogramMetrics[name] = histogramVec
|
me.HistogramMetrics[name] = histogramVec
|
||||||
}
|
}
|
||||||
me.histogramMetrics[name].WithLabelValues(labelValues...).Observe(value)
|
me.HistogramMetrics[name].WithLabelValues(labelValues...).Observe(value)
|
||||||
log.Printf("Histogram обновлён: %s = %f\n", name, value)
|
log.Printf("Histogram обновлён: %s = %f\n", name, value)
|
||||||
}
|
}
|
||||||
|
|
||||||
// Обновление `Summary` метрик
|
// Обновление `Summary` метрик
|
||||||
func (me *MetricsExporter) updateSummary(name string, labels []string, labelValues []string, value float64) {
|
func (me *MetricsExporter) updateSummary(name string, labels []string, labelValues []string, value float64) {
|
||||||
if _, exists := me.summaryMetrics[name]; !exists {
|
if _, exists := me.SummaryMetrics[name]; !exists {
|
||||||
summaryVec := prometheus.NewSummaryVec(
|
summaryVec := prometheus.NewSummaryVec(
|
||||||
prometheus.SummaryOpts{
|
prometheus.SummaryOpts{
|
||||||
Name: name,
|
Name: name,
|
||||||
|
|
@ -141,10 +139,10 @@ func (me *MetricsExporter) updateSummary(name string, labels []string, labelValu
|
||||||
},
|
},
|
||||||
labels,
|
labels,
|
||||||
)
|
)
|
||||||
customRegistry.MustRegister(summaryVec)
|
CustomRegistry.MustRegister(summaryVec)
|
||||||
me.summaryMetrics[name] = summaryVec
|
me.SummaryMetrics[name] = summaryVec
|
||||||
}
|
}
|
||||||
me.summaryMetrics[name].WithLabelValues(labelValues...).Observe(value)
|
me.SummaryMetrics[name].WithLabelValues(labelValues...).Observe(value)
|
||||||
log.Printf("Summary обновлён: %s = %f\n", name, value)
|
log.Printf("Summary обновлён: %s = %f\n", name, value)
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
@ -181,22 +179,3 @@ func (me *MetricsExporter) JSONHandler(w http.ResponseWriter, r *http.Request) {
|
||||||
w.Write([]byte("Metrics updated"))
|
w.Write([]byte("Metrics updated"))
|
||||||
log.Printf("Метрики обновлены успешно для ID: %s\n", request.ID)
|
log.Printf("Метрики обновлены успешно для ID: %s\n", request.ID)
|
||||||
}
|
}
|
||||||
|
|
||||||
func main() {
|
|
||||||
log.SetOutput(os.Stdout) // Логируем в стандартный вывод
|
|
||||||
log.SetFlags(log.Ldate | log.Ltime | log.Lshortfile)
|
|
||||||
|
|
||||||
exporter := NewMetricsExporter()
|
|
||||||
|
|
||||||
// Используем кастомный реестр в обработчике /metrics
|
|
||||||
http.Handle("/metrics", promhttp.HandlerFor(customRegistry, promhttp.HandlerOpts{}))
|
|
||||||
http.HandleFunc("/update", exporter.JSONHandler) // Обработчик для приёма JSON
|
|
||||||
|
|
||||||
port := ":9101"
|
|
||||||
log.Printf("Starting server on port %s\n", port)
|
|
||||||
if err := http.ListenAndServe(port, nil); err != nil {
|
|
||||||
log.Fatalf("Error starting server: %s\n", err)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
//TODO: сделать переменные окружения, настроить канал, дописать тесты
|
|
||||||
|
|
@ -1,7 +0,0 @@
|
||||||
global:
|
|
||||||
scrape_interval: 1s # Интервал сбора метрик
|
|
||||||
|
|
||||||
scrape_configs:
|
|
||||||
- job_name: "exporter"
|
|
||||||
static_configs:
|
|
||||||
- targets: ["exporter:9101"] # Сервис экспортёра
|
|
||||||
12
test_api.txt
12
test_api.txt
|
|
@ -1,12 +0,0 @@
|
||||||
curl -X POST -H "Content-Type: application/json" -d '{
|
|
||||||
"id": "gauge_test",
|
|
||||||
"name": "Gauge Metric",
|
|
||||||
"url": "http://127.0.0.1:8081/gauge",
|
|
||||||
"method": "GET",
|
|
||||||
"type": "gauge",
|
|
||||||
"metrics": {
|
|
||||||
"temperature": 36.6
|
|
||||||
}
|
|
||||||
}' "http://localhost:9101/update"
|
|
||||||
|
|
||||||
curl http://localhost:9101/metrics
|
|
||||||
|
|
@ -1,8 +1,9 @@
|
||||||
package main
|
package test
|
||||||
|
|
||||||
import (
|
import (
|
||||||
"bytes"
|
"bytes"
|
||||||
"encoding/json"
|
"encoding/json"
|
||||||
|
"exporter/internal/app"
|
||||||
"io"
|
"io"
|
||||||
"net/http"
|
"net/http"
|
||||||
"net/http/httptest"
|
"net/http/httptest"
|
||||||
|
|
@ -15,12 +16,12 @@ import (
|
||||||
|
|
||||||
// Тест: успешное обновление метрик через /update
|
// Тест: успешное обновление метрик через /update
|
||||||
func TestJSONHandler_ValidRequest(t *testing.T) {
|
func TestJSONHandler_ValidRequest(t *testing.T) {
|
||||||
exporter := NewMetricsExporter()
|
exporter := app.NewMetricsExporter()
|
||||||
server := httptest.NewServer(http.HandlerFunc(exporter.JSONHandler))
|
server := httptest.NewServer(http.HandlerFunc(exporter.JSONHandler))
|
||||||
defer server.Close()
|
defer server.Close()
|
||||||
|
|
||||||
// JSON-запрос
|
// JSON-запрос
|
||||||
requestData := MetricRequest{
|
requestData := app.MetricRequest{
|
||||||
ID: "test_api",
|
ID: "test_api",
|
||||||
Name: "Test API",
|
Name: "Test API",
|
||||||
URL: "http://localhost/api",
|
URL: "http://localhost/api",
|
||||||
|
|
@ -48,7 +49,7 @@ func TestJSONHandler_ValidRequest(t *testing.T) {
|
||||||
|
|
||||||
// Тест: успешная обработка некорректного JSON
|
// Тест: успешная обработка некорректного JSON
|
||||||
func TestJSONHandler_InvalidJSON(t *testing.T) {
|
func TestJSONHandler_InvalidJSON(t *testing.T) {
|
||||||
exporter := NewMetricsExporter()
|
exporter := app.NewMetricsExporter()
|
||||||
server := httptest.NewServer(http.HandlerFunc(exporter.JSONHandler))
|
server := httptest.NewServer(http.HandlerFunc(exporter.JSONHandler))
|
||||||
defer server.Close()
|
defer server.Close()
|
||||||
|
|
||||||
|
|
@ -60,9 +61,9 @@ func TestJSONHandler_InvalidJSON(t *testing.T) {
|
||||||
|
|
||||||
// Тест: проверка создания и обновления Gauge метрики
|
// Тест: проверка создания и обновления Gauge метрики
|
||||||
func TestUpdateMetric_Gauge(t *testing.T) {
|
func TestUpdateMetric_Gauge(t *testing.T) {
|
||||||
exporter := NewMetricsExporter()
|
exporter := app.NewMetricsExporter()
|
||||||
|
|
||||||
request := MetricRequest{
|
request := app.MetricRequest{
|
||||||
ID: "test",
|
ID: "test",
|
||||||
Name: "Test Gauge",
|
Name: "Test Gauge",
|
||||||
URL: "http://test.com",
|
URL: "http://test.com",
|
||||||
|
|
@ -77,16 +78,16 @@ func TestUpdateMetric_Gauge(t *testing.T) {
|
||||||
exporter.UpdateMetric(request)
|
exporter.UpdateMetric(request)
|
||||||
|
|
||||||
// Проверяем, что метрика существует
|
// Проверяем, что метрика существует
|
||||||
metric, exists := exporter.gaugeMetrics["vks_test_load"]
|
metric, exists := exporter.GaugeMetrics["vks_test_load"]
|
||||||
assert.True(t, exists, "Метрика должна быть зарегистрирована")
|
assert.True(t, exists, "Метрика должна быть зарегистрирована")
|
||||||
assert.NotNil(t, metric, "Метрика не должна быть nil")
|
assert.NotNil(t, metric, "Метрика не должна быть nil")
|
||||||
}
|
}
|
||||||
|
|
||||||
// Тест: проверка создания и увеличения Counter метрики
|
// Тест: проверка создания и увеличения Counter метрики
|
||||||
func TestUpdateMetric_Counter(t *testing.T) {
|
func TestUpdateMetric_Counter(t *testing.T) {
|
||||||
exporter := NewMetricsExporter()
|
exporter := app.NewMetricsExporter()
|
||||||
|
|
||||||
request := MetricRequest{
|
request := app.MetricRequest{
|
||||||
ID: "test",
|
ID: "test",
|
||||||
Name: "Test Counter",
|
Name: "Test Counter",
|
||||||
URL: "http://test.com",
|
URL: "http://test.com",
|
||||||
|
|
@ -102,7 +103,7 @@ func TestUpdateMetric_Counter(t *testing.T) {
|
||||||
exporter.UpdateMetric(request)
|
exporter.UpdateMetric(request)
|
||||||
|
|
||||||
// Проверяем, что метрика существует
|
// Проверяем, что метрика существует
|
||||||
metric, exists := exporter.counterMetrics["vks_test_requests"]
|
metric, exists := exporter.CounterMetrics["vks_test_requests"]
|
||||||
assert.True(t, exists, "Метрика должна быть зарегистрирована")
|
assert.True(t, exists, "Метрика должна быть зарегистрирована")
|
||||||
assert.NotNil(t, metric, "Метрика не должна быть nil")
|
assert.NotNil(t, metric, "Метрика не должна быть nil")
|
||||||
}
|
}
|
||||||
|
|
@ -114,10 +115,10 @@ func TestMetricsEndpoint(t *testing.T) {
|
||||||
Name: "test_metric",
|
Name: "test_metric",
|
||||||
Help: "Test metric for /metrics endpoint",
|
Help: "Test metric for /metrics endpoint",
|
||||||
})
|
})
|
||||||
customRegistry.MustRegister(gauge)
|
app.CustomRegistry.MustRegister(gauge)
|
||||||
gauge.Set(50.5)
|
gauge.Set(50.5)
|
||||||
|
|
||||||
server := httptest.NewServer(promhttp.HandlerFor(customRegistry, promhttp.HandlerOpts{}))
|
server := httptest.NewServer(promhttp.HandlerFor(app.CustomRegistry, promhttp.HandlerOpts{}))
|
||||||
defer server.Close()
|
defer server.Close()
|
||||||
|
|
||||||
// Делаем GET-запрос к /metrics
|
// Делаем GET-запрос к /metrics
|
||||||
Loading…
Reference in New Issue