diff options
Diffstat (limited to 'internal/source/domains.go')
-rw-r--r-- | internal/source/domains.go | 15 |
1 files changed, 15 insertions, 0 deletions
diff --git a/internal/source/domains.go b/internal/source/domains.go index 22187855..79357766 100644 --- a/internal/source/domains.go +++ b/internal/source/domains.go @@ -4,7 +4,10 @@ import ( "errors" "time" + log "github.com/sirupsen/logrus" + "gitlab.com/gitlab-org/gitlab-pages/internal/domain" + "gitlab.com/gitlab-org/gitlab-pages/internal/rollout" "gitlab.com/gitlab-org/gitlab-pages/internal/source/disk" "gitlab.com/gitlab-org/gitlab-pages/internal/source/domains/gitlabsourceconfig" "gitlab.com/gitlab-org/gitlab-pages/internal/source/gitlab" @@ -81,5 +84,17 @@ func (d *Domains) source(domain string) Source { } } + r := gitlabSourceConfig.Domains.Rollout + + enabled, err := rollout.Rollout(domain, r.Percentage, r.Stickiness) + if err != nil { + log.WithError(err).Error("Rollout error") + return d.disk + } + + if enabled { + return d.gitlab + } + return d.disk } |