VOL-1376 Resource manager for openolt adapter
Change-Id: I1fe5f0d06d3ea95044d9ba0757b2bb3740864161
diff --git a/adaptercore/resourcemanager/resourcemanager.go b/adaptercore/resourcemanager/resourcemanager.go
new file mode 100644
index 0000000..3045578
--- /dev/null
+++ b/adaptercore/resourcemanager/resourcemanager.go
@@ -0,0 +1,642 @@
+/*
+ * Copyright 2019-present Open Networking Foundation
+
+ * 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.
+ */
+
+package resourcemanager
+
+import (
+ "encoding/json"
+ "errors"
+ "fmt"
+ "strconv"
+ "strings"
+
+ ponrmgr "github.com/opencord/voltha-go/adapters/common/ponresourcemanager"
+ "github.com/opencord/voltha-go/common/log"
+ "github.com/opencord/voltha-go/db/kvstore"
+ "github.com/opencord/voltha-go/db/model"
+ "github.com/opencord/voltha-go/protos/openolt"
+)
+
+CONST KVSTORE_TIMEOUT = 5
+const BASE_PATH_KV_STORE = "service/voltha/openolt/{%s}" // service/voltha/openolt/<device_id>
+
+type OpenOltResourceMgr struct {
+ DeviceID string //OLT device id
+ HostAndPort string // Host and port of the kv store to connect to
+ Args string // args
+ KVStore *model.Backend // backend kv store connection handle
+ DeviceType string
+ Host string // Host ip of the kv store
+ Port int // port of the kv store
+ DevInfo *openolt.DeviceInfo // device information
+ // array of pon resource managers per interface technology
+ ResourceMgrs map[uint32]*ponrmgr.PONResourceManager
+}
+
+func newKVClient(storeType string, address string, timeout uint32) (kvstore.Client, error) {
+ log.Infow("kv-store-type", log.Fields{"store": storeType})
+ switch storeType {
+ case "consul":
+ return kvstore.NewConsulClient(address, int(timeout))
+ case "etcd":
+ return kvstore.NewEtcdClient(address, int(timeout))
+ }
+ return nil, errors.New("unsupported-kv-store")
+}
+
+func SetKVClient(Backend string, Host string, Port int, DeviceID string) *model.Backend {
+ addr := Host + ":" + strconv.Itoa(Port)
+ // TODO : Make sure direct call to NewBackend is working fine with backend , currently there is some
+ // issue between kv store and backend , core is not calling NewBackend directly
+ kvClient, err := newKVClient(Backend, addr, KVSTORE_TIMEOUT)
+ if err != nil {
+ log.Fatalw("Failed to init KV client\n", log.Fields{"err": err})
+ return nil
+ }
+ kvbackend := &model.Backend{
+ Client: kvClient,
+ StoreType: Backend,
+ Host: Host,
+ Port: Port,
+ Timeout: KVSTORE_TIMEOUT,
+ PathPrefix: fmt.Sprintf(BASE_PATH_KV_STORE, DeviceID)}
+
+ return kvbackend
+}
+
+func NewResourceMgr(DeviceID string, HostPort string, DeviceType string, DevInfo *openolt.DeviceInfo) *OpenOltResourceMgr {
+
+ /* init a New resource maanger instance which in turn instantiates pon resource manager
+ instances according to technology. Initializes the default resource ranges for all
+ the reources.
+ */
+ var ResourceMgr OpenOltResourceMgr
+ log.Debugf("Init new resource manager , host_port: %s, deviceid: %s", HostPort, DeviceID)
+ ResourceMgr.HostAndPort = HostPort
+ ResourceMgr.DeviceType = DeviceType
+ ResourceMgr.DevInfo = DevInfo
+ IpPort := strings.Split(HostPort, ":")
+ ResourceMgr.Host = IpPort[0]
+ ResourceMgr.Port, _ = strconv.Atoi(IpPort[1])
+
+ Backend := "etcd" /* TODO remove this once we get backend store from registry*/
+ ResourceMgr.KVStore = SetKVClient(Backend, ResourceMgr.Host,
+ ResourceMgr.Port, DeviceID)
+ if ResourceMgr.KVStore == nil {
+ log.Error("Failed to setup KV store")
+ }
+ Ranges := make(map[string]*openolt.DeviceInfo_DeviceResourceRanges)
+ RsrcMgrsByTech := make(map[string]*ponrmgr.PONResourceManager)
+ ResourceMgr.ResourceMgrs = make(map[uint32]*ponrmgr.PONResourceManager)
+
+ // TODO self.args = registry('main').get_args()
+
+ /*
+ If a legacy driver returns protobuf without any ranges,s synthesize one from
+ the legacy global per-device informaiton. This, in theory, is temporary until
+ the legacy drivers are upgrade to support pool ranges.
+ */
+ if DevInfo.Ranges == nil {
+ var ranges openolt.DeviceInfo_DeviceResourceRanges
+ ranges.Technology = DevInfo.GetTechnology()
+
+ NumPONPorts := DevInfo.GetPonPorts()
+ var index uint32
+ for index = 0; index < NumPONPorts; index++ {
+ ranges.IntfIds = append(ranges.IntfIds, index)
+ }
+
+ var Pool *openolt.DeviceInfo_DeviceResourceRanges_Pool
+ Pool.Type = openolt.DeviceInfo_DeviceResourceRanges_Pool_ONU_ID
+ Pool.Start = DevInfo.OnuIdStart
+ Pool.End = DevInfo.OnuIdEnd
+ Pool.Sharing = openolt.DeviceInfo_DeviceResourceRanges_Pool_DEDICATED_PER_INTF
+ ranges.Pools = append(ranges.Pools, Pool)
+
+ Pool.Type = openolt.DeviceInfo_DeviceResourceRanges_Pool_ALLOC_ID
+ Pool.Start = DevInfo.AllocIdStart
+ Pool.End = DevInfo.AllocIdEnd
+ Pool.Sharing = openolt.DeviceInfo_DeviceResourceRanges_Pool_SHARED_BY_ALL_INTF_ALL_TECH
+ ranges.Pools = append(ranges.Pools, Pool)
+
+ Pool.Type = openolt.DeviceInfo_DeviceResourceRanges_Pool_GEMPORT_ID
+ Pool.Start = DevInfo.GemportIdStart
+ Pool.End = DevInfo.GemportIdEnd
+ Pool.Sharing = openolt.DeviceInfo_DeviceResourceRanges_Pool_SHARED_BY_ALL_INTF_ALL_TECH
+ ranges.Pools = append(ranges.Pools, Pool)
+
+ Pool.Type = openolt.DeviceInfo_DeviceResourceRanges_Pool_FLOW_ID
+ Pool.Start = DevInfo.FlowIdStart
+ Pool.End = DevInfo.FlowIdEnd
+ Pool.Sharing = openolt.DeviceInfo_DeviceResourceRanges_Pool_SHARED_BY_ALL_INTF_ALL_TECH
+ ranges.Pools = append(ranges.Pools, Pool)
+ }
+
+ //Create a separate Resource Manager instance for each range. This assumes that
+ // each technology is represented by only a single range
+ var GlobalPONRsrcMgr *ponrmgr.PONResourceManager
+ var err error
+ for _, TechRange := range DevInfo.Ranges {
+ technology := TechRange.Technology
+ log.Debugf("Device info technology %s", technology)
+ Ranges[technology] = TechRange
+ RsrcMgrsByTech[technology], err = ponrmgr.NewPONResourceManager(technology, DeviceType, DeviceID,
+ Backend, ResourceMgr.Host, ResourceMgr.Port)
+ if err != nil {
+ log.Errorf("Failed to create pon resource manager instacnce for technology %s", technology)
+ return nil
+ }
+ //resource_mgrs_by_tech[technology] = resource_mgr
+ if GlobalPONRsrcMgr == nil {
+ GlobalPONRsrcMgr = RsrcMgrsByTech[technology]
+ }
+ for IntfId := range TechRange.IntfIds {
+ ResourceMgr.ResourceMgrs[uint32(IntfId)] = RsrcMgrsByTech[technology]
+ }
+ //self.initialize_device_resource_range_and_pool(resource_mgr, global_resource_mgr, arange)
+ InitializeDeviceResourceRangeAndPool(RsrcMgrsByTech[technology], GlobalPONRsrcMgr,
+ TechRange, DevInfo)
+ }
+ // After we have initialized resource ranges, initialize the
+ // resource pools accordingly.
+ for _, PONRMgr := range RsrcMgrsByTech {
+ PONRMgr.InitDeviceResourcePool()
+ }
+ return &ResourceMgr
+}
+
+func InitializeDeviceResourceRangeAndPool(PONRMgr *ponrmgr.PONResourceManager, GlobalPONRMgr *ponrmgr.PONResourceManager,
+ TechRange *openolt.DeviceInfo_DeviceResourceRanges, DevInfo *openolt.DeviceInfo) {
+
+ // init the resource range pool according to the sharing type
+
+ log.Debugf("Resource range pool init for technology %s", PONRMgr.Technology)
+ //first load from KV profiles
+ status := PONRMgr.InitResourceRangesFromKVStore()
+ if status == false {
+ log.Debugf("Failed to load resource ranges from KV store for tech %s", PONRMgr.Technology)
+ }
+
+ /*
+ Then apply device specific information. If KV doesn't exist
+ or is broader than the device, the device's informationw ill
+ dictate the range limits
+ */
+ log.Debugf("Using device info to init pon resource ranges for tech", PONRMgr.Technology)
+
+ ONUIDStart := DevInfo.OnuIdStart
+ ONUIDEnd := DevInfo.OnuIdEnd
+ ONUIDShared := openolt.DeviceInfo_DeviceResourceRanges_Pool_DEDICATED_PER_INTF
+ ONUIDSharedPoolID := uint32(0)
+ AllocIDStart := DevInfo.AllocIdStart
+ AllocIDEnd := DevInfo.AllocIdEnd
+ AllocIDShared := openolt.DeviceInfo_DeviceResourceRanges_Pool_SHARED_BY_ALL_INTF_ALL_TECH // TODO EdgeCore/BAL limitation
+ AllocIDSharedPoolID := uint32(0)
+ GEMPortIDStart := DevInfo.GemportIdStart
+ GEMPortIDEnd := DevInfo.GemportIdEnd
+ GEMPortIDShared := openolt.DeviceInfo_DeviceResourceRanges_Pool_SHARED_BY_ALL_INTF_ALL_TECH // TODO EdgeCore/BAL limitation
+ GEMPortIDSharedPoolID := uint32(0)
+ FlowIDStart := DevInfo.FlowIdStart
+ FlowIDEnd := DevInfo.FlowIdEnd
+ FlowIDShared := openolt.DeviceInfo_DeviceResourceRanges_Pool_SHARED_BY_ALL_INTF_ALL_TECH // TODO EdgeCore/BAL limitation
+ FlowIDSharedPoolID := uint32(0)
+
+ var GlobalPoolID uint32
+ var FirstIntfPoolID uint32
+ var SharedPoolID uint32
+
+ for _, FirstIntfPoolID = range TechRange.IntfIds {
+ break
+ }
+
+ for _, RangePool := range TechRange.Pools {
+ if RangePool.Sharing == openolt.DeviceInfo_DeviceResourceRanges_Pool_SHARED_BY_ALL_INTF_ALL_TECH {
+ SharedPoolID = GlobalPoolID
+ } else if RangePool.Sharing == openolt.DeviceInfo_DeviceResourceRanges_Pool_SHARED_BY_ALL_INTF_SAME_TECH {
+ SharedPoolID = FirstIntfPoolID
+ } else {
+ SharedPoolID = 0
+ }
+ if RangePool.Type == openolt.DeviceInfo_DeviceResourceRanges_Pool_ONU_ID {
+ ONUIDStart = RangePool.Start
+ ONUIDEnd = RangePool.End
+ ONUIDShared = RangePool.Sharing
+ ONUIDSharedPoolID = SharedPoolID
+ } else if RangePool.Type == openolt.DeviceInfo_DeviceResourceRanges_Pool_ALLOC_ID {
+ AllocIDStart = RangePool.Start
+ AllocIDEnd = RangePool.End
+ AllocIDShared = RangePool.Sharing
+ AllocIDSharedPoolID = SharedPoolID
+ } else if RangePool.Type == openolt.DeviceInfo_DeviceResourceRanges_Pool_GEMPORT_ID {
+ GEMPortIDStart = RangePool.Start
+ GEMPortIDEnd = RangePool.End
+ GEMPortIDShared = RangePool.Sharing
+ GEMPortIDSharedPoolID = SharedPoolID
+ } else if RangePool.Type == openolt.DeviceInfo_DeviceResourceRanges_Pool_FLOW_ID {
+ FlowIDStart = RangePool.Start
+ FlowIDEnd = RangePool.End
+ FlowIDShared = RangePool.Sharing
+ FlowIDSharedPoolID = SharedPoolID
+ }
+ }
+
+ log.Debugw("Device info init", log.Fields{"technology": TechRange.Technology,
+ "onu_id_start": ONUIDStart, "onu_id_end": ONUIDEnd, "onu_id_shared_pool_id": ONUIDSharedPoolID,
+ "alloc_id_start": AllocIDStart, "alloc_id_end": AllocIDEnd,
+ "alloc_id_shared_pool_id": AllocIDSharedPoolID,
+ "gemport_id_start": GEMPortIDStart, "gemport_id_end": GEMPortIDEnd,
+ "gemport_id_shared_pool_id": GEMPortIDSharedPoolID,
+ "flow_id_start": FlowIDStart,
+ "flow_id_end_idx": FlowIDEnd,
+ "flow_id_shared_pool_id": FlowIDSharedPoolID,
+ "intf_ids": TechRange.IntfIds,
+ "uni_id_start": 0,
+ "uni_id_end_idx":/*MaxUNIIDperONU()*/ 1})
+
+ PONRMgr.InitDefaultPONResourceRanges(ONUIDStart, ONUIDEnd, ONUIDSharedPoolID,
+ AllocIDStart, AllocIDEnd, AllocIDSharedPoolID,
+ GEMPortIDStart, GEMPortIDEnd, GEMPortIDSharedPoolID,
+ FlowIDStart, FlowIDEnd, FlowIDSharedPoolID, 0, 1,
+ DevInfo.PonPorts, TechRange.IntfIds)
+
+ // For global sharing, make sure to refresh both local and global resource manager instances' range
+
+ if ONUIDShared == openolt.DeviceInfo_DeviceResourceRanges_Pool_SHARED_BY_ALL_INTF_ALL_TECH {
+ GlobalPONRMgr.UpdateRanges(ponrmgr.ONU_ID_START_IDX, ONUIDStart, ponrmgr.ONU_ID_END_IDX, ONUIDEnd,
+ "", 0, nil)
+ PONRMgr.UpdateRanges(ponrmgr.ONU_ID_START_IDX, ONUIDStart, ponrmgr.ONU_ID_END_IDX, ONUIDEnd,
+ "", 0, GlobalPONRMgr)
+ }
+ if AllocIDShared == openolt.DeviceInfo_DeviceResourceRanges_Pool_SHARED_BY_ALL_INTF_ALL_TECH {
+ GlobalPONRMgr.UpdateRanges(ponrmgr.ALLOC_ID_START_IDX, AllocIDStart, ponrmgr.ALLOC_ID_END_IDX, AllocIDEnd,
+ "", 0, nil)
+
+ PONRMgr.UpdateRanges(ponrmgr.ALLOC_ID_START_IDX, AllocIDStart, ponrmgr.ALLOC_ID_END_IDX, AllocIDEnd,
+ "", 0, GlobalPONRMgr)
+ }
+ if GEMPortIDShared == openolt.DeviceInfo_DeviceResourceRanges_Pool_SHARED_BY_ALL_INTF_ALL_TECH {
+ GlobalPONRMgr.UpdateRanges(ponrmgr.GEMPORT_ID_START_IDX, GEMPortIDStart, ponrmgr.GEMPORT_ID_END_IDX, GEMPortIDEnd,
+ "", 0, nil)
+ PONRMgr.UpdateRanges(ponrmgr.GEMPORT_ID_START_IDX, GEMPortIDStart, ponrmgr.GEMPORT_ID_END_IDX, GEMPortIDEnd,
+ "", 0, GlobalPONRMgr)
+ }
+ if FlowIDShared == openolt.DeviceInfo_DeviceResourceRanges_Pool_SHARED_BY_ALL_INTF_ALL_TECH {
+ GlobalPONRMgr.UpdateRanges(ponrmgr.FLOW_ID_START_IDX, FlowIDStart, ponrmgr.FLOW_ID_END_IDX, FlowIDEnd,
+ "", 0, nil)
+ PONRMgr.UpdateRanges(ponrmgr.FLOW_ID_START_IDX, FlowIDStart, ponrmgr.FLOW_ID_END_IDX, FlowIDEnd,
+ "", 0, GlobalPONRMgr)
+ }
+
+ // Make sure loaded range fits the platform bit encoding ranges
+ PONRMgr.UpdateRanges(ponrmgr.UNI_ID_START_IDX, 0, ponrmgr.UNI_ID_END_IDX /* TODO =OpenOltPlatform.MAX_UNIS_PER_ONU-1*/, 1, "", 0, nil)
+}
+
+/* TODO
+def __del__(self):
+ self.log.info("clearing-device-resource-pool")
+ for key, resource_mgr in self.resource_mgrs.iteritems():
+ resource_mgr.clear_device_resource_pool()
+
+ def assert_pon_id_limit(self, pon_intf_id):
+ assert pon_intf_id in self.resource_mgrs
+
+ def assert_onu_id_limit(self, pon_intf_id, onu_id):
+ self.assert_pon_id_limit(pon_intf_id)
+ self.resource_mgrs[pon_intf_id].assert_resource_limits(onu_id, PONResourceManager.ONU_ID)
+
+ @property
+ def max_uni_id_per_onu(self):
+ return 0 #OpenOltPlatform.MAX_UNIS_PER_ONU-1, zero-based indexing Uncomment or override to make default multi-uni
+
+ def assert_uni_id_limit(self, pon_intf_id, onu_id, uni_id):
+ self.assert_onu_id_limit(pon_intf_id, onu_id)
+ self.resource_mgrs[pon_intf_id].assert_resource_limits(uni_id, PONResourceManager.UNI_ID)
+*/
+
+func (RsrcMgr *OpenOltResourceMgr) GetONUID(PONIntfID uint32) (uint32, error) {
+
+ // Get ONU id for a provided pon interface ID.
+
+ ONUID, err := RsrcMgr.ResourceMgrs[PONIntfID].GetResourceID(PONIntfID,
+ ponrmgr.ONU_ID, 1)
+ if err != nil {
+ log.Errorf("Failed to get resource for interface %d for type %s",
+ PONIntfID, ponrmgr.ONU_ID)
+ return ONUID[0], err
+ }
+ if ONUID != nil {
+ RsrcMgr.ResourceMgrs[PONIntfID].InitResourceMap(fmt.Sprintf("%d,%d", PONIntfID, ONUID))
+ }
+
+ return ONUID[0], err
+}
+
+func (RsrcMgr *OpenOltResourceMgr) GetFlowID(PONIntfID uint32, ONUID uint32, UNIID uint32,
+ FlowStoreCookie interface{},
+ FlowCategory interface{}) (uint32, error) {
+
+ // Get flow ID for a given pon interface id, onu id and uni id.
+
+ var err error
+ FlowPath := fmt.Sprintf("%d,%d,%d", PONIntfID, ONUID, UNIID)
+ FlowIDs := RsrcMgr.ResourceMgrs[PONIntfID].GetCurrentFlowIDsForOnu(FlowPath)
+ if FlowIDs != nil {
+ /* TODO once the flow id info structure is known
+ for Flow := range FlowIDs {
+ FlowInfo := RsrcMgr.ResourceMgrs[PONIntfID].GetFlowIDInfo(FlowPath, Flow)
+ for Info := range FlowInfo {
+ if FlowCategory != nil &&
+ Info[FlowCategory] == FlowCategory {
+ return 0, Flow
+ }
+ if FlowStoreCookie != nil &&
+ Info[FlowStoreCookie] == FlowStoreCookie {
+ return 0, Flow
+ }
+ }
+ }
+ */
+ }
+ FlowIDs, err = RsrcMgr.ResourceMgrs[PONIntfID].GetResourceID(PONIntfID,
+ ponrmgr.FLOW_ID, 1)
+ if err != nil {
+ log.Errorf("Failed to get resource for interface %d for type %s",
+ PONIntfID, ponrmgr.FLOW_ID)
+ return FlowIDs[0], err
+ }
+ if FlowIDs != nil {
+ RsrcMgr.ResourceMgrs[PONIntfID].UpdateFlowIDForOnu(FlowPath, FlowIDs[0], true)
+ }
+
+ return FlowIDs[0], err
+}
+
+func (RsrcMgr *OpenOltResourceMgr) GetAllocID(IntfID uint32, ONUID uint32) uint32 {
+
+ // Get alloc id for a given pon interface id and onu id.
+ var err error
+ IntfONUID := fmt.Sprintf("%d,%d", IntfID, ONUID)
+ AllocID := RsrcMgr.ResourceMgrs[IntfID].GetCurrentAllocIDForOnu(IntfONUID)
+ if AllocID != nil {
+ // Since we support only one alloc_id for the ONU at the moment,
+ // return the first alloc_id in the list, if available, for that
+ // ONU.
+ return AllocID[0]
+ }
+ AllocID, err = RsrcMgr.ResourceMgrs[IntfID].GetResourceID(IntfID,
+ ponrmgr.ALLOC_ID, 1)
+
+ if AllocID == nil || err != nil {
+ log.Error("Failed to allocate alloc id")
+ return 0
+ }
+ // update the resource map on KV store with the list of alloc_id
+ // allocated for the pon_intf_onu_id tuple
+ err = RsrcMgr.ResourceMgrs[IntfID].UpdateAllocIdsForOnu(IntfONUID, AllocID)
+ if err != nil {
+ log.Error("Failed to update Alloc ID")
+ return 0
+ }
+ return AllocID[0]
+}
+
+func (RsrcMgr *OpenOltResourceMgr) UpdateAllocIdsForOnu(PONPort uint32, ONUID uint32,
+ UNIID uint32, AllocID []uint32) error {
+
+ /* update alloc ids in kv store for a given pon interface id,
+ onu id and uni id.
+ */
+ IntfONUID := fmt.Sprintf("%d,%d,%d", PONPort, ONUID, UNIID)
+ return RsrcMgr.ResourceMgrs[PONPort].UpdateAllocIdsForOnu(IntfONUID,
+ AllocID)
+}
+func (RsrcMgr *OpenOltResourceMgr) GetCurrentGEMPortIDsForOnu(IntfID uint32, ONUID uint32,
+ UNIID uint32) []uint32 {
+
+ /* Get gem ports for given pon interface , onu id and uni id. */
+
+ IntfONUID := fmt.Sprintf("%d,%d,%d", IntfID, ONUID, UNIID)
+ GEMPortID := RsrcMgr.ResourceMgrs[IntfID].GetCurrentGEMPortIDsForOnu(IntfONUID)
+ if GEMPortID != nil {
+ // Since we support only one alloc_id for the ONU at the moment,
+ // return the first alloc_id in the list, if available, for that
+ // ONU.
+ return GEMPortID
+ }
+ return nil
+}
+
+func (RsrcMgr *OpenOltResourceMgr) GetCurrentAllocIDForOnu(IntfID uint32, ONUID uint32) uint32 {
+
+ /* Get alloc ids for given pon interface and onu id. */
+
+ IntfONUID := fmt.Sprintf("%d,%d", IntfID, ONUID)
+ AllocID := RsrcMgr.ResourceMgrs[IntfID].GetCurrentAllocIDForOnu(IntfONUID)
+ if AllocID != nil {
+ // Since we support only one alloc_id for the ONU at the moment,
+ // return the first alloc_id in the list, if available, for that
+ // ONU.
+ return AllocID[0]
+ }
+ return 0
+}
+
+func (RsrcMgr *OpenOltResourceMgr) UpdateGEMportsPonportToOnuMapOnKVStore(GEMPorts []uint32, PonPort uint32,
+ ONUID uint32, UNIID uint32) error {
+
+ /* Update onu and uni id associated with the gem port to the kv store. */
+ var IntfGEMPortPath string
+ Data := fmt.Sprintf("%d %d", ONUID, UNIID)
+ for _, GEM := range GEMPorts {
+ IntfGEMPortPath = fmt.Sprintf("%d,%d", PonPort, GEM)
+ Val, err := json.Marshal(Data)
+ if err != nil {
+ log.Error("failed to Marshal")
+ return err
+ }
+ // This information is used when packet_indication is received and
+ // we need to derive the ONU Id for which the packet arrived based
+ // on the pon_intf and gemport available in the packet_indication
+ if err = RsrcMgr.KVStore.Put(IntfGEMPortPath, Val); err != nil {
+ log.Errorf("Failed to update resource %s", IntfGEMPortPath)
+ return err
+ }
+ }
+ return nil
+}
+
+func (RsrcMgr *OpenOltResourceMgr) GetONUUNIfromPONPortGEMPort(PONPort uint32, GEMPort uint32) []uint32 {
+
+ /* get the onu and uni id for a given gem port. */
+ IntfGEMPortPath := fmt.Sprintf("%d,%d", PONPort, GEMPort)
+ var GEMPortIDs []uint32
+ var Data string
+ Value, err := RsrcMgr.KVStore.Get(IntfGEMPortPath)
+ if err == nil {
+ if Value != nil {
+ Val, _ := kvstore.ToByte(Value.Value)
+ if err = json.Unmarshal(Val, &Data); err != nil {
+ log.Error("Failed to unmarshal")
+ return nil
+ }
+ IDs := strings.Split(Data, " ")
+ for _, port := range IDs {
+ Intport, _ := strconv.Atoi(port)
+ GEMPortIDs = append(GEMPortIDs, uint32(Intport))
+ }
+ }
+ }
+ return GEMPortIDs
+}
+
+func (RsrcMgr *OpenOltResourceMgr) GetGEMPortID(PONPort uint32, ONUID uint32,
+ UNIID uint32, NumOfPorts uint32) ([]uint32, error) {
+
+ /* Get gem port id for a particular pon port, onu id
+ and uni id.
+ */
+
+ var err error
+ IntfONUID := fmt.Sprintf("%d,%d,%d", PONPort, ONUID, UNIID)
+
+ GEMPortList := RsrcMgr.ResourceMgrs[PONPort].GetCurrentGEMPortIDsForOnu(IntfONUID)
+ if GEMPortList != nil {
+ return GEMPortList, nil
+ }
+
+ GEMPortList, err = RsrcMgr.ResourceMgrs[PONPort].GetResourceID(PONPort,
+ ponrmgr.GEMPORT_ID, NumOfPorts)
+ if err != nil && GEMPortList == nil {
+ log.Errorf("Failed to get gem port id for %s", IntfONUID)
+ return nil, err
+ }
+
+ // update the resource map on KV store with the list of gemport_id
+ // allocated for the pon_intf_onu_id tuple
+ err = RsrcMgr.ResourceMgrs[PONPort].UpdateGEMPortIDsForOnu(IntfONUID,
+ GEMPortList)
+ if err != nil {
+ log.Errorf("Failed to update GEM ports to kv store for %s", IntfONUID)
+ return nil, err
+ }
+ RsrcMgr.UpdateGEMportsPonportToOnuMapOnKVStore(GEMPortList, PONPort,
+ ONUID, UNIID)
+ return GEMPortList, err
+}
+
+func (RsrcMgr *OpenOltResourceMgr) UpdateGEMPortIDsForOnu(PONPort uint32, ONUID uint32,
+ UNIID uint32, GEMPortList []uint32) error {
+
+ /* Update gemport ids on to kv store for a given pon port,
+ onu id and uni id.
+ */
+ IntfONUID := fmt.Sprintf("%d,%d,%d", PONPort, ONUID, UNIID)
+ return RsrcMgr.ResourceMgrs[PONPort].UpdateGEMPortIDsForOnu(IntfONUID,
+ GEMPortList)
+
+}
+func (RsrcMgr *OpenOltResourceMgr) FreeONUID(IntfID uint32, ONUID []uint32) {
+
+ /* Free onu id for a particular interface.*/
+ RsrcMgr.ResourceMgrs[IntfID].FreeResourceID(IntfID, ponrmgr.ONU_ID, ONUID)
+
+ var IntfONUID string
+ for _, onu := range ONUID {
+ IntfONUID = fmt.Sprintf("%d,%d", IntfID, onu)
+ RsrcMgr.ResourceMgrs[IntfID].RemoveResourceMap(IntfONUID)
+ }
+ return
+}
+
+func (RsrcMgr *OpenOltResourceMgr) FreeFlowID(IntfID uint32, ONUID uint32,
+ UNIID uint32, FlowID []uint32) {
+
+ /* Free flow id for a given interface, onu id and uni id.*/
+
+ RsrcMgr.ResourceMgrs[IntfID].FreeResourceID(IntfID, ponrmgr.FLOW_ID, FlowID)
+
+ var IntfONUID string
+ var err error
+ for _, flow := range FlowID {
+ IntfONUID = fmt.Sprintf("%d,%d,%d", IntfID, ONUID, UNIID)
+ err = RsrcMgr.ResourceMgrs[IntfID].UpdateFlowIDForOnu(IntfONUID, flow, false)
+ if err != nil {
+ log.Error("Failed to Update flow id infor for %s", IntfONUID)
+ }
+ RsrcMgr.ResourceMgrs[IntfID].RemoveFlowIDInfo(IntfONUID, flow)
+ }
+ return
+}
+
+func (RsrcMgr *OpenOltResourceMgr) FreePONResourcesForONU(IntfID uint32, ONUID uint32) {
+
+ /* Free pon resources for a given pon interface and onu id. */
+
+ var ONUIDs []uint32
+ ONUIDs = append(ONUIDs, ONUID)
+ IntfONUID := fmt.Sprintf("%d,%d", IntfID, ONUID)
+
+ AllocIDs := RsrcMgr.ResourceMgrs[IntfID].GetCurrentAllocIDForOnu(IntfONUID)
+
+ RsrcMgr.ResourceMgrs[IntfID].FreeResourceID(IntfID,
+ ponrmgr.ALLOC_ID,
+ AllocIDs)
+
+ GEMPortIDs := RsrcMgr.ResourceMgrs[IntfID].GetCurrentGEMPortIDsForOnu(IntfONUID)
+ RsrcMgr.ResourceMgrs[IntfID].FreeResourceID(IntfID,
+ ponrmgr.GEMPORT_ID,
+ GEMPortIDs)
+
+ FlowIDs := RsrcMgr.ResourceMgrs[IntfID].GetCurrentFlowIDsForOnu(IntfONUID)
+ RsrcMgr.ResourceMgrs[IntfID].FreeResourceID(IntfID,
+ ponrmgr.FLOW_ID,
+ FlowIDs)
+ RsrcMgr.ResourceMgrs[IntfID].FreeResourceID(IntfID,
+ ponrmgr.ONU_ID,
+ ONUIDs)
+
+ // Clear resource map associated with (pon_intf_id, gemport_id) tuple.
+ RsrcMgr.ResourceMgrs[IntfID].RemoveResourceMap(IntfONUID)
+
+ // Clear the ONU Id associated with the (pon_intf_id, gemport_id) tuple.
+ for _, GEM := range GEMPortIDs {
+ RsrcMgr.KVStore.Delete(fmt.Sprintf("%d,%d", IntfID, GEM))
+ }
+}
+
+/* TODO once the flow id info structure is known
+def is_flow_cookie_on_kv_store(self, intf_id, onu_id, uni_id, flow_store_cookie):
+ '''
+ Note: For flows which trap from the NNI and not really associated with any particular
+ ONU (like LLDP), the onu_id and uni_id is set as -1. The intf_id is the NNI intf_id.
+ '''
+ intf_onu_id = (intf_id, onu_id, uni_id)
+ try:
+ flow_ids = self.resource_mgrs[intf_id]. \
+ get_current_flow_ids_for_onu(intf_onu_id)
+ if flow_ids is not None:
+ for flow_id in flow_ids:
+ flows = self.get_flow_id_info(intf_id, onu_id, uni_id, flow_id)
+ assert (isinstance(flows, list))
+ for flow in flows:
+ if flow['flow_store_cookie'] == flow_store_cookie:
+ return True
+ except Exception as e:
+ self.log.error("error-retrieving-flow-info", e=e)
+
+ return False
+*/
+