You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
56 lines
1.1 KiB
56 lines
1.1 KiB
5 years ago
|
package paladin
|
||
|
|
|
||
|
|
import (
|
||
|
|
"fmt"
|
||
|
|
"sort"
|
||
|
|
"sync"
|
||
|
|
)
|
||
|
|
|
||
|
|
var (
|
||
|
|
driversMu sync.RWMutex
|
||
|
|
drivers = make(map[string]Driver)
|
||
|
|
)
|
||
|
|
|
||
|
|
// Register makes a paladin driver available by the provided name.
|
||
|
|
// If Register is called twice with the same name or if driver is nil,
|
||
|
|
// it panics.
|
||
|
|
func Register(name string, driver Driver) {
|
||
|
|
driversMu.Lock()
|
||
|
|
defer driversMu.Unlock()
|
||
|
|
|
||
|
|
if driver == nil {
|
||
|
|
panic("paladin: driver is nil")
|
||
|
|
}
|
||
|
|
|
||
|
|
if _, dup := drivers[name]; dup {
|
||
|
|
panic("paladin: Register called twice for driver " + name)
|
||
|
|
}
|
||
|
|
|
||
|
|
drivers[name] = driver
|
||
|
|
}
|
||
|
|
|
||
|
|
// Drivers returns a sorted list of the names of the registered paladin driver.
|
||
|
|
func Drivers() []string {
|
||
|
|
driversMu.RLock()
|
||
|
|
defer driversMu.RUnlock()
|
||
|
|
|
||
|
|
var list []string
|
||
|
|
for name := range drivers {
|
||
|
|
list = append(list, name)
|
||
|
|
}
|
||
|
|
|
||
|
|
sort.Strings(list)
|
||
|
|
return list
|
||
|
|
}
|
||
|
|
|
||
|
|
// GetDriver returns a driver implement by name.
|
||
|
|
func GetDriver(name string) (Driver, error) {
|
||
|
|
driversMu.RLock()
|
||
|
|
driveri, ok := drivers[name]
|
||
|
|
driversMu.RUnlock()
|
||
|
|
if !ok {
|
||
|
|
return nil, fmt.Errorf("paladin: unknown driver %q (forgotten import?)", name)
|
||
|
|
}
|
||
|
|
return driveri, nil
|
||
|
|
}
|