添加redis序列化方式：Protostuff和Kryo两种方式

LinkinStars · LinkinStars · commit 965a1e051eb5 · 2018-08-16T17:21:43.000+08:00
diff --git a/build.gradle b/build.gradle
@@ -40,8 +40,10 @@ dependencies {
 	compile('org.springframework.boot:spring-boot-starter-aop')
 	compile group: 'org.springframework', name: 'spring-tx', version: '4.3.12.RELEASE'
 	compile group: 'org.springframework', name: 'spring-jdbc', version: '4.3.12.RELEASE'
-	compile("org.springframework.boot:spring-boot-devtools")
-	testCompile('org.springframework.boot:spring-boot-starter-test')
+    testCompile('org.springframework.boot:spring-boot-starter-test')
+
+    //  热部署相关
+    //	compile("org.springframework.boot:spring-boot-devtools")
 
 	//数据库相关
 	compile group: 'mysql', name: 'mysql-connector-java', version: '6.0.6'
@@ -64,6 +66,8 @@ dependencies {
 
 	//redis相关配置
 	compile group: 'org.springframework.session', name: 'spring-session-data-redis', version: '1.3.1.RELEASE'
+    //用于redis序列化用户
+    compile group: 'com.esotericsoftware', name: 'kryo', version: '4.0.0'
 
 	//让thymeleaf解析标签没有闭合时通过
 	compile group: 'net.sourceforge.nekohtml', name: 'nekohtml', version: '1.9.22'
@@ -73,4 +77,8 @@ dependencies {
 
     //log4jdbc
     compile group: 'org.bgee.log4jdbc-log4j2', name: 'log4jdbc-log4j2-jdbc4.1', version: '1.16'
+
+    // https://mvnrepository.com/artifact/io.protostuff/protostuff-core
+    compile group: 'io.protostuff', name: 'protostuff-runtime', version: '1.6.0'
+    compile group: 'io.protostuff', name: 'protostuff-core', version: '1.6.0'
 }
diff --git a/src/main/java/com/linkinstars/springBootTemplate/bean/UserEntity.java b/src/main/java/com/linkinstars/springBootTemplate/bean/UserEntity.java
@@ -25,4 +25,11 @@ public void setVal(String val) {
         this.val = val;
     }
 
+    @Override
+    public String toString() {
+        return "UserEntity{" +
+                "id=" + id +
+                ", val='" + val + '\'' +
+                '}';
+    }
 }
diff --git a/src/main/java/com/linkinstars/springBootTemplate/config/RedisConfig.java b/src/main/java/com/linkinstars/springBootTemplate/config/RedisConfig.java
@@ -0,0 +1,37 @@
+package com.linkinstars.springBootTemplate.config;
+
+import com.linkinstars.springBootTemplate.util.ProtostuffSerializer;
+import org.springframework.context.annotation.Bean;
+import org.springframework.context.annotation.Configuration;
+import org.springframework.data.redis.connection.RedisConnectionFactory;
+import org.springframework.data.redis.core.RedisTemplate;
+import org.springframework.data.redis.serializer.StringRedisSerializer;
+import org.springframework.session.data.redis.config.annotation.web.http.EnableRedisHttpSession;
+
+/**
+ * redisTemplate初始化,开启spring-session redis存储支持
+ * @author LinkinStar
+ */
+@Configuration
+@EnableRedisHttpSession
+public class RedisConfig {
+
+    /**
+     * redisTemplate 序列化使用的Serializeable, 存储二进制字节码, 所以自定义序列化类
+     * @Rparam redisConnectionFactory
+     * @return redisTemplate
+     */
+    @Bean
+    public RedisTemplate<Object, Object> protoStuffTemplate(RedisConnectionFactory redisConnectionFactory) {
+        RedisTemplate<Object, Object> template = new RedisTemplate<>();
+        template.setConnectionFactory(redisConnectionFactory);
+
+        // redis value使用的序列化器
+        template.setValueSerializer(new ProtostuffSerializer());
+        // redis key使用的序列化器
+        template.setKeySerializer(new StringRedisSerializer());
+
+        template.afterPropertiesSet();
+        return template;
+    }
+}
diff --git a/src/main/java/com/linkinstars/springBootTemplate/config/RedisSessionConfig.java b/src/main/java/com/linkinstars/springBootTemplate/config/RedisSessionConfig.java
diff --git a/src/main/java/com/linkinstars/springBootTemplate/controller/UserController.java b/src/main/java/com/linkinstars/springBootTemplate/controller/UserController.java
@@ -69,16 +69,17 @@ public String test(HttpServletRequest request, @RequestParam(required = false) I
 
 
         //测试redis
-        redisUtil.setString("xxx","xxx");
-        System.out.println("redis数据获取为: " + redisUtil.getString("xxx"));
-        redisUtil.delete("xxx");
-        System.out.println("redis数据获取为: " + redisUtil.getString("xxx"));
+        UserEntity user = new UserEntity();
+        user.setId(1);
+        user.setVal("xxx");
 
-        redisUtil.setHash("xxxx", "a", "1");
-        redisUtil.setHash("xxxx", "b", "2");
-        redisUtil.setHash("xxxx", "c", "3");
+        redisUtil.set("xxx", user);
+        Object object = redisUtil.get("xxx");
+        UserEntity userTemp = (UserEntity) object;
 
-        System.out.println("redis中hash的数据为： " + redisUtil.getHash("xxxx","a"));
+        System.out.println("redis数据获取为: " + userTemp);
+        redisUtil.delete("xxx");
+        System.out.println("redis删除数据之后获取为: " + redisUtil.get("xxx"));
 
 
         //测试事务回滚
diff --git a/src/main/java/com/linkinstars/springBootTemplate/util/KryoRedisSerializer.java b/src/main/java/com/linkinstars/springBootTemplate/util/KryoRedisSerializer.java
@@ -0,0 +1,72 @@
+package com.linkinstars.springBootTemplate.util;
+
+import com.esotericsoftware.kryo.Kryo;
+import com.esotericsoftware.kryo.io.Input;
+import com.esotericsoftware.kryo.io.Output;
+import org.slf4j.Logger;
+import org.slf4j.LoggerFactory;
+import org.springframework.data.redis.serializer.RedisSerializer;
+import org.springframework.data.redis.serializer.SerializationException;
+
+import java.io.ByteArrayOutputStream;
+
+/**
+ * Kryo序列化工具
+ * @param <T>
+ */
+public class KryoRedisSerializer<T> implements RedisSerializer<T> {
+
+    private static final Logger logger = LoggerFactory.getLogger(KryoRedisSerializer.class);
+
+    public static final byte[] EMPTY_BYTE_ARRAY = new byte[0];
+
+    private static final ThreadLocal<Kryo> KRYOS = ThreadLocal.withInitial(Kryo::new);
+
+    private Class<T> clazz;
+
+    public KryoRedisSerializer(Class<T> clazz) {
+        super();
+        this.clazz = clazz;
+    }
+
+    @Override
+    public byte[] serialize(T t) throws SerializationException {
+        if (t == null) {
+            return EMPTY_BYTE_ARRAY;
+        }
+
+        Kryo kryo = KRYOS.get();
+        kryo.setReferences(false);
+        kryo.register(clazz);
+
+        try (ByteArrayOutputStream baos = new ByteArrayOutputStream();
+             Output output = new Output(baos)) {
+            kryo.writeClassAndObject(output, t);
+            output.flush();
+            return baos.toByteArray();
+        } catch (Exception e) {
+            logger.error(e.getMessage(), e);
+        }
+
+        return EMPTY_BYTE_ARRAY;
+    }
+
+    @Override
+    public T deserialize(byte[] bytes) throws SerializationException {
+        if (bytes == null || bytes.length <= 0) {
+            return null;
+        }
+
+        Kryo kryo = KRYOS.get();
+        kryo.setReferences(false);
+        kryo.register(clazz);
+
+        try (Input input = new Input(bytes)) {
+            return (T) kryo.readClassAndObject(input);
+        } catch (Exception e) {
+            logger.error(e.getMessage(), e);
+        }
+
+        return null;
+    }
+}
diff --git a/src/main/java/com/linkinstars/springBootTemplate/util/ProtostuffSerializer.java b/src/main/java/com/linkinstars/springBootTemplate/util/ProtostuffSerializer.java
@@ -0,0 +1,58 @@
+package com.linkinstars.springBootTemplate.util;
+
+import io.protostuff.LinkedBuffer;
+import io.protostuff.ProtostuffIOUtil;
+import io.protostuff.Schema;
+import io.protostuff.runtime.RuntimeSchema;
+import org.springframework.data.redis.serializer.RedisSerializer;
+import org.springframework.data.redis.serializer.SerializationException;
+
+/**
+ * ProtoStuff序列化工具
+ * @author LinkinStar
+ */
+public class ProtostuffSerializer<T> implements RedisSerializer<T> {
+
+    private boolean isEmpty(byte[] data) {
+        return (data == null || data.length == 0);
+    }
+
+    private final Schema<ProtoWrapper> schema;
+
+    private final ProtoWrapper wrapper;
+
+    private final LinkedBuffer buffer;
+
+    public ProtostuffSerializer() {
+        this.wrapper = new ProtoWrapper();
+        this.schema = RuntimeSchema.getSchema(ProtoWrapper.class);
+        this.buffer = LinkedBuffer.allocate(LinkedBuffer.DEFAULT_BUFFER_SIZE);
+    }
+
+    @Override
+    public byte[] serialize(Object t) throws SerializationException {
+        if (t == null) {
+            return new byte[0];
+        }
+        wrapper.data = t;
+        try {
+            return ProtostuffIOUtil.toByteArray(wrapper, schema, buffer);
+        } finally {
+            buffer.clear();
+        }
+    }
+
+    @Override
+    public T deserialize(byte[] bytes) throws SerializationException {
+        if (isEmpty(bytes)) {
+            return null;
+        }
+        ProtoWrapper newMessage = schema.newMessage();
+        ProtostuffIOUtil.mergeFrom(bytes, newMessage, schema);
+        return (T) newMessage.data;
+    }
+
+    private static class ProtoWrapper {
+        private Object data;
+    }
+}
diff --git a/src/main/java/com/linkinstars/springBootTemplate/util/RedisUtil.java b/src/main/java/com/linkinstars/springBootTemplate/util/RedisUtil.java
@@ -1,27 +1,22 @@
 package com.linkinstars.springBootTemplate.util;
 
 import org.springframework.beans.factory.annotation.Autowired;
-import org.springframework.data.redis.core.StringRedisTemplate;
+import org.springframework.beans.factory.annotation.Qualifier;
+import org.springframework.data.redis.core.RedisTemplate;
+import org.springframework.stereotype.Component;
 
 import java.util.concurrent.TimeUnit;
 
 /**
  * Redis操作工具类
  * @author LinkinStar
  */
+@Component
 public class RedisUtil {
 
-    /**
-     * RedisTemplate 使用的是 JdkSerializationRedisSerializer
-     * StringRedisTemplate 使用的是 StringRedisSerializer
-     * 之后可以通过构建redisTemplate来替换序列化的方式
-     */
-//    @Autowired
-//    private RedisTemplate<String, Object> redisTemplate;
-
     @Autowired
-    private StringRedisTemplate redisTemplate;
-
+    @Qualifier("protoStuffTemplate")
+    private RedisTemplate protoStuffTemplate;
 
     /**
      * 设置过期时间，单位秒
@@ -30,15 +25,15 @@ public class RedisUtil {
      * @return 成功：true，失败：false
      */
     public boolean setExpireTime(String key, long timeout) {
-        return redisTemplate.expire(key, timeout, TimeUnit.SECONDS);
+        return protoStuffTemplate.expire(key, timeout, TimeUnit.SECONDS);
     }
 
     /**
      * 通过键删除一个值
      * @param key 键的名称
      */
     public void delete(String key) {
-        redisTemplate.delete(key);
+        protoStuffTemplate.delete(key);
     }
 
     /**
@@ -47,63 +42,34 @@ public void delete(String key) {
      * @return 存在：true，不存在：false
      */
     public boolean hasKey(String key) {
-        return redisTemplate.hasKey(key);
+        return protoStuffTemplate.hasKey(key);
     }
 
     /**
-     * 字符串存储
+     * 数据存储
      * @param key 键
      * @param value 值
      */
-    public void setString(String key, String value) {
-        redisTemplate.opsForValue().set(key, value);
+    public void set(String key, Object value) {
+        protoStuffTemplate.boundValueOps(key).set(value);
     }
 
     /**
-     * 字符串存储（同时设置超时时间）
+     * 数据存储的同时设置过期时间
      * @param key 键
      * @param value 值
-     * @param timeout 超时时间（以秒为单位）
+     * @param expireTime 过期时间
      */
-    public void setStringAndExpireTime(String key, String value, long timeout) {
-        redisTemplate.opsForValue().set(key, value, timeout, TimeUnit.SECONDS);
+    public void set(String key, Object value, Long expireTime) {
+        protoStuffTemplate.boundValueOps(key).set(value, expireTime, TimeUnit.SECONDS);
     }
 
     /**
-     * 字符串取值
+     * 数据取值
      * @param key 键
      * @return 查询成功：值，查询失败，null
      */
-    public String getString(String key) {
-        return (String) redisTemplate.opsForValue().get(key);
-    }
-
-    /**
-     * 存储哈希表
-     * @param key 整个哈希表的键
-     * @param field 表中的键
-     * @param value 表中的值
-     */
-    public void setHash(String key, String field, Object value) {
-        redisTemplate.opsForHash().put(key, field, value);
-    }
-
-    /**
-     * 获取哈希表
-     * @param key 整个哈希表的键
-     * @param field 表中的键
-     * @return 查询成功：值，查询失败，null
-     */
-    public Object getHash(String key, String field) {
-        return redisTemplate.opsForHash().get(key, field);
-    }
-
-    /**
-     * 删除哈希表中的某个元素
-     * @param key 整个哈希表的键
-     * @param fields 表中的键
-     */
-    public void deleteHash(String key, Object... fields) {
-        redisTemplate.opsForHash().delete(key, fields);
+    public Object get(String key) {
+        return protoStuffTemplate.boundValueOps(key).get();
     }
 }
diff --git a/src/main/resources/application.yml b/src/main/resources/application.yml

Original file line number	Diff line number	Diff line change
`@@ -25,4 +25,11 @@ public void setVal(String val) {`
`25`	`25`	`this.val = val;`
`26`	`26`	`}`
`27`	`27`
	`28`	`+ @Override`
	`29`	`+ public String toString() {`
	`30`	`+ return "UserEntity{" +`
	`31`	`+ "id=" + id +`
	`32`	`+ ", val='" + val + '\'' +`
	`33`	`+ '}';`
	`34`	`+ }`
`28`	`35`	`}`