Divide the MSB source codes into two repos

[msb/apigateway.git] / apiroute / apiroute-service / src / main / java / org / onap / msb / apiroute / wrapper / consulextend / expose / WatchServiceHealthTask.java

diff --git a/apiroute/apiroute-service/src/main/java/org/onap/msb/apiroute/wrapper/consulextend/expose/WatchServiceHealthTask.java b/apiroute/apiroute-service/src/main/java/org/onap/msb/apiroute/wrapper/consulextend/expose/WatchServiceHealthTask.java
new file mode 100644 (file)
index 0000000..b0d64a7
--- /dev/null
+++ b/apiroute/apiroute-service/src/main/java/org/onap/msb/apiroute/wrapper/consulextend/expose/WatchServiceHealthTask.java
@@ -0,0 +1,128 @@
+package org.onap.msb.apiroute.wrapper.consulextend.expose;
+
+import java.math.BigInteger;
+import java.util.List;
+
+import org.onap.msb.apiroute.wrapper.consulextend.HealthClient;
+import org.onap.msb.apiroute.wrapper.consulextend.cache.ServiceHealthCache;
+import org.onap.msb.apiroute.wrapper.consulextend.cache.ConsulCache.Listener;
+import org.onap.msb.apiroute.wrapper.consulextend.model.health.ServiceHealth;
+import org.slf4j.Logger;
+import org.slf4j.LoggerFactory;
+
+import com.orbitz.consul.option.CatalogOptions;
+import com.orbitz.consul.option.QueryOptions;
+
+public class WatchServiceHealthTask extends WatchTask<List<ServiceHealth>> {
+       private final static Logger LOGGER = LoggerFactory
+                       .getLogger(WatchServiceHealthTask.class);
+       
+       private ServiceHealthCache serviceHealthCache = null;
+       private String serviceName="";
+
+       public String getServiceName() {
+               return serviceName;
+       }
+
+       public WatchServiceHealthTask(final HealthClient healthClient,
+                       final String serviceName,final boolean passing,
+                       final CatalogOptions catalogOptions, final int watchSeconds,
+                       final QueryOptions queryOptions) {
+               initCache(healthClient, serviceName, passing, catalogOptions,
+                               watchSeconds, queryOptions);
+       }
+
+       public WatchServiceHealthTask(final HealthClient healthClient,
+                       final String serviceName,final boolean passing,
+                       final int watchSeconds)
+
+       {
+               initCache(healthClient, serviceName, passing, CatalogOptions.BLANK,
+                               watchSeconds, QueryOptions.BLANK);
+       }
+
+       public WatchServiceHealthTask(final HealthClient healthClient,
+                       final String serviceName, final int watchSeconds)
+
+       {
+               initCache(healthClient, serviceName, true, CatalogOptions.BLANK,
+                               watchSeconds, QueryOptions.BLANK);
+       }
+
+       private ServiceHealthCache initCache(final HealthClient healthClient,
+                       final String serviceName,final boolean passing,
+                       final CatalogOptions catalogOptions, final int watchSeconds,
+                       final QueryOptions queryOptions) {
+//             LOGGER.info("************create {} watch task*****************",serviceName);
+               this.serviceName = serviceName;         
+               serviceHealthCache = ServiceHealthCache.newCache(healthClient,
+                               serviceName, passing, catalogOptions, watchSeconds,
+                               queryOptions);
+
+               serviceHealthCache
+                               .addListener((Listener<List<ServiceHealth>>) new InternalListener());
+
+               return serviceHealthCache;
+       }
+
+       public boolean startWatch() {
+               
+               if(serviceHealthCache!=null)
+               {
+                       try {
+                               serviceHealthCache.start();
+                               LOGGER.info("************start {} watch task*****************",serviceName);
+                               return true;
+                       } catch (Exception e) {
+                               // TODO Auto-generated catch block
+                               LOGGER.warn("start service watch failed:", e);
+                       }
+               }
+               
+               return false;
+               
+       }
+
+       public boolean stopWatch(){
+               if (serviceHealthCache != null) {
+                       try {
+                               serviceHealthCache.stop();
+                               LOGGER.info("************stop {} watch task*****************",serviceName);
+                               return true;
+                       } catch (Exception e) {
+                               // TODO Auto-generated catch block
+                               LOGGER.warn("stop service watch failed:", e);
+                       }
+               }
+               
+               return false;
+       }
+
+
+       public boolean resetIndex()
+       {
+               if (LOGGER.isDebugEnabled()) {
+                       LOGGER.debug("reset " + serviceName + " consul index");
+               }
+               
+               //reset consul index
+               serviceHealthCache.updateIndex(BigInteger.valueOf(0));
+               
+               
+               //reset modify index
+               for (WatchTask.Filter<List<ServiceHealth>> filter : getAllFilters()) {
+                       if (filter instanceof ServiceModifyIndexFilter) {
+                               if (LOGGER.isDebugEnabled()) {
+                                       LOGGER.debug("reset " + serviceName + " modify index");
+                               }
+                               return ((ServiceModifyIndexFilter) filter).resetModifyIndex();
+                       }
+               }
+               
+               if (LOGGER.isDebugEnabled()) {
+                       LOGGER.debug("reset modify index.did not find filter:" + serviceName);
+               }
+               
+               return false;
+       }
+}