mirror of
https://github.com/fnproject/fn.git
synced 2022-10-28 21:29:17 +03:00
1ad27f4f0d
* initial Db helper split - make SQL and datastore packages optional * abstracting log store * break out DB, MQ and log drivers as extensions * cleanup * fewer deps * fixing docker test * hmm dbness * updating db startup * Consolidate all your extensions into one convenient package * cleanup * clean up dep constraints
44 lines
1.3 KiB
Go
44 lines
1.3 KiB
Go
// dbhelpers wrap SQL and specific capabilities of an SQL db
|
|
package dbhelper
|
|
|
|
import (
|
|
"fmt"
|
|
"github.com/jmoiron/sqlx"
|
|
"github.com/sirupsen/logrus"
|
|
"net/url"
|
|
)
|
|
|
|
var sqlHelpers []Helper
|
|
|
|
//Add registers a new SQL helper
|
|
func Add(helper Helper) {
|
|
logrus.Infof("Registering DB helper %s", helper)
|
|
sqlHelpers = append(sqlHelpers, helper)
|
|
}
|
|
|
|
//Helper provides DB-specific SQL capabilities
|
|
type Helper interface {
|
|
fmt.Stringer
|
|
//Supports indicates if this helper supports this driver name
|
|
Supports(driverName string) bool
|
|
//PreConnect calculates the connect URL for the db from a canonical URL used in Fn config
|
|
PreConnect(url *url.URL) (string, error)
|
|
//PostCreate Apply any configuration to the DB prior to use
|
|
PostCreate(db *sqlx.DB) (*sqlx.DB, error)
|
|
//CheckTableExists checks if a table exists in the DB
|
|
CheckTableExists(tx *sqlx.Tx, table string) (bool, error)
|
|
//IsDuplicateKeyError determines if an error indicates if the prior error was caused by a duplicate key insert
|
|
IsDuplicateKeyError(err error) bool
|
|
}
|
|
|
|
//GetHelper returns a helper for a specific driver
|
|
func GetHelper(driverName string) (Helper, bool) {
|
|
for _, helper := range sqlHelpers {
|
|
if helper.Supports(driverName) {
|
|
return helper, true
|
|
}
|
|
logrus.Printf("%s does not support %s", helper, driverName)
|
|
}
|
|
return nil, false
|
|
}
|