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

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