tablet/gorm
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Allow disabling of automatic Ping

Browse Source
This commit is contained in:
pjebs 2016-08-08 14:54:40 +10:00
parent a646b13548
commit 10c8478254
1 changed files with 8 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
main.go
View File

@ -47,6 +47,7 @@ func Open(dialect string, args ...interface{}) (*DB, error) {
} else { } else {
var source string var source string
var dbSQL sqlCommon var dbSQL sqlCommon
dontPing := false
switch value := args[0].(type) { switch value := args[0].(type) {
case string: case string:
@ -61,6 +62,12 @@ func Open(dialect string, args ...interface{}) (*DB, error) {
case sqlCommon: case sqlCommon:
source = reflect.Indirect(reflect.ValueOf(value)).FieldByName("dsn").String() source = reflect.Indirect(reflect.ValueOf(value)).FieldByName("dsn").String()
dbSQL = value dbSQL = value
if len(args) == 2 {
aBool, correct := args[1].(bool)
if correct == true {
dontPing = aBool
}
}
} }
db = DB{ db = DB{
@ -73,7 +80,7 @@ func Open(dialect string, args ...interface{}) (*DB, error) {
} }
db.parent = &db db.parent = &db
if err == nil { if err == nil && !dontPing {
err = db.DB().Ping() // Send a ping to make sure the database connection is alive. err = db.DB().Ping() // Send a ping to make sure the database connection is alive.
} }
} }