vendor/github.com/cboling/omci/generated/me.go - bbsim - Gitiles

 /*
  * Copyright (c) 2018 - present.  Boling Consulting Solutions (bcsw.net)
  *
  * Licensed under the Apache License, Version 2.0 (the "License");
  * you may not use this file except in compliance with the License.
  * You may obtain a copy of the License at
  * http://www.apache.org/licenses/LICENSE-2.0
  * Unless required by applicable law or agreed to in writing, software
  * distributed under the License is distributed on an "AS IS" BASIS,
  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  * See the License for the specific language governing permissions and
  * limitations under the License.
  */
 /*
  * NOTE: This file was generated, manual edits will be overwritten!
  *
  * Generated by 'goCodeGenerator.py':
  *              https://github.com/cboling/OMCI-parser/README.md
  */
 package generated

 import (
 	"encoding/binary"
 	"errors"
 	"fmt"
 	"github.com/deckarep/golang-set"
 	"github.com/google/gopacket"
 )

 type ManagedEntity struct {
 	definition    *ManagedEntityDefinition
 	attributeMask uint16
 	attributes    AttributeValueMap
 }

 // String provides a simple string that describes this struct
 func (entity *ManagedEntity) String() string {
 	return fmt.Sprintf("ManagedEntity: %v, EntityID: (%d/%#x): Attributes: %v",
 		entity.GetClassID(), entity.GetEntityID(), entity.GetEntityID(), entity.attributes)
 }

 func NewManagedEntity(definition *ManagedEntityDefinition, params ...ParamData) (*ManagedEntity, OmciErrors) {
 	entity := &ManagedEntity{
 		definition: definition,
 		attributes: make(map[string]interface{}),
 	}
 	if params != nil {
 		err := entity.setAttributes(params...)
 		if err != nil {
 			return nil, err
 		}
 	}
 	return entity, nil
 }

 func (entity *ManagedEntity) GetManagedEntityDefinition() *ManagedEntityDefinition {
 	return entity.definition
 }

 func (entity *ManagedEntity) SetManagedEntityDefinition(def *ManagedEntityDefinition) {
 	entity.definition = def
 }

 func (entity *ManagedEntity) GetName() string {
 	return entity.definition.GetName()
 }

 func (entity *ManagedEntity) GetClassID() ClassID {
 	return entity.definition.GetClassID()
 }

 func (entity *ManagedEntity) GetMessageTypes() mapset.Set {
 	return entity.definition.GetMessageTypes()
 }

 func (entity *ManagedEntity) GetAllowedAttributeMask() uint16 {
 	return entity.definition.GetAllowedAttributeMask()
 }

 func (entity *ManagedEntity) GetAttributeDefinitions() *AttributeDefinitionMap {
 	return entity.definition.GetAttributeDefinitions()
 }

 func (entity *ManagedEntity) DecodeAttributes(mask uint16, data []byte, p gopacket.PacketBuilder, msgType byte) (AttributeValueMap, error) {
 	return entity.definition.DecodeAttributes(mask, data, p, msgType)
 }

 func (entity *ManagedEntity) SerializeAttributes(attr AttributeValueMap, mask uint16,
 	b gopacket.SerializeBuffer, msgType byte, bytesAvailable int) error {
 	return entity.definition.SerializeAttributes(attr, mask, b, msgType, bytesAvailable)
 }

 func (entity *ManagedEntity) GetEntityID() uint16 {
 	if eid, err := entity.GetAttributeByIndex(0); err == nil {
 		return eid.(uint16)
 	}
 	return 0
 }

 func (entity *ManagedEntity) SetEntityID(eid uint16) error {
 	return entity.SetAttributeByIndex(0, eid)
 }

 func (entity *ManagedEntity) GetAttributeMask() uint16 {
 	return entity.attributeMask
 }

 func (entity *ManagedEntity) GetAttributeValueMap() *AttributeValueMap {
 	return &entity.attributes
 }

 func (entity *ManagedEntity) GetAttribute(name string) (interface{}, error) {
 	value, ok := entity.attributes[name]
 	if !ok {
 		return 0, errors.New(fmt.Sprintf("attribute '%v' not found", name))
 	}
 	return value, nil
 }

 func (entity *ManagedEntity) GetAttributeByIndex(index uint) (interface{}, error) {
 	if len(entity.attributes) == 0 {
 		return nil, errors.New("attributes have already been set")
 	}
 	if _, ok := entity.definition.AttributeDefinitions[index]; !ok {
 		return nil, errors.New(fmt.Sprintf("invalid attribute index: %d, should be 0..%d",
 			index, len(entity.definition.AttributeDefinitions)-1))
 	}
 	return entity.GetAttribute(entity.definition.AttributeDefinitions[index].Name)
 }

 func (entity *ManagedEntity) setAttributes(params ...ParamData) OmciErrors {
 	if entity.attributes == nil {
 		entity.attributes = make(map[string]interface{})
 	} else if len(entity.attributes) > 0 {
 		return NewNonStatusError("attributes have already been set")
 	}
 	eidName := entity.definition.AttributeDefinitions[0].Name
 	if len(params) == 0 {
 		entity.attributes[eidName] = uint16(0)
 		return nil
 	}
 	entity.attributes[eidName] = params[0].EntityID

 	for name, value := range params[0].Attributes {
 		if name == eidName {
 			continue
 		}
 		if err := entity.SetAttribute(name, value); err != nil {
 			return err
 		}
 	}
 	return nil
 }

 func (entity *ManagedEntity) SetAttribute(name string, value interface{}) OmciErrors {
 	attrDef, err := GetAttributeDefinitionByName(entity.definition.GetAttributeDefinitions(), name)
 	if err != nil {
 		return err
 	} else if entity.attributes == nil {
 		entity.attributes = make(map[string]interface{})
 	}
 	mask := uint16(1 << (16 - attrDef.GetIndex()))
 	// check any constraints
 	if constraintCheck := attrDef.GetConstraints(); constraintCheck != nil {
 		err = constraintCheck(value)
 		if err != nil {
 			return NewParameterError(mask, err)
 		}
 	}
 	entity.attributes[name] = value
 	entity.attributeMask |= mask
 	return nil
 }

 func (entity *ManagedEntity) SetAttributeByIndex(index uint, value interface{}) error {
 	attrDef, ok := entity.definition.AttributeDefinitions[index]
 	if !ok {
 		return errors.New(fmt.Sprintf("invalid attribute index: %d, should be 0..%d",
 			index, len(entity.definition.AttributeDefinitions)-1))
 	} else if entity.attributes == nil {
 		entity.attributes = make(map[string]interface{})
 	}
 	mask := uint16(1 << (16 - attrDef.GetIndex()))
 	// check any constraints
 	if constraintCheck := attrDef.GetConstraints(); constraintCheck != nil {
 		err := constraintCheck(value)
 		if err != nil {
 			return NewParameterError(mask, err)
 		}
 	}
 	entity.attributes[attrDef.Name] = value
 	entity.attributeMask |= mask
 	return nil
 }

 func (entity *ManagedEntity) DeleteAttribute(name string) error {
 	attrDef, err := GetAttributeDefinitionByName(entity.definition.GetAttributeDefinitions(), name)
 	if err != nil {
 		return err
 	}
 	if entity.attributes != nil {
 		delete(entity.attributes, name)
 		entity.attributeMask &= ^uint16(1 << (16 - attrDef.GetIndex()))
 	}
 	return nil
 }

 func (entity *ManagedEntity) DeleteAttributeByIndex(index uint) error {
 	attrDef, ok := entity.definition.AttributeDefinitions[index]
 	if !ok {
 		return errors.New(fmt.Sprintf("invalid attribute index: %d, should be 0..%d",
 			index, len(entity.definition.AttributeDefinitions)-1))
 	}
 	if entity.attributes != nil {
 		delete(entity.attributes, attrDef.Name)
 		entity.attributeMask &= ^uint16(1 << (16 - attrDef.GetIndex()))
 	}
 	return nil
 }

 func (entity *ManagedEntity) DecodeFromBytes(data []byte, p gopacket.PacketBuilder, msgType byte) error {
 	if len(data) < 6 {
 		p.SetTruncated()
 		return errors.New("frame too small")
 	}
 	classID := ClassID(binary.BigEndian.Uint16(data[0:2]))
 	entityID := binary.BigEndian.Uint16(data[2:4])
 	parameters := ParamData{EntityID: entityID}

 	meDefinition, omciErr := LoadManagedEntityDefinition(classID, parameters)
 	if omciErr != nil {
 		return omciErr.GetError()
 	}
 	entity.definition = meDefinition.definition
 	entity.attributeMask = binary.BigEndian.Uint16(data[4:6])
 	entity.attributes = make(map[string]interface{})
 	entity.SetEntityID(entityID)
 	packetAttributes, err := entity.DecodeAttributes(entity.GetAttributeMask(), data[6:], p, msgType)
 	if err != nil {
 		return err
 	}
 	for name, value := range packetAttributes {
 		entity.attributes[name] = value
 	}
 	return nil
 }

 func (entity *ManagedEntity) SerializeTo(b gopacket.SerializeBuffer, msgType byte, bytesAvailable int) error {
 	// Add class ID and entity ID
 	bytes, err := b.AppendBytes(6)
 	if err != nil {
 		return err
 	}
 	binary.BigEndian.PutUint16(bytes, uint16(entity.GetClassID()))
 	binary.BigEndian.PutUint16(bytes[2:], entity.GetEntityID())
 	binary.BigEndian.PutUint16(bytes[4:], entity.GetAttributeMask())

 	// TODO: Need to limit number of bytes appended to not exceed packet size
 	// Is there space/metadata info in 'b' parameter to allow this?
 	err = entity.SerializeAttributes(entity.attributes, entity.GetAttributeMask(), b, msgType, bytesAvailable)
 	return err
 }
	/*
	* Copyright (c) 2018 - present. Boling Consulting Solutions (bcsw.net)
	*
	* Licensed under the Apache License, Version 2.0 (the "License");
	* you may not use this file except in compliance with the License.
	* You may obtain a copy of the License at
	* http://www.apache.org/licenses/LICENSE-2.0
	* Unless required by applicable law or agreed to in writing, software
	* distributed under the License is distributed on an "AS IS" BASIS,
	* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
	* See the License for the specific language governing permissions and
	* limitations under the License.
	*/
	/*
	* NOTE: This file was generated, manual edits will be overwritten!
	*
	* Generated by 'goCodeGenerator.py':
	* https://github.com/cboling/OMCI-parser/README.md
	*/
	package generated

	import (
	"encoding/binary"
	"errors"
	"fmt"
	"github.com/deckarep/golang-set"
	"github.com/google/gopacket"
	)

	type ManagedEntity struct {
	definition *ManagedEntityDefinition
	attributeMask uint16
	attributes AttributeValueMap
	}

	// String provides a simple string that describes this struct
	func (entity *ManagedEntity) String() string {
	return fmt.Sprintf("ManagedEntity: %v, EntityID: (%d/%#x): Attributes: %v",
	entity.GetClassID(), entity.GetEntityID(), entity.GetEntityID(), entity.attributes)
	}

	func NewManagedEntity(definition ManagedEntityDefinition, params ...ParamData) (ManagedEntity, OmciErrors) {
	entity := &ManagedEntity{
	definition: definition,
	attributes: make(map[string]interface{}),
	}
	if params != nil {
	err := entity.setAttributes(params...)
	if err != nil {
	return nil, err
	}
	}
	return entity, nil
	}

	func (entity ManagedEntity) GetManagedEntityDefinition() ManagedEntityDefinition {
	return entity.definition
	}

	func (entity ManagedEntity) SetManagedEntityDefinition(def ManagedEntityDefinition) {
	entity.definition = def
	}

	func (entity *ManagedEntity) GetName() string {
	return entity.definition.GetName()
	}

	func (entity *ManagedEntity) GetClassID() ClassID {
	return entity.definition.GetClassID()
	}

	func (entity *ManagedEntity) GetMessageTypes() mapset.Set {
	return entity.definition.GetMessageTypes()
	}

	func (entity *ManagedEntity) GetAllowedAttributeMask() uint16 {
	return entity.definition.GetAllowedAttributeMask()
	}

	func (entity ManagedEntity) GetAttributeDefinitions() AttributeDefinitionMap {
	return entity.definition.GetAttributeDefinitions()
	}

	func (entity *ManagedEntity) DecodeAttributes(mask uint16, data []byte, p gopacket.PacketBuilder, msgType byte) (AttributeValueMap, error) {
	return entity.definition.DecodeAttributes(mask, data, p, msgType)
	}

	func (entity *ManagedEntity) SerializeAttributes(attr AttributeValueMap, mask uint16,
	b gopacket.SerializeBuffer, msgType byte, bytesAvailable int) error {
	return entity.definition.SerializeAttributes(attr, mask, b, msgType, bytesAvailable)
	}

	func (entity *ManagedEntity) GetEntityID() uint16 {
	if eid, err := entity.GetAttributeByIndex(0); err == nil {
	return eid.(uint16)
	}
	return 0
	}

	func (entity *ManagedEntity) SetEntityID(eid uint16) error {
	return entity.SetAttributeByIndex(0, eid)
	}

	func (entity *ManagedEntity) GetAttributeMask() uint16 {
	return entity.attributeMask
	}

	func (entity ManagedEntity) GetAttributeValueMap() AttributeValueMap {
	return &entity.attributes
	}

	func (entity *ManagedEntity) GetAttribute(name string) (interface{}, error) {
	value, ok := entity.attributes[name]
	if !ok {
	return 0, errors.New(fmt.Sprintf("attribute '%v' not found", name))
	}
	return value, nil
	}

	func (entity *ManagedEntity) GetAttributeByIndex(index uint) (interface{}, error) {
	if len(entity.attributes) == 0 {
	return nil, errors.New("attributes have already been set")
	}
	if _, ok := entity.definition.AttributeDefinitions[index]; !ok {
	return nil, errors.New(fmt.Sprintf("invalid attribute index: %d, should be 0..%d",
	index, len(entity.definition.AttributeDefinitions)-1))
	}
	return entity.GetAttribute(entity.definition.AttributeDefinitions[index].Name)
	}

	func (entity *ManagedEntity) setAttributes(params ...ParamData) OmciErrors {
	if entity.attributes == nil {
	entity.attributes = make(map[string]interface{})
	} else if len(entity.attributes) > 0 {
	return NewNonStatusError("attributes have already been set")
	}
	eidName := entity.definition.AttributeDefinitions[0].Name
	if len(params) == 0 {
	entity.attributes[eidName] = uint16(0)
	return nil
	}
	entity.attributes[eidName] = params[0].EntityID

	for name, value := range params[0].Attributes {
	if name == eidName {
	continue
	}
	if err := entity.SetAttribute(name, value); err != nil {
	return err
	}
	}
	return nil
	}

	func (entity *ManagedEntity) SetAttribute(name string, value interface{}) OmciErrors {
	attrDef, err := GetAttributeDefinitionByName(entity.definition.GetAttributeDefinitions(), name)
	if err != nil {
	return err
	} else if entity.attributes == nil {
	entity.attributes = make(map[string]interface{})
	}
	mask := uint16(1 << (16 - attrDef.GetIndex()))
	// check any constraints
	if constraintCheck := attrDef.GetConstraints(); constraintCheck != nil {
	err = constraintCheck(value)
	if err != nil {
	return NewParameterError(mask, err)
	}
	}
	entity.attributes[name] = value
	entity.attributeMask \|= mask
	return nil
	}

	func (entity *ManagedEntity) SetAttributeByIndex(index uint, value interface{}) error {
	attrDef, ok := entity.definition.AttributeDefinitions[index]
	if !ok {
	return errors.New(fmt.Sprintf("invalid attribute index: %d, should be 0..%d",
	index, len(entity.definition.AttributeDefinitions)-1))
	} else if entity.attributes == nil {
	entity.attributes = make(map[string]interface{})
	}
	mask := uint16(1 << (16 - attrDef.GetIndex()))
	// check any constraints
	if constraintCheck := attrDef.GetConstraints(); constraintCheck != nil {
	err := constraintCheck(value)
	if err != nil {
	return NewParameterError(mask, err)
	}
	}
	entity.attributes[attrDef.Name] = value
	entity.attributeMask \|= mask
	return nil
	}

	func (entity *ManagedEntity) DeleteAttribute(name string) error {
	attrDef, err := GetAttributeDefinitionByName(entity.definition.GetAttributeDefinitions(), name)
	if err != nil {
	return err
	}
	if entity.attributes != nil {
	delete(entity.attributes, name)
	entity.attributeMask &= ^uint16(1 << (16 - attrDef.GetIndex()))
	}
	return nil
	}

	func (entity *ManagedEntity) DeleteAttributeByIndex(index uint) error {
	attrDef, ok := entity.definition.AttributeDefinitions[index]
	if !ok {
	return errors.New(fmt.Sprintf("invalid attribute index: %d, should be 0..%d",
	index, len(entity.definition.AttributeDefinitions)-1))
	}
	if entity.attributes != nil {
	delete(entity.attributes, attrDef.Name)
	entity.attributeMask &= ^uint16(1 << (16 - attrDef.GetIndex()))
	}
	return nil
	}

	func (entity *ManagedEntity) DecodeFromBytes(data []byte, p gopacket.PacketBuilder, msgType byte) error {
	if len(data) < 6 {
	p.SetTruncated()
	return errors.New("frame too small")
	}
	classID := ClassID(binary.BigEndian.Uint16(data[0:2]))
	entityID := binary.BigEndian.Uint16(data[2:4])
	parameters := ParamData{EntityID: entityID}

	meDefinition, omciErr := LoadManagedEntityDefinition(classID, parameters)
	if omciErr != nil {
	return omciErr.GetError()
	}
	entity.definition = meDefinition.definition
	entity.attributeMask = binary.BigEndian.Uint16(data[4:6])
	entity.attributes = make(map[string]interface{})
	entity.SetEntityID(entityID)
	packetAttributes, err := entity.DecodeAttributes(entity.GetAttributeMask(), data[6:], p, msgType)
	if err != nil {
	return err
	}
	for name, value := range packetAttributes {
	entity.attributes[name] = value
	}
	return nil
	}

	func (entity *ManagedEntity) SerializeTo(b gopacket.SerializeBuffer, msgType byte, bytesAvailable int) error {
	// Add class ID and entity ID
	bytes, err := b.AppendBytes(6)
	if err != nil {
	return err
	}
	binary.BigEndian.PutUint16(bytes, uint16(entity.GetClassID()))
	binary.BigEndian.PutUint16(bytes[2:], entity.GetEntityID())
	binary.BigEndian.PutUint16(bytes[4:], entity.GetAttributeMask())

	// TODO: Need to limit number of bytes appended to not exceed packet size
	// Is there space/metadata info in 'b' parameter to allow this?
	err = entity.SerializeAttributes(entity.attributes, entity.GetAttributeMask(), b, msgType, bytesAvailable)
	return err
	}