Compare commits
2 Commits
909c1151d4
...
4896fb6385
| Author | SHA1 | Date |
|---|---|---|
 SavelyG
|
4896fb6385 | |
|
SavelyG
|
90a3848c72 |
11
.air.toml
11
.air.toml
|
|
@ -1,10 +1,7 @@
|
|||
root = "." # Корень проекта
|
||||
root = "."
|
||||
tmp_dir = "tmp"
|
||||
|
||||
[build]
|
||||
binary = "app" # Имя исполняемого файла
|
||||
cmd = "go build -o app main.go"
|
||||
include_ext = ["go", "tpl", "tmpl", "html"]
|
||||
|
||||
[run]
|
||||
cmd = "./app"
|
||||
cmd = "go build -o ./tmp/app ."
|
||||
bin = "./tmp/app"
|
||||
include_ext = ["go"]
|
||||
|
|
@ -3,9 +3,10 @@ services:
|
|||
image: cosmtrek/air:v1.61.7
|
||||
volumes:
|
||||
- .:/app
|
||||
working_dir: /app
|
||||
ports:
|
||||
- "9101:9101"
|
||||
command: air -c .air.toml
|
||||
command: ["air", "-c", ".air.toml"]
|
||||
|
||||
prometheus:
|
||||
image: prom/prometheus:v3.1.0
|
||||
|
|
|
|||
|
|
@ -0,0 +1,132 @@
|
|||
package main
|
||||
|
||||
import (
|
||||
"bytes"
|
||||
"encoding/json"
|
||||
"io"
|
||||
"net/http"
|
||||
"net/http/httptest"
|
||||
"testing"
|
||||
|
||||
"github.com/prometheus/client_golang/prometheus"
|
||||
"github.com/prometheus/client_golang/prometheus/promhttp"
|
||||
"github.com/stretchr/testify/assert"
|
||||
)
|
||||
|
||||
// Тест: успешное обновление метрик через /update
|
||||
func TestJSONHandler_ValidRequest(t *testing.T) {
|
||||
exporter := NewMetricsExporter()
|
||||
server := httptest.NewServer(http.HandlerFunc(exporter.JSONHandler))
|
||||
defer server.Close()
|
||||
|
||||
// JSON-запрос
|
||||
requestData := MetricRequest{
|
||||
ID: "test_api",
|
||||
Name: "Test API",
|
||||
URL: "http://localhost/api",
|
||||
Method: "GET",
|
||||
Type: "gauge",
|
||||
Metrics: map[string]float64{
|
||||
"response_time": 200.5,
|
||||
},
|
||||
}
|
||||
|
||||
// Маршалим JSON и проверяем ошибку
|
||||
jsonData, err := json.Marshal(requestData)
|
||||
assert.NoError(t, err, "Ошибка при маршалинге JSON")
|
||||
|
||||
// Отправляем запрос
|
||||
resp, err := http.Post(server.URL, "application/json", bytes.NewBuffer(jsonData))
|
||||
assert.NoError(t, err, "Ошибка при отправке запроса на /update")
|
||||
assert.Equal(t, http.StatusOK, resp.StatusCode)
|
||||
|
||||
// Проверяем тело ответа
|
||||
body, err := io.ReadAll(resp.Body)
|
||||
assert.NoError(t, err, "Ошибка чтения тела ответа")
|
||||
assert.Equal(t, "Metrics updated", string(body))
|
||||
}
|
||||
|
||||
// Тест: успешная обработка некорректного JSON
|
||||
func TestJSONHandler_InvalidJSON(t *testing.T) {
|
||||
exporter := NewMetricsExporter()
|
||||
server := httptest.NewServer(http.HandlerFunc(exporter.JSONHandler))
|
||||
defer server.Close()
|
||||
|
||||
// Отправляем некорректный JSON
|
||||
resp, err := http.Post(server.URL, "application/json", bytes.NewBuffer([]byte("{invalid json}")))
|
||||
assert.NoError(t, err)
|
||||
assert.Equal(t, http.StatusBadRequest, resp.StatusCode)
|
||||
}
|
||||
|
||||
// Тест: проверка создания и обновления Gauge метрики
|
||||
func TestUpdateMetric_Gauge(t *testing.T) {
|
||||
exporter := NewMetricsExporter()
|
||||
|
||||
request := MetricRequest{
|
||||
ID: "test",
|
||||
Name: "Test Gauge",
|
||||
URL: "http://test.com",
|
||||
Method: "GET",
|
||||
Type: "gauge",
|
||||
Metrics: map[string]float64{
|
||||
"load": 75.5,
|
||||
},
|
||||
}
|
||||
|
||||
// Вызываем обновление метрики
|
||||
exporter.UpdateMetric(request)
|
||||
|
||||
// Проверяем, что метрика существует
|
||||
metric, exists := exporter.gaugeMetrics["vks_test_load"]
|
||||
assert.True(t, exists, "Метрика должна быть зарегистрирована")
|
||||
assert.NotNil(t, metric, "Метрика не должна быть nil")
|
||||
}
|
||||
|
||||
// Тест: проверка создания и увеличения Counter метрики
|
||||
func TestUpdateMetric_Counter(t *testing.T) {
|
||||
exporter := NewMetricsExporter()
|
||||
|
||||
request := MetricRequest{
|
||||
ID: "test",
|
||||
Name: "Test Counter",
|
||||
URL: "http://test.com",
|
||||
Method: "POST",
|
||||
Type: "counter",
|
||||
Metrics: map[string]float64{
|
||||
"requests": 1,
|
||||
},
|
||||
}
|
||||
|
||||
// Вызываем обновление метрики дважды
|
||||
exporter.UpdateMetric(request)
|
||||
exporter.UpdateMetric(request)
|
||||
|
||||
// Проверяем, что метрика существует
|
||||
metric, exists := exporter.counterMetrics["vks_test_requests"]
|
||||
assert.True(t, exists, "Метрика должна быть зарегистрирована")
|
||||
assert.NotNil(t, metric, "Метрика не должна быть nil")
|
||||
}
|
||||
|
||||
// Тест: получение метрик по /metrics
|
||||
func TestMetricsEndpoint(t *testing.T) {
|
||||
// Регистрируем метрику
|
||||
gauge := prometheus.NewGauge(prometheus.GaugeOpts{
|
||||
Name: "test_metric",
|
||||
Help: "Test metric for /metrics endpoint",
|
||||
})
|
||||
customRegistry.MustRegister(gauge)
|
||||
gauge.Set(50.5)
|
||||
|
||||
server := httptest.NewServer(promhttp.HandlerFor(customRegistry, promhttp.HandlerOpts{}))
|
||||
defer server.Close()
|
||||
|
||||
// Делаем GET-запрос к /metrics
|
||||
resp, err := http.Get(server.URL)
|
||||
assert.NoError(t, err, "Ошибка при запросе к /metrics")
|
||||
assert.Equal(t, http.StatusOK, resp.StatusCode)
|
||||
|
||||
// Проверяем, что в ответе есть test_metric
|
||||
body, err := io.ReadAll(resp.Body)
|
||||
assert.NoError(t, err, "Ошибка чтения тела ответа")
|
||||
assert.Contains(t, string(body), "test_metric")
|
||||
}
|
||||
11
go.mod
11
go.mod
|
|
@ -1,17 +1,24 @@
|
|||
module prometheus_exporter
|
||||
module exporter
|
||||
|
||||
go 1.23.5
|
||||
|
||||
require github.com/prometheus/client_golang v1.20.5
|
||||
require (
|
||||
github.com/prometheus/client_golang v1.20.5
|
||||
github.com/stretchr/testify v1.9.0
|
||||
)
|
||||
|
||||
require (
|
||||
github.com/beorn7/perks v1.0.1 // indirect
|
||||
github.com/cespare/xxhash/v2 v2.3.0 // indirect
|
||||
github.com/davecgh/go-spew v1.1.1 // indirect
|
||||
github.com/klauspost/compress v1.17.9 // indirect
|
||||
github.com/kr/text v0.2.0 // indirect
|
||||
github.com/munnerz/goautoneg v0.0.0-20191010083416-a7dc8b61c822 // indirect
|
||||
github.com/pmezard/go-difflib v1.0.0 // indirect
|
||||
github.com/prometheus/client_model v0.6.1 // indirect
|
||||
github.com/prometheus/common v0.55.0 // indirect
|
||||
github.com/prometheus/procfs v0.15.1 // indirect
|
||||
golang.org/x/sys v0.22.0 // indirect
|
||||
google.golang.org/protobuf v1.34.2 // indirect
|
||||
gopkg.in/yaml.v3 v3.0.1 // indirect
|
||||
)
|
||||
|
|
|
|||
18
go.sum
18
go.sum
|
|
@ -2,14 +2,23 @@ github.com/beorn7/perks v1.0.1 h1:VlbKKnNfV8bJzeqoa4cOKqO6bYr3WgKZxO8Z16+hsOM=
|
|||
github.com/beorn7/perks v1.0.1/go.mod h1:G2ZrVWU2WbWT9wwq4/hrbKbnv/1ERSJQ0ibhJ6rlkpw=
|
||||
github.com/cespare/xxhash/v2 v2.3.0 h1:UL815xU9SqsFlibzuggzjXhog7bL6oX9BbNZnL2UFvs=
|
||||
github.com/cespare/xxhash/v2 v2.3.0/go.mod h1:VGX0DQ3Q6kWi7AoAeZDth3/j3BFtOZR5XLFGgcrjCOs=
|
||||
github.com/creack/pty v1.1.9/go.mod h1:oKZEueFk5CKHvIhNR5MUki03XCEU+Q6VDXinZuGJ33E=
|
||||
github.com/davecgh/go-spew v1.1.1 h1:vj9j/u1bqnvCEfJOwUhtlOARqs3+rkHYY13jYWTU97c=
|
||||
github.com/davecgh/go-spew v1.1.1/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38=
|
||||
github.com/google/go-cmp v0.6.0 h1:ofyhxvXcZhMsU5ulbFiLKl/XBFqE1GSq7atu8tAmTRI=
|
||||
github.com/google/go-cmp v0.6.0/go.mod h1:17dUlkBOakJ0+DkrSSNjCkIjxS6bF9zb3elmeNGIjoY=
|
||||
github.com/klauspost/compress v1.17.9 h1:6KIumPrER1LHsvBVuDa0r5xaG0Es51mhhB9BQB2qeMA=
|
||||
github.com/klauspost/compress v1.17.9/go.mod h1:Di0epgTjJY877eYKx5yC51cX2A2Vl2ibi7bDH9ttBbw=
|
||||
github.com/kr/pretty v0.3.1 h1:flRD4NNwYAUpkphVc1HcthR4KEIFJ65n8Mw5qdRn3LE=
|
||||
github.com/kr/pretty v0.3.1/go.mod h1:hoEshYVHaxMs3cyo3Yncou5ZscifuDolrwPKZanG3xk=
|
||||
github.com/kr/text v0.2.0 h1:5Nx0Ya0ZqY2ygV366QzturHI13Jq95ApcVaJBhpS+AY=
|
||||
github.com/kr/text v0.2.0/go.mod h1:eLer722TekiGuMkidMxC/pM04lWEeraHUUmBw8l2grE=
|
||||
github.com/kylelemons/godebug v1.1.0 h1:RPNrshWIDI6G2gRW9EHilWtl7Z6Sb1BR0xunSBf0SNc=
|
||||
github.com/kylelemons/godebug v1.1.0/go.mod h1:9/0rRGxNHcop5bhtWyNeEfOS8JIWk580+fNqagV/RAw=
|
||||
github.com/munnerz/goautoneg v0.0.0-20191010083416-a7dc8b61c822 h1:C3w9PqII01/Oq1c1nUAm88MOHcQC9l5mIlSMApZMrHA=
|
||||
github.com/munnerz/goautoneg v0.0.0-20191010083416-a7dc8b61c822/go.mod h1:+n7T8mK8HuQTcFwEeznm/DIxMOiR9yIdICNftLE1DvQ=
|
||||
github.com/pmezard/go-difflib v1.0.0 h1:4DBwDE0NGyQoBHbLQYPwSUPoCMWR5BEzIk/f1lZbAQM=
|
||||
github.com/pmezard/go-difflib v1.0.0/go.mod h1:iKH77koFhYxTK1pcRnkKkqfTogsbg7gZNVY4sRDYZ/4=
|
||||
github.com/prometheus/client_golang v1.20.5 h1:cxppBPuYhUnsO6yo/aoRol4L7q7UFfdm+bR9r+8l63Y=
|
||||
github.com/prometheus/client_golang v1.20.5/go.mod h1:PIEt8X02hGcP8JWbeHyeZ53Y/jReSnHgO035n//V5WE=
|
||||
github.com/prometheus/client_model v0.6.1 h1:ZKSh/rekM+n3CeS952MLRAdFwIKqeY8b62p8ais2e9E=
|
||||
|
|
@ -18,7 +27,16 @@ github.com/prometheus/common v0.55.0 h1:KEi6DK7lXW/m7Ig5i47x0vRzuBsHuvJdi5ee6Y3G
|
|||
github.com/prometheus/common v0.55.0/go.mod h1:2SECS4xJG1kd8XF9IcM1gMX6510RAEL65zxzNImwdc8=
|
||||
github.com/prometheus/procfs v0.15.1 h1:YagwOFzUgYfKKHX6Dr+sHT7km/hxC76UB0learggepc=
|
||||
github.com/prometheus/procfs v0.15.1/go.mod h1:fB45yRUv8NstnjriLhBQLuOUt+WW4BsoGhij/e3PBqk=
|
||||
github.com/rogpeppe/go-internal v1.10.0 h1:TMyTOH3F/DB16zRVcYyreMH6GnZZrwQVAoYjRBZyWFQ=
|
||||
github.com/rogpeppe/go-internal v1.10.0/go.mod h1:UQnix2H7Ngw/k4C5ijL5+65zddjncjaFoBhdsK/akog=
|
||||
github.com/stretchr/testify v1.9.0 h1:HtqpIVDClZ4nwg75+f6Lvsy/wHu+3BoSGCbBAcpTsTg=
|
||||
github.com/stretchr/testify v1.9.0/go.mod h1:r2ic/lqez/lEtzL7wO/rwa5dbSLXVDPFyf8C91i36aY=
|
||||
golang.org/x/sys v0.22.0 h1:RI27ohtqKCnwULzJLqkv897zojh5/DwS/ENaMzUOaWI=
|
||||
golang.org/x/sys v0.22.0/go.mod h1:/VUhepiaJMQUp4+oa/7Zr1D23ma6VTLIYjOOTFZPUcA=
|
||||
google.golang.org/protobuf v1.34.2 h1:6xV6lTsCfpGD21XK49h7MhtcApnLqkfYgPcdHftf6hg=
|
||||
google.golang.org/protobuf v1.34.2/go.mod h1:qYOHts0dSfpeUzUFpOMr/WGzszTmLH+DiWniOlNbLDw=
|
||||
gopkg.in/check.v1 v0.0.0-20161208181325-20d25e280405/go.mod h1:Co6ibVJAznAaIkqp8huTwlJQCZ016jof/cbN4VW5Yz0=
|
||||
gopkg.in/check.v1 v1.0.0-20201130134442-10cb98267c6c h1:Hei/4ADfdWqJk1ZMxUNpqntNwaWcugrBjAiHlqqRiVk=
|
||||
gopkg.in/check.v1 v1.0.0-20201130134442-10cb98267c6c/go.mod h1:JHkPIbrfpd72SG/EVd6muEfDQjcINNoR0C8j2r3qZ4Q=
|
||||
gopkg.in/yaml.v3 v3.0.1 h1:fxVm/GzAzEWqLHuvctI91KS9hhNmmWOoWu0XTYJS7CA=
|
||||
gopkg.in/yaml.v3 v3.0.1/go.mod h1:K4uyk7z7BCEPqu6E+C64Yfv1cQ7kz7rIZviUmN+EgEM=
|
||||
|
|
|
|||
167
main.go
167
main.go
|
|
@ -4,7 +4,9 @@ import (
|
|||
"encoding/json"
|
||||
"fmt"
|
||||
"io"
|
||||
"log"
|
||||
"net/http"
|
||||
"os"
|
||||
"sync"
|
||||
|
||||
"github.com/prometheus/client_golang/prometheus"
|
||||
|
|
@ -16,31 +18,35 @@ var customRegistry = prometheus.NewRegistry()
|
|||
|
||||
// Структура JSON
|
||||
type MetricRequest struct {
|
||||
ID string `json:"id"`
|
||||
Name string `json:"name"`
|
||||
URL string `json:"url"`
|
||||
Method string `json:"method"`
|
||||
Type string `json:"type"` // "gauge", "counter" и т.д.
|
||||
Metrics map[string]any `json:"metrics"`
|
||||
ID string `json:"id"`
|
||||
Name string `json:"name"`
|
||||
URL string `json:"url"`
|
||||
Method string `json:"method"`
|
||||
Type string `json:"type"`
|
||||
Metrics map[string]float64 `json:"metrics"`
|
||||
}
|
||||
|
||||
// Экспортёр метрик
|
||||
type MetricsExporter struct {
|
||||
gaugeMetrics map[string]*prometheus.GaugeVec
|
||||
counterMetrics map[string]*prometheus.CounterVec
|
||||
mu sync.Mutex // Защита от одновременного доступа
|
||||
gaugeMetrics map[string]*prometheus.GaugeVec
|
||||
counterMetrics map[string]*prometheus.CounterVec
|
||||
histogramMetrics map[string]*prometheus.HistogramVec
|
||||
summaryMetrics map[string]*prometheus.SummaryVec
|
||||
mu sync.Mutex // Защита от одновременного доступа
|
||||
}
|
||||
|
||||
// Создаём новый экспортёр
|
||||
func NewMetricsExporter() *MetricsExporter {
|
||||
return &MetricsExporter{
|
||||
gaugeMetrics: make(map[string]*prometheus.GaugeVec),
|
||||
counterMetrics: make(map[string]*prometheus.CounterVec),
|
||||
gaugeMetrics: make(map[string]*prometheus.GaugeVec),
|
||||
counterMetrics: make(map[string]*prometheus.CounterVec),
|
||||
histogramMetrics: make(map[string]*prometheus.HistogramVec),
|
||||
summaryMetrics: make(map[string]*prometheus.SummaryVec),
|
||||
}
|
||||
}
|
||||
|
||||
// Обновление или создание метрики
|
||||
func (me *MetricsExporter) UpdateMetric(name string, value interface{}) {
|
||||
func (me *MetricsExporter) UpdateMetric(request MetricRequest) {
|
||||
me.mu.Lock()
|
||||
defer me.mu.Unlock()
|
||||
|
||||
|
|
@ -52,44 +58,100 @@ func (me *MetricsExporter) UpdateMetric(name string, value interface{}) {
|
|||
labels := []string{"name", "url", "method"}
|
||||
labelValues := []string{request.Name, request.URL, request.Method}
|
||||
|
||||
if request.Type == "gauge" {
|
||||
// Регистрируем Gauge, если он ещё не существует
|
||||
if _, exists := me.gaugeMetrics[fullMetricName]; !exists {
|
||||
gaugeVec := prometheus.NewGaugeVec(
|
||||
prometheus.GaugeOpts{
|
||||
Name: fullMetricName,
|
||||
Help: fmt.Sprintf("Metric %s from %s", metricName, request.ID),
|
||||
},
|
||||
labels,
|
||||
)
|
||||
customRegistry.MustRegister(gaugeVec)
|
||||
me.gaugeMetrics[fullMetricName] = gaugeVec
|
||||
}
|
||||
// Обновляем значение Gauge
|
||||
me.gaugeMetrics[fullMetricName].WithLabelValues(labelValues...).Set(floatVal)
|
||||
|
||||
} else if request.Type == "counter" {
|
||||
// Регистрируем Counter, если он ещё не существует
|
||||
if _, exists := me.counterMetrics[fullMetricName]; !exists {
|
||||
counterVec := prometheus.NewCounterVec(
|
||||
prometheus.CounterOpts{
|
||||
Name: fullMetricName,
|
||||
Help: fmt.Sprintf("Metric %s from %s", metricName, request.ID),
|
||||
},
|
||||
labels,
|
||||
)
|
||||
customRegistry.MustRegister(counterVec)
|
||||
me.counterMetrics[fullMetricName] = counterVec
|
||||
}
|
||||
// Инкрементируем Counter
|
||||
me.counterMetrics[fullMetricName].WithLabelValues(labelValues...).Add(floatVal)
|
||||
// Обработка метрик в зависимости от типа
|
||||
switch request.Type {
|
||||
case "gauge":
|
||||
me.updateGauge(fullMetricName, labels, labelValues, value)
|
||||
case "counter":
|
||||
me.updateCounter(fullMetricName, labels, labelValues, value)
|
||||
case "histogram":
|
||||
me.updateHistogram(fullMetricName, labels, labelValues, value)
|
||||
case "summary":
|
||||
me.updateSummary(fullMetricName, labels, labelValues, value)
|
||||
default:
|
||||
log.Printf("Неподдерживаемый тип метрики: %s\n", request.Type)
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
// Обновление `Gauge` метрик
|
||||
func (me *MetricsExporter) updateGauge(name string, labels []string, labelValues []string, value float64) {
|
||||
if _, exists := me.gaugeMetrics[name]; !exists {
|
||||
gaugeVec := prometheus.NewGaugeVec(
|
||||
prometheus.GaugeOpts{
|
||||
Name: name,
|
||||
Help: fmt.Sprintf("Gauge metric %s", name),
|
||||
},
|
||||
labels,
|
||||
)
|
||||
customRegistry.MustRegister(gaugeVec)
|
||||
me.gaugeMetrics[name] = gaugeVec
|
||||
}
|
||||
me.gaugeMetrics[name].WithLabelValues(labelValues...).Set(value)
|
||||
log.Printf("Gauge обновлён: %s = %f\n", name, value)
|
||||
}
|
||||
|
||||
// Обновление `Counter` метрик
|
||||
func (me *MetricsExporter) updateCounter(name string, labels []string, labelValues []string, value float64) {
|
||||
if value < 0 {
|
||||
log.Printf("Ошибка: Counter %s не может быть отрицательным\n", name)
|
||||
return
|
||||
}
|
||||
if _, exists := me.counterMetrics[name]; !exists {
|
||||
counterVec := prometheus.NewCounterVec(
|
||||
prometheus.CounterOpts{
|
||||
Name: name,
|
||||
Help: fmt.Sprintf("Counter metric %s", name),
|
||||
},
|
||||
labels,
|
||||
)
|
||||
customRegistry.MustRegister(counterVec)
|
||||
me.counterMetrics[name] = counterVec
|
||||
}
|
||||
me.counterMetrics[name].WithLabelValues(labelValues...).Add(value)
|
||||
log.Printf("Counter обновлён: %s += %f\n", name, value)
|
||||
}
|
||||
|
||||
// Обновление `Histogram` метрик
|
||||
func (me *MetricsExporter) updateHistogram(name string, labels []string, labelValues []string, value float64) {
|
||||
if _, exists := me.histogramMetrics[name]; !exists {
|
||||
histogramVec := prometheus.NewHistogramVec(
|
||||
prometheus.HistogramOpts{
|
||||
Name: name,
|
||||
Help: fmt.Sprintf("Histogram metric %s", name),
|
||||
Buckets: prometheus.LinearBuckets(10, 10, 10), // Пример диапазонов
|
||||
},
|
||||
labels,
|
||||
)
|
||||
customRegistry.MustRegister(histogramVec)
|
||||
me.histogramMetrics[name] = histogramVec
|
||||
}
|
||||
me.histogramMetrics[name].WithLabelValues(labelValues...).Observe(value)
|
||||
log.Printf("Histogram обновлён: %s = %f\n", name, value)
|
||||
}
|
||||
|
||||
// Обновление `Summary` метрик
|
||||
func (me *MetricsExporter) updateSummary(name string, labels []string, labelValues []string, value float64) {
|
||||
if _, exists := me.summaryMetrics[name]; !exists {
|
||||
summaryVec := prometheus.NewSummaryVec(
|
||||
prometheus.SummaryOpts{
|
||||
Name: name,
|
||||
Help: fmt.Sprintf("Summary metric %s", name),
|
||||
Objectives: map[float64]float64{0.5: 0.05, 0.9: 0.01, 0.99: 0.001}, // Пример целей
|
||||
},
|
||||
labels,
|
||||
)
|
||||
customRegistry.MustRegister(summaryVec)
|
||||
me.summaryMetrics[name] = summaryVec
|
||||
}
|
||||
me.summaryMetrics[name].WithLabelValues(labelValues...).Observe(value)
|
||||
log.Printf("Summary обновлён: %s = %f\n", name, value)
|
||||
}
|
||||
|
||||
// Обработчик для приёма JSON
|
||||
func (me *MetricsExporter) JSONHandler(w http.ResponseWriter, r *http.Request) {
|
||||
if r.Method != http.MethodPost {
|
||||
log.Printf("Неверный метод: %s (ожидался POST)", r.Method)
|
||||
http.Error(w, "Only POST method is allowed", http.StatusMethodNotAllowed)
|
||||
return
|
||||
}
|
||||
|
|
@ -97,26 +159,33 @@ func (me *MetricsExporter) JSONHandler(w http.ResponseWriter, r *http.Request) {
|
|||
// Читаем тело запроса
|
||||
body, err := io.ReadAll(r.Body)
|
||||
if err != nil {
|
||||
log.Printf("Ошибка чтения тела запроса: %s\n", err)
|
||||
http.Error(w, "Failed to read request body", http.StatusInternalServerError)
|
||||
return
|
||||
}
|
||||
defer r.Body.Close()
|
||||
|
||||
// Парсим JSON
|
||||
var data MetricRequest
|
||||
if err := json.Unmarshal(body, &data); err != nil {
|
||||
var request MetricRequest
|
||||
if err := json.Unmarshal(body, &request); err != nil {
|
||||
log.Printf("Ошибка парсинга JSON: %s\nТело запроса: %s\n", err, string(body))
|
||||
http.Error(w, "Invalid JSON format", http.StatusBadRequest)
|
||||
return
|
||||
}
|
||||
|
||||
// Обновляем метрики
|
||||
me.UpdateMetric(name, value)
|
||||
log.Printf("Обновление метрик для запроса ID: %s\n", request.ID)
|
||||
me.UpdateMetric(request)
|
||||
|
||||
w.WriteHeader(http.StatusOK)
|
||||
w.Write([]byte("Metrics updated"))
|
||||
log.Printf("Метрики обновлены успешно для ID: %s\n", request.ID)
|
||||
}
|
||||
|
||||
func main() {
|
||||
log.SetOutput(os.Stdout) // Логируем в стандартный вывод
|
||||
log.SetFlags(log.Ldate | log.Ltime | log.Lshortfile)
|
||||
|
||||
exporter := NewMetricsExporter()
|
||||
|
||||
// Используем кастомный реестр в обработчике /metrics
|
||||
|
|
@ -124,8 +193,10 @@ func main() {
|
|||
http.HandleFunc("/update", exporter.JSONHandler) // Обработчик для приёма JSON
|
||||
|
||||
port := ":9101"
|
||||
fmt.Println("Starting server on port", port)
|
||||
log.Printf("Starting server on port %s\n", port)
|
||||
if err := http.ListenAndServe(port, nil); err != nil {
|
||||
fmt.Println("Error starting server:", err)
|
||||
log.Fatalf("Error starting server: %s\n", err)
|
||||
}
|
||||
}
|
||||
|
||||
//TODO: сделать переменные окружения, настроить канал, дописать тесты
|
||||
|
|
|
|||
|
|
@ -0,0 +1,12 @@
|
|||
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,11 +0,0 @@
|
|||
curl -X POST -H "Content-Type: application/json" -d '{
|
||||
"id": "mock_api_2",
|
||||
"name": "Mock /ping",
|
||||
"url": "http://127.0.0.1:8081/ping",
|
||||
"method": "GET",
|
||||
"type": "gauge",
|
||||
"metrics": {
|
||||
"response_time": 120.5,
|
||||
"status_code": 200,
|
||||
}
|
||||
}' http://localhost:9101/update
|
||||
Loading…
Reference in New Issue