Change healthchecker to return error

2021-11-06 00:41:27 +01:00
parent 8ba601a2b1
commit db8372f6db
5 changed files with 18 additions and 9 deletions
--- a/honeypot/ssh/store/postgres.go
+++ b/honeypot/ssh/store/postgres.go
@@ -126,8 +126,11 @@ func (s *PostgresStore) Stats(statType LoginStats, limit int) ([]StatsResult, er
 	return results, nil
 }

-func (s *PostgresStore) IsHealthy() bool {
-	return s.db.Ping() == nil
+func (s *PostgresStore) IsHealthy() error {
+	if err := s.db.Ping(); err != nil {
+		return ErrStoreUnhealthy
+	}
+	return nil
 }

 func (s *PostgresStore) statsTotal(limit int) ([]StatsResult, error) {