File: worker.go

package info (click to toggle)
gitlab-agent 16.1.3-2
  • links: PTS, VCS
  • area: contrib
  • in suites: forky, sid, trixie
  • size: 6,324 kB
  • sloc: makefile: 175; sh: 52; ruby: 3
file content (34 lines) | stat: -rw-r--r-- 976 bytes parent folder | download
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
package chartops

import (
	"context"
	"net/http"

	"gitlab.com/gitlab-org/cluster-integration/gitlab-agent/v16/internal/module/gitops/rpc"
	"gitlab.com/gitlab-org/cluster-integration/gitlab-agent/v16/internal/tool/retry"
	"gitlab.com/gitlab-org/cluster-integration/gitlab-agent/v16/pkg/agentcfg"
	"go.uber.org/zap"
	"k8s.io/apimachinery/pkg/util/wait"
)

type worker struct {
	log               *zap.Logger
	chartCfg          *agentcfg.ChartCF
	installPollConfig retry.PollConfig
	helm              Helm
	httpClient        http.RoundTripper
	objWatcher        rpc.ObjectsToSynchronizeWatcherInterface
}

func (w *worker) Run(ctx context.Context) {
	// Data flow: fetch_and_load() -> installOrUpgrade()
	jobs := make(chan job)

	var wg wait.Group
	defer wg.Wait()   // Wait for all pipeline stages to finish
	defer close(jobs) // Close jobs to signal installOrUpgrade() there is no more work to be done.
	wg.Start(func() {
		w.installOrUpgrade(jobs)
	})
	w.fetch(ctx, jobs)
}