public class JedisBeanPostProcessor implements BeanPostProcessor {

@Autowired
ShardedJedisPool shardedJedisPool;

static final Logger logger = Logger.getLogger(JedisBeanPostProcessor.class);

@Override
public Object postProcessAfterInitialization(Object bean, String beanName) throws BeansException {
if (bean instanceof JedisManageSupport) {
Enhancer enhancer = new Enhancer();
enhancer.setSuperclass(bean.getClass());
enhancer.setCallback(new JedisInterceptor(shardedJedisPool, bean));
Object targetBean = enhancer.create();
return targetBean;
}
else {
return bean;
}
}

@Override
public Object postProcessBeforeInitialization(Object bean, String beanName) throws BeansException {
return bean;
}
}

class JedisInterceptor implements MethodInterceptor {

static final Logger logger = Logger.getLogger(JedisInterceptor.class);

ShardedJedisPool pool;

Object src;

public JedisInterceptor(ShardedJedisPool pool, Object src) {
this.pool = pool;
this.src = src;
}

@Override
public Object intercept(Object target, Method method, Object[] arguments, MethodProxy methodProxy) throws Throwable {
Object result = null;
if (target instanceof JedisManageSupport) {
if (this.isDeclaredMethod(target, method)) {
ShardedJedis jedis = null;
try {
JedisManageSupport support = (JedisManageSupport) src;
jedis = pool.getResource();
support.setShardedJedis(jedis);
// logger.debug("調(diào)用之前注入jedis對(duì)象,method:" + method);
/**
* 下面代碼可以使用 method.invoke(src,arguments)。 不能使用
* methodProxy.invokeSuper(target,arguments);
* 因?yàn)锳類(lèi)中用Autowired注入的屬性，生成代理的子類(lèi)B后,因?yàn)樽宇?lèi)B是新建的類(lèi)。從父類(lèi)繼承的屬性沒(méi)有被初始化，
* 使用methodProxy.invokeSuper（）執(zhí)行是，會(huì)報(bào)空指針異常.
*/
result = methodProxy.invoke(src, arguments);
support.setShardedJedis(null);
}
catch (Exception e) {
pool.returnBrokenResource(jedis);
e.printStackTrace();
}
finally {
if (jedis != null) {
pool.returnResource(jedis);
}
// logger.debug("調(diào)用之后歸還jedis對(duì)象,method:" + method);
}
}
else {
result = methodProxy.invoke(src, arguments);
}
}
else {
throw new Exception("使用該代理必須繼承JedisManageSupport");
}
return result;
}

/**
* 是否是target類(lèi)本身定義的非私有的方法，還是繼承的父類(lèi)
* @return true是target自己類(lèi)的并且不是私有的的，
*/
private boolean isDeclaredMethod(Object target, Method arg1) {
Method temp = null;
try {
temp = target.getClass().getDeclaredMethod(arg1.getName(), arg1.getParameterTypes());
}
catch (SecurityException e) {
e.printStackTrace();
}
catch (NoSuchMethodException e) {
e.printStackTrace();
}
/**
* 不為null，并且是非私有的，返回true
*/
if (temp != null) {

return true;
}
else {
return false;
}
}
}

public abstract class JedisManageSupport {
ThreadLocalShardedJedis> jedisHolder = new ThreadLocalShardedJedis>();

public final ShardedJedis getShardedJedis() {
return jedisHolder.get();
}

public final void setShardedJedis(ShardedJedis jedis) {
jedisHolder.set(jedis);
}

/**
* 如果某個(gè)鍵不同單位之間也不會(huì)重復(fù)，可以使用這個(gè)方法生成redis的鍵
*/
public final byte[] assemKey(String baseKey) {
Assert.isTrue(StringUtils.isNotBlank(baseKey), "參數(shù)不能為空");
return baseKey.getBytes();
}

/**
* 根據(jù)tableName+prefix 構(gòu)造唯一key與assemKey(String baseKey, String tableName)
* 規(guī)則一致
*/
public final byte[] assemKeyByPrefix(String tableName, String baseKey) {
Assert.isTrue(StringUtils.isNotBlank(baseKey), "參數(shù)不能為空");
Assert.isTrue(StringUtils.isNotBlank(tableName), "參數(shù)不能為空");
UnitInfo unit = WebService.getUnitInfo();
Assert.isTrue(unit != null, "單位信息獲取不到");
return (tableName + "-" + unit.getPrefix() + "-" + baseKey).getBytes();
}

/**
*
* 不同前綴的表中可能有相同的鍵，同一個(gè)表中也可能是有重復(fù)的baseKey時(shí)，用這個(gè)生成redis的key 比如 用戶(hù)信息表的
* username字段，不同的用戶(hù)信息表允許重復(fù)的username，mooc_t_userinfo
* 也允許有相同的賬號(hào)，所以生成redis的key時(shí)，需要用到單位的mooc_school 放入redis中
*/
public final byte[] assemKeyByFid(String tableName, String baseKey) {
UnitInfo unit = WebService.getUnitInfo();
Assert.isTrue(unit != null, "單位信息獲取不到");
return (tableName + "-" + unit.getMoocSchool() + "-" + baseKey).getBytes();
}

}