X-Git-Url: https://gerrit.o-ran-sc.org/r/gitweb?a=blobdiff_plain;ds=sidebyside;f=pkg%2Frestful%2Frestful.go;h=4255e2b02cb376d1ed1e87bab894e58ec2aada7c;hb=cd6cc6ad28c32ccce6a2833917f7408a798e45fc;hp=4693f56c53464b3fb37adb0a909a7fc758dbf739;hpb=8333c8b0a4fa46bcd1cf414662c602922cff8f57;p=ric-plt%2Fricdms.git

diff --git a/pkg/restful/restful.go b/pkg/restful/restful.go
index 4693f56..4255e2b 100644
--- 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())
 	}