X-Git-Url: https://gerrit.o-ran-sc.org/r/gitweb?a=blobdiff_plain;f=a1-go%2Fpkg%2Frestful%2Frestful.go;h=f34a5ad208a730688c168ac5fe29bfe7f9be69e0;hb=6242a3ff1b95eb8fc89931b11115ec9c06d08e96;hp=66b66fb97becb26a5948d2e90331dcce707b8623;hpb=327d9dba437b30a06161d4f21cc2a33034d75bac;p=ric-plt%2Fa1.git

diff --git a/a1-go/pkg/restful/restful.go b/a1-go/pkg/restful/restful.go
index 66b66fb..f34a5ad 100644
--- a/a1-go/pkg/restful/restful.go
+++ b/a1-go/pkg/restful/restful.go
@@ -21,16 +21,17 @@
 package restful
 
 import (
-	"fmt"
 	"log"
 	"os"
 
+	"gerrit.o-ran-sc.org/r/ric-plt/a1/pkg/a1"
+	"gerrit.o-ran-sc.org/r/ric-plt/a1/pkg/models"
+	"gerrit.o-ran-sc.org/r/ric-plt/a1/pkg/restapi"
+	"gerrit.o-ran-sc.org/r/ric-plt/a1/pkg/restapi/operations"
+	"gerrit.o-ran-sc.org/r/ric-plt/a1/pkg/restapi/operations/a1_mediator"
+	"gerrit.o-ran-sc.org/r/ric-plt/a1/pkg/resthooks"
 	"github.com/go-openapi/loads"
 	"github.com/go-openapi/runtime/middleware"
-	"subh.com/a1-go/pkg/restapi"
-	"subh.com/a1-go/pkg/restapi/operations"
-	"subh.com/a1-go/pkg/restapi/operations/a1_mediator"
-	"subh.com/a1-go/pkg/resthooks"
 )
 
 func NewRestful() *Restful {
@@ -49,9 +50,65 @@ func (r *Restful) setupHandler() *operations.A1API {
 
 	api := operations.NewA1API(swaggerSpec)
 	api.A1MediatorA1ControllerGetAllPolicyTypesHandler = a1_mediator.A1ControllerGetAllPolicyTypesHandlerFunc(func(param a1_mediator.A1ControllerGetAllPolicyTypesParams) middleware.Responder {
-		fmt.Printf("\n---- handler for get all all policy type --- \n")
+		a1.Logger.Debug("handler for get all policy type")
 		return a1_mediator.NewA1ControllerGetAllPolicyTypesOK().WithPayload(r.rh.GetAllPolicyType())
 	})
+
+	api.A1MediatorA1ControllerCreatePolicyTypeHandler = a1_mediator.A1ControllerCreatePolicyTypeHandlerFunc(func(params a1_mediator.A1ControllerCreatePolicyTypeParams) middleware.Responder {
+		a1.Logger.Debug("handler for Create policy type ")
+		if err = r.rh.CreatePolicyType(models.PolicyTypeID(params.PolicyTypeID), *params.Body); err == nil {
+			//Increase prometheus counter
+			return a1_mediator.NewA1ControllerCreatePolicyTypeCreated()
+		}
+		if r.rh.IsTypeAlready(err) || r.rh.IsTypeMismatch(err) {
+			return a1_mediator.NewA1ControllerCreatePolicyTypeBadRequest()
+		}
+		return a1_mediator.NewA1ControllerCreatePolicyTypeServiceUnavailable()
+
+	})
+
+	api.A1MediatorA1ControllerGetPolicyTypeHandler = a1_mediator.A1ControllerGetPolicyTypeHandlerFunc(func(params a1_mediator.A1ControllerGetPolicyTypeParams) middleware.Responder {
+		a1.Logger.Debug("handler for get policy type from policytypeID")
+		return a1_mediator.NewA1ControllerGetPolicyTypeOK().WithPayload(r.rh.GetPolicyType(models.PolicyTypeID(params.PolicyTypeID)))
+	})
+
+	api.A1MediatorA1ControllerCreateOrReplacePolicyInstanceHandler = a1_mediator.A1ControllerCreateOrReplacePolicyInstanceHandlerFunc(func(params a1_mediator.A1ControllerCreateOrReplacePolicyInstanceParams) middleware.Responder {
+		a1.Logger.Debug("handler for create policy type instance ")
+		if err = r.rh.CreatePolicyInstance(models.PolicyTypeID(params.PolicyTypeID), models.PolicyInstanceID(params.PolicyInstanceID), params.Body); err == nil {
+			return a1_mediator.NewA1ControllerCreateOrReplacePolicyInstanceAccepted()
+		}
+		if r.rh.IsValidJson(err) {
+			return a1_mediator.NewA1ControllerCreateOrReplacePolicyInstanceBadRequest()
+		}
+		return a1_mediator.NewA1ControllerCreateOrReplacePolicyInstanceServiceUnavailable()
+
+	})
+
+	api.A1MediatorA1ControllerGetPolicyInstanceHandler = a1_mediator.A1ControllerGetPolicyInstanceHandlerFunc(func(params a1_mediator.A1ControllerGetPolicyInstanceParams) middleware.Responder {
+		a1.Logger.Debug("handler for get policy instance from policytypeID")
+		if resp, err := r.rh.GetPolicyInstance(models.PolicyTypeID(params.PolicyTypeID), models.PolicyInstanceID(params.PolicyInstanceID)); err == nil {
+			return a1_mediator.NewA1ControllerGetPolicyInstanceOK().WithPayload(resp)
+		}
+		if r.rh.IsPolicyInstanceNotFound(err) {
+			return a1_mediator.NewA1ControllerGetPolicyInstanceNotFound()
+		}
+		return a1_mediator.NewA1ControllerGetPolicyInstanceServiceUnavailable()
+	})
+
+	api.A1MediatorA1ControllerGetAllInstancesForTypeHandler = a1_mediator.A1ControllerGetAllInstancesForTypeHandlerFunc(func(params a1_mediator.A1ControllerGetAllInstancesForTypeParams) middleware.Responder {
+		a1.Logger.Debug("handler for get all policy instance")
+		if resp, err := r.rh.GetAllPolicyInstance(models.PolicyTypeID(params.PolicyTypeID)); err == nil {
+			if resp != nil {
+				return a1_mediator.NewA1ControllerGetAllInstancesForTypeOK().WithPayload(resp)
+			}
+		}
+		if r.rh.IsPolicyInstanceNotFound(err) {
+			return a1_mediator.NewA1ControllerGetPolicyInstanceNotFound()
+		}
+		return a1_mediator.NewA1ControllerGetAllInstancesForTypeServiceUnavailable()
+
+	})
+
 	return api
 
 }