mirror of
https://github.com/harness/drone.git
synced 2025-05-08 07:21:10 +08:00
81 lines
3.3 KiB
Go
81 lines
3.3 KiB
Go
// Source: https://github.com/distribution/distribution
|
|
|
|
// Copyright 2014 https://github.com/distribution/distribution Authors
|
|
//
|
|
// Licensed under the Apache License, Version 2.0 (the "License");
|
|
// you may not use this file except in compliance with the License.
|
|
// You may obtain a copy of the License at
|
|
//
|
|
// http://www.apache.org/licenses/LICENSE-2.0
|
|
//
|
|
// Unless required by applicable law or agreed to in writing, software
|
|
// distributed under the License is distributed on an "AS IS" BASIS,
|
|
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
// See the License for the specific language governing permissions and
|
|
// limitations under the License.
|
|
|
|
package factory
|
|
|
|
import (
|
|
"fmt"
|
|
|
|
"github.com/harness/gitness/registry/app/driver"
|
|
)
|
|
|
|
// driverFactories stores an internal mapping between storage driver names and their respective
|
|
// factories.
|
|
var driverFactories = make(map[string]StorageDriverFactory)
|
|
|
|
// StorageDriverFactory is a factory interface for creating storagedriver.StorageDriver interfaces
|
|
// Storage drivers should call Register() with a factory to make the driver available by name.
|
|
// Individual StorageDriver implementations generally register with the factory via the Register
|
|
// func (below) in their init() funcs, and as such they should be imported anonymously before use.
|
|
// See below for an example of how to register and get a StorageDriver for S3
|
|
//
|
|
// import _ "github.com/distribution/distribution/v3/registry/storage/driver/s3-aws"
|
|
// s3Driver, err = factory.Create("s3", storageParams)
|
|
// // assuming no error, s3Driver is the StorageDriver that communicates with S3 according to storageParams
|
|
type StorageDriverFactory interface {
|
|
// Create returns a new storagedriver.StorageDriver with the given parameters
|
|
// Parameters will vary by driver and may be ignored
|
|
// Each parameter key must only consist of lowercase letters and numbers
|
|
Create(parameters map[string]interface{}) (driver.StorageDriver, error)
|
|
}
|
|
|
|
// Register makes a storage driver available by the provided name.
|
|
// If Register is called twice with the same name or if driver factory is nil, it panics.
|
|
// Additionally, it is not concurrency safe. Most Storage Drivers call this function
|
|
// in their init() functions. See the documentation for StorageDriverFactory for more.
|
|
func Register(name string, factory StorageDriverFactory) {
|
|
if factory == nil {
|
|
panic("Must not provide nil StorageDriverFactory")
|
|
}
|
|
_, registered := driverFactories[name]
|
|
if registered {
|
|
panic(fmt.Sprintf("StorageDriverFactory named %s already registered", name))
|
|
}
|
|
|
|
driverFactories[name] = factory
|
|
}
|
|
|
|
// Create a new storagedriver.StorageDriver with the given name and
|
|
// parameters. To use a driver, the StorageDriverFactory must first be
|
|
// registered with the given name. If no drivers are found, an
|
|
// InvalidStorageDriverError is returned.
|
|
func Create(name string, parameters map[string]interface{}) (driver.StorageDriver, error) {
|
|
driverFactory, ok := driverFactories[name]
|
|
if !ok {
|
|
return nil, InvalidStorageDriverError{name}
|
|
}
|
|
return driverFactory.Create(parameters)
|
|
}
|
|
|
|
// InvalidStorageDriverError records an attempt to construct an unregistered storage driver.
|
|
type InvalidStorageDriverError struct {
|
|
Name string
|
|
}
|
|
|
|
func (err InvalidStorageDriverError) Error() string {
|
|
return fmt.Sprintf("StorageDriver not registered: %s", err.Name)
|
|
}
|