Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
- Loading branch information
1 parent
a90c4fc
commit 71ec6dd
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
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -1,5 +1,4 @@ | ||
| .direnv | ||
| .devenv | ||
| coverage.out | ||
|
|
||
|
|
||
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,196 @@ | ||
| package exporter | ||
|
|
||
| import ( | ||
| "log/slog" | ||
| "time" | ||
|
|
||
| "github.com/google/go-github/v66/github" | ||
| "github.com/prometheus/client_golang/prometheus" | ||
| "github.com/promhippie/github_exporter/pkg/config" | ||
| "github.com/promhippie/github_exporter/pkg/store" | ||
| ) | ||
|
|
||
| // WorkflowCollector collects metrics about the servers. | ||
| type WorkflowJobCollector struct { | ||
| client *github.Client | ||
| logger *slog.Logger | ||
| db store.Store | ||
| failures *prometheus.CounterVec | ||
| duration *prometheus.HistogramVec | ||
| config config.Target | ||
|
|
||
| Status *prometheus.Desc | ||
| Duration *prometheus.Desc | ||
| Creation *prometheus.Desc | ||
| Created *prometheus.Desc | ||
| Started *prometheus.Desc | ||
| } | ||
|
|
||
| // NewWorkflowCollector returns a new WorkflowCollector. | ||
| func NewWorkflowJobCollector(logger *slog.Logger, client *github.Client, db store.Store, failures *prometheus.CounterVec, duration *prometheus.HistogramVec, cfg config.Target) *WorkflowJobCollector { | ||
| if failures != nil { | ||
| failures.WithLabelValues("action").Add(0) | ||
| } | ||
|
|
||
| labels := cfg.Workflows.Labels.Value() | ||
| return &WorkflowJobCollector{ | ||
| client: client, | ||
| logger: logger.With("collector", "workflow_job"), | ||
| db: db, | ||
| failures: failures, | ||
| duration: duration, | ||
| config: cfg, | ||
|
|
||
| Status: prometheus.NewDesc( | ||
| "github_workflow_job_status", | ||
| "Status of workflow jobs", | ||
| labels, | ||
| nil, | ||
| ), | ||
| Duration: prometheus.NewDesc( | ||
| "github_workflow_job_duration_ms", | ||
| "Duration of workflow runs", | ||
| labels, | ||
| nil, | ||
| ), | ||
| Creation: prometheus.NewDesc( | ||
| "github_workflow_job_duration_run_created_minutes", | ||
| "Duration since the workflow run creation time in minutes", | ||
| labels, | ||
| nil, | ||
| ), | ||
| Created: prometheus.NewDesc( | ||
| "github_workflow_job_created_timestamp", | ||
| "Timestamp when the workflow job have been created", | ||
| labels, | ||
| nil, | ||
| ), | ||
| Started: prometheus.NewDesc( | ||
| "github_workflow_job_started_timestamp", | ||
| "Timestamp when the workflow job have been started", | ||
| labels, | ||
| nil, | ||
| ), | ||
| } | ||
| } | ||
|
|
||
| // Metrics simply returns the list metric descriptors for generating a documentation. | ||
| func (c *WorkflowJobCollector) Metrics() []*prometheus.Desc { | ||
| return []*prometheus.Desc{ | ||
| c.Status, | ||
| c.Duration, | ||
| c.Creation, | ||
| c.Created, | ||
| c.Started, | ||
| } | ||
| } | ||
|
|
||
| // Describe sends the super-set of all possible descriptors of metrics collected by this Collector. | ||
| func (c *WorkflowJobCollector) Describe(ch chan<- *prometheus.Desc) { | ||
| ch <- c.Status | ||
| ch <- c.Duration | ||
| ch <- c.Creation | ||
| ch <- c.Created | ||
| ch <- c.Started | ||
| } | ||
|
|
||
| // Collect is called by the Prometheus registry when collecting metrics. | ||
| func (c *WorkflowJobCollector) Collect(ch chan<- prometheus.Metric) { | ||
| if err := c.db.PruneWorkflowJobs( | ||
| c.config.WorkflowJobs.Window, | ||
| ); err != nil { | ||
| c.logger.Error( | ||
| "Failed to prune workflow jobs", | ||
| "err", err, | ||
| ) | ||
| } | ||
|
|
||
| now := time.Now() | ||
| records, err := c.db.GetWorkflowJobs() | ||
| c.duration.WithLabelValues("workflow_job").Observe(time.Since(now).Seconds()) | ||
|
|
||
| if err != nil { | ||
| c.logger.Error( | ||
| "Failed to fetch workflows", | ||
| "err", err, | ||
| ) | ||
|
|
||
| c.failures.WithLabelValues("workflow_job").Inc() | ||
| return | ||
| } | ||
|
|
||
| c.logger.Debug( | ||
| "Fetched workflow jobs", | ||
| "count", len(records), | ||
| "duration", time.Since(now), | ||
| ) | ||
|
|
||
| for _, record := range records { | ||
| c.logger.Debug( | ||
| "Collecting workflow job", | ||
| "owner", record.Owner, | ||
| "repo", record.Repo, | ||
| "id", record.Identifier, | ||
| "run_id", record.RunID, | ||
| ) | ||
|
|
||
| labels := []string{} | ||
|
|
||
| for _, label := range c.config.WorkflowJobs.Labels.Value() { | ||
| labels = append( | ||
| labels, | ||
| record.ByLabel(label), | ||
| ) | ||
| } | ||
|
|
||
| ch <- prometheus.MustNewConstMetric( | ||
| c.Status, | ||
| prometheus.GaugeValue, | ||
| jobStatusToGauge(record.Status), | ||
| labels..., | ||
| ) | ||
|
|
||
| ch <- prometheus.MustNewConstMetric( | ||
| c.Duration, | ||
| prometheus.GaugeValue, | ||
| float64((record.CompletedAt-record.StartedAt)*1000), | ||
| labels..., | ||
| ) | ||
|
|
||
| ch <- prometheus.MustNewConstMetric( | ||
| c.Creation, | ||
| prometheus.GaugeValue, | ||
| time.Since(time.Unix(record.StartedAt, 0)).Minutes(), | ||
| labels..., | ||
| ) | ||
|
|
||
| ch <- prometheus.MustNewConstMetric( | ||
| c.Created, | ||
| prometheus.GaugeValue, | ||
| float64(record.CreatedAt), | ||
| labels..., | ||
| ) | ||
|
|
||
| ch <- prometheus.MustNewConstMetric( | ||
| c.Started, | ||
| prometheus.GaugeValue, | ||
| float64(record.StartedAt), | ||
| labels..., | ||
| ) | ||
| } | ||
| } | ||
|
|
||
| func jobStatusToGauge(conclusion string) float64 { | ||
| switch conclusion { | ||
| case "queued": | ||
| return 1.0 | ||
| case "waiting": | ||
| return 2.0 | ||
| case "in_progress": | ||
| return 3.0 | ||
| case "completed": | ||
| return 4.0 | ||
| } | ||
|
|
||
| return 0.0 | ||
| } |
Oops, something went wrong.