Files
object-browser/api/admin_info.go
Alex 80c03839a4 Fixed lint issues with files (#3234)
Signed-off-by: Benjamin Perez <benjamin@bexsoft.net>
2024-02-12 21:55:17 -08:00

1193 lines
27 KiB
Go

// This file is part of MinIO Console Server
// Copyright (c) 2021 MinIO, Inc.
//
// This program is free software: you can redistribute it and/or modify
// it under the terms of the GNU Affero General Public License as published by
// the Free Software Foundation, either version 3 of the License, or
// (at your option) any later version.
//
// This program is distributed in the hope that it will be useful,
// but WITHOUT ANY WARRANTY; without even the implied warranty of
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
// GNU Affero General Public License for more details.
//
// You should have received a copy of the GNU Affero General Public License
// along with this program. If not, see <http://www.gnu.org/licenses/>.
package api
import (
"context"
"encoding/json"
"errors"
"fmt"
"net/http"
"net/url"
"regexp"
"strings"
"sync"
"time"
"github.com/minio/console/pkg/utils"
"github.com/go-openapi/runtime/middleware"
"github.com/minio/console/api/operations"
systemApi "github.com/minio/console/api/operations/system"
"github.com/minio/console/models"
)
func registerAdminInfoHandlers(api *operations.ConsoleAPI) {
// return usage stats
api.SystemAdminInfoHandler = systemApi.AdminInfoHandlerFunc(func(params systemApi.AdminInfoParams, session *models.Principal) middleware.Responder {
infoResp, err := getAdminInfoResponse(session, params)
if err != nil {
return systemApi.NewAdminInfoDefault(err.Code).WithPayload(err.APIError)
}
return systemApi.NewAdminInfoOK().WithPayload(infoResp)
})
// return single widget results
api.SystemDashboardWidgetDetailsHandler = systemApi.DashboardWidgetDetailsHandlerFunc(func(params systemApi.DashboardWidgetDetailsParams, _ *models.Principal) middleware.Responder {
infoResp, err := getAdminInfoWidgetResponse(params)
if err != nil {
return systemApi.NewDashboardWidgetDetailsDefault(err.Code).WithPayload(err.APIError)
}
return systemApi.NewDashboardWidgetDetailsOK().WithPayload(infoResp)
})
}
type UsageInfo struct {
Buckets int64
Objects int64
Usage int64
DrivesUsage int64
Servers []*models.ServerProperties
EndpointNotReady bool
Backend *models.BackendProperties
}
// GetAdminInfo invokes admin info and returns a parsed `UsageInfo` structure
func GetAdminInfo(ctx context.Context, client MinioAdmin) (*UsageInfo, error) {
serverInfo, err := client.serverInfo(ctx)
if err != nil {
return nil, err
}
// we are trimming uint64 to int64 this will report an incorrect measurement for numbers greater than
// 9,223,372,036,854,775,807
backendType := serverInfo.Backend.Type
rrSCParity := serverInfo.Backend.RRSCParity
standardSCParity := serverInfo.Backend.StandardSCParity
onlineDrives := serverInfo.Backend.OnlineDisks
offlineDrives := serverInfo.Backend.OfflineDisks
var usedSpace int64
// serverArray contains the serverProperties which describe the servers in the network
var serverArray []*models.ServerProperties
for _, serv := range serverInfo.Servers {
drives := []*models.ServerDrives{}
for _, drive := range serv.Disks {
usedSpace += int64(drive.UsedSpace)
drives = append(drives, &models.ServerDrives{
State: drive.State,
UUID: drive.UUID,
Endpoint: drive.Endpoint,
RootDisk: drive.RootDisk,
DrivePath: drive.DrivePath,
Healing: drive.Healing,
Model: drive.Model,
TotalSpace: int64(drive.TotalSpace),
UsedSpace: int64(drive.UsedSpace),
AvailableSpace: int64(drive.AvailableSpace),
})
}
newServer := &models.ServerProperties{
State: serv.State,
Endpoint: serv.Endpoint,
Uptime: serv.Uptime,
Version: serv.Version,
CommitID: serv.CommitID,
PoolNumber: int64(serv.PoolNumber),
Network: serv.Network,
Drives: drives,
}
serverArray = append(serverArray, newServer)
}
backendData := &models.BackendProperties{
BackendType: string(backendType),
RrSCParity: int64(rrSCParity),
StandardSCParity: int64(standardSCParity),
OnlineDrives: int64(onlineDrives),
OfflineDrives: int64(offlineDrives),
}
return &UsageInfo{
Buckets: int64(serverInfo.Buckets.Count),
Objects: int64(serverInfo.Objects.Count),
Usage: int64(serverInfo.Usage.Size),
DrivesUsage: usedSpace,
Servers: serverArray,
Backend: backendData,
}, nil
}
type Target struct {
Expr string
Interval string
LegendFormat string
Step int32
InitialTime int64
}
type ReduceOptions struct {
Calcs []string
}
type MetricOptions struct {
ReduceOptions ReduceOptions
}
type Metric struct {
ID int32
Title string
Type string
Options MetricOptions
Targets []Target
GridPos GridPos
MaxDataPoints int32
}
type GridPos struct {
H int32
W int32
X int32
Y int32
}
type WidgetLabel struct {
Name string
}
var labels = []WidgetLabel{
{Name: "instance"},
{Name: "drive"},
{Name: "server"},
{Name: "api"},
}
var widgets = []Metric{
{
ID: 1,
Title: "Uptime",
Type: "stat",
MaxDataPoints: 100,
GridPos: GridPos{
H: 6,
W: 3,
X: 0,
Y: 0,
},
Options: MetricOptions{
ReduceOptions: ReduceOptions{
Calcs: []string{
"mean",
},
},
},
Targets: []Target{
{
Expr: `time() - max(minio_node_process_starttime_seconds{$__query})`,
LegendFormat: "{{instance}}",
Step: 60,
},
},
},
{
ID: 65,
Title: "Total S3 Traffic Inbound",
Type: "stat",
MaxDataPoints: 100,
GridPos: GridPos{
H: 3,
W: 3,
X: 3,
Y: 0,
},
Options: MetricOptions{
ReduceOptions: ReduceOptions{
Calcs: []string{
"last",
},
},
},
Targets: []Target{
{
Expr: `sum by (instance) (minio_s3_traffic_received_bytes{$__query})`,
LegendFormat: "{{instance}}",
Step: 60,
},
},
},
{
ID: 50,
Title: "Current Usable Free Capacity",
Type: "gauge",
MaxDataPoints: 100,
GridPos: GridPos{
H: 6,
W: 3,
X: 6,
Y: 0,
},
Options: MetricOptions{
ReduceOptions: ReduceOptions{
Calcs: []string{
"lastNotNull",
},
},
},
Targets: []Target{
{
Expr: `topk(1, sum(minio_cluster_capacity_usable_total_bytes{$__query}) by (instance))`,
LegendFormat: "Total Usable",
Step: 300,
},
{
Expr: `topk(1, sum(minio_cluster_capacity_usable_free_bytes{$__query}) by (instance))`,
LegendFormat: "Usable Free",
Step: 300,
},
{
Expr: `topk(1, sum(minio_cluster_capacity_usable_total_bytes{$__query}) by (instance)) - topk(1, sum(minio_cluster_capacity_usable_free_bytes{$__query}) by (instance))`,
LegendFormat: "Used Space",
Step: 300,
},
},
},
{
ID: 51,
Title: "Current Usable Total Bytes",
Type: "gauge",
MaxDataPoints: 100,
GridPos: GridPos{
H: 6,
W: 3,
X: 6,
Y: 0,
},
Options: MetricOptions{
ReduceOptions: ReduceOptions{
Calcs: []string{
"lastNotNull",
},
},
},
Targets: []Target{
{
Expr: `topk(1, sum(minio_cluster_capacity_usable_total_bytes{$__query}) by (instance))`,
LegendFormat: "",
Step: 300,
},
},
},
{
ID: 68,
Title: "Data Usage Growth",
Type: "graph",
GridPos: GridPos{
H: 6,
W: 7,
X: 9,
Y: 0,
},
Targets: []Target{
{
Expr: `minio_cluster_usage_total_bytes{$__query}`,
LegendFormat: "Used Capacity",
InitialTime: -180,
Step: 10,
},
},
},
{
ID: 52,
Title: "Object size distribution",
Type: "bargauge",
GridPos: GridPos{
H: 6,
W: 5,
X: 16,
Y: 0,
},
Options: MetricOptions{
ReduceOptions: ReduceOptions{
Calcs: []string{
"mean",
},
},
},
Targets: []Target{
{
Expr: `minio_cluster_objects_size_distribution{$__query}`,
LegendFormat: "{{range}}",
Step: 300,
},
},
},
{
ID: 61,
Title: "Total Open FDs",
Type: "stat",
MaxDataPoints: 100,
GridPos: GridPos{
H: 3,
W: 3,
X: 21,
Y: 0,
},
Options: MetricOptions{
ReduceOptions: ReduceOptions{
Calcs: []string{
"last",
},
},
},
Targets: []Target{
{
Expr: `sum(minio_node_file_descriptor_open_total{$__query})`,
LegendFormat: "",
Step: 60,
},
},
},
{
ID: 64,
Title: "Total S3 Traffic Outbound",
Type: "stat",
MaxDataPoints: 100,
GridPos: GridPos{
H: 3,
W: 3,
X: 3,
Y: 3,
},
Options: MetricOptions{
ReduceOptions: ReduceOptions{
Calcs: []string{
"last",
},
},
},
Targets: []Target{
{
Expr: `sum by (instance) (minio_s3_traffic_sent_bytes{$__query})`,
LegendFormat: "",
Step: 60,
},
},
},
{
ID: 62,
Title: "Total Goroutines",
Type: "stat",
MaxDataPoints: 100,
GridPos: GridPos{
H: 3,
W: 3,
X: 21,
Y: 3,
},
Options: MetricOptions{
ReduceOptions: ReduceOptions{
Calcs: []string{
"last",
},
},
},
Targets: []Target{
{
Expr: `sum without (server,instance) (minio_node_go_routine_total{$__query})`,
LegendFormat: "",
Step: 60,
},
},
},
{
ID: 53,
Title: "Total Online Servers",
Type: "stat",
MaxDataPoints: 100,
GridPos: GridPos{
H: 2,
W: 3,
X: 0,
Y: 6,
},
Options: MetricOptions{
ReduceOptions: ReduceOptions{
Calcs: []string{
"mean",
},
},
},
Targets: []Target{
{
Expr: `minio_cluster_nodes_online_total{$__query}`,
LegendFormat: "",
Step: 60,
},
},
},
{
ID: 9,
Title: "Total Online Drives",
Type: "stat",
MaxDataPoints: 100,
GridPos: GridPos{
H: 2,
W: 3,
X: 3,
Y: 6,
},
Options: MetricOptions{
ReduceOptions: ReduceOptions{
Calcs: []string{
"mean",
},
},
},
Targets: []Target{
{
Expr: `minio_cluster_drive_online_total{$__query}`,
LegendFormat: "Total online drives in MinIO Cluster",
Step: 60,
},
},
},
{
ID: 66,
Title: "Number of Buckets",
Type: "stat",
MaxDataPoints: 5,
GridPos: GridPos{
H: 3,
W: 3,
X: 6,
Y: 6,
},
Options: MetricOptions{
ReduceOptions: ReduceOptions{
Calcs: []string{
"lastNotNull",
},
},
},
Targets: []Target{
{
Expr: `minio_cluster_bucket_total{$__query}`,
LegendFormat: "",
Step: 100,
},
},
},
{
ID: 63,
Title: "S3 API Data Received Rate ",
Type: "graph",
GridPos: GridPos{
H: 6,
W: 7,
X: 9,
Y: 6,
},
Targets: []Target{
{
Expr: `sum by (server) (rate(minio_s3_traffic_received_bytes{$__query}[$__rate_interval]))`,
LegendFormat: "Data Received [{{server}}]",
},
},
},
{
ID: 70,
Title: "S3 API Data Sent Rate ",
Type: "graph",
GridPos: GridPos{
H: 6,
W: 8,
X: 16,
Y: 6,
},
Targets: []Target{
{
Expr: `sum by (server) (rate(minio_s3_traffic_sent_bytes{$__query}[$__rate_interval]))`,
LegendFormat: "Data Sent [{{server}}]",
},
},
},
{
ID: 69,
Title: "Total Offline Servers",
Type: "stat",
MaxDataPoints: 100,
GridPos: GridPos{
H: 2,
W: 3,
X: 0,
Y: 8,
},
Options: MetricOptions{
ReduceOptions: ReduceOptions{
Calcs: []string{
"mean",
},
},
},
Targets: []Target{
{
Expr: `minio_cluster_nodes_offline_total{$__query}`,
LegendFormat: "",
Step: 60,
},
},
},
{
ID: 78,
Title: "Total Offline Drives",
Type: "stat",
MaxDataPoints: 100,
GridPos: GridPos{
H: 2,
W: 3,
X: 3,
Y: 8,
},
Options: MetricOptions{
ReduceOptions: ReduceOptions{
Calcs: []string{
"mean",
},
},
},
Targets: []Target{
{
Expr: `minio_cluster_drive_offline_total{$__query}`,
LegendFormat: "",
Step: 60,
},
},
},
{
ID: 44,
Title: "Number of Objects",
Type: "stat",
MaxDataPoints: 100,
GridPos: GridPos{
H: 3,
W: 3,
X: 6,
Y: 9,
},
Options: MetricOptions{
ReduceOptions: ReduceOptions{
Calcs: []string{
"lastNotNull",
},
},
},
Targets: []Target{
{
Expr: `minio_cluster_usage_object_total{$__query}`,
LegendFormat: "",
},
},
},
{
ID: 80,
Title: "Time Since Last Heal Activity",
Type: "stat",
MaxDataPoints: 100,
GridPos: GridPos{
H: 2,
W: 3,
X: 0,
Y: 10,
},
Options: MetricOptions{
ReduceOptions: ReduceOptions{
Calcs: []string{
"last",
},
},
},
Targets: []Target{
{
Expr: `minio_heal_time_last_activity_nano_seconds{$__query}`,
LegendFormat: "{{server}}",
Step: 60,
},
},
},
{
ID: 81,
Title: "Time Since Last Scan Activity",
Type: "stat",
MaxDataPoints: 100,
GridPos: GridPos{
H: 2,
W: 3,
X: 3,
Y: 10,
},
Options: MetricOptions{
ReduceOptions: ReduceOptions{
Calcs: []string{
"last",
},
},
},
Targets: []Target{
{
Expr: `minio_usage_last_activity_nano_seconds{$__query}`,
LegendFormat: "{{server}}",
Step: 60,
},
},
},
{
ID: 60,
Title: "S3 API Request Rate",
Type: "graph",
GridPos: GridPos{
H: 10,
W: 12,
X: 0,
Y: 12,
},
Targets: []Target{
{
Expr: `sum by (server,api) (increase(minio_s3_requests_total{$__query}[$__rate_interval]))`,
LegendFormat: "{{server,api}}",
},
},
},
{
ID: 71,
Title: "S3 API Request Error Rate",
Type: "graph",
GridPos: GridPos{
H: 10,
W: 12,
X: 12,
Y: 12,
},
Targets: []Target{
{
Expr: `sum by (server,api) (increase(minio_s3_requests_errors_total{$__query}[$__rate_interval]))`,
LegendFormat: "{{server,api}}",
},
},
},
{
ID: 17,
Title: "Internode Data Transfer",
Type: "graph",
GridPos: GridPos{
H: 8,
W: 24,
X: 0,
Y: 22,
},
Targets: []Target{
{
Expr: `rate(minio_inter_node_traffic_sent_bytes{$__query}[$__rate_interval])`,
LegendFormat: "Internode Bytes Received [{{server}}]",
Step: 4,
},
{
Expr: `rate(minio_inter_node_traffic_sent_bytes{$__query}[$__rate_interval])`,
LegendFormat: "Internode Bytes Received [{{server}}]",
},
},
},
{
ID: 77,
Title: "Node CPU Usage",
Type: "graph",
GridPos: GridPos{
H: 9,
W: 12,
X: 0,
Y: 30,
},
Targets: []Target{
{
Expr: `rate(minio_node_process_cpu_total_seconds{$__query}[$__rate_interval])`,
LegendFormat: "CPU Usage Rate [{{server}}]",
},
},
},
{
ID: 76,
Title: "Node Memory Usage",
Type: "graph",
GridPos: GridPos{
H: 9,
W: 12,
X: 12,
Y: 30,
},
Targets: []Target{
{
Expr: `minio_node_process_resident_memory_bytes{$__query}`,
LegendFormat: "Memory Used [{{server}}]",
},
},
},
{
ID: 74,
Title: "Drive Used Capacity",
Type: "graph",
GridPos: GridPos{
H: 8,
W: 12,
X: 0,
Y: 39,
},
Targets: []Target{
{
Expr: `minio_node_drive_used_bytes{$__query}`,
LegendFormat: "Used Capacity [{{server}}:{{drive}}]",
},
},
},
{
ID: 82,
Title: "Drives Free Inodes",
Type: "graph",
GridPos: GridPos{
H: 8,
W: 12,
X: 12,
Y: 39,
},
Targets: []Target{
{
Expr: `minio_cluster_drive_free_inodes{$__query}`,
LegendFormat: "Free Inodes [{{server}}:{{drive}}]",
},
},
},
{
ID: 11,
Title: "Node Syscalls",
Type: "graph",
GridPos: GridPos{
H: 9,
W: 12,
X: 0,
Y: 47,
},
Targets: []Target{
{
Expr: `rate(minio_node_syscall_read_total{$__query}[$__rate_interval])`,
LegendFormat: "Read Syscalls [{{server}}]",
Step: 60,
},
{
Expr: `rate(minio_node_syscall_read_total{$__query}[$__rate_interval])`,
LegendFormat: "Read Syscalls [{{server}}]",
},
},
},
{
ID: 8,
Title: "Node File Descriptors",
Type: "graph",
GridPos: GridPos{
H: 9,
W: 12,
X: 12,
Y: 47,
},
Targets: []Target{
{
Expr: `minio_node_file_descriptor_open_total{$__query}`,
LegendFormat: "Open FDs [{{server}}]",
},
},
},
{
ID: 73,
Title: "Node IO",
Type: "graph",
GridPos: GridPos{
H: 8,
W: 24,
X: 0,
Y: 56,
},
Targets: []Target{
{
Expr: `rate(minio_node_io_rchar_bytes{$__query}[$__rate_interval])`,
LegendFormat: "Node RChar [{{server}}]",
},
{
Expr: `rate(minio_node_io_wchar_bytes{$__query}[$__rate_interval])`,
LegendFormat: "Node WChar [{{server}}]",
},
},
},
}
type Widget struct {
Title string
Type string
}
type DataResult struct {
Metric map[string]string `json:"metric"`
Values []interface{} `json:"values"`
}
type PromRespData struct {
ResultType string `json:"resultType"`
Result []DataResult `json:"result"`
}
type PromResp struct {
Status string `json:"status"`
Data PromRespData `json:"data"`
}
type LabelResponse struct {
Status string `json:"status"`
Data []string `json:"data"`
}
type LabelResults struct {
Label string
Response LabelResponse
}
// getAdminInfoResponse returns the response containing total buckets, objects and usage.
func getAdminInfoResponse(session *models.Principal, params systemApi.AdminInfoParams) (*models.AdminInfoResponse, *CodedAPIError) {
ctx, cancel := context.WithCancel(params.HTTPRequest.Context())
defer cancel()
prometheusURL := ""
if !*params.DefaultOnly {
promURL := getPrometheusURL()
if promURL != "" {
prometheusURL = promURL
}
}
mAdmin, err := NewMinioAdminClient(params.HTTPRequest.Context(), session)
if err != nil {
return nil, ErrorWithContext(ctx, err)
}
sessionResp, err2 := getUsageWidgetsForDeployment(ctx, prometheusURL, AdminClient{Client: mAdmin})
if err2 != nil {
return nil, ErrorWithContext(ctx, err2)
}
return sessionResp, nil
}
func getUsageWidgetsForDeployment(ctx context.Context, prometheusURL string, adminClient MinioAdmin) (*models.AdminInfoResponse, error) {
prometheusStatus := models.AdminInfoResponseAdvancedMetricsStatusAvailable
if prometheusURL == "" {
prometheusStatus = models.AdminInfoResponseAdvancedMetricsStatusNotConfigured
}
if prometheusURL != "" && !testPrometheusURL(ctx, prometheusURL) {
prometheusStatus = models.AdminInfoResponseAdvancedMetricsStatusUnavailable
}
sessionResp := &models.AdminInfoResponse{
AdvancedMetricsStatus: prometheusStatus,
}
doneCh := make(chan error)
go func() {
defer close(doneCh)
// serialize output
usage, err := GetAdminInfo(ctx, adminClient)
if err != nil {
doneCh <- err
}
if usage != nil {
sessionResp.Buckets = usage.Buckets
sessionResp.Objects = usage.Objects
sessionResp.Usage = usage.Usage
sessionResp.Servers = usage.Servers
sessionResp.Backend = usage.Backend
}
}()
var wdgts []*models.Widget
if prometheusStatus == models.AdminInfoResponseAdvancedMetricsStatusAvailable {
// We will tell the frontend about a list of widgets so it can fetch the ones it wants
for _, m := range widgets {
wdgtResult := models.Widget{
ID: m.ID,
Title: m.Title,
Type: m.Type,
}
if len(m.Options.ReduceOptions.Calcs) > 0 {
wdgtResult.Options = &models.WidgetOptions{
ReduceOptions: &models.WidgetOptionsReduceOptions{
Calcs: m.Options.ReduceOptions.Calcs,
},
}
}
wdgts = append(wdgts, &wdgtResult)
}
sessionResp.Widgets = wdgts
}
// wait for mc admin info
err := <-doneCh
if err != nil {
return nil, err
}
return sessionResp, nil
}
func unmarshalPrometheus(ctx context.Context, httpClnt *http.Client, endpoint string, data interface{}) bool {
req, err := http.NewRequestWithContext(ctx, http.MethodGet, endpoint, nil)
if err != nil {
ErrorWithContext(ctx, fmt.Errorf("Unable to create the request to fetch labels from prometheus: %w", err))
return true
}
prometheusBearer := getPrometheusAuthToken()
if prometheusBearer != "" {
req.Header.Add("Authorization", fmt.Sprintf("Bearer %s", prometheusBearer))
}
resp, err := httpClnt.Do(req)
if err != nil {
ErrorWithContext(ctx, fmt.Errorf("Unable to fetch labels from prometheus: %w", err))
return true
}
defer resp.Body.Close()
if resp.StatusCode != http.StatusOK {
ErrorWithContext(ctx, fmt.Errorf("Unexpected status code from prometheus (%s)", resp.Status))
return true
}
if err = json.NewDecoder(resp.Body).Decode(data); err != nil {
ErrorWithContext(ctx, fmt.Errorf("Unexpected error from prometheus: %w", err))
return true
}
return false
}
func testPrometheusURL(ctx context.Context, url string) bool {
clientIP := utils.ClientIPFromContext(ctx)
httpClnt := GetConsoleHTTPClient(url, clientIP)
req, err := http.NewRequestWithContext(ctx, http.MethodGet, url+"/-/healthy", nil)
if err != nil {
ErrorWithContext(ctx, fmt.Errorf("error Building Request: (%v)", err))
return false
}
prometheusBearer := getPrometheusAuthToken()
if prometheusBearer != "" {
req.Header.Add("Authorization", fmt.Sprintf("Bearer %s", prometheusBearer))
}
response, err := httpClnt.Do(req)
if err != nil {
ErrorWithContext(ctx, fmt.Errorf("default Prometheus URL not reachable, trying root testing: (%v)", err))
newTestURL := req.URL.Scheme + "://" + req.URL.Host + "/-/healthy"
req2, err := http.NewRequestWithContext(ctx, http.MethodGet, newTestURL, nil)
if err != nil {
ErrorWithContext(ctx, fmt.Errorf("error Building Root Request: (%v)", err))
return false
}
rootResponse, err := httpClnt.Do(req2)
if err != nil {
// URL & Root tests didn't work. Prometheus not reachable
ErrorWithContext(ctx, fmt.Errorf("root Prometheus URL not reachable: (%v)", err))
return false
}
return rootResponse.StatusCode == http.StatusOK
}
return response.StatusCode == http.StatusOK
}
func getAdminInfoWidgetResponse(params systemApi.DashboardWidgetDetailsParams) (*models.WidgetDetails, *CodedAPIError) {
ctx, cancel := context.WithCancel(params.HTTPRequest.Context())
defer cancel()
prometheusURL := getPrometheusURL()
prometheusJobID := getPrometheusJobID()
prometheusExtraLabels := getPrometheusExtraLabels()
selector := fmt.Sprintf(`job="%s"`, prometheusJobID)
if strings.TrimSpace(prometheusExtraLabels) != "" {
selector = fmt.Sprintf(`job="%s",%s`, prometheusJobID, prometheusExtraLabels)
}
clientIP := getClientIP(params.HTTPRequest)
ctx = context.WithValue(ctx, utils.ContextClientIP, clientIP)
return getWidgetDetails(ctx, prometheusURL, selector, params.WidgetID, params.Step, params.Start, params.End)
}
func getWidgetDetails(ctx context.Context, prometheusURL string, selector string, widgetID int32, step *int32, start *int64, end *int64) (*models.WidgetDetails, *CodedAPIError) {
// We test if prometheus URL is reachable. this is meant to avoid unuseful calls and application hang.
if !testPrometheusURL(ctx, prometheusURL) {
return nil, ErrorWithContext(ctx, errors.New("prometheus URL is unreachable"))
}
clientIP := utils.ClientIPFromContext(ctx)
httpClnt := GetConsoleHTTPClient(prometheusURL, clientIP)
labelResultsCh := make(chan LabelResults)
for _, lbl := range labels {
go func(lbl WidgetLabel) {
endpoint := fmt.Sprintf("%s/api/v1/label/%s/values", prometheusURL, lbl.Name)
var response LabelResponse
if unmarshalPrometheus(ctx, httpClnt, endpoint, &response) {
return
}
labelResultsCh <- LabelResults{Label: lbl.Name, Response: response}
}(lbl)
}
labelMap := make(map[string][]string)
// wait for as many goroutines that come back in less than 1 second
LabelsWaitLoop:
for {
select {
case <-time.After(1 * time.Second):
break LabelsWaitLoop
case res := <-labelResultsCh:
labelMap[res.Label] = res.Response.Data
if len(labelMap) >= len(labels) {
break LabelsWaitLoop
}
}
}
// launch a goroutines per widget
for _, m := range widgets {
if m.ID != widgetID {
continue
}
var (
wg sync.WaitGroup
targetResults = make([]*models.ResultTarget, len(m.Targets))
)
// for each target we will launch another goroutine to fetch the values
for idx, target := range m.Targets {
wg.Add(1)
go func(idx int, target Target, inStep *int32, inStart *int64, inEnd *int64) {
defer wg.Done()
apiType := "query_range"
now := time.Now()
var initTime int64 = -15
if target.InitialTime != 0 {
initTime = target.InitialTime
}
timeCalculated := time.Duration(initTime * int64(time.Minute))
extraParamters := fmt.Sprintf("&start=%d&end=%d", now.Add(timeCalculated).Unix(), now.Unix())
var step int32 = 60
if target.Step > 0 {
step = target.Step
}
if inStep != nil && *inStep > 0 {
step = *inStep
}
if step > 0 {
extraParamters = fmt.Sprintf("%s&step=%d", extraParamters, step)
}
if inStart != nil && inEnd != nil {
extraParamters = fmt.Sprintf("&start=%d&end=%d&step=%d", *inStart, *inEnd, *inStep)
}
// replace the `$__rate_interval` global for step with unit (s for seconds)
queryExpr := strings.ReplaceAll(target.Expr, "$__rate_interval", fmt.Sprintf("%ds", 240))
if strings.Contains(queryExpr, "$") {
re := regexp.MustCompile(`\$([a-z]+)`)
for _, match := range re.FindAllStringSubmatch(queryExpr, -1) {
if val, ok := labelMap[match[1]]; ok {
queryExpr = strings.ReplaceAll(queryExpr, "$"+match[1], fmt.Sprintf("(%s)", strings.Join(val, "|")))
}
}
}
queryExpr = strings.ReplaceAll(queryExpr, "$__query", selector)
endpoint := fmt.Sprintf("%s/api/v1/%s?query=%s%s", prometheusURL, apiType, url.QueryEscape(queryExpr), extraParamters)
var response PromResp
if unmarshalPrometheus(ctx, httpClnt, endpoint, &response) {
return
}
targetResult := models.ResultTarget{
LegendFormat: target.LegendFormat,
ResultType: response.Data.ResultType,
}
for _, r := range response.Data.Result {
targetResult.Result = append(targetResult.Result, &models.WidgetResult{
Metric: r.Metric,
Values: r.Values,
})
}
targetResults[idx] = &targetResult
}(idx, target, step, start, end)
}
wg.Wait()
wdgtResult := models.WidgetDetails{
ID: m.ID,
Title: m.Title,
Type: m.Type,
}
if len(m.Options.ReduceOptions.Calcs) > 0 {
wdgtResult.Options = &models.WidgetDetailsOptions{
ReduceOptions: &models.WidgetDetailsOptionsReduceOptions{
Calcs: m.Options.ReduceOptions.Calcs,
},
}
}
for _, res := range targetResults {
if res != nil {
wdgtResult.Targets = append(wdgtResult.Targets, res)
}
}
return &wdgtResult, nil
}
return nil, &CodedAPIError{Code: 404, APIError: &models.APIError{Message: "Widget not found"}}
}