X-Git-Url: https://gerrit.o-ran-sc.org/r/gitweb?a=blobdiff_plain;f=a1-go%2Fpkg%2Fresthooks%2Fresthooks.go;h=54ca39d74ef4c125b7ecea62357ec64834e0cd1b;hb=b5c9dd5ecf69928bb998662b12c01672895e20d3;hp=db8bd6e26271023ce7240eea438bd49f5aff92f8;hpb=f8e5619649c0fb2ddfc466b970e8338a45e55cb8;p=ric-plt%2Fa1.git

diff --git a/a1-go/pkg/resthooks/resthooks.go b/a1-go/pkg/resthooks/resthooks.go
index db8bd6e..54ca39d 100644
--- a/a1-go/pkg/resthooks/resthooks.go
+++ b/a1-go/pkg/resthooks/resthooks.go
@@ -21,14 +21,31 @@
 package resthooks
 
 import (
+	"encoding/json"
+	"errors"
 	"fmt"
 	"strconv"
 	"strings"
 
+	"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/sdlgo"
-       "gerrit.o-ran-sc.org/r/ric-plt/a1/pkg/models"
 )
 
+const (
+	a1PolicyPrefix = "a1.policy_type."
+	a1MediatorNs   = "A1m_ns"
+)
+
+var typeAlreadyError = errors.New("Policy Type already exists")
+var typeMismatchError = errors.New("Policytype Mismatch")
+
+func (rh *Resthook) IsTypeAlready(err error) bool {
+	return err == typeAlreadyError
+}
+func (rh *Resthook) IsTypeMismatch(err error) bool {
+	return err == typeMismatchError
+}
 func NewResthook() *Resthook {
 	return createResthook(sdlgo.NewSyncStorage())
 }
@@ -46,19 +63,95 @@ func (rh *Resthook) GetAllPolicyType() []models.PolicyTypeID {
 	keys, err := rh.db.GetAll("A1m_ns")
 
 	if err != nil {
-		fmt.Printf("error in retrieving policy. err: %v", err)
+		a1.Logger.Error("error in retrieving policy. err: %v", err)
 		return policyTypeIDs
 	}
-	fmt.Printf("keys : %+v", keys)
+	a1.Logger.Debug("keys : %+v", keys)
 
 	for _, key := range keys {
-		if strings.HasPrefix(strings.TrimLeft(key, " "), "a1.policy_type.") {
-			pti := strings.Split(strings.Trim(key, " "), "a1.policy_type.")[1]
+		if strings.HasPrefix(strings.TrimLeft(key, " "), a1PolicyPrefix) {
+			pti := strings.Split(strings.Trim(key, " "), a1PolicyPrefix)[1]
 			ptii, _ := strconv.ParseInt(pti, 10, 64)
 			policyTypeIDs = append(policyTypeIDs, models.PolicyTypeID(ptii))
 		}
 	}
 
-	fmt.Printf("return : %+v", policyTypeIDs)
+	a1.Logger.Debug("return : %+v", policyTypeIDs)
 	return policyTypeIDs
 }
+
+func (rh *Resthook) GetPolicyType(policyTypeId models.PolicyTypeID) *models.PolicyTypeSchema {
+	a1.Logger.Debug("GetPolicyType1")
+
+	var policytypeschema *models.PolicyTypeSchema
+	var keys [1]string
+
+	key := a1PolicyPrefix + strconv.FormatInt((int64(policyTypeId)), 10)
+	keys[0] = key
+
+	a1.Logger.Debug("key : %+v", key)
+
+	valmap, err := rh.db.Get(a1MediatorNs, keys[:])
+
+	a1.Logger.Debug("policytype map : ", valmap)
+
+	if len(valmap) == 0 {
+		a1.Logger.Error("policy type Not Present for policyid : %v", policyTypeId)
+		return policytypeschema
+	}
+
+	if err != nil {
+		a1.Logger.Error("error in retrieving policy type. err: %v", err)
+		return policytypeschema
+	}
+
+	if valmap[key] == nil {
+		a1.Logger.Error("policy type Not Present for policyid : %v", policyTypeId)
+		return policytypeschema
+	}
+
+	a1.Logger.Debug("keysmap : %+v", valmap[key])
+
+	var item models.PolicyTypeSchema
+	valStr := fmt.Sprint(valmap[key])
+
+	a1.Logger.Debug("Policy type for %+v :  %+v", key, valStr)
+	valkey := "`" + valStr + "`"
+	valToUnmarshall, err := strconv.Unquote(valkey)
+	if err != nil {
+		panic(err)
+	}
+
+	a1.Logger.Debug("Policy type for %+v :  %+v", key, string(b))
+	errunm := json.Unmarshal([]byte(valToUnmarshall), &item)
+
+	a1.Logger.Debug(" Unmarshalled json : %+v", (errunm))
+	a1.Logger.Debug("Policy type Name :  %v", (item.Name))
+
+	return &item
+}
+
+func (rh *Resthook) CreatePolicyType(policyTypeId models.PolicyTypeID, httprequest models.PolicyTypeSchema) error {
+	a1.Logger.Debug("CreatePolicyType function")
+	if policyTypeId != models.PolicyTypeID(*httprequest.PolicyTypeID) {
+		//error message
+		a1.Logger.Debug("Policytype Mismatch")
+		return typeMismatchError
+	}
+	key := a1PolicyPrefix + strconv.FormatInt((int64(policyTypeId)), 10)
+	a1.Logger.Debug("key %+v ", key)
+	if data, err := httprequest.MarshalBinary(); err == nil {
+		a1.Logger.Debug("Marshaled String : %+v", string(data))
+		success, err1 := rh.db.SetIfNotExists(a1MediatorNs, key, string(data))
+		a1.Logger.Info("success:%+v", success)
+		if err1 != nil {
+			a1.Logger.Error("error :%+v", err1)
+			return err1
+		}
+		if !success {
+			a1.Logger.Debug("Policy type %+v already exist", policyTypeId)
+			return typeAlreadyError
+		}
+	}
+	return nil
+}