wip
This commit is contained in:
@@ -1,8 +1,11 @@
|
||||
package main
|
||||
|
||||
import (
|
||||
"context"
|
||||
"database/sql"
|
||||
"flag"
|
||||
"os"
|
||||
"time"
|
||||
)
|
||||
|
||||
type Config struct {
|
||||
@@ -20,3 +23,36 @@ func NewConfig() (Config, error) {
|
||||
err := fs.Parse(os.Args[1:])
|
||||
return c, err
|
||||
}
|
||||
|
||||
type DB struct {
|
||||
scheme string
|
||||
conn string
|
||||
}
|
||||
|
||||
func NewDB(ctx context.Context, scheme, conn string) (DB, error) {
|
||||
ctx, can := context.WithTimeout(ctx, time.Second*10)
|
||||
defer can()
|
||||
|
||||
db := DB{
|
||||
scheme: scheme,
|
||||
conn: conn,
|
||||
}
|
||||
|
||||
sql, err := db.dial(ctx)
|
||||
if sql != nil {
|
||||
sql.Close()
|
||||
}
|
||||
|
||||
return db, err
|
||||
}
|
||||
|
||||
func (db DB) dial(ctx context.Context) (*sql.DB, error) {
|
||||
c, err := sql.Open(db.scheme, db.conn)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
if err := c.PingContext(ctx); err != nil {
|
||||
return nil, err
|
||||
}
|
||||
return c, nil
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user