分析“备忘使用spring-data-redis中的redistemplate的一个大坑”

lishenghan

　　前几天刚刚粗略看了一下spring-data-redis的源码 (1.0.1-RELEASE)
　　今天一早看到了 “备忘使用spring-data-redis中的redistemplate的一个大坑” http://www.iyunv.com/topic/1125295
　　又针对这部分分析了下源码，总结整理如下
　　spring-data-redis的各种Operations实现类，如RedisTemplate，DefaultSetOperations，DefaultListOperations等，对Redis命令的封闭都是通过如下结构调用的
　　RedisTemplate中的hasKey

public Boolean hasKey(K key) {
final byte[] rawKey = rawKey(key);
return execute(new RedisCallback<Boolean>() {
public Boolean doInRedis(RedisConnection connection) {
return connection.exists(rawKey);
}
}, true);
}
　　要实现一个指定返回类型的RedisCallback接口，这个接口只有一个函数doInRedis
　　doInRedis的参数是RedisConnection接口，spring-data-redis针对不的Redis Java Client都有对应的 RedisConnection实现：
　　JedisConnection
　　JredisConnection
　　RjcConnection
　　SrpConnection
　　目的就是将不同Client的API统一了一下，可以看成是Adapter吧 。
　　execute这个是模板函数，主是处理连接的问题。
　　spring-data-redis 针对不的 Redis Java Client 也都实现了相应的 RedisConnectionFactory，来获取连接。  
　　比如 JedisConnectionFactory 内部是通过 JedisPool 来实现连接工厂。
　　execute模板函数源码：

public <T> T execute(RedisCallback<T> action, boolean exposeConnection, boolean pipeline) {
Assert.notNull(action, "Callback object must not be null");
RedisConnectionFactory factory = getConnectionFactory();
RedisConnection conn = RedisConnectionUtils.getConnection(factory);
boolean existingConnection = TransactionSynchronizationManager.hasResource(factory);
preProcessConnection(conn, existingConnection);
boolean pipelineStatus = conn.isPipelined();
if (pipeline && !pipelineStatus) {
conn.openPipeline();
}
try {
RedisConnection connToExpose = (exposeConnection ? conn : createRedisConnectionProxy(conn));
T result = action.doInRedis(connToExpose);
// close pipeline
if (pipeline && !pipelineStatus) {
conn.closePipeline();
}
// TODO: any other connection processing?
return postProcessResult(result, conn, existingConnection);
} finally {
RedisConnectionUtils.releaseConnection(conn, factory);
}
}
　　正常情况下RedisCallback中doInRedis获取的RedisConnection每次都是新的
　　这也就出现了前面帖子中说的问题了“

这时候肯定会有个疑问，既然这个template每次都会生成新连接，那这个multi和exec命令还有个蛋用？？

 

”
　　使用RedisTemplate确实会有这个问题，但是作者对官方回答的理解有些出入。
　　The methods are exposed in case the connection is shared across methods. Currently we don't provide any out of the box support for connection binding but the RedisTemplate supports it - just like with the rest of the templates, one connection could be bound to the running thread and the RT will use it without creating new ones. 
　　回答中确实承认RedisTemplate 不支持连接绑定，但后半段说的是“在其它的模板类中，连接是可以绑定到当前线程，这样RedisTemplate 就会使用这个连接，而不会重新创建了”。
　　OK！ 下面分析源码，看一下这个绑定是怎么实现的。
　　org.springframework.data.redis.support.collections.AbstractRedisCollection 是 spring-data-redis 对redis的几个集合类型的数据结构封装类的抽象类
　　

　　其中有 rename 这个函数

public void rename(final String newKey) {
CollectionUtils.rename(key, newKey, operations);
key = newKey;
}
　　但它的实现并没有使用内部持有的RedisOperations，而是CollectionUtils.rename，那我们就来看一个这个CollectionUtils.rename
　　这个rename是通过事务来保证进行rename操作时，原key一定存在

static <K> void rename(final K key, final K newKey, RedisOperations<K, ?> operations) {
operations.execute(new SessionCallback<Object>() {
@SuppressWarnings("unchecked")
public Object execute(RedisOperations operations) throws DataAccessException {
do {
operations.watch(key);
if (operations.hasKey(key)) {
operations.multi();
operations.rename(key, newKey);
}
else {
operations.multi();
}
} while (operations.exec() == null);
return null;
}
});
}
　　这是一个静态函数，对Redis的操作是通过传入的RedisOperations来完成的
　　rename调用了RedisOperations的execute，传入了一个匿名的回调接口，实现具体操作，这和前面说的execute模板看起来是一样的。但这里的RedisOperations为什么可以使用watch 和 multi 能？ 前面不是说 RedisTemplate 不能直接watch 和 multi 吗？  (RedisOperations是RedisTemplate的接口，针对不同数据结构的操作类和集合都是从RedisTemplate生成的，也就是说都是通过RedisTemplate操作Redis的)
　　其实这个execute模板和前面的是不一样的，这里的参数是SessionCallback 而前面的是RedisCallback，从名字应该能看出，这个里的 execute 调用是会保持会话，也就是连接。
　　看一下这里的execute源码

public <T> T execute(SessionCallback<T> session) {
RedisConnectionFactory factory = getConnectionFactory();
// bind connection
RedisConnectionUtils.bindConnection(factory);
try {
return session.execute(this);
} finally {
RedisConnectionUtils.unbindConnection(factory);
}
}
 　　不用多解释了吧，在调用 SessionCallback 的实现进行具体操作前后，对连接进行了绑定和解绑。
　　然后在session.execute中，会调用operations.watch(key); 等着操作，这些操作和前面分析的RedisTemplate中的hasKey的流程是一样的，会调用前面分析的execute进行操作。
　　回到前面看下源码，那里的RedisConnection 是通过RedisConnectionUtils取到的

RedisConnection conn = RedisConnectionUtils.getConnection(factory);
　　连接绑定和解绑也是通过 RedisConnectionUtils 完成的，里面是通过TransactionSynchronizationManager将连接绑定到当前线程的，和spring的DB事务管理是一样的，这里就不详细分析了(主要是使用ThreadLocal)。
　　这样一分析，spring-data-redis 是可以使用watch 和 multi 的，关键是怎么使用的问题。 
　　总结：
　　

RedisTemplate和其它特定类型的操作类，主要是实现了基本的操作功能，也有部分高级功能（如上面的rename和

RedisAtomicInteger等高级操作）。
　　通过 SessionCallback 是可以获取到绑定连接的操作类的，它上面的操作都是在一个连接上的，这样可以实现高级功能。
　　在spring-data-redis中使用watch和multi，可以参照源码中的CollectionUtils.rename
　　spring-data-redis也确实有些问题，参看这篇博文 http://ldd600.iyunv.com/blog/1115196
　　我在项目中目前还是直接使用Jedis，它的源码写得很好 http://jimgreat.iyunv.com/blog/1586671