feat(backend): add redis interface for registering docId -> broadcast server route

Author: BrotherJing

scalable-ot-core/src/main/java/com/brotherjing/Const.java

@@ -9,6 +9,12 @@ public interface Const {
 
     String REVISION_CONSUMER_GROUP_ID = "revision_consumer";
 
+    /**
+     * Redis
+     */
+    String ROUTE_PREFIX = "route:";
+    int ROUTE_EXPIRE_MINUTES = 30;
+
     /**
      * Zookeeper
      */

scalable-ot-core/src/main/java/com/brotherjing/config/RedisConfig.java

@@ -4,7 +4,12 @@
 import org.springframework.context.annotation.Configuration;
 import org.springframework.data.redis.connection.jedis.JedisConnectionFactory;
 import org.springframework.data.redis.core.RedisTemplate;
+import org.springframework.data.redis.core.ValueOperations;
 import org.springframework.data.redis.repository.configuration.EnableRedisRepositories;
+import org.springframework.data.redis.serializer.Jackson2JsonRedisSerializer;
+import org.springframework.data.redis.serializer.StringRedisSerializer;
+
+import com.brotherjing.core.loadbalance.ServerEntity;
 
 @Configuration
 @EnableRedisRepositories
@@ -15,9 +20,19 @@ JedisConnectionFactory jedisConnectionFactory() {
     }
 
     @Bean
-    public RedisTemplate<String, Object> redisTemplate() {
-        RedisTemplate<String, Object> template = new RedisTemplate<>();
+    public RedisTemplate<String, ServerEntity> routeTemplate() {
+        RedisTemplate<String, ServerEntity> template = new RedisTemplate<>();
         template.setConnectionFactory(jedisConnectionFactory());
+
+        template.setKeySerializer(new StringRedisSerializer());
+        template.setValueSerializer(new Jackson2JsonRedisSerializer<>(ServerEntity.class));
+        template.afterPropertiesSet();
+
         return template;
     }
+
+    @Bean
+    public ValueOperations<String, ServerEntity> routeValueOperations() {
+        return routeTemplate().opsForValue();
+    }
 }

scalable-ot-core/src/main/java/com/brotherjing/core/dao/RedisDao.java

@@ -0,0 +1,37 @@
+package com.brotherjing.core.dao;
+
+import java.util.concurrent.TimeUnit;
+
+import org.springframework.beans.factory.annotation.Autowired;
+import org.springframework.data.redis.core.RedisTemplate;
+import org.springframework.data.redis.core.ValueOperations;
+import org.springframework.stereotype.Repository;
+
+import com.brotherjing.Const;
+import com.brotherjing.core.loadbalance.ServerEntity;
+
+@Repository
+public class RedisDao {
+
+    @Autowired
+    private RedisTemplate<String, ServerEntity> routeTemplate;
+
+    @Autowired
+    private ValueOperations<String, ServerEntity> routeValueOperations;
+
+    public void addRoute(String docId, ServerEntity server) {
+        routeValueOperations.set(getRouteKey(docId), server, Const.ROUTE_EXPIRE_MINUTES, TimeUnit.MINUTES);
+    }
+
+    public ServerEntity getRoute(String docId) {
+        return routeValueOperations.get(docId);
+    }
+
+    public void removeRoute(String docId) {
+        routeTemplate.delete(getRouteKey(docId));
+    }
+
+    private String getRouteKey(String docId) {
+        return Const.ROUTE_PREFIX + docId;
+    }
+}