diff --git a/logdb.go b/logdb.go index 2cddea8..d3a040f 100644 --- a/logdb.go +++ b/logdb.go @@ -40,6 +40,11 @@ func NewLogDB(filename string) *LogDB { return &logdb } +// Ping 是否Ping通数据库 +func (logdb *LogDB) Ping() error { + return logdb.driver.Ping() +} + // Connect 重连 func (logdb *LogDB) Connect() { defer func() { @@ -67,5 +72,6 @@ func (logdb *LogDB) Connect() { if err != nil { panic(err) } + db.Ping() logdb.driver = db } diff --git a/logdb_test.go b/logdb_test.go index 43038aa..c8ccdba 100644 --- a/logdb_test.go +++ b/logdb_test.go @@ -14,7 +14,8 @@ import ( // user: spider func TestSelect(t *testing.T) { + logdb := NewLogDB("logdb.yaml") - t.Error(logdb) logdb.Connect() + t.Error(logdb) }