Implement health check for xApp

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

diff --git a/pkg/restful/restful.go b/pkg/restful/restful.go
index acfd65a..b17c7ad 100644 (file)
--- a/pkg/restful/restful.go
+++ b/pkg/restful/restful.go
@@ -20,20 +20,26 @@
 package restful
 
 import (
+       "fmt"
+       "io"
+       "io/ioutil"
        "log"
        "os"
 
        ch "gerrit.o-ran-sc.org/r/ric-plt/ricdms/pkg/charts"
+       dm "gerrit.o-ran-sc.org/r/ric-plt/ricdms/pkg/deploy"
        ph "gerrit.o-ran-sc.org/r/ric-plt/ricdms/pkg/health"
        po "gerrit.o-ran-sc.org/r/ric-plt/ricdms/pkg/onboard"
        "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/charts"
+       "gerrit.o-ran-sc.org/r/ric-plt/ricdms/pkg/restapi/operations/deploy"
        "gerrit.o-ran-sc.org/r/ric-plt/ricdms/pkg/restapi/operations/health"
        "gerrit.o-ran-sc.org/r/ric-plt/ricdms/pkg/restapi/operations/onboard"
        "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"
        "github.com/go-openapi/runtime/middleware"
 )
 
@@ -43,6 +49,7 @@ func NewRestful() *Restful {
                        ph.NewHealthChecker(),
                        po.NewOnboarder(),
                        ch.NewChartmgr(),
+                       dm.NewDeploymentManager(),
                ),
        }
        r.setupHandler()
@@ -63,6 +70,12 @@ func (r *Restful) setupHandler() {
                return resp
        })
 
+       api.HealthGetHealthcheckXAppXAppNameNamespaceNamespaceHandler = health.GetHealthcheckXAppXAppNameNamespaceNamespaceHandlerFunc(func(param health.GetHealthcheckXAppXAppNameNamespaceNamespaceParams) middleware.Responder {
+               ricdms.Logger.Debug("==> Healthcheck for xApp is invoked")
+               resp := r.rh.GetxAppHealth(param.XAppName, param.Namespace)
+               return resp
+       })
+
        api.OnboardPostOnboardxAppsHandler = onboard.PostOnboardxAppsHandlerFunc(func(poap onboard.PostOnboardxAppsParams) middleware.Responder {
                ricdms.Logger.Debug("==> onboard API invoked.")
                resp := r.rh.OnBoard(poap.Body)
@@ -75,6 +88,49 @@ func (r *Restful) setupHandler() {
                return resp
        })
 
+       api.ChartsDownloadHelmChartHandler = charts.DownloadHelmChartHandlerFunc(func(param charts.DownloadHelmChartParams) middleware.Responder {
+               ricdms.Logger.Debug("==> Download helm chart")
+               resp := r.rh.DownloadChart(param.XAppName, param.Version)
+               return resp
+       })
+
+       api.ChartsGetChartHandler = charts.GetChartHandlerFunc(func(param charts.GetChartParams) middleware.Responder {
+               ricdms.Logger.Debug("==> Get Charts by name is invoked")
+               resp := r.rh.GetChartsByName(param.XAppName)
+               return resp
+       })
+
+       api.ChartsGetChartsFetcherHandler = charts.GetChartsFetcherHandlerFunc(func(param charts.GetChartsFetcherParams) middleware.Responder {
+               ricdms.Logger.Debug("==> Get Charts by name and version is invoked")
+               resp := r.rh.GetChartByNameAndVersion(param.XAppName, param.Version)
+               return resp
+       })
+
+       api.DeployPostDeployHandler = deploy.PostDeployHandlerFunc(func(param deploy.PostDeployParams) middleware.Responder {
+               ricdms.Logger.Debug("==> deployment of xApp")
+               resp := r.rh.DownloadAndInstallChart(param.Body.XAppname, param.Body.Version, *param.Body.Namespace)
+               return resp
+       })
+
+       api.ApplicationZipProducer = runtime.ProducerFunc(func(w io.Writer, data interface{}) error {
+               if zp, ok := data.(io.ReadCloser); ok {
+                       defer zp.Close()
+                       b, err := ioutil.ReadAll(zp)
+
+                       if err != nil {
+                               ricdms.Logger.Error("error: %v", err)
+                               return err
+                       }
+                       _, err = w.Write(b)
+
+                       if err != nil {
+                               ricdms.Logger.Error("error: %v", err)
+                               return err
+                       }
+                       return nil
+               }
+               return fmt.Errorf("not support")
+       })
        r.api = api
 }