gitwrapper.go

package metrics

import (
	"time"

	"github.com/argoproj/argo-cd/util/git"
)

type gitClientWrapper struct {
	repo          string
	client        git.Client
	metricsServer *MetricsServer
}

func WrapGitClient(repo string, metricsServer *MetricsServer, client git.Client) git.Client {
	return &gitClientWrapper{repo: repo, client: client, metricsServer: metricsServer}
}

func (w *gitClientWrapper) Fetch() error {
	startTime := time.Now()
	w.metricsServer.IncGitRequest(w.repo, GitRequestTypeFetch)
	defer w.metricsServer.ObserveGitRequestDuration(w.repo, GitRequestTypeFetch, time.Since(startTime))
	return w.client.Fetch()
}

func (w *gitClientWrapper) LsRemote(revision string) (string, error) {
	startTime := time.Now()
	sha, err := w.client.LsRemote(revision)
	if sha != revision {
		// This is true only if specified revision is a tag, branch or HEAD and client had to use 'ls-remote'
		w.metricsServer.IncGitRequest(w.repo, GitRequestTypeLsRemote)
		defer w.metricsServer.ObserveGitRequestDuration(w.repo, GitRequestTypeLsRemote, time.Since(startTime))
	}
	return sha, err
}

func (w *gitClientWrapper) LsRefs() (*git.Refs, error) {
	return w.client.LsRefs()
}

func (w *gitClientWrapper) LsFiles(path string) ([]string, error) {
	return w.client.LsFiles(path)
}

func (w *gitClientWrapper) LsLargeFiles() ([]string, error) {
	return w.client.LsLargeFiles()
}

func (w *gitClientWrapper) Checkout(revision string) error {
	return w.client.Checkout(revision)
}

func (w *gitClientWrapper) CommitSHA() (string, error) {
	return w.client.CommitSHA()
}

func (w *gitClientWrapper) Root() string {
	return w.client.Root()
}

func (w *gitClientWrapper) Init() error {
	return w.client.Init()
}

func (w *gitClientWrapper) RevisionMetadata(revision string) (*git.RevisionMetadata, error) {
	return w.client.RevisionMetadata(revision)
}

func (w *gitClientWrapper) VerifyCommitSignature(revision string) (string, error) {
	return w.client.VerifyCommitSignature(revision)
}