sdnr/wt/data-provider/provider/src/main/java/org/onap/ccsdk/features/sdnr/wt/dataprovider/data/MediatorServerDataProvider.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.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.onap.ccsdk.features.sdnr.wt.dataprovider.data.acessor.DataObjectAcessorWithId;
  34 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.data.provider.rev201110.Entity;
  35 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.data.provider.rev201110.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 DataObjectAcessorWithId<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, false);
  52     }
  53
  54     public MediatorServerDataProvider(HostInfo[] hosts, String authUsername, String authPassword, boolean trustAll) throws Exception {
  55         super();
  56         LOG.info("Start {}", this.getClass().getName());
  57         this.entries = new HashMap<>();
  58         this.dbClient = HtDatabaseClient.getClient(hosts, authUsername, authPassword, trustAll);
  59         this.mediatorserverRW = new DataObjectAcessorWithId<>(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 }