Feature: 사용자 정보를 레디스 대기열에 보관할 수 있다. (#58)

build.gradle

@@ -32,6 +32,7 @@ ext {
 
 dependencies {
     implementation 'org.springframework.boot:spring-boot-starter-data-jpa'
+    implementation 'org.springframework.boot:spring-boot-starter-data-redis'
     implementation 'org.springframework.boot:spring-boot-starter-validation'
     implementation 'org.springframework.boot:spring-boot-starter-web'
     implementation 'org.springframework.boot:spring-boot-starter-security'
@@ -55,8 +56,11 @@ dependencies {
     testImplementation 'org.springframework.security:spring-security-test'
     testRuntimeOnly 'org.junit.platform:junit-platform-launcher'
 
+    //test
     testImplementation 'org.springframework.restdocs:spring-restdocs-mockmvc'
     asciidoctorExt 'org.springframework.restdocs:spring-restdocs-asciidoctor'
+    testImplementation "org.testcontainers:testcontainers:1.20.1"
+    testImplementation 'org.testcontainers:junit-jupiter:1.20.1'
 }
 
 spotless {

src/main/java/com/thirdparty/ticketing/domain/common/ErrorCode.java

@@ -45,7 +45,13 @@ public enum ErrorCode {
     /*
        Payment Error
     */
-    PAYMENT_FAILED(HttpStatus.INTERNAL_SERVER_ERROR, "P500-1", "결제에 실패했습니다.");
+    PAYMENT_FAILED(HttpStatus.INTERNAL_SERVER_ERROR, "P500-1", "결제에 실패했습니다."),
+
+    /*
+       Waiting Error
+    */
+    WAITING_WRITE_ERROR(HttpStatus.INTERNAL_SERVER_ERROR, "W500-1", "대기열 쓰기에 실패했습니다."),
+    WAITING_READ_ERROR(HttpStatus.INTERNAL_SERVER_ERROR, "W500-2", "대기열 읽기에 실패했습니다.");
 
     ErrorCode(HttpStatus httpStatus, String code, String message) {
         this.httpStatus = httpStatus;

src/main/java/com/thirdparty/ticketing/domain/waiting/WaitingAspect.java

@@ -20,12 +20,14 @@ public class WaitingAspect {
     private final WaitingManager waitingManager;
 
     private Object waitingRequest(ProceedingJoinPoint joinPoint) throws Throwable {
-        Authentication authentication = SecurityContextHolder.getContext().getAuthentication();
-        String email = (String) authentication.getPrincipal();
         HttpServletRequest request =
-                ((ServletRequestAttributes) RequestContextHolder.getRequestAttributes())
+                ((ServletRequestAttributes) RequestContextHolder.currentRequestAttributes())
                         .getRequest();
         Long performanceId = Long.valueOf(request.getHeader("performanceId"));
+
+        Authentication authentication = SecurityContextHolder.getContext().getAuthentication();
+        String email = (String) authentication.getPrincipal();
+
         WaitingMember waitingMember = new WaitingMember(email, performanceId);
         if (waitingManager.isReadyToHandle(waitingMember)) {
             return joinPoint.proceed();

src/main/java/com/thirdparty/ticketing/domain/waiting/WaitingMember.java

@@ -3,13 +3,28 @@
 import java.time.ZonedDateTime;
 
 import lombok.Data;
-import lombok.RequiredArgsConstructor;
+import lombok.NoArgsConstructor;
 
 @Data
-@RequiredArgsConstructor
+@NoArgsConstructor
 public class WaitingMember {
-    private final String email;
-    private final Long performanceId;
-    private long waitingCounter;
+    private String email;
+    private long performanceId;
+    private long waitingCount;
     private ZonedDateTime enteredAt;
+
+    public WaitingMember(String email, String performanceId) {
+        this.email = email;
+        this.performanceId = Long.parseLong(performanceId);
+    }
+
+    public WaitingMember(String email, Long performanceId) {
+        this.email = email;
+        this.performanceId = performanceId;
+    }
+
+    public void updateWaitingInfo(long waitingCount, ZonedDateTime enteredAt) {
+        this.waitingCount = waitingCount;
+        this.enteredAt = enteredAt;
+    }
 }

src/main/java/com/thirdparty/ticketing/domain/waiting/manager/DefaultWaitingManager.java

@@ -29,7 +29,7 @@ public void moveWaitingMemberToRunningRoom(long performanceId, long count) {
         List<WaitingMember> waitingMembers = waitingRoom.pollWaitingMembers(performanceId, count);
         long maxCount = 0L;
         for (WaitingMember waitingMember : waitingMembers) {
-            maxCount = Math.max(maxCount, waitingMember.getWaitingCounter());
+            maxCount = Math.max(maxCount, waitingMember.getWaitingCount());
         }
         map.put(performanceId, maxCount);
         runningRoom.put(performanceId, waitingMembers);

src/main/java/com/thirdparty/ticketing/domain/waiting/room/DefaultWaitingCounter.java

@@ -4,12 +4,15 @@
 import java.util.Map;
 import java.util.concurrent.atomic.AtomicLong;
 
+import com.thirdparty.ticketing.domain.waiting.WaitingMember;
+
 public class DefaultWaitingCounter implements WaitingCounter {
 
     private final Map<Long, AtomicLong> map = new HashMap<>();
 
     @Override
-    public long getNextCount(Long performanceId) {
+    public long getNextCount(WaitingMember waitingMember) {
+        long performanceId = waitingMember.getPerformanceId();
         if (!map.containsKey(performanceId)) {
             map.put(performanceId, new AtomicLong());
         }

src/main/java/com/thirdparty/ticketing/domain/waiting/room/DefaultWaitingRoom.java

@@ -32,10 +32,10 @@ public synchronized long enter(WaitingMember waitingMember) {
             map.put(performanceId, new ConcurrentHashMap<>());
         }
         if (map.get(performanceId).containsKey(email)) {
-            return waitingMember.getWaitingCounter();
+            return waitingMember.getWaitingCount();
         }
-        long counter = waitingCounter.getNextCount(performanceId);
-        waitingMember.setWaitingCounter(counter);
+        long counter = waitingCounter.getNextCount(waitingMember);
+        waitingMember.setWaitingCount(counter);
         map.get(performanceId).put(email, waitingMember);
         waitingLine.enter(waitingMember);
         return counter;

src/main/java/com/thirdparty/ticketing/domain/waiting/room/WaitingCounter.java

@@ -1,9 +1,11 @@
 package com.thirdparty.ticketing.domain.waiting.room;
 
+import com.thirdparty.ticketing.domain.waiting.WaitingMember;
+
 public interface WaitingCounter {
 
     /**
      * @return 사용자에게 부여되는 고유한 카운트를 반환한다.
      */
-    long getNextCount(Long performanceId);
+    long getNextCount(WaitingMember waitingMember);
 }

src/main/java/com/thirdparty/ticketing/global/config/LettuceConfig.java

@@ -18,7 +18,7 @@ public class LettuceConfig {
 
     @Bean
     public LettuceConnectionFactory lettuceConnectionFactory() {
-        return new LettuceConnectionFactory();
+        return new LettuceConnectionFactory(host, port);
     }
 
     @Bean
@@ -28,6 +28,8 @@ public StringRedisTemplate lettuceRedisTemplate(
         redisTemplate.setConnectionFactory(lettuceConnectionFactory);
         redisTemplate.setKeySerializer(new StringRedisSerializer());
         redisTemplate.setValueSerializer(new StringRedisSerializer());
+        redisTemplate.setHashKeySerializer(new StringRedisSerializer());
+        redisTemplate.setHashValueSerializer(new StringRedisSerializer());
         return redisTemplate;
     }
 }

src/main/java/com/thirdparty/ticketing/global/config/WaitingConfig.java

@@ -0,0 +1,58 @@
+package com.thirdparty.ticketing.global.config;
+
+import org.springframework.beans.factory.annotation.Qualifier;
+import org.springframework.context.annotation.Bean;
+import org.springframework.context.annotation.Configuration;
+import org.springframework.data.redis.core.StringRedisTemplate;
+
+import com.fasterxml.jackson.databind.ObjectMapper;
+import com.thirdparty.ticketing.domain.waiting.manager.WaitingManager;
+import com.thirdparty.ticketing.domain.waiting.room.RunningRoom;
+import com.thirdparty.ticketing.domain.waiting.room.WaitingCounter;
+import com.thirdparty.ticketing.domain.waiting.room.WaitingLine;
+import com.thirdparty.ticketing.domain.waiting.room.WaitingRoom;
+import com.thirdparty.ticketing.global.waiting.manager.RedisWaitingManager;
+import com.thirdparty.ticketing.global.waiting.room.RedisRunningRoom;
+import com.thirdparty.ticketing.global.waiting.room.RedisWaitingCounter;
+import com.thirdparty.ticketing.global.waiting.room.RedisWaitingLine;
+import com.thirdparty.ticketing.global.waiting.room.RedisWaitingRoom;
+
+@Configuration
+public class WaitingConfig {
+
+    @Bean
+    public WaitingManager waitingManager(
+            RunningRoom runningRoom,
+            WaitingRoom waitingRoom,
+            @Qualifier("lettuceRedisTemplate") StringRedisTemplate redisTemplate) {
+        return new RedisWaitingManager(runningRoom, waitingRoom, redisTemplate);
+    }
+
+    @Bean
+    public WaitingRoom waitingRoom(
+            WaitingLine waitingLine,
+            WaitingCounter waitingCounter,
+            @Qualifier("lettuceRedisTemplate") StringRedisTemplate redisTemplate,
+            ObjectMapper objectMapper) {
+        return new RedisWaitingRoom(waitingLine, waitingCounter, redisTemplate, objectMapper);
+    }
+
+    @Bean
+    public WaitingLine waitingLine(
+            ObjectMapper objectMapper,
+            @Qualifier("lettuceRedisTemplate") StringRedisTemplate redisTemplate) {
+        return new RedisWaitingLine(objectMapper, redisTemplate);
+    }
+
+    @Bean
+    public WaitingCounter waitingCounter(
+            @Qualifier("lettuceRedisTemplate") StringRedisTemplate redisTemplate) {
+        return new RedisWaitingCounter(redisTemplate);
+    }
+
+    @Bean
+    public RunningRoom runningRoom(
+            @Qualifier("lettuceRedisTemplate") StringRedisTemplate redisTemplate) {
+        return new RedisRunningRoom(redisTemplate);
+    }
+}

src/main/java/com/thirdparty/ticketing/global/waiting/ObjectMapperUtils.java

@@ -0,0 +1,25 @@
+package com.thirdparty.ticketing.global.waiting;
+
+import com.fasterxml.jackson.core.JsonProcessingException;
+import com.fasterxml.jackson.databind.ObjectMapper;
+import com.thirdparty.ticketing.domain.common.ErrorCode;
+import com.thirdparty.ticketing.domain.common.TicketingException;
+
+public class ObjectMapperUtils {
+
+    public static String writeValueAsString(ObjectMapper objectMapper, Object value) {
+        try {
+            return objectMapper.writeValueAsString(value);
+        } catch (JsonProcessingException e) {
+            throw new TicketingException(ErrorCode.WAITING_WRITE_ERROR);
+        }
+    }
+
+    public static <T> T readValue(ObjectMapper objectMapper, String value, Class<T> valueType) {
+        try {
+            return objectMapper.readValue(value, valueType);
+        } catch (JsonProcessingException e) {
+            throw new TicketingException(ErrorCode.WAITING_READ_ERROR);
+        }
+    }
+}

src/main/java/com/thirdparty/ticketing/global/waiting/manager/RedisWaitingManager.java

@@ -0,0 +1,33 @@
+package com.thirdparty.ticketing.global.waiting.manager;
+
+import org.springframework.data.redis.core.StringRedisTemplate;
+import org.springframework.data.redis.core.ValueOperations;
+
+import com.thirdparty.ticketing.domain.waiting.WaitingMember;
+import com.thirdparty.ticketing.domain.waiting.manager.WaitingManager;
+import com.thirdparty.ticketing.domain.waiting.room.RunningRoom;
+import com.thirdparty.ticketing.domain.waiting.room.WaitingRoom;
+
+public class RedisWaitingManager extends WaitingManager {
+
+    private static final String MANAGED_MEMBER_COUNTER_KEY = "managed_member_counter:";
+
+    private final ValueOperations<String, String> managedMemberCounter;
+
+    public RedisWaitingManager(
+            RunningRoom runningRoom, WaitingRoom waitingRoom, StringRedisTemplate redisTemplate) {
+        super(runningRoom, waitingRoom);
+        managedMemberCounter = redisTemplate.opsForValue();
+    }
+
+    @Override
+    protected long countManagedMember(WaitingMember waitingMember) {
+        String key = getPerformanceManagedMemberCounterKey(waitingMember);
+        managedMemberCounter.setIfAbsent(key, "0"); // todo: 불필요하게 네트워크를 탐. 추후 개선 필요
+        return Long.parseLong(managedMemberCounter.get(key));
+    }
+
+    private String getPerformanceManagedMemberCounterKey(WaitingMember waitingMember) {
+        return MANAGED_MEMBER_COUNTER_KEY + waitingMember.getPerformanceId();
+    }
+}

src/main/java/com/thirdparty/ticketing/global/waiting/room/RedisRunningRoom.java

@@ -0,0 +1,33 @@
+package com.thirdparty.ticketing.global.waiting.room;
+
+import java.util.List;
+
+import org.springframework.data.redis.core.RedisTemplate;
+import org.springframework.data.redis.core.SetOperations;
+
+import com.thirdparty.ticketing.domain.waiting.WaitingMember;
+import com.thirdparty.ticketing.domain.waiting.room.RunningRoom;
+
+public class RedisRunningRoom implements RunningRoom {
+
+    private static final String RUNNING_ROOM_KEY = "running_room:";
+
+    private final SetOperations<String, String> runningRoom;
+
+    public RedisRunningRoom(RedisTemplate<String, String> redisTemplate) {
+        runningRoom = redisTemplate.opsForSet();
+    }
+
+    @Override
+    public boolean contains(WaitingMember waitingMember) {
+        return runningRoom.isMember(
+                getPerformanceRunningRoomKey(waitingMember), waitingMember.getEmail());
+    }
+
+    @Override
+    public void put(long performanceId, List<WaitingMember> waitingMembers) {}
+
+    private String getPerformanceRunningRoomKey(WaitingMember waitingMember) {
+        return RUNNING_ROOM_KEY + waitingMember.getPerformanceId();
+    }
+}

src/main/java/com/thirdparty/ticketing/global/waiting/room/RedisWaitingCounter.java

@@ -0,0 +1,25 @@
+package com.thirdparty.ticketing.global.waiting.room;
+
+import org.springframework.data.redis.core.StringRedisTemplate;
+import org.springframework.data.redis.core.ValueOperations;
+
+import com.thirdparty.ticketing.domain.waiting.WaitingMember;
+import com.thirdparty.ticketing.domain.waiting.room.WaitingCounter;
+
+public class RedisWaitingCounter implements WaitingCounter {
+
+    private static final String WAITING_COUNTER_KEY = "waiting_counter";
+
+    private final ValueOperations<String, String> counter;
+
+    public RedisWaitingCounter(StringRedisTemplate redisTemplate) {
+        this.counter = redisTemplate.opsForValue();
+    }
+
+    @Override
+    public long getNextCount(WaitingMember waitingMember) {
+        String performanceWaitingCounterKey =
+                WAITING_COUNTER_KEY + waitingMember.getPerformanceId();
+        return counter.increment(performanceWaitingCounterKey, 1);
+    }
+}

src/main/java/com/thirdparty/ticketing/global/waiting/room/RedisWaitingLine.java

@@ -0,0 +1,38 @@
+package com.thirdparty.ticketing.global.waiting.room;
+
+import java.util.List;
+
+import org.springframework.data.redis.core.StringRedisTemplate;
+import org.springframework.data.redis.core.ZSetOperations;
+
+import com.fasterxml.jackson.databind.ObjectMapper;
+import com.thirdparty.ticketing.domain.waiting.WaitingMember;
+import com.thirdparty.ticketing.domain.waiting.room.WaitingLine;
+import com.thirdparty.ticketing.global.waiting.ObjectMapperUtils;
+
+public class RedisWaitingLine implements WaitingLine {
+
+    private static final String WAITING_LINE_KEY = "waiting_line:";
+
+    private final ObjectMapper objectMapper;
+    private final ZSetOperations<String, String> waitingLine;
+
+    public RedisWaitingLine(ObjectMapper objectMapper, StringRedisTemplate redisTemplate) {
+        this.objectMapper = objectMapper;
+        this.waitingLine = redisTemplate.opsForZSet();
+    }
+
+    @Override
+    public void enter(WaitingMember waitingMember) {
+        String performanceWaitingLineKey = WAITING_LINE_KEY + waitingMember.getPerformanceId();
+        String waitingMemberValue =
+                ObjectMapperUtils.writeValueAsString(objectMapper, waitingMember);
+        waitingLine.add(
+                performanceWaitingLineKey, waitingMemberValue, waitingMember.getWaitingCount());
+    }
+
+    @Override
+    public List<WaitingMember> pollWaitingMembers(long performanceId, long count) {
+        return List.of();
+    }
+}