Redis 客户端 Jedis 的特性和原理

Redis 作为目前通用的缓存选型,因其高性能而倍受欢迎。Redis 的 2.x 版本仅支持单机模式,从 3.0 版本开始引入集群模式。

Redis 的 Java 生态的客户端当中包含 Jedis、Redisson、Lettuce,不同的客户端具备不同的能力是使用方式,本文主要分析 Jedis 客户端。

Jedis 客户端同时支持单机模式、分片模式、集群模式的访问模式,通过构建 Jedis 类对象实现单机模式下的数据访问,通过构建 ShardedJedis 类对象实现分片模式的数据访问,通过构建 JedisCluster 类对象实现集群模式下的数据访问。

Jedis 客户端支持单命令和 Pipeline 方式访问 Redis 集群,通过 Pipeline 的方式能够提高集群访问的效率。

本文的整体分析基于 Jedis 的 3.5.0 版本进行分析,相关源码均参考此版本。

一、Jedis 访问模式对比

Jedis 客户端操作 Redis 主要分为三种模式,分表是单机模式、分片模式、集群模式。

  • 单机模式主要是创建 Jedis 对象来操作单节点的 Redis,只适用于访问单个 Redis 节点。
  • 分片模式(ShardedJedis)主要是通过创建 ShardedJedisPool 对象来访问分片模式的多个 Redis 节点,是 Redis 没有集群功能之前客户端实现的一个数据分布式方案,本质上是客户端通过一致性哈希来实现数据分布式存储。
  • 集群模式(JedisCluster)主要是通过创建 JedisCluster 对象来访问集群模式下的多个 Redis 节点,是 Redis3.0 引入集群模式后客户端实现的集群访问访问,本质上是通过引入槽(slot)概念以及通过 CRC16 哈希槽算法来实现数据分布式存储。

单机模式不涉及任何分片的思想,所以我们着重分析分片模式和集群模式的理念。

1.1 分片模式

  • 分片模式本质属于基于客户端的分片,在客户端实现如何根据一个 key 找到 Redis 集群中对应的节点的方案。
  • Jedis 的客户端分片模式采用一致性 Hash 来实现,一致性 Hash 算法的好处是当 Redis 节点进行增减时只会影响新增或删除节点前后的小部分数据,相对于取模等算法来说对数据的影响范围较小。
  • Redis 在大部分场景下作为缓存进行使用,所以不用考虑数据丢失致使缓存穿透造成的影响,在 Redis 节点增减时可以不用考虑部分数据无法命中的问题。

分片模式的整体应用如下图所示,核心在于客户端的一致性 Hash 策略。


1.2 集群模式

集群模式本质属于服务器分片技术,由 Redis 集群本身提供分片功能,从 Redis 3.0 版本开始正式提供。

集群的原理是:一个 Redis 集群包含 16384 个哈希槽(Hash slot), Redis 保存的每个键都属于这 16384 个哈希槽的其中一个, 集群使用公式 CRC16(key)%16384 来计算键 key 属于哪个槽, 其中 CRC16(key) 语句用于计算键 key 的 CRC16 校验和 。

集群中的每个节点负责处理一部分哈希槽。举个例子, 一个集群可以有三个哈希槽, 其中:

  • 节点 A 负责处理 0 号至 5500 号哈希槽。
  • 节点 B 负责处理 5501 号至 11000 号哈希槽。
  • 节点 C 负责处理 11001 号至 16383 号哈希槽。

Redis 在集群模式下对于 key 的读写过程首先将对应的 key 值进行 CRC16 计算得到对应的哈希值,将哈希值对槽位总数取模映射到对应的槽位,最终映射到对应的节点进行读写。以命令 set("key", "value")为例子,它会使用 CRC16 算法对 key 进行计算得到哈希值 28989,然后对 16384 进行取模得到 12605,最后找到 12605 对应的 Redis 节点,最终跳转到该节点执行 set 命令。

集群模式的整体应用如下图所示,核心在于集群哈希槽的设计以及重定向命令。

二、Jedis 的基础用法

// Jedis单机模式的访问
public void main(String[] args) {
    // 创建Jedis对象
    jedis = new Jedis("localhost", 6379);
    // 执行hmget操作
    jedis.hmget("foobar", "foo");
    // 关闭Jedis对象
    jedis.close();
}
 
// Jedis分片模式的访问
public void main(String[] args) {
    HostAndPort redis1 = HostAndPortUtil.getRedisServers().get(0);
    HostAndPort redis2 = HostAndPortUtil.getRedisServers().get(1);
    List shards = new ArrayList(2);
    JedisShardInfo shard1 = new JedisShardInfo(redis1);
    JedisShardInfo shard2 = new JedisShardInfo(redis2);
    // 创建ShardedJedis对象
    ShardedJedis shardedJedis = new ShardedJedis(shards);
    // 通过ShardedJedis对象执行set操作
    shardedJedis.set("a", "bar");
}
 
// Jedis集群模式的访问
public void main(String[] args) {
    // 构建redis的集群池
    Set nodes = new HashSet<>();
    nodes.add(new HostAndPort("127.0.0.1", 7001));
    nodes.add(new HostAndPort("127.0.0.1", 7002));
    nodes.add(new HostAndPort("127.0.0.1", 7003));
 
    // 创建JedisCluster
    JedisCluster cluster = new JedisCluster(nodes);
 
    // 执行JedisCluster对象中的方法
    cluster.set("cluster-test", "my jedis cluster test");
    String result = cluster.get("cluster-test");
}

Jedis 通过创建 Jedis 的类对象来实现单机模式下的数据访问,通过构建 JedisCluster 类对象来实现集群模式下的数据访问。

要理解 Jedis 的访问 Redis 的整个过程,可以通过先理解单机模式下的访问流程,在这个基础上再分析集群模式的访问流程会比较合适。

三、Jedis 单机模式的访问

Jedis 访问单机模式 Redis 的整体流程图如下所示,从图中可以看出核心的流程包含 Jedis 对象的创建以及通过 Jedis 对象实现 Redis 的访问。

熟悉 Jedis 访问单机 Redis 的过程,本身就是需要了解 Jedis 的创建过程以及执行 Redis 命令的过程。

  • Jedis 的创建过程核心在于创建 Jedis 对象以及 Jedis 内部变量 Client 对象。
  • Jedis 访问 Redis 的过程在于通过 Jedis 内部的 Client 对象访问 Redis。


3.1 创建过程

Jedis 本身的类关系图如下图所示,从图中我们能够看到 Jedis 继承自 BinaryJedis 类。

在 BinaryJedis 类中存在和 Redis 对接的 Client 类对象,Jedis 通过父类的 BinaryJedis 的 Client 对象实现 Redis 的读写。



Jedis 类在创建过程中通过父类 BinaryJedis 创建了 Client 对象,而了解 Client 对象是进一步理解访问过程的关键。

public class Jedis extends BinaryJedis implements JedisCommands, MultiKeyCommands,
    AdvancedJedisCommands, ScriptingCommands, BasicCommands, ClusterCommands, SentinelCommands, ModuleCommands {
 
  protected JedisPoolAbstract dataSource = null;
 
  public Jedis(final String host, final int port) {
    // 创建父类BinaryJedis对象
    super(host, port);
  }
}
 
public class BinaryJedis implements BasicCommands, BinaryJedisCommands, MultiKeyBinaryCommands,
    AdvancedBinaryJedisCommands, BinaryScriptingCommands, Closeable {
 
  // 访问redis的Client对象
  protected Client client = null;
 
  public BinaryJedis(final String host, final int port) {
    // 创建Client对象访问redis
    client = new Client(host, port);
  }
}

Client 类的类关系图如下图所示,Client 对象继承自 BinaryClient 和 Connection 类。在 BinaryClient 类中存在 Redis 访问密码等相关参数,在 Connection 类在存在访问 Redis 的 socket 对象以及对应的输入输出流。本质上 Connection 是和 Redis 进行通信的核心类。



Client 类在创建过程中初始化核心父类 Connection 对象,而 Connection 是负责和 Redis 直接进行通信。

public class Client extends BinaryClient implements Commands {
  public Client(final String host, final int port) {
    super(host, port);
  }
}
 
public class BinaryClient extends Connection {
  // 存储和Redis连接的相关信息
  private boolean isInMulti;
  private String user;
  private String password;
  private int db;
  private boolean isInWatch;
 
  public BinaryClient(final String host, final int port) {
    super(host, port);
  }
}
 
public class Connection implements Closeable {
  // 管理和Redis连接的socket信息及对应的输入输出流
  private JedisSocketFactory jedisSocketFactory;
  private Socket socket;
  private RedisOutputStream outputStream;
  private RedisInputStream inputStream;
  private int infiniteSoTimeout = 0;
  private boolean broken = false;
 
  public Connection(final String host, final int port, final boolean ssl,
      SSLSocketFactory sslSocketFactory, SSLParameters sslParameters,
      HostnameVerifier hostnameVerifier) {
    // 构建DefaultJedisSocketFactory来创建和Redis连接的Socket对象
    this(new DefaultJedisSocketFactory(host, port, Protocol.DEFAULT_TIMEOUT,
        Protocol.DEFAULT_TIMEOUT, ssl, sslSocketFactory, sslParameters, hostnameVerifier));
  }
}

3.2 访问过程

以 Jedis 执行 set 命令为例,整个过程如下:

  • Jedis 的 set 操作是通过 Client 的 set 操作来实现的。
  • Client 的 set 操作是通过父类 Connection 的 sendCommand 来实现。
public class Jedis extends BinaryJedis implements JedisCommands, MultiKeyCommands,
    AdvancedJedisCommands, ScriptingCommands, BasicCommands, ClusterCommands, SentinelCommands, ModuleCommands {
  @Override
  public String set(final String key, final String value) {
    checkIsInMultiOrPipeline();
    // client执行set操作
    client.set(key, value);
    return client.getStatusCodeReply();
  }
}
 
public class Client extends BinaryClient implements Commands {
  @Override
  public void set(final String key, final String value) {
    // 执行set命令
    set(SafeEncoder.encode(key), SafeEncoder.encode(value));
  }
}
 
public class BinaryClient extends Connection {
  public void set(final byte[] key, final byte[] value) {
    // 发送set指令
    sendCommand(SET, key, value);
  }
}
 
public class Connection implements Closeable {
  public void sendCommand(final ProtocolCommand cmd, final byte[]... args) {
    try {
      // socket连接redis
      connect();
      // 按照redis的协议发送命令
      Protocol.sendCommand(outputStream, cmd, args);
    } catch (JedisConnectionException ex) {
    }
  }
}

四、Jedis 分片模式的访问

基于前面已经介绍的 Redis 分片模式的一致性 Hash 的原理来理解 Jedis 的分片模式的访问。

关于 Redis 分片模式的概念:Redis 在 3.0 版本之前没有集群模式的概念,这导致单节点能够存储的数据有限,通过 Redis 的客户端如 Jedis 在客户端通过一致性 Hash 算法来实现数据的分片存储。

本质上 Redis 的分片模式跟 Redis 本身没有任何关系,只是通过客户端来解决单节点数据有限存储的问题。

ShardedJedis 访问 Redis 的核心在于构建对象的时候初始化一致性 Hash 对象,构建一致性 Hash 经典的 Hash 值和 node 的映射关系。构建完映射关系后执行 set 等操作就是 Hash 值到 node 的寻址过程,寻址完成后直接进行单节点的操作。

4.1 创建过程

ShardedJedis 的创建过程在于父类的 Sharded 中关于一致性 Hash 相关的初始化过程,核心在于构建一致性的虚拟节点以及虚拟节点和 Redis 节点的映射关系。

源码中最核心的部分代码在于根据根据权重映射成未 160 个虚拟节点,通过虚拟节点来定位到具体的 Redis 节点。

public class Sharded> {
 
  public static final int DEFAULT_WEIGHT = 1;
  // 保存虚拟节点和redis的node节点的映射关系
  private TreeMap nodes;
  // hash算法
  private final Hashing algo;
  // 保存redis节点和访问该节点的Jedis的连接信息
  private final Map, R> resources = new LinkedHashMap<>();
 
  public Sharded(List shards, Hashing algo) {
    this.algo = algo;
    initialize(shards);
  }
 
  private void initialize(List shards) {
    nodes = new TreeMap<>();
    // 遍历每个redis的节点并设置hash值到节点的映射关系
    for (int i = 0; i != shards.size(); ++i) {
      final S shardInfo = shards.get(i);
      // 根据权重映射成未160个虚拟节点
      int N =  160 * shardInfo.getWeight();
      if (shardInfo.getName() == null) for (int n = 0; n < N; n++) {
        // 构建hash值和节点映射关系
        nodes.put(this.algo.hash("SHARD-" + i + "-NODE-" + n), shardInfo);
      }
      else for (int n = 0; n < N; n++) {
        nodes.put(this.algo.hash(shardInfo.getName() + "*" + n), shardInfo);
      }
      // 保存每个节点的访问对象
      resources.put(shardInfo, shardInfo.createResource());
    }
  }
}

4.2 访问过程

ShardedJedis 的访问过程就是一致性 Hash 的计算过程,核心的逻辑就是:通过 Hash 算法对访问的 key 进行 Hash 计算生成 Hash 值,根据 Hash 值获取对应 Redis 节点,根据对应的 Redis 节点获取对应的访问对象 Jedis。

获取访问对象 Jedis 之后就可以直接进行命令操作。

public class Sharded> {
 
  public static final int DEFAULT_WEIGHT = 1;
  private TreeMap nodes;
  private final Hashing algo;
  // 保存redis节点和访问该节点的Jedis的连接信息
  private final Map, R> resources = new LinkedHashMap<>();
 
  public R getShard(String key) {
    // 根据redis节点找到对应的访问对象Jedis
    return resources.get(getShardInfo(key));
  }
 
  public S getShardInfo(String key) {
    return getShardInfo(SafeEncoder.encode(getKeyTag(key)));
  }
 
  public S getShardInfo(byte[] key) {
    // 针对访问的key生成对应的hash值
    // 根据hash值找到对应的redis节点
    SortedMap tail = nodes.tailMap(algo.hash(key));
    if (tail.isEmpty()) {
      return nodes.get(nodes.firstKey());
    }
    return tail.get(tail.firstKey());
  }
}

五、Jedis 集群模式的访问

基于前面介绍的 Redis 的集群原理来理解 Jedis 的集群模式的访问。

Jedis 能够实现 key 和哈希槽的定位的核心机制在于哈希槽和 Redis 节点的映射,而这个发现过程基于 Redis 的 cluster slot 命令。

关于 Redis 集群操作的命令:Redis 通过 cluster slots 会返回 Redis 集群的整体状况。返回每一个 Redis 节点的信息包含:

  • 哈希槽起始编号
  • 哈希槽结束编号
  • 哈希槽对应 master 节点,节点使用 IP/Port 表示
  • master 节点的第一个副本
  • master 节点的第二个副本
127.0.0.1:30001> cluster slots
1) 1) (integer) 0 // 开始槽位
   2) (integer) 5460 // 结束槽位
   3) 1) "127.0.0.1" // master节点的host
      2) (integer) 30001 // master节点的port
      3) "09dbe9720cda62f7865eabc5fd8857c5d2678366" // 节点的编码
   4) 1) "127.0.0.1" // slave节点的host
      2) (integer) 30004 // slave节点的port
      3) "821d8ca00d7ccf931ed3ffc7e3db0599d2271abf" // 节点的编码
2) 1) (integer) 5461
   2) (integer) 10922
   3) 1) "127.0.0.1"
      2) (integer) 30002
      3) "c9d93d9f2c0c524ff34cc11838c2003d8c29e013"
   4) 1) "127.0.0.1"
      2) (integer) 30005
      3) "faadb3eb99009de4ab72ad6b6ed87634c7ee410f"
3) 1) (integer) 10923
   2) (integer) 16383
   3) 1) "127.0.0.1"
      2) (integer) 30003
      3) "044ec91f325b7595e76dbcb18cc688b6a5b434a1"
   4) 1) "127.0.0.1"
      2) (integer) 30006
      3) "58e6e48d41228013e5d9c1c37c5060693925e97e"

Jedis 访问集群模式 Redis 的整体流程图如下所示,从图中可以看出核心的流程包含 JedisCluster 对象的创建以及通过 JedisCluster 对象实现 Redis 的访问。

JedisCluster 对象的创建核心在于创建 JedisClusterInfoCache 对象并通过集群发现来建立 slot 和集群节点的映射关系。

JedisCluster 对 Redis 集群的访问在于获取 key 所在的 Redis 节点并通过 Jedis 对象进行访问。

5.1 创建过程

JedisCluster 的类关系如下图所示,在图中可以看到核心变量 JedisSlotBasedConnectionHandler 对象。


JedisCluster 的父类 BinaryJedisCluster 创建了 JedisSlotBasedConnectionHandler 对象,该对象负责和 Redis 的集群进行通信。

public class JedisCluster extends BinaryJedisCluster implements JedisClusterCommands,
    MultiKeyJedisClusterCommands, JedisClusterScriptingCommands {
  public JedisCluster(Set jedisClusterNode, int connectionTimeout, int soTimeout,
      int maxAttempts, String password, String clientName, final GenericObjectPoolConfig poolConfig,
      boolean ssl, SSLSocketFactory sslSocketFactory, SSLParameters sslParameters,
      HostnameVerifier hostnameVerifier, JedisClusterHostAndPortMap hostAndPortMap) {
 
    // 访问父类BinaryJedisCluster
    super(jedisClusterNode, connectionTimeout, soTimeout, maxAttempts, password, clientName, poolConfig,
        ssl, sslSocketFactory, sslParameters, hostnameVerifier, hostAndPortMap);
  }
}
 
public class BinaryJedisCluster implements BinaryJedisClusterCommands,
    MultiKeyBinaryJedisClusterCommands, JedisClusterBinaryScriptingCommands, Closeable {
  public BinaryJedisCluster(Set jedisClusterNode, int connectionTimeout, int soTimeout,
      int maxAttempts, String user, String password, String clientName, GenericObjectPoolConfig poolConfig,
      boolean ssl, SSLSocketFactory sslSocketFactory, SSLParameters sslParameters,
      HostnameVerifier hostnameVerifier, JedisClusterHostAndPortMap hostAndPortMap) {
 
    // 创建JedisSlotBasedConnectionHandler对象
    this.connectionHandler = new JedisSlotBasedConnectionHandler(jedisClusterNode, poolConfig,
        connectionTimeout, soTimeout, user, password, clientName, ssl, sslSocketFactory, sslParameters, hostnameVerifier, hostAndPortMap);
 
    this.maxAttempts = maxAttempts;
  }
}

JedisSlotBasedConnectionHandler 的核心在于创建并初始化 JedisClusterInfoCache 对象,该对象缓存了 Redis 集群的信息。

JedisClusterInfoCache 对象的初始化过程通过 initializeSlotsCache 来完成,主要目的用于实现集群节点和槽位发现。

public class JedisSlotBasedConnectionHandler extends JedisClusterConnectionHandler {
  public JedisSlotBasedConnectionHandler(Set nodes, GenericObjectPoolConfig poolConfig,
      int connectionTimeout, int soTimeout, String user, String password, String clientName,
      boolean ssl, SSLSocketFactory sslSocketFactory, SSLParameters sslParameters,
      HostnameVerifier hostnameVerifier, JedisClusterHostAndPortMap portMap) {
 
    super(nodes, poolConfig, connectionTimeout, soTimeout, user, password, clientName,
        ssl, sslSocketFactory, sslParameters, hostnameVerifier, portMap);
  }
}
 
public abstract class JedisClusterConnectionHandler implements Closeable {
  public JedisClusterConnectionHandler(Set nodes, final GenericObjectPoolConfig poolConfig,
      int connectionTimeout, int soTimeout, int infiniteSoTimeout, String user, String password, String clientName,
      boolean ssl, SSLSocketFactory sslSocketFactory, SSLParameters sslParameters,
      HostnameVerifier hostnameVerifier, JedisClusterHostAndPortMap portMap) {
 
    // 创建JedisClusterInfoCache对象
    this.cache = new JedisClusterInfoCache(poolConfig, connectionTimeout, soTimeout, infiniteSoTimeout,
        user, password, clientName, ssl, sslSocketFactory, sslParameters, hostnameVerifier, portMap);
 
    // 初始化jedis的Slot信息
    initializeSlotsCache(nodes, connectionTimeout, soTimeout, infiniteSoTimeout,
        user, password, clientName, ssl, sslSocketFactory, sslParameters, hostnameVerifier);
  }
 
 
  private void initializeSlotsCache(Set startNodes,
      int connectionTimeout, int soTimeout, int infiniteSoTimeout, String user, String password, String clientName,
      boolean ssl, SSLSocketFactory sslSocketFactory, SSLParameters sslParameters, HostnameVerifier hostnameVerifier) {
    for (HostAndPort hostAndPort : startNodes) {
 
      try (Jedis jedis = new Jedis(hostAndPort.getHost(), hostAndPort.getPort(), connectionTimeout,
          soTimeout, infiniteSoTimeout, ssl, sslSocketFactory, sslParameters, hostnameVerifier)) {
 
        // 通过discoverClusterNodesAndSlots进行集群发现
        cache.discoverClusterNodesAndSlots(jedis);
        return;
      } catch (JedisConnectionException e) {
      }
    }
  }
}

JedisClusterInfoCache 的 nodes 用来保存 Redis 集群的节点信息,slots 用来保存槽位和集群节点的信息。

nodes 和 slots 维持的对象都是 JedisPool 对象,该对象维持了和 Redis 的连接信息。集群的发现过程由 discoverClusterNodesAndSlots 来实现,本质是执行 Redis 的集群发现命令 cluster slots 实现的。

public class JedisClusterInfoCache {
  // 负责保存redis集群的节点信息
  private final Map nodes = new HashMap<>();
  // 负责保存redis的槽位和redis节点的映射关系
  private final Map slots = new HashMap<>();
 
  // 负责集群的发现逻辑
  public void discoverClusterNodesAndSlots(Jedis jedis) {
    w.lock();
 
    try {
      reset();
      List slots = jedis.clusterSlots();
 
      for (Object slotInfoObj : slots) {
        List slotInfo = (List) slotInfoObj;
 
        if (slotInfo.size() <= MASTER_NODE_INDEX) {
          continue;
        }
        // 获取redis节点对应的槽位信息
        List slotNums = getAssignedSlotArray(slotInfo);
 
        // hostInfos
        int size = slotInfo.size();
        for (int i = MASTER_NODE_INDEX; i < size; i++) {
          List hostInfos = (List) slotInfo.get(i);
          if (hostInfos.isEmpty()) {
            continue;
          }
 
          HostAndPort targetNode = generateHostAndPort(hostInfos);
          // 负责保存redis节点信息
          setupNodeIfNotExist(targetNode);
          if (i == MASTER_NODE_INDEX) {
            // 负责保存槽位和redis节点的映射关系
            assignSlotsToNode(slotNums, targetNode);
          }
        }
      }
    } finally {
      w.unlock();
    }
  }
 
  public void assignSlotsToNode(List targetSlots, HostAndPort targetNode) {
    w.lock();
    try {
      JedisPool targetPool = setupNodeIfNotExist(targetNode);
      // 保存槽位和对应的JedisPool对象
      for (Integer slot : targetSlots) {
        slots.put(slot, targetPool);
      }
    } finally {
      w.unlock();
    }
  }
 
  public JedisPool setupNodeIfNotExist(HostAndPort node) {
    w.lock();
    try {
      // 生产redis节点对应的nodeKey
      String nodeKey = getNodeKey(node);
      JedisPool existingPool = nodes.get(nodeKey);
      if (existingPool != null) return existingPool;
      // 生产redis节点对应的JedisPool
      JedisPool nodePool = new JedisPool(poolConfig, node.getHost(), node.getPort(),
          connectionTimeout, soTimeout, infiniteSoTimeout, user, password, 0, clientName,
          ssl, sslSocketFactory, sslParameters, hostnameVerifier);
      // 保存redis节点的key和对应的JedisPool对象
      nodes.put(nodeKey, nodePool);
      return nodePool;
    } finally {
      w.unlock();
    }
  }
}
 
 

JedisPool 的类关系如下图所示,其中内部 internalPool 是通过 apache common pool 来实现的池化。



JedisPool 内部的 internalPool 通过 JedisFactory 的 makeObject 来创建 Jedis 对象。

每个 Redis 节点都会对应一个 JedisPool 对象,通过 JedisPool 来管理 Jedis 的申请释放复用等。

public class JedisPool extends JedisPoolAbstract {
 
  public JedisPool() {
    this(Protocol.DEFAULT_HOST, Protocol.DEFAULT_PORT);
  }
}
 
public class JedisPoolAbstract extends Pool {
 
  public JedisPoolAbstract() {
    super();
  }
}
 
public abstract class Pool implements Closeable {
  protected GenericObjectPool internalPool;
 
  public void initPool(final GenericObjectPoolConfig poolConfig, PooledObjectFactory factory) {
    if (this.internalPool != null) {
      try {
        closeInternalPool();
      } catch (Exception e) {
      }
    }
    this.internalPool = new GenericObjectPool<>(factory, poolConfig);
  }
}
 
class JedisFactory implements PooledObjectFactory {
   
  @Override
  public PooledObject makeObject() throws Exception {
    // 创建Jedis对象
    final HostAndPort hp = this.hostAndPort.get();
    final Jedis jedis = new Jedis(hp.getHost(), hp.getPort(), connectionTimeout, soTimeout,
        infiniteSoTimeout, ssl, sslSocketFactory, sslParameters, hostnameVerifier);
 
    try {
      // Jedis对象连接
      jedis.connect();
      if (user != null) {
        jedis.auth(user, password);
      } else if (password != null) {
        jedis.auth(password);
      }
      if (database != 0) {
        jedis.select(database);
      }
      if (clientName != null) {
        jedis.clientSetname(clientName);
      }
    } catch (JedisException je) {
      jedis.close();
      throw je;
    }
    // 将Jedis对象包装成DefaultPooledObject进行返回
    return new DefaultPooledObject<>(jedis);
  }
}

5.1 访问过程

JedisCluster 访问 Redis 的过程通过 JedisClusterCommand 来实现重试机制,最终通过 Jedis 对象来实现访问。从实现的角度来说 JedisCluster 是在 Jedis 之上封装了一层,进行集群节点定位以及重试机制等。

以 set 命令为例,整个访问通过 JedisClusterCommand 实现如下:

  • 计算 key 所在的 Redis 节点。
  • 获取 Redis 节点对应的 Jedis 对象。
  • 通过 Jedis 对象进行 set 操作。
public class JedisCluster extends BinaryJedisCluster implements JedisClusterCommands,
    MultiKeyJedisClusterCommands, JedisClusterScriptingCommands {
 
  @Override
  public String set(final String key, final String value, final SetParams params) {
    return new JedisClusterCommand(connectionHandler, maxAttempts) {
      @Override
      public String execute(Jedis connection) {
        return connection.set(key, value, params);
      }
    }.run(key);
  }
}

JedisClusterCommand 的 run 方法核心主要定位 Redis 的 key 所在的 Redis 节点,然后获取与该节点对应的 Jedis 对象进行访问。

在 Jedis 对象访问异常后,JedisClusterCommand 会进行重试操作并按照一定策略执行 renewSlotCache 方法进行重集群节点重发现动作。

public abstract class JedisClusterCommand {
  public T run(String key) {
    // 针对key进行槽位的计算
    return runWithRetries(JedisClusterCRC16.getSlot(key), this.maxAttempts, false, null);
  }
   
  private T runWithRetries(final int slot, int attempts, boolean tryRandomNode, JedisRedirectionException redirect) {
 
    Jedis connection = null;
    try {
 
      if (redirect != null) {
        connection = this.connectionHandler.getConnectionFromNode(redirect.getTargetNode());
        if (redirect instanceof JedisAskDataException) {
          connection.asking();
        }
      } else {
        if (tryRandomNode) {
          connection = connectionHandler.getConnection();
        } else {
          // 根据slot去获取Jedis对象
          connection = connectionHandler.getConnectionFromSlot(slot);
        }
      }
      // 执行真正的Redis的命令
      return execute(connection);
    } catch (JedisNoReachableClusterNodeException jnrcne) {
      throw jnrcne;
    } catch (JedisConnectionException jce) {
 
      releaseConnection(connection);
      connection = null;
 
      if (attempts <= 1) {
        // 保证最后两次机会去重新刷新槽位和节点的对应的信息
        this.connectionHandler.renewSlotCache();
      }
      // 按照重试次数进行重试操作
      return runWithRetries(slot, attempts - 1, tryRandomNode, redirect);
    } catch (JedisRedirectionException jre) {
      // 针对返回Move命令立即触发重新刷新槽位和节点的对应信息
      if (jre instanceof JedisMovedDataException) {
        // it rebuilds cluster's slot cache recommended by Redis cluster specification
        this.connectionHandler.renewSlotCache(connection);
      }
 
      releaseConnection(connection);
      connection = null;
 
      return runWithRetries(slot, attempts - 1, false, jre);
    } finally {
      releaseConnection(connection);
    }
  }
}

JedisSlotBasedConnectionHandler 的 cache 对象维持了 slot 和 node 的映射关系,通过 getConnectionFromSlot 方法来获取该 slot 对应的 Jedis 对象。

public class JedisSlotBasedConnectionHandler extends JedisClusterConnectionHandler {
 
  protected final JedisClusterInfoCache cache;
 
  @Override
  public Jedis getConnectionFromSlot(int slot) {
    // 获取槽位对应的JedisPool对象
    JedisPool connectionPool = cache.getSlotPool(slot);
    if (connectionPool != null) {
      // 从JedisPool对象中获取Jedis对象
      return connectionPool.getResource();
    } else {
      // 获取失败就重新刷新槽位信息
      renewSlotCache();
      connectionPool = cache.getSlotPool(slot);
      if (connectionPool != null) {
        return connectionPool.getResource();
      } else {
        //no choice, fallback to new connection to random node
        return getConnection();
      }
    }
  }
}

六、 Jedis 的 Pipeline 实现

Pipeline 的技术核心思想是将多个命令发送到服务器而不用等待回复,最后在一个步骤中读取该答复。这种模式的好处在于节省了请求响应这种模式的网络开销。

Redis 的普通命令如 set 和 Pipeline 批量操作的核心的差别在于 set 命令的操作会直接发送请求到 Redis 并同步等待结果返回,而 Pipeline 的操作会发送请求但不立即同步等待结果返回,具体的实现可以从 Jedis 的源码一探究竟。

原生的 Pipeline 在集群模式下相关的 key 必须 Hash 到同一个节点才能生效,原因在于 Pipeline 下的 Client 对象只能其中的一个节点建立了连接。

在集群模式下归属于不同节点的 key 能够使用 Pipeline 就需要针对每个 key 保存对应的节点的 client 对象,在最后执行获取数据的时候一并获取。本质上可以认为在单节点的 Pipeline 的基础上封装成一个集群式的 Pipeline。

6.1 Pipeline 用法分析

Pipeline 访问单节点的 Redis 的时候,通过 Jedis 对象的 Pipeline 方法返回 Pipeline 对象,其他的命令操作通过该 Pipeline 对象进行访问。

Pipeline 从使用角度来分析,会批量发送多个命令并最后统一使用 syncAndReturnAll 来一次性返回结果。

public void pipeline() {
    jedis = new Jedis(hnp.getHost(), hnp.getPort(), 500);
    Pipeline p = jedis.pipelined();
    // 批量发送命令到redis
    p.set("foo", "bar");
    p.get("foo");
    // 同步等待响应结果
    List results = p.syncAndReturnAll();
 
    assertEquals(2, results.size());
    assertEquals("OK", results.get(0));
    assertEquals("bar", results.get(1));
 }
 
 
public abstract class PipelineBase extends Queable implements BinaryRedisPipeline, RedisPipeline {
 
  @Override
  public Response set(final String key, final String value) {
    // 发送命令
    getClient(key).set(key, value);
    // pipeline的getResponse只是把待响应的请求聚合到pipelinedResponses对象当中
    return getResponse(BuilderFactory.STRING);
  }
}
 
 
public class Queable {
 
  private Queue> pipelinedResponses = new LinkedList<>();
  protected  Response getResponse(Builder builder) {
    Response lr = new Response<>(builder);
    // 统一保存到响应队列当中
    pipelinedResponses.add(lr);
    return lr;
  }
}
 
 
public class Pipeline extends MultiKeyPipelineBase implements Closeable {
 
  public List syncAndReturnAll() {
    if (getPipelinedResponseLength() > 0) {
      // 根据批量发送命令的个数即需要批量返回命令的个数,通过client对象进行批量读取
      List unformatted = client.getMany(getPipelinedResponseLength());
      List formatted = new ArrayList<>();
      for (Object o : unformatted) {
        try {
          // 格式化每个返回的结果并最终保存在列表中进行返回
          formatted.add(generateResponse(o).get());
        } catch (JedisDataException e) {
          formatted.add(e);
        }
      }
      return formatted;
    } else {
      return java.util.Collections. emptyList();
    }
  }
}
 
 

普通 set 命令发送请求给 Redis 后立即通过 getStatusCodeReply 来获取响应结果,所以这是一种请求响应的模式。

getStatusCodeReply 在获取响应结果的时候会通过 flush()命令强制发送报文到 Redis 服务端然后通过读取响应结果。

public class BinaryJedis implements BasicCommands, BinaryJedisCommands, MultiKeyBinaryCommands,
    AdvancedBinaryJedisCommands, BinaryScriptingCommands, Closeable {
 
  @Override
  public String set(final byte[] key, final byte[] value) {
    checkIsInMultiOrPipeline();
    // 发送命令
    client.set(key, value);
    // 等待请求响应
    return client.getStatusCodeReply();
  }
}
 
 
public class Connection implements Closeable {
  public String getStatusCodeReply() {
    // 通过flush立即发送请求
    flush();
    // 处理响应请求
    final byte[] resp = (byte[]) readProtocolWithCheckingBroken();
    if (null == resp) {
      return null;
    } else {
      return SafeEncoder.encode(resp);
    }
  }
}
 
 
public class Connection implements Closeable {
  protected void flush() {
    try {
      // 针对输出流进行flush操作保证报文的发出
      outputStream.flush();
    } catch (IOException ex) {
      broken = true;
      throw new JedisConnectionException(ex);
    }
  }
}

你可能感兴趣的:(Redis 客户端 Jedis 的特性和原理)