mirror of
https://github.com/status-im/consul.git
synced 2025-02-10 04:45:05 +00:00
c112a6632d
* Option to set HCP client at runtime Allows us to initially set a nil HCP client for the telemetry provider and update it later. * Set telemetry provider HCP client in HCP manager Set the telemetry provider as a dependency and pass it to the manager. Update the telemetry provider's HCP client when the HCP manager starts. * Add a provider interface for the metrics client This provider will allow us to configure and reconfigure the retryable HTTP client and the headers for the metrics client. * Move HTTP retryable client to separate file Copied directly from the metrics client. * Abstract HCP specific values in HTTP client Remove HCP specific references and instead initiate with a generic TLS configuration and authentication source. * Set up HTTP client and headers in the provider Move setup from the metrics client to the HCP telemetry provider. * Update the telemetry provider in the HCP manager Initialize the provider without the HCP configs and then update it in the HCP manager to enable it. * Improve test assertion, fix method comment * Move client provider to metrics client * Stop the manager on setup error * Add separate lock for http configuration * Start telemetry provider in HCP manager * Update HCP client and config as part of Run * Remove option to set config at initialization * Simplify and clean up setting HCP configs * Add test for telemetry provider Run method * Fix race condition * Use clone of HTTP headers * Only allow initial update and run once
87 lines
2.3 KiB
Go
87 lines
2.3 KiB
Go
// Copyright (c) HashiCorp, Inc.
|
|
// SPDX-License-Identifier: BUSL-1.1
|
|
|
|
package hcp
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
|
|
"github.com/armon/go-metrics"
|
|
"github.com/hashicorp/go-hclog"
|
|
|
|
"github.com/hashicorp/consul/agent/hcp/client"
|
|
"github.com/hashicorp/consul/agent/hcp/config"
|
|
"github.com/hashicorp/consul/agent/hcp/scada"
|
|
"github.com/hashicorp/consul/agent/hcp/telemetry"
|
|
)
|
|
|
|
// Deps contains the interfaces that the rest of Consul core depends on for HCP integration.
|
|
type Deps struct {
|
|
Client client.Client
|
|
Provider scada.Provider
|
|
Sink metrics.MetricSink
|
|
TelemetryProvider *hcpProviderImpl
|
|
}
|
|
|
|
func NewDeps(cfg config.CloudConfig, logger hclog.Logger) (Deps, error) {
|
|
ctx := context.Background()
|
|
ctx = hclog.WithContext(ctx, logger)
|
|
|
|
hcpClient, err := client.NewClient(cfg)
|
|
if err != nil {
|
|
return Deps{}, fmt.Errorf("failed to init client: %w", err)
|
|
}
|
|
|
|
provider, err := scada.New(logger.Named("scada"))
|
|
if err != nil {
|
|
return Deps{}, fmt.Errorf("failed to init scada: %w", err)
|
|
}
|
|
|
|
metricsProvider := NewHCPProvider(ctx)
|
|
if err != nil {
|
|
logger.Error("failed to init HCP metrics provider", "error", err)
|
|
return Deps{}, fmt.Errorf("failed to init HCP metrics provider: %w", err)
|
|
}
|
|
|
|
metricsClient := client.NewMetricsClient(ctx, metricsProvider)
|
|
|
|
sink, err := sink(ctx, metricsClient, metricsProvider)
|
|
if err != nil {
|
|
// Do not prevent server start if sink init fails, only log error.
|
|
logger.Error("failed to init sink", "error", err)
|
|
}
|
|
|
|
return Deps{
|
|
Client: hcpClient,
|
|
Provider: provider,
|
|
Sink: sink,
|
|
TelemetryProvider: metricsProvider,
|
|
}, nil
|
|
}
|
|
|
|
// sink initializes an OTELSink which forwards Consul metrics to HCP.
|
|
// This step should not block server initialization, errors are returned, only to be logged.
|
|
func sink(
|
|
ctx context.Context,
|
|
metricsClient telemetry.MetricsClient,
|
|
cfgProvider *hcpProviderImpl,
|
|
) (metrics.MetricSink, error) {
|
|
logger := hclog.FromContext(ctx)
|
|
|
|
reader := telemetry.NewOTELReader(metricsClient, cfgProvider)
|
|
sinkOpts := &telemetry.OTELSinkOpts{
|
|
Reader: reader,
|
|
ConfigProvider: cfgProvider,
|
|
}
|
|
|
|
sink, err := telemetry.NewOTELSink(ctx, sinkOpts)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("failed to create OTELSink: %w", err)
|
|
}
|
|
|
|
logger.Debug("initialized HCP metrics sink")
|
|
|
|
return sink, nil
|
|
}
|