31 lines
497 B
Go
31 lines
497 B
Go
package db
|
|
|
|
import (
|
|
"database/sql"
|
|
"fmt"
|
|
)
|
|
|
|
const (
|
|
DriverSqlite3 = "sqlite3"
|
|
)
|
|
|
|
type Config struct {
|
|
Driver string
|
|
Name string
|
|
Host string
|
|
User string
|
|
Password string
|
|
}
|
|
|
|
func New(cfg Config) (*sql.DB, error) {
|
|
switch cfg.Driver {
|
|
case DriverSqlite3:
|
|
name := cfg.Name
|
|
if len(name) == 0 {
|
|
name = "app_db"
|
|
}
|
|
return sql.Open(cfg.Driver, name)
|
|
default:
|
|
return nil, fmt.Errorf("invalid database driver (%s): currently only sqlite3 is supported", cfg.Driver)
|
|
}
|
|
}
|