2019-05-16 10:39:25 -04:00
|
|
|
package handlers
|
|
|
|
|
|
|
|
import (
|
|
|
|
"context"
|
|
|
|
"net/http"
|
|
|
|
|
|
|
|
"geeks-accelerator/oss/saas-starter-kit/example-project/internal/platform/web"
|
2019-05-23 14:32:24 -05:00
|
|
|
"github.com/jmoiron/sqlx"
|
2019-06-24 17:36:42 -08:00
|
|
|
"github.com/pkg/errors"
|
|
|
|
"gopkg.in/DataDog/dd-trace-go.v1/contrib/go-redis/redis"
|
2019-05-16 10:39:25 -04:00
|
|
|
)
|
|
|
|
|
|
|
|
// Check provides support for orchestration health checks.
|
|
|
|
type Check struct {
|
2019-05-23 14:32:24 -05:00
|
|
|
MasterDB *sqlx.DB
|
2019-06-24 17:36:42 -08:00
|
|
|
Redis *redis.Client
|
2019-05-16 10:39:25 -04:00
|
|
|
|
|
|
|
// ADD OTHER STATE LIKE THE LOGGER IF NEEDED.
|
|
|
|
}
|
|
|
|
|
|
|
|
// Health validates the service is healthy and ready to accept requests.
|
|
|
|
func (c *Check) Health(ctx context.Context, w http.ResponseWriter, r *http.Request, params map[string]string) error {
|
2019-06-24 17:36:42 -08:00
|
|
|
|
|
|
|
// check postgres
|
2019-05-23 14:32:24 -05:00
|
|
|
_, err := c.MasterDB.Exec("SELECT 1")
|
|
|
|
if err != nil {
|
2019-06-24 17:36:42 -08:00
|
|
|
return errors.Wrap(err, "Postgres failed")
|
|
|
|
}
|
|
|
|
|
|
|
|
// check redis
|
2019-06-26 01:16:57 -08:00
|
|
|
//err = c.Redis.Ping().Err()
|
|
|
|
//if err != nil {
|
|
|
|
// return errors.Wrap(err, "Redis failed")
|
|
|
|
//}
|
2019-05-16 10:39:25 -04:00
|
|
|
|
|
|
|
status := struct {
|
|
|
|
Status string `json:"status"`
|
|
|
|
}{
|
|
|
|
Status: "ok",
|
|
|
|
}
|
|
|
|
|
2019-05-23 14:32:24 -05:00
|
|
|
return web.RespondJson(ctx, w, status, http.StatusOK)
|
2019-05-16 10:39:25 -04:00
|
|
|
}
|