update: 토큰 저장소 Redis로 변경 (sharetreats-team#43)

Author: tlqkrus012345

src/main/java/com/sharetreats/chatbot/infra/config/RedisConfig.java

@@ -3,9 +3,11 @@
 import org.springframework.beans.factory.annotation.Value;
 import org.springframework.context.annotation.Bean;
 import org.springframework.context.annotation.Configuration;
+import org.springframework.context.annotation.Primary;
 import org.springframework.data.redis.connection.lettuce.LettuceConnectionFactory;
 import org.springframework.data.redis.core.RedisTemplate;
 import org.springframework.data.redis.serializer.GenericJackson2JsonRedisSerializer;
+import org.springframework.data.redis.serializer.StringRedisSerializer;
 
 @Configuration
 public class RedisConfig {
@@ -16,15 +18,36 @@ public class RedisConfig {
     @Value("${spring.redis.gift-history.port}")
     private int redisPort;
 
+    @Value("${spring.redis.token.host}")
+    private String tokenRedisHost;
+
+    @Value("${spring.redis.token.port}")
+    private int tokenRedisPort;
+
     @Bean
+    @Primary
     public LettuceConnectionFactory connectionFactory() {
         return new LettuceConnectionFactory(redisHost, redisPort);
     }
+
+    @Bean
+    public LettuceConnectionFactory tokenConnectionFactory() {
+        return new LettuceConnectionFactory(tokenRedisHost, tokenRedisPort);
+    }
     @Bean
     public RedisTemplate<String, Object> redisTemplate() {
         RedisTemplate<String, Object> redisTemplate = new RedisTemplate<>();
         redisTemplate.setDefaultSerializer(new GenericJackson2JsonRedisSerializer());
         redisTemplate.setConnectionFactory(connectionFactory());
         return redisTemplate;
     }
+
+    @Bean
+    @Primary
+    public RedisTemplate<String, String> tokenRedisTemplate() {
+        RedisTemplate<String, String> redisTemplate = new RedisTemplate<>();
+        redisTemplate.setDefaultSerializer(new StringRedisSerializer());
+        redisTemplate.setConnectionFactory(tokenConnectionFactory());
+        return redisTemplate;
+    }
 }

src/main/java/com/sharetreats/chatbot/infra/config/TokenConfig.java

@@ -1,38 +1,41 @@
 package com.sharetreats.chatbot.infra.config;
 
 import lombok.Getter;
+import lombok.RequiredArgsConstructor;
+import lombok.extern.slf4j.Slf4j;
+import org.springframework.data.redis.core.RedisTemplate;
 import org.springframework.stereotype.Component;
 
+import java.time.Duration;
 import java.util.UUID;
 import java.util.concurrent.ConcurrentHashMap;
 
+@Slf4j
 @Component
+@RequiredArgsConstructor
 public class TokenConfig {
-    private static final long TOKEN_EXPIRE_TIME = 30 * 60 * 1000;
-    private static final ConcurrentHashMap<String, TokenInfo> tokenStore = new ConcurrentHashMap<>();
+
+    private final RedisTemplate<String, String> redisTemplate;
     public void generateToken(String userId) {
+
         String token = UUID.randomUUID().toString();
-        long expireTime = System.currentTimeMillis() + TOKEN_EXPIRE_TIME;
-        TokenInfo tokenInfo = new TokenInfo(token, expireTime);
-        tokenStore.put(userId, tokenInfo);
-    }
+        Duration expireDuration = Duration.ofMinutes(20);
+        redisTemplate.opsForValue().set(userId, token, expireDuration);
 
+    }
     public boolean validateToken(String userId) {
-        TokenInfo tokenInfo = tokenStore.get(userId);
-        if (tokenInfo != null && tokenInfo.getExpireTime() >= System.currentTimeMillis()) {
-            return true;
-        } else {
-            tokenStore.remove(userId);
+
+        try {
+            String token = redisTemplate.opsForValue().get(userId);
+            if (token != null) {
+                Long expireTime = redisTemplate.getExpire(userId);
+                if (expireTime != null && expireTime > 0) {
+                    return true;
+                }
+            }
+        } catch (IllegalArgumentException e) {
             return false;
         }
-    }
-    @Getter
-    private class TokenInfo {
-        private String token;
-        private long expireTime;
-        public TokenInfo(String token, long expireTime) {
-            this.token = token;
-            this.expireTime = expireTime;
-        }
+        return false;
     }
 }