dex/db/health.go
2015-08-24 15:43:15 -07:00

34 lines
519 B
Go

package db
import (
"errors"
"fmt"
"github.com/go-gorp/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
}