Ajouter ping de database à healthcheck

This commit is contained in:
Victor Lacasse-Beaudoin 2023-06-02 03:17:43 -04:00
parent b4af26d3dd
commit 7823541f0e

View file

@ -1,8 +1,10 @@
package handlers package handlers
import ( import (
"fmt"
"net/http" "net/http"
"git.agecem.com/agecem/bottin/v4/data"
"github.com/labstack/echo/v4" "github.com/labstack/echo/v4"
) )
@ -11,7 +13,23 @@ type GetHealthResponse struct {
} }
func GetHealth(c echo.Context) error { func GetHealth(c echo.Context) error {
response := GetHealthResponse{"Bottin API v4 is ready"} response := GetHealthResponse{
Message: "Bottin API v4 is ready",
}
dataClient, err := data.NewDataClientFromViper()
if err != nil {
response.Message = fmt.Sprintf("Error during data.NewDataClientFromViper(): %s", err)
return c.JSON(http.StatusInternalServerError, response)
}
defer dataClient.DB.Close()
if err = dataClient.DB.Ping(); err != nil {
response.Message = fmt.Sprintf("Error during dataClient.DB.Ping(): %s", err)
return c.JSON(http.StatusInternalServerError, response)
}
return c.JSON(http.StatusOK, response) return c.JSON(http.StatusOK, response)
} }