redis中使用pipeline批量执行命令,提升性能

发布于:2024-04-30 ⋅ 阅读:(36) ⋅ 点赞:(0)

注意:此操作非原子性 
       将一批要执行的redis命令提交到pipeline中,pipeline一次性的将数据发送给服务器,服务器再逐条执行命令。

        redisTemplate中已经提供了对应方法executePipelined()可以直接调用,它支持两个类型的参数:RedisCallback更接近redis原生命令,但是需要自己将键和值都转换为字节码传递过去;SessionCallback对操作进行了封装,可以根据操作不同的数据类型进行转换,方便api使用

代码示例

import lombok.extern.slf4j.Slf4j;
import org.example.service_a.service_a_App;
import org.junit.jupiter.api.Test;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.boot.test.context.SpringBootTest;
import org.springframework.dao.DataAccessException;
import org.springframework.data.redis.connection.RedisConnection;
import org.springframework.data.redis.core.*;
import java.nio.charset.StandardCharsets;
import java.util.List;

@SpringBootTest(classes={service_a_App.class})
@Slf4j
public class Test_Pipelined {

    @Autowired
    private StringRedisTemplate redisTemplate;


    @Test
    void executePipelined_RedisCallback() {

        List<Object> datas = redisTemplate.executePipelined(new RedisCallback<Object>() {
            @Override
            public Object doInRedis(RedisConnection connection) throws DataAccessException {
                connection.set("key1".getBytes(StandardCharsets.UTF_8), "value1".getBytes(StandardCharsets.UTF_8));
                connection.set("key2".getBytes(StandardCharsets.UTF_8), "value2".getBytes(StandardCharsets.UTF_8));
                connection.set("key3".getBytes(StandardCharsets.UTF_8), "value3".getBytes(StandardCharsets.UTF_8));
                connection.set("key4".getBytes(StandardCharsets.UTF_8), "value4".getBytes(StandardCharsets.UTF_8));
                connection.set("key5".getBytes(StandardCharsets.UTF_8), "value5".getBytes(StandardCharsets.UTF_8));
                connection.set("key6".getBytes(StandardCharsets.UTF_8), "value6".getBytes(StandardCharsets.UTF_8));
                connection.get("key1".getBytes(StandardCharsets.UTF_8));

                // 这里必须返回null,在 connection.closePipeline() 时覆盖原来的返回值,所以返回值没有必要设置,设置会报错
                return null;
            }
        });

        System.out.println("datas = " + datas);
    }


    @Test
    void executePipelined_SessionCallback() {
        List<Object> datas = redisTemplate.executePipelined(new SessionCallback<Object>() {
            @Override
            public <K, V> Object execute(RedisOperations<K, V> operations) throws DataAccessException {

                ValueOperations<String, String> op1 = (ValueOperations<String, String>) operations.opsForValue();
                op1.set("key7", "value7");
                op1.set("key8", "value8");
                op1.get("key2");

                SetOperations<String, String> op2 = (SetOperations<String, String>) operations.opsForSet();
                op2.add("set_demo", "value1", "value2", "value3");
                op2.randomMember("set_demo");

                return null;
            }
        });
        System.out.println("datas = " + datas);
    }
}


网站公告

今日签到

点亮在社区的每一天
去签到