X-Git-Url: https://gerrit.o-ran-sc.org/r/gitweb?a=blobdiff_plain;f=a1-go%2Fpkg%2Fresthooks%2Fresthooks.go;h=8d30b3d61bcb7a6a74bcd387d965e87ec64dc51b;hb=b513650252f1a739e3e9e2d02f2598d506aa86cf;hp=43011a467d51ae2531e173d22f754ada3a1b4a54;hpb=3ad6de4df1dfe70e467064e6477f2e6068e87048;p=ric-plt%2Fa1.git diff --git a/a1-go/pkg/resthooks/resthooks.go b/a1-go/pkg/resthooks/resthooks.go index 43011a4..8d30b3d 100644 --- a/a1-go/pkg/resthooks/resthooks.go +++ b/a1-go/pkg/resthooks/resthooks.go @@ -26,18 +26,22 @@ import ( "fmt" "strconv" "strings" + "time" "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/rmr" "gerrit.o-ran-sc.org/r/ric-plt/sdlgo" "github.com/santhosh-tekuri/jsonschema/v5" "gopkg.in/yaml.v2" ) const ( - a1PolicyPrefix = "a1.policy_type." - a1MediatorNs = "A1m_ns" - a1InstancePrefix = "a1.policy_instance." + a1PolicyPrefix = "a1.policy_type." + a1MediatorNs = "A1m_ns" + a1InstancePrefix = "a1.policy_instance." + a1InstanceMetadataPrefix = "a1.policy_inst_metadata." + a1HandlerPrefix = "a1.policy_handler." ) var typeAlreadyError = errors.New("Policy Type already exists") @@ -46,6 +50,11 @@ var typeMismatchError = errors.New("Policytype Mismatch") var invalidJsonSchema = errors.New("Invalid Json ") var policyInstanceNotFoundError = errors.New("Policy Instance Not Found") var policyTypeNotFoundError = errors.New("Policy Type Not Found") +var policyTypeCanNotBeDeletedError = errors.New("tried to delete a type that isn't empty") + +func (rh *Resthook) CanPolicyTypeBeDeleted(err error) bool { + return err == policyTypeCanNotBeDeletedError +} func (rh *Resthook) IsPolicyTypePresent(err error) bool { return err == policyTypeNotFoundError @@ -69,12 +78,13 @@ func (rh *Resthook) IsValidJson(err error) bool { return err == invalidJsonSchema } func NewResthook() *Resthook { - return createResthook(sdlgo.NewSyncStorage()) + return createResthook(sdlgo.NewSyncStorage(), rmr.NewRMRSender()) } -func createResthook(sdlInst iSdl) *Resthook { +func createResthook(sdlInst iSdl, rmrSenderInst rmr.IRmrSender) *Resthook { return &Resthook{ - db: sdlInst, + db: sdlInst, + iRmrSenderInst: rmrSenderInst, } } @@ -115,7 +125,7 @@ func (rh *Resthook) GetPolicyType(policyTypeId models.PolicyTypeID) *models.Poli valmap, err := rh.db.Get(a1MediatorNs, keys[:]) - a1.Logger.Debug("policytype map : ", valmap) + a1.Logger.Debug("policytype map : %+v", valmap) if len(valmap) == 0 { a1.Logger.Error("policy type Not Present for policyid : %v", policyTypeId) @@ -239,7 +249,7 @@ func validate(httpBodyString string, schemaString string) bool { return true } -func (rh *Resthook) StorePolicyInstance(policyTypeId models.PolicyTypeID, policyInstanceID models.PolicyInstanceID, httpBody interface{}) (string, error) { +func (rh *Resthook) storePolicyInstance(policyTypeId models.PolicyTypeID, policyInstanceID models.PolicyInstanceID, httpBody interface{}) (string, error) { var keys [1]string operation := "CREATE" typekey := a1PolicyPrefix + strconv.FormatInt((int64(policyTypeId)), 10) @@ -301,6 +311,31 @@ func (rh *Resthook) StorePolicyInstance(policyTypeId models.PolicyTypeID, policy return operation, nil } +func (rh *Resthook) storePolicyInstanceMetadata(policyTypeId models.PolicyTypeID, policyInstanceID models.PolicyInstanceID) (bool, error) { + + creation_timestamp := time.Now() + instanceMetadataKey := a1InstanceMetadataPrefix + strconv.FormatInt((int64(policyTypeId)), 10) + "." + string(policyInstanceID) + + a1.Logger.Debug("key : %+v", instanceMetadataKey) + + var metadatajson []interface{} + metadatajson = append(metadatajson, map[string]string{"created_at": creation_timestamp.Format("2006-01-02 15:04:05"), "has_been_deleted": "False"}) + metadata, _ := json.Marshal(metadatajson) + + a1.Logger.Debug("policyinstanceMetaData to create : %+v", string(metadata)) + + err := rh.db.Set(a1MediatorNs, instanceMetadataKey, string(metadata)) + + if err != nil { + a1.Logger.Error("error :%+v", err) + return false, err + } + + a1.Logger.Debug("Policy Instance Meta Data created at :%+v", creation_timestamp) + + return true, nil +} + func (rh *Resthook) CreatePolicyInstance(policyTypeId models.PolicyTypeID, policyInstanceID models.PolicyInstanceID, httpBody interface{}) error { a1.Logger.Debug("CreatePolicyInstance function") // validate the PUT against the schema @@ -314,15 +349,34 @@ func (rh *Resthook) CreatePolicyInstance(policyTypeId models.PolicyTypeID, polic a1.Logger.Debug("schema to validate %+v", string(schemaStr)) a1.Logger.Debug("httpbody to validate %+v", httpBody) schemaString := fmt.Sprint(string(schemaStr)) - httpBodyString := fmt.Sprint((httpBody)) + httpBodyMarshal, err := json.Marshal(httpBody) + httpBodyString := string((httpBodyMarshal)) + a1.Logger.Debug("schema to validate sprint %+v", (schemaString)) + a1.Logger.Debug("httpbody to validate sprint %+v", httpBodyString) isvalid := validate(httpBodyString, schemaString) if isvalid { - operation, err := rh.StorePolicyInstance(policyTypeId, policyInstanceID, httpBody) + var operation string + operation, err = rh.storePolicyInstance(policyTypeId, policyInstanceID, httpBody) if err != nil { a1.Logger.Error("error :%+v", err) return err } a1.Logger.Debug("policy instance :%+v", operation) + iscreated, errmetadata := rh.storePolicyInstanceMetadata(policyTypeId, policyInstanceID) + if errmetadata != nil { + a1.Logger.Error("error :%+v", errmetadata) + return errmetadata + } + if iscreated { + a1.Logger.Debug("policy instance metadata created") + } + isSent := rh.iRmrSenderInst.RmrSendToXapp(httpBodyString) + if isSent { + a1.Logger.Debug("rmrSendToXapp : message sent") + } else { + a1.Logger.Debug("rmrSendToXapp : message not sent") + } + } else { a1.Logger.Error("%+v", invalidJsonSchema) return invalidJsonSchema @@ -330,3 +384,103 @@ func (rh *Resthook) CreatePolicyInstance(policyTypeId models.PolicyTypeID, polic return nil } + +func (rh *Resthook) GetPolicyInstance(policyTypeId models.PolicyTypeID, policyInstanceID models.PolicyInstanceID) (interface{}, error) { + a1.Logger.Debug("GetPolicyInstance1") + + var keys [1]string + + typekey := a1PolicyPrefix + strconv.FormatInt((int64(policyTypeId)), 10) + keys[0] = typekey + + a1.Logger.Debug("key1 : %+v", typekey) + + valmap, err := rh.db.Get(a1MediatorNs, keys[:]) + if len(valmap) == 0 { + a1.Logger.Debug("policy type Not Present for policyid : %v", policyTypeId) + return "{}", policyTypeNotFoundError + } + + if err != nil { + a1.Logger.Error("error in retrieving policy type. err: %v", err) + return "{}", err + } + + if valmap[typekey] == nil { + a1.Logger.Debug("policy type Not Present for policyid : %v", policyTypeId) + return "{}", policyTypeNotFoundError + } + + a1.Logger.Debug("keysmap : %+v", valmap[typekey]) + + instancekey := a1InstancePrefix + strconv.FormatInt((int64(policyTypeId)), 10) + "." + string(policyInstanceID) + a1.Logger.Debug("key2 : %+v", instancekey) + keys[0] = instancekey + instanceMap, err := rh.db.Get(a1MediatorNs, keys[:]) + if err != nil { + a1.Logger.Error("policy instance error : %v", err) + } + a1.Logger.Debug("policyinstancetype map : %+v", instanceMap) + + if instanceMap[instancekey] == nil { + a1.Logger.Debug("policy instance Not Present for policyinstaneid : %v", policyInstanceID) + return "{}", policyInstanceNotFoundError + } + + valStr := fmt.Sprint(instanceMap[instancekey]) + return valStr, nil +} + +func (rh *Resthook) GetAllPolicyInstance(policyTypeId models.PolicyTypeID) ([]models.PolicyInstanceID, error) { + a1.Logger.Debug("GetAllPolicyInstance") + var policyTypeInstances = []models.PolicyInstanceID{} + + keys, err := rh.db.GetAll("A1m_ns") + + if err != nil { + a1.Logger.Error("error in retrieving policy. err: %v", err) + return policyTypeInstances, err + } + a1.Logger.Debug("keys : %+v", keys) + typekey := a1InstancePrefix + strconv.FormatInt((int64(policyTypeId)), 10) + "." + + for _, key := range keys { + if strings.HasPrefix(strings.TrimLeft(key, " "), typekey) { + pti := strings.Split(strings.Trim(key, " "), typekey)[1] + a1.Logger.Debug("pti %+v", pti) + policyTypeInstances = append(policyTypeInstances, models.PolicyInstanceID(pti)) + } + } + + if len(policyTypeInstances) == 0 { + a1.Logger.Debug("policy instance Not Present ") + } + + a1.Logger.Debug("return : %+v", policyTypeInstances) + return policyTypeInstances, nil +} + +func (rh *Resthook) DeletePolicyType(policyTypeId models.PolicyTypeID) error { + a1.Logger.Debug("DeletePolicyType") + + policyinstances, err := rh.GetAllPolicyInstance(policyTypeId) + if err != nil { + a1.Logger.Error("error in retrieving policy. err: %v", err) + return err + } + + var keys [1]string + key := a1PolicyPrefix + strconv.FormatInt((int64(policyTypeId)), 10) + keys[0] = key + if len(policyinstances) == 0 { + err := rh.db.Remove(a1MediatorNs, keys[:]) + if err != nil { + a1.Logger.Error("error in deleting policy type err: %v", err) + return err + } + } else { + a1.Logger.Error("tried to delete a type that isn't empty") + return policyTypeCanNotBeDeletedError + } + return nil +}