1 files changed, 96 insertions, 0 deletions
diff --git a/apiroute/apiroute-service/src/main/java/org/onap/msb/apiroute/wrapper/consulextend/expose/CheckTagAndAutoStopWatchFilter.java b/apiroute/apiroute-service/src/main/java/org/onap/msb/apiroute/wrapper/consulextend/expose/CheckTagAndAutoStopWatchFilter.java
new file mode 100644
index 0000000..311edce
--- /dev/null
+++ b/apiroute/apiroute-service/src/main/java/org/onap/msb/apiroute/wrapper/consulextend/expose/CheckTagAndAutoStopWatchFilter.java
@@ -0,0 +1,96 @@
+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.ServiceHealth;
+import org.onap.msb.apiroute.wrapper.queue.QueueManager;
+import org.onap.msb.apiroute.wrapper.queue.ServiceData;
+import org.onap.msb.apiroute.wrapper.queue.ServiceData.Operate;
+import org.onap.msb.apiroute.wrapper.util.ServiceFilter;
+import org.slf4j.Logger;
+import org.slf4j.LoggerFactory;
+
+import com.orbitz.consul.model.ConsulResponse;
+
+public class CheckTagAndAutoStopWatchFilter implements
+		WatchTask.Filter<List<ServiceHealth>> {
+
+	private final static Logger LOGGER = LoggerFactory
+			.getLogger(CheckTagAndAutoStopWatchFilter.class);
+
+	private final String serviceName;
+
+	public CheckTagAndAutoStopWatchFilter(final String serviceName) {
+		this.serviceName = serviceName;
+	}
+
+	// from consul,the response data:List<ServiceHealth>
+	// filter ServiceHealth list and find the ServiceHealths which satisfy the
+	// tags conditions
+	// 1)if all ServiceHealth don't satisfy,create delete event and stop watch
+	// 2)if have some ServiceHealths satisfy the tags conditions,create update
+	// event and send these ServiceHealths
+	@Override
+	public boolean filter(ConsulResponse<List<ServiceHealth>> object) {
+		// TODO Auto-generated method stub
+
+		// find #ServiceHealth# which satisfy the tag conditions
+		List<ServiceHealth> satisfyList = getSatisfyList(object);
+
+		// no satisfied ServiceHealth
+		if (satisfyList.isEmpty()) {
+			
+			LOGGER.info("put delete service["
+					+ serviceName
+					+ "] to service queue :because of NO tag meet the conditions");
+			
+			// create delete
+			writeServiceToQueue(object.getResponse(),
+					ServiceData.Operate.delete);
+			// stop watch
+			//SyncDataManager.stopWatchService(serviceName);
+			return false;
+		}
+
+		LOGGER.info("put update service["
+				+ serviceName
+				+ "] to service queue :which tags meet the conditions");
+		
+		// put the satisfy list to queue
+		writeServiceToQueue(satisfyList, ServiceData.Operate.update);
+
+		return true;
+	}
+
+	private List<ServiceHealth> getSatisfyList(
+			ConsulResponse<List<ServiceHealth>> object) {
+		List<ServiceHealth> satisfyList = new ArrayList<ServiceHealth>();
+		for (ServiceHealth health : object.getResponse()) {
+		
+			if (ServiceFilter.getInstance().isFilterCheck(health)) {
+				satisfyList.add(health);
+			}
+		}
+
+		return satisfyList;
+	}
+
+	private void writeServiceToQueue(List<ServiceHealth> serviceData,
+			Operate operate) {
+		ServiceData<List<ServiceHealth>> data = new ServiceData<List<ServiceHealth>>();
+		data.setOperate(operate);
+		data.setDataType(ServiceData.DataType.service);
+		data.setData(serviceData);
+		
+		
+		try {
+			QueueManager.getInstance().putIn(data);
+		} catch (InterruptedException e) {
+			LOGGER.warn("put " + operate + " service[" + serviceName
+					+ "]  to  service queue interrupted ", e);
+		}
+
+	}
+}