-
Notifications
You must be signed in to change notification settings - Fork 47
/
driver.go
66 lines (52 loc) · 1.4 KB
/
driver.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
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
// This Source Code Form is subject to the terms of the MIT License.
// If a copy of the MIT License was not distributed with this
// file, you can obtain one at https://opensource.org/licenses/MIT.
//
// Copyright (c) DUSK NETWORK. All rights reserved.
package database
import (
"errors"
"sort"
"sync"
)
// Replicated from "database/sql" pkg.
var (
driversMu sync.RWMutex
drivers = make(map[string]Driver)
)
// Register makes a database driver available by the provided name.
// If Register is called twice with the same name or if driver is nil,
// it returns error.
func Register(driver Driver) error {
driversMu.Lock()
defer driversMu.Unlock()
if driver == nil {
return errors.New("cannot register a nil driver")
}
name := driver.Name()
if _, dup := drivers[name]; dup {
return errors.New("duplicated driver name: " + name)
}
drivers[name] = driver
return nil
}
// Drivers returns a sorted list of the names of the registered drivers.
func Drivers() []string {
driversMu.RLock()
defer driversMu.RUnlock()
var list []string //nolint
for name := range drivers {
list = append(list, name)
}
sort.Strings(list)
return list
}
// From returns a registered Driver by name.
func From(name string) (Driver, error) {
driversMu.RLock()
defer driversMu.RUnlock()
if _, dup := drivers[name]; dup {
return drivers[name], nil
}
return nil, errors.New("Cannot find driver with name " + name)
}