// 創(chuàng)建 Jedis 實(shí)例
Jedis jedis = new Jedis("localhost", 6379);
 
// 使用 pipelining 方式批量執(zhí)行命令
Pipeline pipeline = jedis.pipelined();
 
// 批量操作：使用 pipeline 來(lái)緩存命令
for (int i = 0; i < 1000; i++) {
    pipeline.set("key" + i, "value" + i);
}
 
// 同步執(zhí)行所有命令
pipeline.sync();

public class JedisClusterTest {
 
    // 聲明一個(gè)JedisCluster對(duì)象，用于與Redis集群進(jìn)行交互
    private JedisCluster jedisCluster;
 
    // 在每個(gè)測(cè)試方法執(zhí)行之前，初始化JedisCluster連接
    @BeforeEach
    void setUp() {
        // 配置Jedis連接池
        JedisPoolConfig poolConfig = new JedisPoolConfig();
        poolConfig.setMaxTotal(8);
        poolConfig.setMaxIdle(8);
        poolConfig.setMinIdle(0);
        poolConfig.setMaxWaitMillis(1000);
 
        // 創(chuàng)建一個(gè)HashSet，用于存儲(chǔ)Redis集群的節(jié)點(diǎn)信息
        HashSet<HostAndPort> nodes = new HashSet<>();
        // 添加Redis集群的節(jié)點(diǎn)信息（IP和端口）
        nodes.add(new HostAndPort("192.168.150.101", 7001));
        nodes.add(new HostAndPort("192.168.150.101", 7002));
        nodes.add(new HostAndPort("192.168.150.101", 7003));
        nodes.add(new HostAndPort("192.168.150.101", 8001));
        nodes.add(new HostAndPort("192.168.150.101", 8002));
        nodes.add(new HostAndPort("192.168.150.101", 8003));
 
        // 使用配置的連接池和節(jié)點(diǎn)信息初始化JedisCluster對(duì)象
        jedisCluster = new JedisCluster(nodes, poolConfig);
    }
 
    // 測(cè)試方法：使用mset命令一次性設(shè)置多個(gè)鍵值對(duì)
    @Test
    void testMSet() {
        // 使用JedisCluster的mset方法，一次性設(shè)置多個(gè)鍵值對(duì)
        // 但是jedisCluster默認(rèn)是無(wú)法解決批處理問(wèn)題的，需要我們手動(dòng)解決
        jedisCluster.mset("name", "Jack", "age", "21", "sex", "male");
    }
 
    // 測(cè)試方法：使用mset命令按槽位分組設(shè)置多個(gè)鍵值對(duì)
    @Test
    void testMSet2() {
        // 創(chuàng)建一個(gè)HashMap，用于存儲(chǔ)多個(gè)鍵值對(duì)
        Map<String, String> map = new HashMap<>(3);
        map.put("name", "Jack");
        map.put("age", "21");
        map.put("sex", "Male");
 
        // 將map中的鍵值對(duì)按照Redis集群的槽位進(jìn)行分組
        Map<Integer, List<Map.Entry<String, String>>> result = map.entrySet()
                .stream()
                .collect(Collectors.groupingBy(
                        // 使用ClusterSlotHashUtil計(jì)算每個(gè)鍵對(duì)應(yīng)的槽位
                        entry -> ClusterSlotHashUtil.calculateSlot(entry.getKey()))
                );
 
        // 遍歷按哈希槽分組后的結(jié)果
        for (List<Map.Entry<String, String>> list : result.values()) {
            // 創(chuàng)建一個(gè)數(shù)組用于批量設(shè)置Redis的鍵值對(duì)
            String[] arr = new String[list.size() * 2];  // 每個(gè)鍵值對(duì)包含兩個(gè)元素
            int j = 0;  // 索引變量，用于在數(shù)組中定位位置
            for (int i = 0; i < list.size(); i++) {
                j = i << 1;  // 通過(guò)位移計(jì)算數(shù)組中的位置
                Map.Entry<String, String> e = list.get(i);  // 獲取當(dāng)前的鍵值對(duì)
                arr[j] = e.getKey();  // 將鍵放入數(shù)組中
                arr[j + 1] = e.getValue();  // 將值放入數(shù)組中
            }
            // 批量設(shè)置Redis集群中的鍵值對(duì)
            jedisCluster.mset(arr);
        }
    }
 
    // 在每個(gè)測(cè)試方法執(zhí)行之后，關(guān)閉JedisCluster連接
    @AfterEach
    void tearDown() {
        // 如果JedisCluster對(duì)象不為空，則關(guān)閉連接
        if (jedisCluster != null) {
            jedisCluster.close();
        }
    }
}

@Service
public class RedisService {
 
    @Autowired
    private RedisTemplate<String, Object> redisTemplate;
 
    /**
     * 跨槽位批量獲取多個(gè)鍵的值
     */
    public Map<String, Object> batchGetCrossSlot(List<String> keys) {
        // 按槽位分組
        Map<Integer, List<String>> slotKeyMap = keys.stream()
                .collect(Collectors.groupingBy(ClusterSlotHashUtil::calculateSlot));
 
        // 存儲(chǔ)最終結(jié)果
        Map<String, Object> result = new HashMap<>();
 
        // 對(duì)每個(gè)槽位的鍵分別調(diào)用multiGet
        for (Map.Entry<Integer, List<String>> entry : slotKeyMap.entrySet()) {
            List<String> slotKeys = entry.getValue();
            List<Object> slotValues = redisTemplate.opsForValue().multiGet(slotKeys);
 
            // 將結(jié)果存入Map
            for (int i = 0; i < slotKeys.size(); i++) {
                result.put(slotKeys.get(i), slotValues.get(i));
            }
        }
 
        return result;
    }
 
    /**
     * 測(cè)試跨槽位批量獲取方法
     */
    public void testBatchGetCrossSlot() {
        List<String> keys = Arrays.asList("name", "age", "sex");
        Map<String, Object> values = batchGetCrossSlot(keys);
 
        // 打印結(jié)果
        values.forEach((key, value) -> {
            System.out.println("Key: " + key + ", Value: " + value);
        });
    }
}