-
Notifications
You must be signed in to change notification settings - Fork 721
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
[tmpnet] Add check for log and metric collection to monitoring action
- Loading branch information
Showing
7 changed files
with
347 additions
and
17 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,253 @@ | ||
| // Copyright (C) 2019-2024, Ava Labs, Inc. All rights reserved. | ||
| // See the file LICENSE for licensing terms. | ||
|
|
||
| package tmpnet | ||
|
|
||
| import ( | ||
| "context" | ||
| "encoding/base64" | ||
| "encoding/json" | ||
| "errors" | ||
| "fmt" | ||
| "io" | ||
| "net/http" | ||
| "net/url" | ||
| "strconv" | ||
| "strings" | ||
| "time" | ||
|
|
||
| "github.com/prometheus/client_golang/api" | ||
| "github.com/prometheus/client_golang/api/prometheus/v1" | ||
| "github.com/prometheus/common/model" | ||
| "go.uber.org/zap" | ||
|
|
||
| "github.com/ava-labs/avalanchego/utils/logging" | ||
| ) | ||
|
|
||
| // CheckLogsExist checks if logs exist for the given network. Github labels are also | ||
| // included if provided as env vars (GH_*). | ||
| func CheckLogsExist(ctx context.Context, log logging.Logger, networkUUID string) error { | ||
| username, password, err := getCollectorCredentials(promtailCmd) | ||
| if err != nil { | ||
| return fmt.Errorf("failed to get collector credentials: %w", err) | ||
| } | ||
| query, err := getCheckLogsQuery(networkUUID) | ||
| if err != nil { | ||
| return err | ||
| } | ||
| url := getLokiURL() | ||
|
|
||
| log.Info("checking if logs exist", | ||
| zap.String("url", url), | ||
| zap.String("query", query), | ||
| ) | ||
|
|
||
| logsCount, err := queryLoki(ctx, url, username, password, query) | ||
| if err != nil { | ||
| return err | ||
| } | ||
|
|
||
| if logsCount > 0 { | ||
| log.Info("logs exist", | ||
| zap.Int("count", logsCount), | ||
| ) | ||
| return nil | ||
| } | ||
|
|
||
| return errors.New("logs not found") | ||
| } | ||
|
|
||
| // getCheckLogsQuery returns the query to check if logs exist. | ||
| func getCheckLogsQuery(networkUUID string) (string, error) { | ||
| selectors, err := getSelectors(networkUUID) | ||
| if err != nil { | ||
| return "", err | ||
| } | ||
| return fmt.Sprintf("sum(count_over_time({%s}[1h]))", selectors), nil | ||
| } | ||
|
|
||
| func queryLoki( | ||
| ctx context.Context, | ||
| lokiURL string, | ||
| username string, | ||
| password string, | ||
| query string, | ||
| ) (int, error) { | ||
| // Compose the URL | ||
| params := url.Values{} | ||
| params.Add("query", query) | ||
| reqURL := fmt.Sprintf("%s/loki/api/v1/query?%s", lokiURL, params.Encode()) | ||
|
|
||
| // Create request | ||
| req, err := http.NewRequestWithContext(ctx, http.MethodGet, reqURL, nil) | ||
| if err != nil { | ||
| return 0, fmt.Errorf("failed to create request: %w", err) | ||
| } | ||
|
|
||
| auth := base64.StdEncoding.EncodeToString([]byte(username + ":" + password)) | ||
| req.Header.Set("Authorization", "Basic "+auth) | ||
|
|
||
| // Execute request | ||
| resp, err := http.DefaultClient.Do(req) | ||
| if err != nil { | ||
| return 0, fmt.Errorf("failed to execute request: %w", err) | ||
| } | ||
| defer resp.Body.Close() | ||
|
|
||
| // Read and parse response | ||
| body, err := io.ReadAll(resp.Body) | ||
| if err != nil { | ||
| return 0, fmt.Errorf("failed to read response: %w", err) | ||
| } | ||
|
|
||
| if resp.StatusCode != http.StatusOK { | ||
| return 0, fmt.Errorf("unexpected status code %d: %s", resp.StatusCode, string(body)) | ||
| } | ||
|
|
||
| // Parse JSON response | ||
| var result struct { | ||
| Status string `json:"status"` | ||
| Data struct { | ||
| Result []struct { | ||
| Value []interface{} `json:"value"` | ||
| } `json:"result"` | ||
| } `json:"data"` | ||
| } | ||
|
|
||
| if err := json.Unmarshal(body, &result); err != nil { | ||
| return 0, fmt.Errorf("failed to parse response: %w", err) | ||
| } | ||
|
|
||
| // Extract count value | ||
| if len(result.Data.Result) == 0 { | ||
| return 0, nil | ||
| } | ||
| if len(result.Data.Result[0].Value) != 2 { | ||
| return 0, errors.New("unexpected value format in response") | ||
| } | ||
| // Convert value to a string | ||
| valueStr, ok := result.Data.Result[0].Value[1].(string) | ||
| if !ok { | ||
| return 0, errors.New("value is not a string") | ||
| } | ||
| // Convert string to float64 first to handle scientific notation | ||
| floatVal, err := strconv.ParseFloat(valueStr, 64) | ||
| if err != nil { | ||
| return 0, fmt.Errorf("parsing count value: %w", err) | ||
| } | ||
| // Round to nearest integer | ||
| return int(floatVal + 0.5), nil | ||
| } | ||
|
|
||
| // CheckMetricsExist checks if metrics exist for the given network. Github labels are also | ||
| // included if provided as env vars (GH_*). | ||
| func CheckMetricsExist(ctx context.Context, log logging.Logger, networkUUID string) error { | ||
| username, password, err := getCollectorCredentials(prometheusCmd) | ||
| if err != nil { | ||
| return fmt.Errorf("failed to get collector credentials: %w", err) | ||
| } | ||
| query, err := getCheckMetricsQuery(networkUUID) | ||
| if err != nil { | ||
| return err | ||
| } | ||
| url := getPrometheusURL() | ||
|
|
||
| log.Info("checking if metrics exist", | ||
| zap.String("url", url), | ||
| zap.String("query", query), | ||
| ) | ||
|
|
||
| results, err := queryPrometheus(ctx, log, url, username, password, query) | ||
| if err != nil { | ||
| return err | ||
| } | ||
|
|
||
| metricsCount := len(results) | ||
| if metricsCount > 0 { | ||
| log.Info("metrics exist", | ||
| zap.Int("count", metricsCount), | ||
| ) | ||
| return nil | ||
| } | ||
|
|
||
| return errors.New("metrics not found") | ||
| } | ||
|
|
||
| // getCheckMetricsQuery returns the query to check if metrics exist. | ||
| func getCheckMetricsQuery(networkUUID string) (string, error) { | ||
| selectors, err := getSelectors(networkUUID) | ||
| if err != nil { | ||
| return "", err | ||
| } | ||
| return fmt.Sprintf("{%s}", selectors), nil | ||
| } | ||
|
|
||
| func queryPrometheus( | ||
| ctx context.Context, | ||
| log logging.Logger, | ||
| url string, | ||
| username string, | ||
| password string, | ||
| query string, | ||
| ) (model.Vector, error) { | ||
| // Create client with basic auth | ||
| client, err := api.NewClient(api.Config{ | ||
| Address: url, | ||
| RoundTripper: &basicAuthRoundTripper{ | ||
| username: username, | ||
| password: password, | ||
| rt: api.DefaultRoundTripper, | ||
| }, | ||
| }) | ||
| if err != nil { | ||
| return nil, fmt.Errorf("failed to create client: %w", err) | ||
| } | ||
|
|
||
| // Query Prometheus | ||
| result, warnings, err := v1.NewAPI(client).Query(ctx, query, time.Now()) | ||
| if err != nil { | ||
| return nil, fmt.Errorf("query failed: %w", err) | ||
| } | ||
| if len(warnings) > 0 { | ||
| log.Warn("prometheus query warnings", | ||
| zap.Strings("warnings", warnings), | ||
| ) | ||
| } | ||
|
|
||
| // Check results | ||
| vector, ok := result.(model.Vector) | ||
| if !ok { | ||
| return nil, fmt.Errorf("unexpected result type: %s", result.Type()) | ||
| } | ||
| return vector, nil | ||
| } | ||
|
|
||
| type basicAuthRoundTripper struct { | ||
| username, password string | ||
| rt http.RoundTripper | ||
| } | ||
|
|
||
| func (b *basicAuthRoundTripper) RoundTrip(req *http.Request) (*http.Response, error) { | ||
| req.SetBasicAuth(b.username, b.password) | ||
| return b.rt.RoundTrip(req) | ||
| } | ||
|
|
||
| // getSelectors returns the comma-separated list of selectors. | ||
| func getSelectors(networkUUID string) (string, error) { | ||
| selectors := []string{} | ||
| if len(networkUUID) > 0 { | ||
| selectors = append(selectors, fmt.Sprintf(`network_uuid="%s"`, networkUUID)) | ||
| } | ||
| githubLabels := githubLabelsFromEnv() | ||
| for label := range githubLabels { | ||
| value, err := githubLabels.GetStringVal(label) | ||
| if err != nil { | ||
| return "", err | ||
| } | ||
| if len(value) == 0 { | ||
| continue | ||
| } | ||
| selectors = append(selectors, fmt.Sprintf(`%s="%s"`, label, value)) | ||
| } | ||
| return strings.Join(selectors, ","), nil | ||
| } |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Oops, something went wrong.