Skip to content
Snippets Groups Projects

Introduce GCS adapter for remote cache

Merged Tomasz Maczukin requested to merge introduce-gcs-cache-support into master
1 unresolved thread
18 files
+ 534
368
Compare changes
  • Side-by-side
  • Inline
Files
18
+ 76
0
package adapter
import (
"fmt"
"net/url"
"sync"
"time"
"github.com/sirupsen/logrus"
"gitlab.com/gitlab-org/gitlab-runner/common"
)
type Adapter interface {
GetDownloadURL() *url.URL
GetUploadURL() *url.URL
}
type Factory func(config *common.CacheConfig, timeout time.Duration, objectName string) (Adapter, error)
type FactoriesMap struct {
sync.Mutex
internal map[string]Factory
}
func (m *FactoriesMap) Register(typeName string, factory Factory) error {
m.Lock()
defer m.Unlock()
if len(m.internal) == 0 {
m.internal = make(map[string]Factory)
}
_, ok := m.internal[typeName]
if ok {
return fmt.Errorf("adapter %q already registered", typeName)
}
m.internal[typeName] = factory
return nil
}
func (m *FactoriesMap) Find(typeName string) (Factory, error) {
factory := m.internal[typeName]
if factory == nil {
return nil, fmt.Errorf("factory for cache adapter %q was not registered", typeName)
}
return factory, nil
}
var factories = &FactoriesMap{}
func Factories() *FactoriesMap {
return factories
}
func FactorizeAdapter(cacheConfig *common.CacheConfig, timeout time.Duration, objectName string) Adapter {
log := logrus.WithField("type", cacheConfig.Type)
factorize, err := Factories().Find(cacheConfig.Type)
if err != nil {
log.WithError(err).Error("Cache factory not found")
return nil
}
adapter, err := factorize(cacheConfig, timeout, objectName)
if err != nil {
log.WithError(err).Error("Cache adapter could not be initialized")
return nil
}
return adapter
}
Loading