harness-drone/plugin/registry/combine.go
2019-02-19 15:56:41 -08:00

49 lines
1.3 KiB
Go

// Copyright 2019 Drone.IO Inc. All rights reserved.
// Use of this source code is governed by the Drone Non-Commercial License
// that can be found in the LICENSE file.
package registry
import (
"context"
"github.com/drone/drone/logger"
"github.com/drone/drone/core"
"github.com/sirupsen/logrus"
)
// Combine combines the registry services, allowing the
// system to source registry credential from multiple sources.
func Combine(services ...core.RegistryService) core.RegistryService {
return &combined{services}
}
type combined struct {
sources []core.RegistryService
}
func (c *combined) List(ctx context.Context, req *core.RegistryArgs) ([]*core.Registry, error) {
var all []*core.Registry
for _, source := range c.sources {
list, err := source.List(ctx, req)
if err != nil {
return all, err
}
all = append(all, list...)
}
// if trace level debugging is enabled we print
// all registry credentials retrieved from the
// various registy sources.
logger := logger.FromContext(ctx)
if logrus.IsLevelEnabled(logrus.TraceLevel) {
if len(all) == 0 {
logger.Traceln("registry: no registry credentials loaded")
}
for _, registry := range all {
logger.WithField("address", registry.Address).
Traceln("registry: registry credentials loaded")
}
}
return all, nil
}