com.weicoder.redis.init.RedisInit Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of redis Show documentation
There is a newer version: 3.6.2
package com.weicoder.redis.init;

import java.lang.reflect.Method;
import java.lang.reflect.Parameter;
import java.util.List;
import java.util.Map;

import com.weicoder.common.concurrent.ExecutorUtil;
import com.weicoder.common.init.Init;
import com.weicoder.common.lang.Lists;
import com.weicoder.common.lang.Maps;
import com.weicoder.common.log.Log;
import com.weicoder.common.log.LogFactory; 
import com.weicoder.common.util.BeanUtil;
import com.weicoder.common.util.ClassUtil;
import com.weicoder.common.U;
import com.weicoder.common.U.C;
import com.weicoder.json.JsonEngine;
import com.weicoder.redis.params.RedisParams;
import com.weicoder.redis.Redis;
import com.weicoder.redis.annotation.Channel;
import com.weicoder.redis.annotation.Subscribes;
import com.weicoder.redis.factory.RedisFactory;

/**
 * redis订阅功能
 * 
 * @author WD
 */
public class RedisInit implements Init {
	// 日志
	private Log LOG = LogFactory.getLog(RedisInit.class);
	// 保存Channel对应对象
	private Map SUBSCRIBES = Maps.newMap();
	// 保存Channel对应方法
	private Map METHODS = Maps.newMap();
	// 保存Redis消费
	private Map REDIS = Maps.newMap();
	// 保存Redis对应消费的Channel
	private Map> CHANNELS = Maps.newMap();

	@Override
	public void init() {
		// 获得所有redis订阅者
		List> subscribes = C.list(Subscribes.class);
		if (U.E.isNotEmpty(subscribes)) {
			// 循环处理所有redis订阅类
			int n = 0;
			for (Class c : subscribes) {
				// 执行对象
//				Object subscribe = ClassUtil.newInstance(c);
				Object subscribe = C.ioc(c);
				Subscribes a = subscribe.getClass().getAnnotation(Subscribes.class);
				String name = a.value();
				if (!REDIS.containsKey(name))
					REDIS.put(name, RedisFactory.getRedis(name));
				// 获得channels列表
				List channels = Maps.getList(CHANNELS, name);
				// 处理所有方法
				ClassUtil.getPublicMethod(c).forEach(m -> {
					// 方法有执行时间注解
					Channel channel = m.getAnnotation(Channel.class);
					if (channel != null) {
						String val = channel.value();
						METHODS.put(val, m);
						channels.add(val);
						SUBSCRIBES.put(val, subscribe);
						LOG.debug("add redis name={} subscribe={} channel={}", name, c.getSimpleName(), val);
					}
				});
				n += channels.size();
			}
			LOG.info("add redis subscribe={} channels={}", subscribes.size(), n);
			// 订阅相关消费数据
			CHANNELS.forEach((key, val) -> {
				// 定时观察订阅信息
				ExecutorUtil.pool(RedisParams.PREFIX).execute(() -> {
					// ScheduledUtil.delay(RedisParams.PREFIX, () -> {
					REDIS.get(key).subscribe((channel, message) -> {
						// 线程池id
						long tid = Thread.currentThread().getId();
						// 获得订阅通道的对象和方法
						long time = System.currentTimeMillis();
						Object s = SUBSCRIBES.get(channel);
						Method m = METHODS.get(channel);
						if (U.E.isNotEmptys(s, m)) {
							// 获得所有参数
							Parameter[] params = m.getParameters();
							Object[] objs = null;
							if (U.E.isEmpty(params))
								// 参数为空直接执行方法
								BeanUtil.invoke(s, m);
							else {
								objs = new Object[params.length];
								// 有参数 现在只支持 1位的参数，1个参数表示message
								Class type = params[0].getType();
								if (params.length == 1)
									if (String.class.equals(type))
										objs[0] = message;
									else
										objs[0] = JsonEngine.toBean(message, type);
								// 执行方法
								BeanUtil.invoke(s, m, objs);
							}
						}
						LOG.debug("redis subscribe={} method={} channel={} message={} time={}  thread={}",
								s.getClass().getSimpleName(), m.getName(), channel, message,
								System.currentTimeMillis() - time, tid);
					}, Lists.toArray(val));
				});
			});
		}
	}
}