X-Git-Url: https://gerrit.o-ran-sc.org/r/gitweb?a=blobdiff_plain;ds=sidebyside;f=pkg%2Frestful%2Frestful.go;h=3f493e794d16a402d8291bc159d49795512973aa;hb=d916d3bfdf78aae92bf54253a20ef7cf73f5b21f;hp=f7ce15bc2c6c616b313ff07142cfe490cade9da1;hpb=5152bd39551f19139752d89a795327ca9dd189db;p=ric-plt%2Fa1.git

diff --git a/pkg/restful/restful.go b/pkg/restful/restful.go
index f7ce15b..3f493e7 100644
--- a/pkg/restful/restful.go
+++ b/pkg/restful/restful.go
@@ -82,7 +82,12 @@ func (r *Restful) setupHandler() *operations.A1API {
 
 	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)))
+                var policyTypeSchema *models.PolicyTypeSchema
+                policyTypeSchema, err = r.rh.GetPolicyType(models.PolicyTypeID(params.PolicyTypeID))
+	        if err != nil {
+		        return a1_mediator.NewA1ControllerGetPolicyTypeNotFound()
+	        }
+		return a1_mediator.NewA1ControllerGetPolicyTypeOK().WithPayload(policyTypeSchema)
 	})
 
 	api.A1MediatorA1ControllerCreateOrReplacePolicyInstanceHandler = a1_mediator.A1ControllerCreateOrReplacePolicyInstanceHandlerFunc(func(params a1_mediator.A1ControllerCreateOrReplacePolicyInstanceParams) middleware.Responder {