-
Notifications
You must be signed in to change notification settings - Fork 0
/
client.go
41 lines (33 loc) · 1.27 KB
/
client.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
package client
import (
"fmt"
"github.com/D3rus/go-library/configurator"
"github.com/D3rus/go-library/container"
"github.com/D3rus/go-library/data_base/client/driver"
"github.com/D3rus/go-library/data_base/client/mysql"
"github.com/D3rus/go-library/data_base/client/sqlite"
"github.com/D3rus/go-library/logger"
"strings"
)
//go:generate mockgen -package=client -destination=mock_configurator_test.go github.com/D3rus/go-library/configurator Configurator
func Configuration(config *Config, configurator configurator.Configurator) *Config {
if config.Driver == "" || config.Driver == DriverDefault {
configurator.SetDefault(DriverFieldName, DriverDefault)
config.Driver = strings.ToLower(configurator.GetString(DriverFieldName))
}
return config
}
func New(config *Config, informer logger.Informer, container container.Container) (DataBase, error) {
var dataBase DataBase
var err error
switch config.Driver {
case driver.MySQL:
err = container.Invoke(func(db *mysql.ReConfiguration) { dataBase = db })
case driver.SQLite:
err = container.Invoke(func(db *sqlite.ReConfiguration) { dataBase = db })
default:
return nil, fmt.Errorf("data_base: driver '%s' unknown", config.Driver)
}
informer.Infof("data_base: usage driver - '%s'", config.Driver)
return dataBase, err
}