migrate sdnr features to sulfur
[ccsdk/features.git] / sdnr / wt / data-provider / provider / src / main / java / org / onap / ccsdk / features / sdnr / wt / dataprovider / database / elasticsearch / data / acessor / DataObjectAcessorStatus.java
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.queries.QueryBuilder;
29 import org.onap.ccsdk.features.sdnr.wt.common.database.requests.SearchRequest;
30 import org.onap.ccsdk.features.sdnr.wt.common.database.responses.AggregationEntries;
31 import org.onap.ccsdk.features.sdnr.wt.common.database.responses.SearchResponse;
32 import org.onap.ccsdk.features.sdnr.wt.dataprovider.database.elasticsearch.data.rpctypehelper.QueryByFilter;
33 import org.onap.ccsdk.features.sdnr.wt.dataprovider.database.elasticsearch.data.rpctypehelper.QueryResult;
34 import org.onap.ccsdk.features.sdnr.wt.dataprovider.model.types.YangHelper2;
35 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.data.provider.rev201110.ConnectionLogStatus;
36 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.data.provider.rev201110.Entity;
37 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.data.provider.rev201110.EntityInput;
38 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.data.provider.rev201110.read.status.output.Data;
39 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.data.provider.rev201110.read.status.output.DataBuilder;
40 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.data.provider.rev201110.status.entity.FaultsBuilder;
41 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.data.provider.rev201110.status.entity.NetworkElementConnectionsBuilder;
42 import org.opendaylight.yangtools.yang.common.Uint32;
43
44 public class DataObjectAcessorStatus extends DataObjectAcessor<Data> {
45
46     final String ESDATATYPE_FAULTCURRENT_SEVERITY_KEY = "severity";
47     final String ESDATATYPE_NECON_CONNECTIONSTATE_KEY = "status";
48
49     private final ExtRestClient dbClient;
50
51     public DataObjectAcessorStatus(HtDatabaseClient dbClient, Entity entity, boolean doFullsizeRequests)
52             throws ClassNotFoundException {
53         super(dbClient, entity, Data.class, doFullsizeRequests);
54         this.dbClient = dbClient;
55
56     }
57
58     public QueryResult<Data> getDataStatus(EntityInput input) throws IOException {
59         SearchRequest request = getNewInstanceOfSearchRequest(Entity.Faultcurrent);
60         QueryByFilter queryByFilter = new QueryByFilter(input);
61         QueryBuilder query = queryByFilter.getQueryBuilderByFilter();
62         query.aggregations(ESDATATYPE_FAULTCURRENT_SEVERITY_KEY).size(0);
63         if(this.doFullsizeRequest) {
64             query.doFullsizeRequest();
65         }
66         request.setQuery(query);
67         SearchResponse response = this.dbClient.search(request);
68         AggregationEntries aggs = response.getAggregations(ESDATATYPE_FAULTCURRENT_SEVERITY_KEY);
69
70         DataBuilder builder = new DataBuilder().setFaults(
71                 new FaultsBuilder().setCriticals(YangHelper2.getLongOrUint32(aggs.getOrDefault("Critical", 0L)))
72                         .setMajors(YangHelper2.getLongOrUint32(aggs.getOrDefault("Major", 0L)))
73                         .setMinors(YangHelper2.getLongOrUint32(aggs.getOrDefault("Minor", 0L)))
74                         .setWarnings(YangHelper2.getLongOrUint32(aggs.getOrDefault("Warning", 0L))).build());
75
76         request = getNewInstanceOfSearchRequest(Entity.NetworkelementConnection);
77         query.aggregations(ESDATATYPE_NECON_CONNECTIONSTATE_KEY).size(0);
78         if(this.doFullsizeRequest) {
79             query.doFullsizeRequest();
80         }
81         request.setQuery(query);
82         response = this.dbClient.search(request);
83         aggs = response.getAggregations(ESDATATYPE_NECON_CONNECTIONSTATE_KEY);
84         builder.setNetworkElementConnections(new NetworkElementConnectionsBuilder()
85                 .setConnected(Uint32.valueOf(aggs.getOrDefault(ConnectionLogStatus.Connected.getName(), 0L)))
86                 .setConnecting(Uint32.valueOf(aggs.getOrDefault(ConnectionLogStatus.Connecting.getName(), 0L)))
87                 .setDisconnected(Uint32.valueOf(aggs.getOrDefault(ConnectionLogStatus.Disconnected.getName(), 0L)))
88                 .setUnableToConnect(
89                         Uint32.valueOf(aggs.getOrDefault(ConnectionLogStatus.UnableToConnect.getName(), 0L)))
90                 .setMounted(Uint32.valueOf(aggs.getOrDefault(ConnectionLogStatus.Mounted.getName(), 0L)))
91                 .setUnmounted(Uint32.valueOf(aggs.getOrDefault(ConnectionLogStatus.Unmounted.getName(), 0L)))
92                 .setUndefined(Uint32.valueOf(aggs.getOrDefault(ConnectionLogStatus.Undefined.getName(), 0L)))
93                 .setTotal(Uint32.valueOf(response.getTotal())).build());
94
95         long toalsize = 1;
96         return new QueryResult<Data>(1L, 1L, new SearchResult<Data>(new Data[] {builder.build()}, toalsize));
97
98     }
99
100
101     private static SearchRequest getNewInstanceOfSearchRequest(Entity entity) {
102         return new SearchRequest(entity.getName(), entity.getName());
103     }
104
105
106 }