com.github.jedis.support.JedisFactory Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of jedis-spring-support Show documentation
using jedis lib, add spring support
The newest version!
package com.github.jedis.support;

import com.github.trace.TraceContext;
import com.github.trace.TraceRecorder;
import com.google.common.collect.ImmutableSet;
import com.google.common.reflect.Reflection;
import org.springframework.beans.factory.FactoryBean;
import redis.clients.jedis.*;
import redis.clients.util.Pool;

import java.lang.reflect.InvocationHandler;
import java.lang.reflect.Method;
import java.lang.reflect.ParameterizedType;
import java.net.ConnectException;
import java.util.Set;

/**
 * 通过cms系统配置redis，并接入监控系统
 * Created by lirui on 15/1/18.
 */
public class JedisFactory extends ConfigurableJedisPool implements FactoryBean {
  private Set names = ImmutableSet.of("hashCode", "toString", "equals");
  private Class clazz;

  @SuppressWarnings("unchecked")
  public JedisFactory() {
    this.clazz = (Class) ((ParameterizedType) getClass().getGenericSuperclass()).getActualTypeArguments()[0];
  }

  @Override
  public T getObject() throws Exception {
    return Reflection.newProxy(clazz, new InvocationHandler() {
      @Override
      public Object invoke(Object proxy, Method method, Object[] args) throws Throwable {
        if (names.contains(method.getName())) {
          JedisPool pool = getPool();
          if (pool != null) {
            return method.invoke(pool.getResource(), args);
          } else {
            ShardedJedisPool sPool = getShardedPool();
            if (sPool == null) {
              return findDefault(method);
            }
            return method.invoke(sPool.getResource(), args);
          }
        }

        String configName = getConfigName();
        TraceContext context = TraceContext.get();
        context.reset();
        context.inc();
        context.setIface(configName);
        context.setMethod(method.getName());
        context.setParameter(getParameters(args));
        context.setServerName("redis:" + configName);
        JedisPool pool = getPool();
        if (pool != null) {
          return exchangeRedis(pool, method, args, configName, context);
        } else {
          ShardedJedisPool sPool = getShardedPool();
          if (sPool == null) {
            return findDefault(method);
          }
          return exchangeRedis(sPool, method, args, configName, context);
        }
      }
    });
  }

  private  Object exchangeRedis(Pool pool, Method method, Object[] args, String configName, TraceContext context) throws ConnectException {
    Object ret = null;
    boolean fail = true;
    long start = System.currentTimeMillis();
    try {
      T redis;
      try {
        redis = pool.getResource();
      } catch (Exception e) {
        log.error("cannot getResource from:{}", configName, e);
        return null;
      }
      if (redis != null) {
        try {
          ret = method.invoke(redis, args);
          pool.returnResource(redis);
          fail = false;
        } catch (Exception e) {
          pool.returnBrokenResource(redis);
          context.setReason(e.getMessage());
          log.error("cannot exchange Redis: " + configName, e);
          return null;
        } finally {
          Client client = null;
          if (redis instanceof Jedis) {
            client = ((Jedis) redis).getClient();
          } else if (args.length > 0){
            Object key = args[0];
            if (key instanceof String) {
              client = ((ShardedJedis) redis).getShard((String) key).getClient();
            } else if (key instanceof byte[]) {
              client = ((ShardedJedis) redis).getShard((byte[]) key).getClient();
            }
          }
          if (client != null) {
            String url = client.getHost() + ':' + client.getPort() + '/' + client.getDB();
            context.setUrl(url);
          }
        }
      }
    } finally {
      long cost = System.currentTimeMillis() - start;
      context.setStamp(start);
      context.setFail(fail);
      context.setCost(cost);
      TraceRecorder.getInstance().post(context.copy());
      context.reset();
    }
    return ret;
  }

  @Override
  public Class getObjectType() {
    return clazz;
  }

  @Override
  public boolean isSingleton() {
    return false;
  }

}