From 29cda9cf324bcc2983751b90f52dadac5771eb23 Mon Sep 17 00:00:00 2001 From: Muneeb Ullah Khan Date: Fri, 16 Jan 2026 15:38:49 +0500 Subject: [PATCH] splitted line like before --- pkg/sources/gitlab/gitlab.go | 15 +++++++++++++-- 1 file changed, 13 insertions(+), 2 deletions(-) diff --git a/pkg/sources/gitlab/gitlab.go b/pkg/sources/gitlab/gitlab.go index 8a669e0a0..e4dccffc6 100644 --- a/pkg/sources/gitlab/gitlab.go +++ b/pkg/sources/gitlab/gitlab.go @@ -8,6 +8,7 @@ import ( "slices" "strings" "sync" + "time" "github.com/trufflesecurity/trufflehog/v3/pkg/common" "github.com/trufflesecurity/trufflehog/v3/pkg/context" @@ -473,7 +474,12 @@ func (s *Source) newClient() (*gitlab.Client, error) { switch s.authMethod { case "OAUTH": ts := oauth2.StaticTokenSource(&oauth2.Token{AccessToken: s.token}) - apiClient, err := gitlab.NewAuthSourceClient(gitlab.OAuthTokenSource{TokenSource: ts}, gitlab.WithBaseURL(s.url)) + apiClient, err := gitlab.NewAuthSourceClient( + gitlab.OAuthTokenSource{TokenSource: ts}, + gitlab.WithBaseURL(s.url), + gitlab.WithCustomRetryWaitMinMax(time.Second, 5*time.Second), + gitlab.WithCustomRetryMax(3), + ) if err != nil { return nil, fmt.Errorf("could not create Gitlab TOKEN client for %q: %w", s.url, err) } @@ -493,7 +499,12 @@ func (s *Source) newClient() (*gitlab.Client, error) { fallthrough case "TOKEN": ts := oauth2.StaticTokenSource(&oauth2.Token{AccessToken: s.token}) - apiClient, err := gitlab.NewAuthSourceClient(gitlab.OAuthTokenSource{TokenSource: ts}, gitlab.WithBaseURL(s.url)) + apiClient, err := gitlab.NewAuthSourceClient( + gitlab.OAuthTokenSource{TokenSource: ts}, + gitlab.WithBaseURL(s.url), + gitlab.WithCustomRetryWaitMinMax(time.Second, 5*time.Second), + gitlab.WithCustomRetryMax(3), + ) if err != nil { return nil, fmt.Errorf("could not create Gitlab TOKEN client for %q: %w", s.url, err) }