com.alibaba.tmq.common.util.DiamondHelper Maven / Gradle / Ivy

Go to download
package com.alibaba.tmq.common.util;

import java.util.HashMap;
import java.util.Map;
import java.util.concurrent.CopyOnWriteArrayList;
import java.util.concurrent.Executor;
import java.util.concurrent.Executors;
import java.util.concurrent.locks.ReentrantLock;

import org.apache.commons.logging.Log;
import org.apache.commons.logging.LogFactory;

import com.taobao.diamond.manager.DiamondManager;
import com.taobao.diamond.manager.ManagerListener;
import com.taobao.diamond.manager.impl.DefaultDiamondManager;


public class DiamondHelper {

	private static final Log logger = LogFactory.getLog(DiamondHelper.class);

	private final static Map contexts = new HashMap();

	private final static Executor executor = Executors.newSingleThreadScheduledExecutor();

	private final static ReentrantLock lock = new ReentrantLock();

	private final static String DEFAULT_GROUP = "DEFAULT_GROUP";

	private final static long DEFAULT_READ_TIMOUT = 3000;


	public static String getData(String dataId) {
		return getData(dataId, DEFAULT_READ_TIMOUT);
	}

	public static String getData(String dataId, long timeOut) {
		DiamondContext context = getContext(dataId);
		return context.getManager().getAvailableConfigureInfomation(timeOut);
	}


	public static DiamondManager getDiamondManager(String dataId) {
		DiamondContext context = getContext(dataId);
		return context.getManager();
	}


	public static void addListener(String dataId, DataListener dataListener) {
		if (null != dataListener) {
			DiamondContext context = getContext(dataId);
			context.getListener().addListener(dataListener);
		}
	}

	public static void removeListener(String dataId, DataListener dataListener) {
		if (null != dataListener) {
			DiamondContext context = getContext(dataId);
			context.getListener().removeListener(dataListener);
		}
	}

	public static void cleanListener(String dataId) {
		DiamondContext context = getContext(dataId);
		context.getListener().cleanListener();
	}

	public static void cleanListener() {
		lock.lock();
		try {
			for (Map.Entry entry : contexts.entrySet()) {
				DiamondContext diamondContext = entry.getValue();
				//�Ƴ�����dataId��Ӧ��listener
				diamondContext.getListener().cleanListener();
			}
		} finally {
			lock.unlock();
		}
	}

	private static DiamondContext buildContext(String dataId) {
		DiamondContext context = contexts.get(dataId);
		if (null == context) {
			lock.lock();
			try {
				context = contexts.get(dataId);
				if (null == context) {
					ControlListener controlListener = new ControlListener(dataId);
					DiamondManager diamondManager = new DefaultDiamondManager(DEFAULT_GROUP, dataId, controlListener);
					context = new DiamondContext(diamondManager, controlListener);
					contexts.put(dataId, context);
				}
			} finally {
				lock.unlock();
			}
		}
		return context;
	}

	private static DiamondContext getContext(String dataId) {
		DiamondContext context = contexts.get(dataId);
		if (null == context) {
			context = buildContext(dataId);
		}
		return context;
	}

	static class DiamondContext {
		private final ControlListener listener;
		private final DiamondManager manager;

		public DiamondContext(DiamondManager manager, ControlListener listener) {
			this.listener = listener;
			this.manager = manager;
		}

		public ControlListener getListener() {
			return listener;
		}

		public DiamondManager getManager() {
			return manager;
		}
	}


	static class ControlListener implements ManagerListener {

		private final String dataId;

		private CopyOnWriteArrayList dataListeners = new CopyOnWriteArrayList();

		public ControlListener(String dataId) {
			this.dataId = dataId;
		}

		public void addListener(DataListener listener) {
			dataListeners.addIfAbsent(listener);
		}

		public void removeListener(DataListener listener) {
			this.dataListeners.remove(listener);
		}

		public void cleanListener() {
			this.dataListeners.clear();
		}

		public void receiveConfigInfo(String configInfo) {
			for (DataListener listener : dataListeners) {
				try {
					listener.receiveConfigInfo(dataId, configInfo);
				} catch (Exception e) {
					logger.error("call dataListener Error dataId: " + dataId, e);
				}
			}
		}

		public Executor getExecutor() {
			return DiamondHelper.executor;
		}
	}

	public interface DataListener {
		public void receiveConfigInfo(String dataId, String configInfo);
	}

	public static void main(String[] args) {
		String dataId = "qihao.qihao.qihao";
		String data = DiamondHelper.getData(dataId);
		System.out.println(data);
		DiamondHelper.addListener(dataId, new DataListener() {
			public void receiveConfigInfo(String dataId, String configInfo) {
				System.out.println(dataId + " " + configInfo + " " + this.toString());
			}
		});
		DiamondHelper.addListener(dataId, new DataListener() {
			public void receiveConfigInfo(String dataId, String configInfo) {
				System.out.println(dataId + " " + configInfo + " " + this.toString());
			}
		});
	}
}