Skip to content

Commit

Permalink
modify: Initializer 에서 UserTracking 데이터 삭제
Browse files Browse the repository at this point in the history
  • Loading branch information
ghkdgus29 committed Apr 10, 2024
1 parent c1276e7 commit 065e7ec
Show file tree
Hide file tree
Showing 2 changed files with 11 additions and 1 deletion.
5 changes: 4 additions & 1 deletion src/main/java/site/youtogether/config/WebSocketConfig.java
Original file line number Diff line number Diff line change
Expand Up @@ -9,12 +9,15 @@
import org.springframework.web.socket.config.annotation.WebSocketMessageBrokerConfigurer;

import lombok.RequiredArgsConstructor;
import site.youtogether.util.interceptor.StompHandshakeInterceptor;

@Configuration
@EnableWebSocketMessageBroker
@RequiredArgsConstructor
public class WebSocketConfig implements WebSocketMessageBrokerConfigurer {

private final StompHandshakeInterceptor stompHandshakeInterceptor;

@Override
public void configureMessageBroker(MessageBrokerRegistry registry) {
registry.enableSimpleBroker("/sub");
Expand All @@ -24,7 +27,7 @@ public void configureMessageBroker(MessageBrokerRegistry registry) {
@Override
public void registerStompEndpoints(StompEndpointRegistry registry) {
registry.addEndpoint(STOMP_ENDPOINT).setAllowedOriginPatterns("http://localhost:3000", "https://you-together-web.vercel.app")
.addInterceptors()
.addInterceptors(stompHandshakeInterceptor)
.withSockJS();
}

Expand Down
7 changes: 7 additions & 0 deletions src/main/java/site/youtogether/util/Initializer.java
Original file line number Diff line number Diff line change
@@ -1,10 +1,14 @@
package site.youtogether.util;

import static site.youtogether.util.AppConstants.*;

import java.time.LocalDateTime;
import java.util.Set;

import org.springframework.boot.ApplicationArguments;
import org.springframework.boot.ApplicationRunner;
import org.springframework.context.annotation.Profile;
import org.springframework.data.redis.core.RedisTemplate;
import org.springframework.stereotype.Component;

import lombok.RequiredArgsConstructor;
Expand All @@ -24,11 +28,14 @@ public class Initializer implements ApplicationRunner {

private final RoomStorage roomStorage;
private final UserStorage userStorage;
private final RedisTemplate<String, String> redisTemplate;

@Override
public void run(ApplicationArguments args) throws Exception {
roomStorage.deleteAll();
userStorage.deleteAll();
Set<String> keys = redisTemplate.keys(USER_TRACKING_KEY_PREFIX + "*");
redisTemplate.delete(keys);

for (long i = 0; i < NO_PASSWORD_ROOM_COUNT; i++) {
User host = User.builder()
Expand Down

0 comments on commit 065e7ec

Please sign in to comment.