82 lines
2 KiB
Go
82 lines
2 KiB
Go
package api
|
|
|
|
import (
|
|
"database/sql"
|
|
"fmt"
|
|
"net/http"
|
|
"os"
|
|
"time"
|
|
|
|
"github.com/ente-io/museum/ente"
|
|
"github.com/ente-io/stacktrace"
|
|
"github.com/sirupsen/logrus"
|
|
"github.com/spf13/viper"
|
|
|
|
"github.com/ente-io/museum/pkg/utils/config"
|
|
"github.com/ente-io/museum/pkg/utils/handler"
|
|
"github.com/gin-gonic/gin"
|
|
)
|
|
|
|
type HealthCheckHandler struct {
|
|
DB *sql.DB
|
|
}
|
|
|
|
func (h *HealthCheckHandler) Ping(c *gin.Context) {
|
|
res := 0
|
|
err := h.DB.QueryRowContext(c, `SELECT 1`).Scan(&res)
|
|
if err != nil {
|
|
handler.Error(c, stacktrace.Propagate(err, ""))
|
|
return
|
|
}
|
|
result := make(map[string]string)
|
|
result["message"] = "pong"
|
|
result["id"] = os.Getenv("GIT_COMMIT")
|
|
if c.Query("host") != "" {
|
|
result["host"], _ = os.Hostname()
|
|
}
|
|
c.JSON(http.StatusOK, result)
|
|
}
|
|
|
|
func (h *HealthCheckHandler) PingDBStats(c *gin.Context) {
|
|
host, _ := os.Hostname()
|
|
stats := h.DB.Stats()
|
|
logrus.WithFields(logrus.Fields{
|
|
"MaxOpenConnections": stats.MaxOpenConnections,
|
|
"Idle": stats.Idle,
|
|
"InUse": stats.InUse,
|
|
"OpenConnections": stats.OpenConnections,
|
|
"WaitCount": stats.WaitCount,
|
|
"WaitDuration": stats.WaitDuration.String(),
|
|
"MaxIdleClosed": stats.MaxIdleClosed,
|
|
"MaxIdleTimeClosed": stats.MaxIdleTimeClosed,
|
|
"MaxLifetimeClosed": stats.MaxLifetimeClosed,
|
|
}).Info("DB STATS")
|
|
|
|
logrus.Info("DB Ping Start")
|
|
err := h.DB.Ping()
|
|
if err != nil {
|
|
logrus.WithError(err).Error("DB Ping failed")
|
|
handler.Error(c, stacktrace.Propagate(ente.NewInternalError(fmt.Sprintf("DB ping failed on %s", host)), ""))
|
|
return
|
|
}
|
|
c.Status(http.StatusOK)
|
|
}
|
|
|
|
func (h *HealthCheckHandler) PerformHealthCheck() {
|
|
logrus.Info("Performing HC");
|
|
healthCheckURL := viper.GetString("internal.health-check-url")
|
|
if healthCheckURL == "" {
|
|
if !config.IsLocalEnvironment() {
|
|
logrus.Error("Could not obtain health check URL in non-local environment")
|
|
}
|
|
return
|
|
}
|
|
var client = &http.Client{
|
|
Timeout: 10 * time.Second,
|
|
}
|
|
_, err := client.Head(healthCheckURL)
|
|
if err != nil {
|
|
logrus.Error("Error performing health check", err)
|
|
}
|
|
}
|