Return list of parsed metrics, for autodiscovery.
This commit is contained in:
parent
529113a41a
commit
7e90814fde
@ -1,12 +1,16 @@
|
||||
package application
|
||||
|
||||
import (
|
||||
"encoding/json"
|
||||
|
||||
"go.dev.pztrn.name/metricator/internal/models"
|
||||
)
|
||||
|
||||
// Responds with needed data. First parameter is a type of data needed (like metric name),
|
||||
// second parameter is actual metric name. Second parameter also can be empty.
|
||||
func (a *Application) respond(rInfo *models.RequestInfo) string {
|
||||
// If metric was requested - return only it.
|
||||
if rInfo.Metric != "" {
|
||||
metric, err := a.storage.Get(rInfo.Metric)
|
||||
if err != nil {
|
||||
return ""
|
||||
@ -14,3 +18,16 @@ func (a *Application) respond(rInfo *models.RequestInfo) string {
|
||||
|
||||
return metric.GetValue()
|
||||
}
|
||||
|
||||
// Otherwise we should get all metrics as slice and return them as string.
|
||||
// This is needed for metrics autodiscovery.
|
||||
metrics := a.storage.GetAsSlice()
|
||||
|
||||
metricsBytes, err := json.Marshal(metrics)
|
||||
if err != nil {
|
||||
// ToDo: log error
|
||||
return ""
|
||||
}
|
||||
|
||||
return string(metricsBytes)
|
||||
}
|
||||
|
@ -3,21 +3,21 @@ package models
|
||||
// Metric is a generic metric structure.
|
||||
type Metric struct {
|
||||
// Metric name.
|
||||
name string
|
||||
Name string
|
||||
// HELP data, if present.
|
||||
description string
|
||||
Description string
|
||||
// Additional parameters, data inside "{}".
|
||||
params []string
|
||||
Params []string
|
||||
// Metric value.
|
||||
value string
|
||||
Value string
|
||||
}
|
||||
|
||||
// NewMetric creates new structure for storing single metric data.
|
||||
func NewMetric(name, description string, params []string) Metric {
|
||||
m := Metric{
|
||||
name: name,
|
||||
description: description,
|
||||
params: params,
|
||||
Name: name,
|
||||
Description: description,
|
||||
Params: params,
|
||||
}
|
||||
|
||||
return m
|
||||
@ -25,10 +25,10 @@ func NewMetric(name, description string, params []string) Metric {
|
||||
|
||||
// GetValue returns metric's value.
|
||||
func (m *Metric) GetValue() string {
|
||||
return m.value
|
||||
return m.Value
|
||||
}
|
||||
|
||||
// SetValue sets value for metric.
|
||||
func (m *Metric) SetValue(value string) {
|
||||
m.value = value
|
||||
m.Value = value
|
||||
}
|
||||
|
@ -7,6 +7,8 @@ import "go.dev.pztrn.name/metricator/internal/models"
|
||||
type GenericStorage interface {
|
||||
// Get returns data from storage by key.
|
||||
Get(string) (models.Metric, error)
|
||||
// GetAsSlice returns all data from storage as slice.
|
||||
GetAsSlice() []models.Metric
|
||||
// GetDoneChan returns a channel which should be used to block execution
|
||||
// until storage's routines are completed.
|
||||
GetDoneChan() chan struct{}
|
||||
|
@ -46,6 +46,17 @@ func (s *Storage) Get(key string) (models.Metric, error) {
|
||||
return data, nil
|
||||
}
|
||||
|
||||
// GetAsSlice returns all data from storage as slice.
|
||||
func (s *Storage) GetAsSlice() []models.Metric {
|
||||
metrics := make([]models.Metric, 0, len(s.data))
|
||||
|
||||
for _, metric := range s.data {
|
||||
metrics = append(metrics, metric)
|
||||
}
|
||||
|
||||
return metrics
|
||||
}
|
||||
|
||||
// GetDoneChan returns a channel which should be used to block execution
|
||||
// until storage's routines are completed.
|
||||
func (s *Storage) GetDoneChan() chan struct{} {
|
||||
|
Loading…
Reference in New Issue
Block a user