34 lines
523 B
Go
34 lines
523 B
Go
package db
|
|
|
|
import (
|
|
"errors"
|
|
"fmt"
|
|
|
|
"github.com/coopernurse/gorp"
|
|
)
|
|
|
|
func NewHealthChecker(dbm *gorp.DbMap) *healthChecker {
|
|
return &healthChecker{dbMap: dbm}
|
|
}
|
|
|
|
type healthChecker struct {
|
|
dbMap *gorp.DbMap
|
|
}
|
|
|
|
func (hc *healthChecker) Healthy() (err error) {
|
|
if err = hc.dbMap.Db.Ping(); err != nil {
|
|
err = fmt.Errorf("database error: %v", err)
|
|
return
|
|
}
|
|
|
|
num, err := hc.dbMap.SelectInt("SELECT 1")
|
|
if err != nil {
|
|
return
|
|
}
|
|
|
|
if num != 1 {
|
|
err = errors.New("unable to connect to database")
|
|
}
|
|
|
|
return
|
|
}
|