examcloud-components/src/main/java/cn/com/qmth/examcloud/web/redis/SimpleRedisClient.java

package cn.com.qmth.examcloud.web.redis;

import java.util.concurrent.TimeUnit;

import org.springframework.data.redis.core.RedisTemplate;

import cn.com.qmth.examcloud.commons.exception.ExamCloudRuntimeException;
import cn.com.qmth.examcloud.commons.logging.ExamCloudLog;
import cn.com.qmth.examcloud.commons.logging.ExamCloudLogFactory;

/**
 * redis client
 *
 * @author WANGWEI
 * @date 2019年2月22日
 * @Copyright (c) 2018-2020 WANGWEI [QQ:522080330] All Rights Reserved.
 */
public final class SimpleRedisClient implements RedisClient {

	private static final ExamCloudLog REDIS_LOG = ExamCloudLogFactory.getLog("REDIS_LOGGER");

	private RedisTemplate<String, Object> redisTemplate;

	private boolean enable = true;

	public SimpleRedisClient(RedisTemplate<String, Object> redisTemplate) {
		super();
		this.redisTemplate = redisTemplate;
	}

	private void beforeMethod() {
		if (!enable) {
			throw new ExamCloudRuntimeException("RedisClient is not enabled");
		}
	}

	private void afterMethod(String method, long startTimeMillis) {
		if (REDIS_LOG.isDebugEnabled()) {
			String s = String.format("[SimpleRedisClient.%s] cost %d ms.", method,
					System.currentTimeMillis() - startTimeMillis);
			REDIS_LOG.debug(s);
		}
	}

	@Override
	public void set(String key, Object value) {
		long s = System.currentTimeMillis();
		beforeMethod();
		redisTemplate.opsForValue().set(key, value);
		afterMethod("set(String key, Object value)", s);
	}

	@Override
	public void set(String key, Object value, int timeout) {
		long s = System.currentTimeMillis();
		beforeMethod();
		set(key, value);
		expire(key, timeout);
		afterMethod("set(String key, Object value, int timeout)", s);
	}

	@Override
	public void expire(String key, int timeout) {
		long s = System.currentTimeMillis();
		beforeMethod();
		redisTemplate.expire(key, timeout, TimeUnit.SECONDS);
		afterMethod("expire(String key, int timeout)", s);
	}

	/**
	 * 方法注释
	 *
	 * @author WANGWEI
	 * @param key
	 * @param timeout
	 * @param unit
	 */
	public void expire(String key, final long timeout, final TimeUnit unit) {
		long s = System.currentTimeMillis();
		beforeMethod();
		redisTemplate.expire(key, timeout, unit);
		afterMethod("(String key, final long timeout, final TimeUnit unit)", s);
	}

	@Override
	public <T> T get(String key, Class<T> c, int timeout) {
		long s = System.currentTimeMillis();
		beforeMethod();
		Object object = redisTemplate.opsForValue().get(key);
		@SuppressWarnings("unchecked")
		T t = (T) object;
		expire(key, timeout);
		afterMethod("get(String key, Class<T> c, int timeout)", s);
		return t;
	}

	@Override
	public <T> T get(String key, Class<T> c) {
		long s = System.currentTimeMillis();
		beforeMethod();
		Object object = redisTemplate.opsForValue().get(key);
		@SuppressWarnings("unchecked")
		T t = (T) object;
		afterMethod("get(String key, Class<T> c)", s);
		return t;
	}

	@Override
	public void delete(String key) {
		long s = System.currentTimeMillis();
		beforeMethod();
		redisTemplate.opsForValue().set(key, null);
		expire(key, 0);
		afterMethod("delete(String key)", s);
	}

	@Override
	public void convertAndSend(String channel, Object message) {
		long s = System.currentTimeMillis();
		beforeMethod();
		redisTemplate.convertAndSend(channel, message);
		afterMethod("convertAndSend(String channel, Object message)", s);
	}

	@Override
	public Boolean setIfAbsent(String key, String value, int timeout) {
		long s = System.currentTimeMillis();
		beforeMethod();
		Boolean b = redisTemplate.opsForValue().setIfAbsent(key, value, timeout, TimeUnit.SECONDS);
		afterMethod("setIfAbsent(String key, String value, int timeout)", s);
		return b;
	}

	@Override
	public boolean isEnable() {
		return enable;
	}

	@Override
	public void setEnable(boolean enable) {
		this.enable = enable;
	}

}