Divide the MSB source codes into two repos

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

diff --git a/apiroute/apiroute-service/src/main/java/org/onap/msb/apiroute/wrapper/consulextend/expose/CheckServiceDataEmptyAndAutoStopWatchFilter.java b/apiroute/apiroute-service/src/main/java/org/onap/msb/apiroute/wrapper/consulextend/expose/CheckServiceDataEmptyAndAutoStopWatchFilter.java
new file mode 100644 (file)
index 0000000..01b5168
--- /dev/null
+++ b/apiroute/apiroute-service/src/main/java/org/onap/msb/apiroute/wrapper/consulextend/expose/CheckServiceDataEmptyAndAutoStopWatchFilter.java
@@ -0,0 +1,98 @@
+package org.onap.msb.apiroute.wrapper.consulextend.expose;
+
+import java.util.ArrayList;
+import java.util.List;
+
+import org.onap.msb.apiroute.SyncDataManager;
+import org.onap.msb.apiroute.wrapper.consulextend.model.health.ImmutableService;
+import org.onap.msb.apiroute.wrapper.consulextend.model.health.ImmutableServiceHealth;
+import org.onap.msb.apiroute.wrapper.consulextend.model.health.Service;
+import org.onap.msb.apiroute.wrapper.consulextend.model.health.ServiceHealth;
+import org.onap.msb.apiroute.wrapper.queue.QueueManager;
+import org.onap.msb.apiroute.wrapper.queue.ServiceData;
+import org.slf4j.Logger;
+import org.slf4j.LoggerFactory;
+
+import com.orbitz.consul.model.ConsulResponse;
+import com.orbitz.consul.model.health.ImmutableNode;
+
+
+
+public class CheckServiceDataEmptyAndAutoStopWatchFilter implements
+               WatchTask.Filter<List<ServiceHealth>> {
+
+       private final static Logger LOGGER = LoggerFactory
+                       .getLogger(CheckServiceDataEmptyAndAutoStopWatchFilter.class);
+       private final String serviceName;
+
+       public CheckServiceDataEmptyAndAutoStopWatchFilter(
+                       final String serviceName) {
+               this.serviceName = serviceName;
+       }
+
+       @Override
+       public boolean filter(ConsulResponse<List<ServiceHealth>> object) {
+               // TODO Auto-generated method stub
+               boolean result = check(object);
+
+               if (!result) {
+                       // create delete
+                       writeServiceToQueue4Del();
+                       // stop watch
+                       SyncDataManager.stopWatchService(serviceName);
+               }
+
+               return result;
+       }
+
+       // when:
+       // 1)service had been deleted
+       // 2)service Health check was not passing
+       // single service return [],size==0
+       // stop this service watching task and create delete event
+       private boolean check(ConsulResponse<List<ServiceHealth>> object) {
+               boolean result = true;
+
+               if (object == null || object.getResponse() == null
+                               || object.getResponse().size() == 0) {
+                       LOGGER.info("check service-{},its data is empty",
+                                       serviceName);
+                       return false;
+               }
+
+               return result;
+       }
+
+       private void writeServiceToQueue4Del() {
+               ServiceData<List<ServiceHealth>> data = new ServiceData<List<ServiceHealth>>();
+               data.setDataType(ServiceData.DataType.service);
+               data.setOperate(ServiceData.Operate.delete);
+               
+               // tell the subsequent operation the service name which will be deleted
+               Service service = ImmutableService.builder().id("").port(0).address("")
+                               .service(serviceName).addTags("").createIndex(0).modifyIndex(0).build();
+               ServiceHealth serviceHealth = ImmutableServiceHealth.builder()
+                               .service(service)
+                               .node(ImmutableNode.builder().node("").address("").build())
+                               .build();
+               List<ServiceHealth> serviceHealthList = new ArrayList<ServiceHealth>();
+               serviceHealthList.add(serviceHealth);
+
+               data.setData(serviceHealthList);
+
+               LOGGER.info("put delete service["
+                               + serviceName
+                               + "] to service queue :because of deleted ");
+
+               try {
+                       QueueManager.getInstance().putIn(data);
+               } catch (InterruptedException e) {
+                       // TODO Auto-generated catch block
+                       LOGGER.warn(
+                                       "put delete service["
+                                                       + serviceName
+                                                       + "]  to  service queue interrupted because of deleted:",
+                                       e);
+               }
+       }
+}