62233340307d52f0d66937503f50225d881d6382
[ccsdk/features.git] /
1 /*
2  * ============LICENSE_START=======================================================
3  * ONAP : ccsdk features
4  * ================================================================================
5  * Copyright (C) 2019 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.elasticsearch.data.acessor;
23
24 import java.io.IOException;
25 import org.onap.ccsdk.features.sdnr.wt.common.database.ExtRestClient;
26 import org.onap.ccsdk.features.sdnr.wt.common.database.HtDatabaseClient;
27 import org.onap.ccsdk.features.sdnr.wt.common.database.SearchResult;
28 import org.onap.ccsdk.features.sdnr.wt.common.database.requests.SearchRequest;
29 import org.onap.ccsdk.features.sdnr.wt.common.database.responses.AggregationEntries;
30 import org.onap.ccsdk.features.sdnr.wt.common.database.responses.SearchResponse;
31 import org.onap.ccsdk.features.sdnr.wt.dataprovider.database.elasticsearch.data.rpctypehelper.QueryByFilter;
32 import org.onap.ccsdk.features.sdnr.wt.dataprovider.database.elasticsearch.data.rpctypehelper.QueryResult;
33 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.data.provider.rev201110.Entity;
34 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.data.provider.rev201110.EntityInput;
35 import org.opendaylight.yangtools.yang.binding.DataObject;
36 import org.slf4j.Logger;
37 import org.slf4j.LoggerFactory;
38
39 public class DataObjectAcessorInventory<T extends DataObject> extends DataObjectAcessorWithId<T> {
40
41     private final Logger LOG = LoggerFactory.getLogger(DataObjectAcessorInventory.class);
42
43     private static final String KEY = "node-id";
44
45     private ExtRestClient dbClient;
46
47     public DataObjectAcessorInventory(HtDatabaseClient dbClient, Entity entity, Class<T> clazz,
48                                       boolean doFullsizeRequest) throws ClassNotFoundException {
49         super(dbClient, entity, clazz, doFullsizeRequest);
50         LOG.info("Create DataObjectAcessorInventory");
51         this.dbClient = dbClient;
52     }
53
54
55     /**
56      * get aggregated devices list
57      *
58      * @param input filter should be empty/no filter handled, only sortorder for KEY ('node-name')
59      * @return
60      * @throws IOException
61      */
62     public QueryResult<String> getDataDeviceList(EntityInput input) {
63
64         QueryByFilter queryByFilter = new QueryByFilter(input);
65         SearchRequest request =
66                 queryByFilter.getSearchRequestAggregated(KEY,this.getDataTypeName(), this.getDataTypeName(), this.doFullsizeRequest);
67         try {
68             SearchResponse response = this.dbClient.search(request);
69             AggregationEntries aggs = response.getAggregations(KEY);
70             String[] uuids =
71                     aggs.getKeysAsPagedStringList(queryByFilter.getPageSize(), queryByFilter.getPageStartIndex());
72             long totalSize = aggs.size();
73             return new QueryResult<String>(queryByFilter.getPage(), queryByFilter.getPageSize(),
74                     new SearchResult<String>(uuids, totalSize));
75         } catch (IOException e) {
76             throw new RuntimeException("problem reading nodes for req=" + request, e);
77         }
78
79     }
80
81 }