gerrit.onap Code Review - ccsdk/features.git/blob

   1 /*
   2  * ============LICENSE_START=======================================================
   3  * ONAP : ccsdk features
   4  * ================================================================================
   5  * Copyright (C) 2023 highstreet technologies GmbH Intellectual Property.
   6  * All rights reserved.
   7  * ================================================================================
   8  * Licensed under the Apache License, Version 2.0 (the "License");
   9  * you may not use this file except in compliance with the License.
  10  * You may obtain a copy of the License at
  11  *
  12  *     http://www.apache.org/licenses/LICENSE-2.0
  13  *
  14  * Unless required by applicable law or agreed to in writing, software
  15  * distributed under the License is distributed on an "AS IS" BASIS,
  16  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  17  * See the License for the specific language governing permissions and
  18  * limitations under the License.
  19  * ============LICENSE_END=========================================================
  20  *
  21  */
  22 package org.onap.ccsdk.features.sdnr.wt.dataprovider.database.sqldb.data;
  23
  24 import java.util.ArrayList;
  25 import java.util.HashMap;
  26 import java.util.List;
  27 import java.util.Map;
  28 import java.util.stream.Collectors;
  29 import org.onap.ccsdk.features.sdnr.wt.common.database.queries.SortOrder;
  30 import org.onap.ccsdk.features.sdnr.wt.dataprovider.database.sqldb.SqlDBClient;
  31 import org.onap.ccsdk.features.sdnr.wt.dataprovider.database.sqldb.database.SqlDBReaderWriter;
  32 import org.onap.ccsdk.features.sdnr.wt.dataprovider.model.BaseInventoryTreeProvider;
  33 import org.onap.ccsdk.features.sdnr.wt.dataprovider.model.types.YangHelper2;
  34 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.data.provider.rev201110.Entity;
  35 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.data.provider.rev201110.EntityInput;
  36 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.data.provider.rev201110.Inventory;
  37 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.data.provider.rev201110.InventoryEntity;
  38 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.data.provider.rev201110.ReadInventoryListInputBuilder;
  39 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.data.provider.rev201110.entity.input.Filter;
  40 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.data.provider.rev201110.entity.input.FilterBuilder;
  41 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.data.provider.rev201110.entity.input.FilterKey;
  42 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.data.provider.rev201110.entity.input.Sortorder;
  43 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.data.provider.rev201110.entity.input.SortorderBuilder;
  44 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.data.provider.rev201110.entity.input.SortorderKey;
  45
  46 public class SqlDbInventoryTreeProvider extends BaseInventoryTreeProvider {
  47     private final SqlDBReaderWriter<Inventory> dbReader;
  48
  49     public SqlDbInventoryTreeProvider(SqlDBClient dbClient, String controllerId) {
  50         this.dbReader = new SqlDBReaderWriter<>(dbClient, Entity.Inventoryequipment, "", Inventory.class, controllerId);
  51     }
  52
  53     @Override
  54     protected List<String> getAllNodeIds() {
  55         return this.dbReader.readAll(INVENTORY_PROPERTY_NODEID);
  56     }
  57
  58     @Override
  59     protected List<InventoryEntity> search(String filter, String sortOrderProperty, SortOrder sortOrder) {
  60         return new ArrayList<>(this.dbReader.searchAll(Inventory.class, null, filter));
  61     }
  62
  63     @Override
  64     protected List<InventoryEntity> search(String filter, String nodeId, String parentUuid, String uuid,
  65             String sortOrderProperty, SortOrder sortOrder) {
  66         List<Filter> filters = new ArrayList<>();
  67         if (nodeId != null) {
  68             filters.add(new FilterBuilder().setProperty(INVENTORY_PROPERTY_NODEID).setFiltervalue(nodeId).build());
  69         }
  70         EntityInput input = new ReadInventoryListInputBuilder()
  71                 .setFilter(YangHelper2.getListOrMap(FilterKey.class, filters)).build();
  72         return new ArrayList<>(this.dbReader.searchAll(Inventory.class, input, filter));
  73     }
  74
  75     @Override
  76     protected List<InventoryEntity> getItemsForNodes(List<String> nodeIds, String sortOrderProperty,
  77             SortOrder sortOrder) {
  78         Map<FilterKey, Filter> nodeFilter = new HashMap<>();
  79         Filter filter = new FilterBuilder().setProperty(INVENTORY_PROPERTY_NODEID)
  80                 .setFiltervalues(nodeIds.stream().collect(Collectors.toSet())).build();
  81         nodeFilter.put(filter.key(), filter);
  82         Map<SortorderKey, Sortorder> so = new HashMap<>();
  83         Sortorder soItem = new SortorderBuilder().setProperty(sortOrderProperty)
  84                 .setSortorder(YangHelper2.getSortOrder(sortOrder)).build();
  85         so.put(soItem.key(), soItem);
  86         EntityInput input = new ReadInventoryListInputBuilder().setFilter(nodeFilter).setSortorder(so).build();
  87         return new ArrayList<>(this.dbReader.readAll(Inventory.class, input));
  88     }
  89 }