Skip to content

Commit

Permalink
feat: scheduleHandler 추가
Browse files Browse the repository at this point in the history
  • Loading branch information
dlrkdus committed Aug 1, 2024
1 parent ec84b0e commit b186ad4
Show file tree
Hide file tree
Showing 2 changed files with 24 additions and 3 deletions.
24 changes: 24 additions & 0 deletions src/main/java/com/goormy/hackathon/handler/ScheduledHandler.java
Original file line number Diff line number Diff line change
@@ -0,0 +1,24 @@
package com.goormy.hackathon.handler;

import com.amazonaws.services.lambda.runtime.Context;
import com.amazonaws.services.lambda.runtime.RequestHandler;
import com.amazonaws.services.lambda.runtime.events.ScheduledEvent;
import com.goormy.hackathon.lambda.FollowFunction;
import org.springframework.context.ApplicationContext;
import org.springframework.context.annotation.AnnotationConfigApplicationContext;

public class ScheduledHandler implements RequestHandler<ScheduledEvent, String> {

private static final ApplicationContext context = new AnnotationConfigApplicationContext(FollowFunction.class);
private final FollowFunction followFunction;

public ScheduledHandler() {
followFunction = context.getBean(FollowFunction.class);
}

@Override
public String handleRequest(ScheduledEvent event, Context context) {
followFunction.migrateData();
return "Data migration completed";
}
}
3 changes: 0 additions & 3 deletions src/main/java/com/goormy/hackathon/lambda/FollowFunction.java
Original file line number Diff line number Diff line change
Expand Up @@ -9,7 +9,6 @@
import com.goormy.hackathon.repository.UserRepository;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;
import org.springframework.scheduling.annotation.Scheduled;

import java.util.List;
import java.util.Map;
Expand Down Expand Up @@ -62,7 +61,6 @@ public Consumer<Map<String, Object>> processFollow() {
}

// Redis 데이터를 RDBMS에 저장하고 Redis 비우기
@Scheduled(cron = "0 0 0 * * *") // 매일 자정에 실행
public void migrateData() {
// Redis에서 모든 팔로우 데이터 가져오기
List<Follow> follows = followRedisRepository.getAllFollows();
Expand All @@ -72,7 +70,6 @@ public void migrateData() {
followRepository.saveAll(follows);

// Redis 비우기

System.out.println("Redis 데이터를 RDBMS로 옮기고 Redis를 초기화했습니다.");
}

Expand Down

0 comments on commit b186ad4

Please sign in to comment.