0d2e39bcb1db40b71c5f66bca9c319d1073f0eb1
[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.data;
23
24 import java.util.HashMap;
25 import java.util.List;
26 import java.util.Map;
27 import java.util.concurrent.Executors;
28 import java.util.concurrent.ScheduledExecutorService;
29 import java.util.concurrent.TimeUnit;
30
31 import org.onap.ccsdk.features.sdnr.wt.common.database.HtDatabaseClient;
32 import org.onap.ccsdk.features.sdnr.wt.common.database.SearchResult;
33 import org.onap.ccsdk.features.sdnr.wt.common.database.config.HostInfo;
34 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.data.provider.rev190801.Entity;
35 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.data.provider.rev190801.read.mediator.server.list.output.Data;
36 import org.slf4j.Logger;
37 import org.slf4j.LoggerFactory;
38
39 public class MediatorServerDataProvider implements AutoCloseable {
40
41         private static final Logger LOG = LoggerFactory.getLogger(MediatorServerDataProvider.class);
42
43         private final HtDatabaseClient dbClient;
44         private final DataObjectAcessor<Data> mediatorserverRW;
45         private final int REFRESH_INTERVAL = 60;
46         private final Map<String, Data> entries;
47         private final ScheduledExecutorService scheduler = Executors.newScheduledThreadPool(1);
48         private boolean isRunning;
49
50         public MediatorServerDataProvider(HostInfo[] hosts) throws Exception {
51                 this(hosts, null, null);
52         }
53
54         public MediatorServerDataProvider(HostInfo[] hosts, String authUsername, String authPassword) throws Exception {
55                 super();
56                 LOG.info("Start {}", this.getClass().getName());
57                 this.entries = new HashMap<>();
58                 this.dbClient = new HtDatabaseClient(hosts, authUsername, authPassword);
59                 this.mediatorserverRW = new DataObjectAcessor<>(dbClient, Entity.MediatorServer, Data.class);
60                 this.scheduler.scheduleAtFixedRate(onTick, this.REFRESH_INTERVAL, this.REFRESH_INTERVAL, TimeUnit.SECONDS);
61         }
62
63         private final Runnable onTick = new Runnable() {
64
65                 @Override
66                 public void run() {
67                         isRunning = true;
68                         runIt();
69                         isRunning = false;
70                 }
71
72         };
73
74         private void runIt() {
75                 SearchResult<Data> result = MediatorServerDataProvider.this.mediatorserverRW.doReadAll();
76                 List<Data> data = result.getHits();
77                 for (Data item : data) {
78                         MediatorServerDataProvider.this.entries.put(item.getId(), item);
79                 }
80         }
81
82         /**
83          * 
84          * @param dbServerId
85          * @return url or null if not exists
86          */
87         public String getHostUrl(String dbServerId) {
88                 Data info = this.entries.getOrDefault(dbServerId, null);
89                 return info == null ? null : info.getUrl();
90         }
91
92         public boolean triggerReloadSync() {
93                 if (!isRunning) {
94                         runIt();
95                 }
96                 return true;
97         }
98
99         @Override
100         public void close() throws Exception {
101                 this.scheduler.shutdown();
102         }
103 }