Health check API for service

[ric-plt/ricdms.git] / pkg / restful / restful.go

diff --git a/pkg/restful/restful.go b/pkg/restful/restful.go
index 4693f56..4255e2b 100644 (file)
--- a/pkg/restful/restful.go
+++ b/pkg/restful/restful.go
@@ -23,13 +23,22 @@ import (
        "log"
        "os"
 
+       ph "gerrit.o-ran-sc.org/r/ric-plt/ricdms/pkg/health"
        "gerrit.o-ran-sc.org/r/ric-plt/ricdms/pkg/restapi"
        "gerrit.o-ran-sc.org/r/ric-plt/ricdms/pkg/restapi/operations"
+       "gerrit.o-ran-sc.org/r/ric-plt/ricdms/pkg/restapi/operations/health"
+       "gerrit.o-ran-sc.org/r/ric-plt/ricdms/pkg/resthooks"
+       "gerrit.o-ran-sc.org/r/ric-plt/ricdms/pkg/ricdms"
        "github.com/go-openapi/loads"
+       "github.com/go-openapi/runtime/middleware"
 )
 
 func NewRestful() *Restful {
-       r := &Restful{}
+       r := &Restful{
+               rh: resthooks.NewResthook(
+                       ph.NewHealthChecker(),
+               ),
+       }
        r.setupHandler()
        return r
 }
@@ -41,6 +50,13 @@ func (r *Restful) setupHandler() {
        }
 
        api := operations.NewRICDMSAPI(swaggerSpec)
+
+       api.HealthGetHealthCheckHandler = health.GetHealthCheckHandlerFunc(func(ghcp health.GetHealthCheckParams) middleware.Responder {
+               ricdms.Logger.Debug("==> HealthCheck API invoked.")
+               resp := r.rh.GetDMSHealth()
+               return resp
+       })
+
        r.api = api
 }
 
@@ -49,6 +65,7 @@ func (r *Restful) Run() {
        defer server.Shutdown()
        server.Port = 8000
        server.Host = "0.0.0.0"
+       ricdms.Logger.Info("Starting server at : %s:%d", server.Host, server.Port)
        if err := server.Serve(); err != nil {
                log.Fatal(err.Error())
        }