com.sharksharding.resources.watcher.RedisWatcher Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of shark Show documentation
shark mysql sharding
The newest version!
/*
 * Copyright 2015-2101 gaoxianglong
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *      http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */
package com.sharksharding.resources.watcher;

import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.scheduling.annotation.Scheduled;
import org.springframework.stereotype.Component;
import com.sharksharding.resources.register.bean.RegisterDataSource;
import com.sharksharding.util.MD5Util;
import redis.clients.jedis.JedisCluster;

/**
 * sharding、数据源相关的redis节点watcher
 * 
 * @author gaoxianglong
 * 
 * @version 1.3.7
 */
public class RedisWatcher {
	private JedisCluster jedisCluster;
	private String key;
	private int type;
	private static int version = 1;
	private static String md5Code;
	private String resourceType = "redis";
	private Logger logger = LoggerFactory.getLogger(RedisWatcher.class);

	public void init(JedisCluster jedisCluster, String key, int type) {
		this.jedisCluster = jedisCluster;
		this.key = key;
		this.type = type;
	}

	@Scheduled(cron = "0/10 * * * * ?")
	private void run() {
		switch (type) {
		case 0:
			versionCheck();
			break;
		case 1:
			md5Check();
			break;
		default:
			break;
		}
	}

	/**
	 * 版本号吗检查,如果不一致就意味着配置中心修改了数据
	 * 
	 * @author gaoxianglong
	 * 
	 * @return void
	 */
	private void versionCheck() {
		if (null != jedisCluster && null != key) {
			logger.debug("redisWatch run...");
			final String[] values = jedisCluster.get(key).split("(%@%)");
			final int version = Integer.valueOf(values[0]);
			final String resource = values[1];
			if (RedisWatcher.getVersion() != version) {
				/* 如果版本发生变化，则重新向ioc容器中注册相关bean实例 */
				RegisterDataSource.register(resource, resourceType);
				RedisWatcher.setVersion(version);
				logger.debug("resource version-->" + version);
			}
		}
	}

	/**
	 * md5吗检查,如果不一致就意味着配置中心修改了数据
	 * 
	 * @author gaoxianglong
	 * 
	 * @return void
	 */
	private void md5Check() {
		if (null != jedisCluster && null != key) {
			logger.debug("redisWatch执行中...");
			final String value = jedisCluster.get(key);
			final String md5Code = MD5Util.toMd5Code(value);
			if (!md5Code.equals(RedisWatcher.getMd5Code())) {
				/* 如果版本发生变化，则重新向ioc容器中注册相关bean实例 */
				RegisterDataSource.register(value, resourceType);
				RedisWatcher.setMd5Code(md5Code);
				logger.debug("md5 code-->" + md5Code);
			}
		}
	}

	public static String getMd5Code() {
		return md5Code;
	}

	public static void setMd5Code(String md5Code) {
		RedisWatcher.md5Code = md5Code;
	}

	public static int getVersion() {
		return version;
	}

	public static void setVersion(int version) {
		RedisWatcher.version = version;
	}
}