add fixes for wt sulfur
[ccsdk/features.git] / sdnr / wt / data-provider / dblib / src / main / java / org / onap / ccsdk / features / sdnr / wt / dataprovider / database / sqldb / database / SqlDBStatusReader.java
1 /*
2  * ============LICENSE_START=======================================================
3  * ONAP : ccsdk features
4  * ================================================================================
5  * Copyright (C) 2021 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.database;
23
24 import java.sql.ResultSet;
25 import java.sql.SQLException;
26 import java.util.Arrays;
27 import java.util.Map;
28 import org.onap.ccsdk.features.sdnr.wt.dataprovider.database.sqldb.SqlDBClient;
29 import org.onap.ccsdk.features.sdnr.wt.dataprovider.database.sqldb.data.rpctypehelper.QueryResult;
30 import org.onap.ccsdk.features.sdnr.wt.dataprovider.database.sqldb.query.SqlQuery;
31 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.data.provider.rev201110.ConnectionLogStatus;
32 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.data.provider.rev201110.Entity;
33 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.data.provider.rev201110.EntityInput;
34 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.data.provider.rev201110.SeverityType;
35 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.data.provider.rev201110.entity.input.Filter;
36 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.data.provider.rev201110.entity.input.FilterKey;
37 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.data.provider.rev201110.read.status.output.Data;
38 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.data.provider.rev201110.read.status.output.DataBuilder;
39 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.data.provider.rev201110.status.entity.FaultsBuilder;
40 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.data.provider.rev201110.status.entity.NetworkElementConnectionsBuilder;
41 import org.opendaylight.yangtools.yang.common.Uint32;
42 import org.slf4j.Logger;
43 import org.slf4j.LoggerFactory;
44
45 public class SqlDBStatusReader {
46
47     private final Logger LOG = LoggerFactory.getLogger(SqlDBStatusReader.class);
48
49     private final SqlDBClient dbClient;
50     private final String controllerId;
51
52     public SqlDBStatusReader(SqlDBClient dbClient, String controllerId) {
53         this.dbClient = dbClient;
54         this.controllerId = controllerId;
55     }
56
57     public QueryResult<Data> getDataStatus(EntityInput input) {
58         String selectQuery = createCountQuery("severity", Entity.Faultcurrent, this.controllerId, input);
59         long criticalCount = 0;
60         long majorCount = 0;
61         long minorCount = 0;
62         long warningCount = 0;
63         ResultSet data;
64         if(LOG.isTraceEnabled()){
65             LOG.trace("query={}", selectQuery);
66         }
67         try {
68             data = this.dbClient.read(selectQuery);
69             String severity;
70             while (data.next()) {
71                 severity = data.getString(1);
72                 if (severity != null) {
73                     if (severity.equals(SeverityType.Critical.getName())) {
74                         criticalCount = data.getLong(2);
75                     } else if (severity.equals(SeverityType.Major.getName())) {
76                         majorCount = data.getLong(2);
77                     } else if (severity.equals(SeverityType.Minor.getName())) {
78                         minorCount = data.getLong(2);
79                     } else if (severity.equals(SeverityType.Warning.getName())) {
80                         warningCount = data.getLong(2);
81                     }
82
83                 }
84             }
85             try { data.close(); } catch (SQLException ignore) { }
86         } catch (SQLException e) {
87             LOG.warn("problem reading status:", e);
88         }
89         DataBuilder builder = new DataBuilder().setFaults(
90                 new FaultsBuilder().setCriticals(Uint32.valueOf(criticalCount)).setMajors(Uint32.valueOf(majorCount))
91                         .setMinors(Uint32.valueOf(minorCount)).setWarnings(Uint32.valueOf(warningCount)).build());
92         selectQuery = createCountQuery("status", Entity.NetworkelementConnection, this.controllerId, input);
93         NetworkElementConnectionsBuilder neBuilder = new NetworkElementConnectionsBuilder();
94         String state;
95         long connectedCount = 0, connectingCount = 0, disconnectedCount = 0, mountedCount = 0, unableToConnectCount = 0,
96                 undefinedCount = 0, unmountedCount = 0;
97         long cnt;
98         if(LOG.isTraceEnabled()){
99             LOG.trace("query={}", selectQuery);
100         }
101         try {
102             data = this.dbClient.read(selectQuery);
103             while (data.next()) {
104                 state = data.getString(1);
105                 cnt = data.getLong(2);
106                 if (state != null) {
107                     if (state.equals(ConnectionLogStatus.Connected.getName())) {
108                         connectedCount = cnt;
109                     } else if (state.equals(ConnectionLogStatus.Connecting.getName())) {
110                         connectingCount = cnt;
111                     } else if (state.equals(ConnectionLogStatus.Disconnected.getName())) {
112                         disconnectedCount = cnt;
113                     } else if (state.equals(ConnectionLogStatus.Mounted.getName())) {
114                         mountedCount = cnt;
115                     } else if (state.equals(ConnectionLogStatus.UnableToConnect.getName())) {
116                         unableToConnectCount = cnt;
117                     } else if (state.equals(ConnectionLogStatus.Undefined.getName())) {
118                         undefinedCount = cnt;
119                     } else if (state.equals(ConnectionLogStatus.Unmounted.getName())) {
120                         unmountedCount = cnt;
121                     }
122                 }
123             }
124         } catch (SQLException e) {
125             LOG.warn("problem reading status:", e);
126         }
127         neBuilder.setConnected(Uint32.valueOf(connectedCount)).setConnecting(Uint32.valueOf(connectingCount))
128                 .setDisconnected(Uint32.valueOf(disconnectedCount)).setMounted(Uint32.valueOf(mountedCount))
129                 .setTotal(Uint32.valueOf(connectedCount + connectingCount + disconnectedCount + mountedCount
130                         + unableToConnectCount + undefinedCount + unmountedCount))
131                 .setUnableToConnect(Uint32.valueOf(unableToConnectCount)).setUndefined(Uint32.valueOf(undefinedCount))
132                 .setUnmounted(Uint32.valueOf(unmountedCount));
133         builder.setNetworkElementConnections(neBuilder.build());
134         return new QueryResult<Data>(Arrays.asList(builder.build()), 1, 1, 1);
135     }
136
137     private static String createCountQuery(String key, Entity e, String controllerId, EntityInput input) {
138         Map<FilterKey, Filter> filter = input != null ? input.getFilter() : null;
139         return String.format("SELECT `%s`, COUNT(`%s`) " + "FROM `%s` " + "%s " + "GROUP BY `%s`;", key, key,
140                 e.getName(),
141                 SqlQuery.getWhereExpression(filter!=null?filter.values():null, controllerId),
142                 key);
143     }
144
145 }