Add healthcheck to all stores

2021-11-06 00:37:28 +01:00
parent a19fb3086b
commit 8ba601a2b1
5 changed files with 21 additions and 0 deletions
--- a/honeypot/ssh/store/postgres.go
+++ b/honeypot/ssh/store/postgres.go
@@ -126,6 +126,10 @@ 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) statsTotal(limit int) ([]StatsResult, error) {
 	uniquePasswordStmt := `select count(*) from (select distinct password from login_attempts) as temp`
 	uniqueUsernameStmt := `select count(*) from (select distinct username from login_attempts) as temp`