mirror of
https://github.com/fnproject/fn.git
synced 2022-10-28 21:29:17 +03:00
faaf5846ce
* Use retry func while trying to ping SQL datastore - implements retry func specifically for SQL datastore ping - fmt fixes - using sqlx.Db.PingContext instead of sqlx.Db.Ping - propogate context to SQL datastore * Rely on context from ServerOpt * Consolidate log instances * Cleanup * Fix server usage in API tests
38 lines
967 B
Go
38 lines
967 B
Go
package datastore
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"net/url"
|
|
|
|
"github.com/fnproject/fn/api/common"
|
|
"github.com/fnproject/fn/api/datastore/internal/datastoreutil"
|
|
"github.com/fnproject/fn/api/datastore/sql"
|
|
"github.com/fnproject/fn/api/models"
|
|
"github.com/sirupsen/logrus"
|
|
)
|
|
|
|
func New(ctx context.Context, dbURL string) (models.Datastore, error) {
|
|
ds, err := newds(ctx, dbURL) // teehee
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
return datastoreutil.MetricDS(datastoreutil.NewValidator(ds)), nil
|
|
}
|
|
|
|
func newds(ctx context.Context, dbURL string) (models.Datastore, error) {
|
|
log := common.Logger(ctx)
|
|
u, err := url.Parse(dbURL)
|
|
if err != nil {
|
|
log.WithError(err).WithFields(logrus.Fields{"url": dbURL}).Fatal("bad DB URL")
|
|
}
|
|
log.WithFields(logrus.Fields{"db": u.Scheme}).Debug("creating new datastore")
|
|
switch u.Scheme {
|
|
case "sqlite3", "postgres", "mysql":
|
|
return sql.New(ctx, u)
|
|
default:
|
|
return nil, fmt.Errorf("db type not supported %v", u.Scheme)
|
|
}
|
|
}
|