forked from gemnasium/migrate
-
Notifications
You must be signed in to change notification settings - Fork 0
/
registry.go
43 lines (38 loc) · 967 Bytes
/
registry.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
package driver
import (
"sort"
"sync"
)
var driversMu sync.Mutex
var drivers = make(map[string]Driver)
// Registers a driver so it can be created from its name. Drivers should call
// this from an init() function so that they registers themselves on import.
func RegisterDriver(name string, driver Driver) {
driversMu.Lock()
defer driversMu.Unlock()
if driver == nil {
panic("driver: Register driver is nil")
}
if _, dup := drivers[name]; dup {
panic("sql: Register called twice for driver " + name)
}
drivers[name] = driver
}
// Retrieves a registered driver by name.
func GetDriver(name string) Driver {
driversMu.Lock()
defer driversMu.Unlock()
driver := drivers[name]
return driver
}
// Drivers returns a sorted list of the names of the registered drivers.
func Drivers() []string {
driversMu.Lock()
defer driversMu.Unlock()
var list []string
for name := range drivers {
list = append(list, name)
}
sort.Strings(list)
return list
}