add redis util in lettuce sample sync io

ipipman · ipipman · commit c7fe8b728392 · 2021-01-20T20:35:19.000+08:00
diff --git a/springboot-lettuce-sample/src/main/java/com/ipman/springboot/lettuce/sample/config/LettucePoolConfig.java b/springboot-lettuce-sample/src/main/java/com/ipman/springboot/lettuce/sample/config/LettucePoolConfig.java
@@ -3,6 +3,7 @@
 import io.lettuce.core.RedisClient;
 import io.lettuce.core.api.StatefulRedisConnection;
 import io.lettuce.core.support.ConnectionPoolSupport;
+import lombok.Getter;
 import org.apache.commons.pool2.impl.GenericObjectPool;
 import org.apache.commons.pool2.impl.GenericObjectPoolConfig;
 import org.springframework.beans.factory.annotation.Value;
@@ -45,11 +46,12 @@ public class LettucePoolConfig {
     //Apache-Common-Pool是一个对象池，用于缓存Redis连接，
     //因为Letture本身基于Netty的异步驱动，但基于Servlet模型的同步访问时，连接池是必要的
     //连接池可以很好的复用连接，减少重复的IO消耗与RedisURI创建实例的性能消耗
+    @Getter
     GenericObjectPool<StatefulRedisConnection<String, String>> redisConnectionPool;
 
     //Servlet初始化时先初始化Lettuce连接池
     @PostConstruct
-    public void init() {
+    private void init() {
         GenericObjectPoolConfig<StatefulRedisConnection<String, String>> redisPoolConfig
                 = new GenericObjectPoolConfig<>();
         redisPoolConfig.setMaxIdle(this.maxIdle);
@@ -62,7 +64,7 @@ public void init() {
 
     //Servlet销毁时先销毁Lettuce连接池
     @PreDestroy
-    public void destroy() {
+    private void destroy() {
         redisConnectionPool.close();
         redisClient.shutdown();
     }
diff --git a/springboot-lettuce-sample/src/main/java/com/ipman/springboot/lettuce/sample/utils/LettuceUtil.java b/springboot-lettuce-sample/src/main/java/com/ipman/springboot/lettuce/sample/utils/LettuceUtil.java
@@ -0,0 +1,52 @@
+package com.ipman.springboot.lettuce.sample.utils;
+
+import com.ipman.springboot.lettuce.sample.config.LettucePoolConfig;
+import io.lettuce.core.api.StatefulRedisConnection;
+import io.lettuce.core.api.sync.RedisCommands;
+import lombok.extern.slf4j.Slf4j;
+import org.springframework.beans.factory.annotation.Autowired;
+import org.springframework.stereotype.Component;
+
+/**
+ * Created by ipipman on 2021/1/20.
+ *
+ * @version V1.0
+ * @Package com.ipman.springboot.lettuce.sample.utils
+ * @Description: (用一句话描述该文件做什么)
+ * @date 2021/1/20 7:04 下午
+ */
+@Component
+@Slf4j
+public class LettuceUtil {
+
+    @Autowired
+    LettucePoolConfig lettucePoolConfig;
+
+    //编写executeSync方法，在方法中，获取Redis连接，利用Callback操作Redis，最后释放连接，并返回结果
+    //这里使用的同步的方式执行cmd指令
+    public <T> T executeSync(SyncCommandCallback<T> callback) {
+        //这里利用try的语法糖，执行完，自动给释放连接
+        try (StatefulRedisConnection<String, String> connection = lettucePoolConfig.getRedisConnectionPool().borrowObject()) {
+            //开启自动提交，如果false，命令会被缓冲，调用flushCommand()方法发出
+            connection.setAutoFlushCommands(true);
+            //设置为同步模式
+            RedisCommands<String, String> commands = connection.sync();
+            //执行传入的实现类
+            return callback.doInConnection(commands);
+        } catch (Exception e) {
+            log.error(e.getMessage());
+            throw new RuntimeException(e);
+        }
+    }
+
+    //分装一个set方法
+    public String set(final String key, final String val) {
+        return executeSync(commands -> commands.set(key, val));
+    }
+
+    //分装一个get方法
+    public String get(final String key) {
+        return executeSync(commands -> commands.get(key));
+    }
+
+}
diff --git a/springboot-lettuce-sample/src/main/java/com/ipman/springboot/lettuce/sample/utils/SyncCommandCallback.java b/springboot-lettuce-sample/src/main/java/com/ipman/springboot/lettuce/sample/utils/SyncCommandCallback.java
@@ -0,0 +1,18 @@
+package com.ipman.springboot.lettuce.sample.utils;
+
+import io.lettuce.core.api.sync.RedisCommands;
+
+/**
+ * Created by ipipman on 2021/1/20.
+ *
+ * @version V1.0
+ * @Package com.ipman.springboot.lettuce.sample.utils
+ * @Description: (用一句话描述该文件做什么)
+ * @date 2021/1/20 7:02 下午
+ */
+@FunctionalInterface
+public interface SyncCommandCallback<T> {
+
+    //抽象方法，为了简化代码，便于传入回调函数
+    T doInConnection(RedisCommands<String, String> commands);
+}
