47 lines
959 B
Go
47 lines
959 B
Go
package controllers
|
|
|
|
import (
|
|
"otaupdate/services"
|
|
"time"
|
|
|
|
"github.com/fiskerinc/cloud-services/pkg/health"
|
|
"github.com/fiskerinc/cloud-services/pkg/logger"
|
|
)
|
|
|
|
func HealthCheck() {
|
|
redis := health.NewRedisHealth(services.RedisClientPool())
|
|
server := health.HealthCheckServer{}
|
|
|
|
err := server.Serve([]health.Config{
|
|
{
|
|
Name: "db",
|
|
Check: health.NewPostgresCheck(services.GetDB().GetDBClient().GetConn()),
|
|
Timeout: time.Second * 1,
|
|
},
|
|
{
|
|
Name: "redis",
|
|
Check: redis.Check,
|
|
Timeout: time.Second * 1,
|
|
Info: redis.RedisStatus,
|
|
},
|
|
{
|
|
Name: "mongodb",
|
|
Check: health.NewMongoDBCheck(getMongoClient),
|
|
Timeout: time.Second * 1,
|
|
},
|
|
})
|
|
if err != nil {
|
|
logger.Error().Err(err).Send()
|
|
}
|
|
}
|
|
|
|
func getMongoClient() (health.MongoConnCheckInterface, error) {
|
|
client, err := services.GetMongoClient()
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
conn := client.(health.MongoConnCheckInterface)
|
|
return conn, nil
|
|
}
|