#3 增加自定义序列化功能

Author: zhou-hao

.gitignore

@@ -25,3 +25,4 @@ hs_err_pid*
 /upload
 /ui/upload/
 build/
+!/lib/*.jar

.travis.yml

@@ -7,7 +7,7 @@ service:
 before_script:
     - sudo redis-server /etc/redis/redis.conf --port 6379
 script:
-  - mvn test
+  - mvn -Dcodec.lib.dir=${PWD}/lib/ test
 after_success:
   - bash <(curl -s https://codecov.io/bash)
 cache:

lib/redis-mananger-beans-1.0-SNAPSHOT.jar

@@ -203,14 +203,9 @@
                 <executions>
                     <execution>
                         <goals>
-                            <goal>addSources</goal>
                             <goal>addTestSources</goal>
-                            <goal>generateStubs</goal>
                             <goal>compile</goal>
-                            <goal>generateTestStubs</goal>
                             <goal>compileTests</goal>
-                            <goal>removeStubs</goal>
-                            <goal>removeTestStubs</goal>
                         </goals>
                     </execution>
                 </executions>

pom.xml

@@ -16,6 +16,26 @@
             <groupId>org.redisson</groupId>
             <artifactId>redisson</artifactId>
         </dependency>
+
+        <dependency>
+            <groupId>net.openhft</groupId>
+            <artifactId>zero-allocation-hashing</artifactId>
+            <version>0.8</version>
+        </dependency>
+
+        <dependency>
+            <groupId>de.ruedigermoeller</groupId>
+            <artifactId>fst</artifactId>
+            <version>2.55</version>
+        </dependency>
+
+        <!-- https://mvnrepository.com/artifact/com.esotericsoftware/kryo -->
+        <dependency>
+            <groupId>com.esotericsoftware</groupId>
+            <artifactId>kryo</artifactId>
+            <version>4.0.2</version>
+        </dependency>
+
         <dependency>
             <groupId>io.vavr</groupId>
             <artifactId>vavr</artifactId>

redis-manager-core/pom.xml

@@ -1,10 +1,34 @@
 package org.hswebframework.redis.manager.codec;
 
+import io.vavr.API;
+import io.vavr.CheckedFunction1;
+
+import java.io.File;
 import java.net.URL;
 import java.net.URLClassLoader;
+import java.util.stream.Stream;
 
 public class CodecClassLoader extends URLClassLoader {
 
+    public CodecClassLoader(File... urls) {
+        super(Stream.of(urls)
+                .flatMap(API.unchecked(file -> {
+                    if (file.isDirectory()) {
+                        File[] children = file.listFiles();
+                        return children == null ? Stream.empty() : Stream.of(children);
+                    }
+                    return Stream.of(file);
+                }))
+                .map(API.unchecked(file -> file.toURI().toURL()))
+                .toArray(URL[]::new));
+    }
+
+    public CodecClassLoader(String... urls) {
+        super(Stream.of(urls)
+                .map(API.<String, URL>unchecked(URL::new))
+                .toArray(URL[]::new));
+    }
+
     public CodecClassLoader(URL... urls) {
         super(urls);
     }

redis-manager-core/src/main/java/org/hswebframework/redis/manager/codec/CodecClassLoader.java

@@ -7,30 +7,27 @@
 import org.redisson.codec.SerializationCodec;
 
 public enum CodecType {
-    string(){
+    string() {
         @Override
         public Codec createCodec(ClassLoader classLoader) {
             return StringCodec.INSTANCE;
         }
     },
-    jdk(){
+    jdk() {
         @Override
         public Codec createCodec(ClassLoader classLoader) {
             return new SerializationCodec(classLoader);
         }
     },
-    fst(){
+    fst() {
         @Override
         public Codec createCodec(ClassLoader classLoader) {
-
-            return null;//new FstCodec(classLoader);
+            return new FstCodec(classLoader);
         }
     },
-    kryo(){
+    kryo() {
         @Override
         public Codec createCodec(ClassLoader classLoader) {
-
-
             return new KryoCodec(classLoader);
         }
     };

redis-manager-core/src/main/java/org/hswebframework/redis/manager/codec/CodecType.java

@@ -0,0 +1,21 @@
+package org.hswebframework.redis.manager.codec
+
+import spock.lang.Specification
+
+/**
+ * @author zhouhao
+ * @since 1.0.0
+ */
+class CodecClassLoaderTest extends Specification {
+
+    def "测试自定义类加载"() {
+        given:
+        def classLoader = new CodecClassLoader(new File(System.getProperty("codec.lib.dir", "./lib")))
+
+        and:
+        def testBean = classLoader.loadClass("org.hswebframework.redis.manager.beans.TestBean");
+
+        expect:
+        testBean != null
+    }
+}

redis-manager-core/src/test/groovy/org/hswebframework/redis/manager/codec/CodecClassLoaderTest.groovy

@@ -0,0 +1,43 @@
+package org.hswebframework.redis.manager.codec
+
+import io.netty.buffer.ByteBuf
+import org.redisson.client.codec.Codec
+import org.redisson.client.handler.State
+import spock.lang.Shared
+import spock.lang.Specification
+
+/**
+ * @author zhouhao
+ * @since 1.0.0
+ */
+class CodecTypeTest extends Specification {
+
+    @Shared
+    def string = CodecType.string.createCodec(this.class.getClassLoader());
+    @Shared
+    def jdk = CodecType.jdk.createCodec(this.class.getClassLoader());
+    @Shared
+    def fst = CodecType.fst.createCodec(this.class.getClassLoader());
+    @Shared
+    def kryo = CodecType.kryo.createCodec(this.class.getClassLoader());
+
+    @Shared
+    def bean = new TestClass(id: "test", name: "测试", age: 18, enabled: true);
+
+    def doEncodeDecode(Codec c, Object o) {
+        ByteBuf buf = c.getValueEncoder().encode(o)
+        def decode = c.getValueDecoder().decode(buf, new State(false))
+        return decode == o
+    }
+
+    def "测试序列化"() {
+        given: "准备执行"
+        doEncodeDecode(codec, data) == success
+        where: "序列化,反序列成功"
+        codec  | data | success
+        string | bean | true
+        fst    | bean | true
+        jdk    | bean | true
+        kryo   | bean | true
+    }
+}

redis-manager-core/src/test/groovy/org/hswebframework/redis/manager/codec/CodecTypeTest.groovy

@@ -0,0 +1,24 @@
+package org.hswebframework.redis.manager.codec;
+
+import lombok.EqualsAndHashCode;
+import lombok.Getter;
+import lombok.Setter;
+
+import java.io.Serializable;
+
+/**
+ * @author zhouhao
+ * @since 1.0
+ */
+@EqualsAndHashCode
+@Getter
+@Setter
+public class TestClass implements Serializable {
+    private String id;
+
+    private String name;
+
+    private int age;
+
+    private boolean enabled;
+}