-
Notifications
You must be signed in to change notification settings - Fork 2k
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Collect count of VirtualServer, VirtualServerRoute and TransportServe…
…r resources (#5095)
- Loading branch information
Showing
10 changed files
with
631 additions
and
218 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
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,104 @@ | ||
| // Package telemetry provides functionality for collecting and exporting NIC telemetry data. | ||
| package telemetry | ||
|
|
||
| import ( | ||
| "context" | ||
| "io" | ||
| "time" | ||
|
|
||
| "github.com/nginxinc/kubernetes-ingress/internal/configs" | ||
|
|
||
| k8s_nginx "github.com/nginxinc/kubernetes-ingress/pkg/client/clientset/versioned" | ||
| "k8s.io/apimachinery/pkg/util/wait" | ||
| "k8s.io/client-go/kubernetes" | ||
|
|
||
| "github.com/golang/glog" | ||
| ) | ||
|
|
||
| // Option is a functional option used for configuring TraceReporter. | ||
| type Option func(*Collector) error | ||
|
|
||
| // WithExporter configures telemetry collector to use given exporter. | ||
| // | ||
| // This may change in the future when we use exporter implemented | ||
| // in the external module. | ||
| func WithExporter(e Exporter) Option { | ||
| return func(c *Collector) error { | ||
| c.Exporter = e | ||
| return nil | ||
| } | ||
| } | ||
|
|
||
| // Collector is NIC telemetry data collector. | ||
| type Collector struct { | ||
| // Exporter is a temp exporter for exporting telemetry data. | ||
| // The concrete implementation will be implemented in a separate module. | ||
| Exporter Exporter | ||
|
|
||
| // Configuration for the collector. | ||
| Config CollectorConfig | ||
| } | ||
|
|
||
| // CollectorConfig contains configuration options for a Collector | ||
| type CollectorConfig struct { | ||
| // K8sClientReader is a kubernetes client. | ||
| K8sClientReader kubernetes.Interface | ||
|
|
||
| // CustomK8sClientReader is a kubernetes client for our CRDs. | ||
| // Note: May not need this client. | ||
| CustomK8sClientReader k8s_nginx.Interface | ||
|
|
||
| // Period to collect telemetry | ||
| Period time.Duration | ||
|
|
||
| Configurator *configs.Configurator | ||
| } | ||
|
|
||
| // NewCollector takes 0 or more options and creates a new TraceReporter. | ||
| // If no options are provided, NewReporter returns TraceReporter | ||
| // configured to gather data every 24h. | ||
| func NewCollector(cfg CollectorConfig, opts ...Option) (*Collector, error) { | ||
| c := Collector{ | ||
| Exporter: &StdoutExporter{Endpoint: io.Discard}, | ||
| Config: cfg, | ||
| } | ||
| for _, o := range opts { | ||
| if err := o(&c); err != nil { | ||
| return nil, err | ||
| } | ||
| } | ||
| return &c, nil | ||
| } | ||
|
|
||
| // Start starts running NIC Telemetry Collector. | ||
| func (c *Collector) Start(ctx context.Context) { | ||
| wait.JitterUntilWithContext(ctx, c.Collect, c.Config.Period, 0.1, true) | ||
| } | ||
|
|
||
| // Collect collects and exports telemetry data. | ||
| // It exports data using provided exporter. | ||
| func (c *Collector) Collect(ctx context.Context) { | ||
| glog.V(3).Info("Collecting telemetry data") | ||
| // TODO: Re-add ctx to BuildReport when collecting Node Count. | ||
| data, err := c.BuildReport() | ||
| if err != nil { | ||
| glog.Errorf("Error collecting telemetry data: %v", err) | ||
| } | ||
| err = c.Exporter.Export(ctx, data) | ||
| if err != nil { | ||
| glog.Errorf("Error exporting telemetry data: %v", err) | ||
| } | ||
| glog.V(3).Infof("Exported telemetry data: %+v", data) | ||
| } | ||
|
|
||
| // BuildReport takes context and builds report from gathered telemetry data. | ||
| func (c *Collector) BuildReport() (Data, error) { | ||
| d := Data{} | ||
| var err error | ||
|
|
||
| if c.Config.Configurator != nil { | ||
| d.NICResourceCounts.VirtualServers, d.NICResourceCounts.VirtualServerRoutes = c.Config.Configurator.GetVirtualServerCounts() | ||
| d.NICResourceCounts.TransportServers = c.Config.Configurator.GetTransportServerCounts() | ||
| } | ||
| return d, err | ||
| } |
Oops, something went wrong.