리팩토링

Daemawiki · Jan 29, 2024 · 7f07ac8 · 7f07ac8
1 parent 983e169
commit 7f07ac8
Showing 1 changed file with 6 additions and 4 deletions.
diff --git a/src/main/java/com/example/daemawiki/domain/mail/service/MailSend.java b/src/main/java/com/example/daemawiki/domain/mail/service/MailSend.java
@@ -12,7 +12,7 @@
 import org.springframework.mail.javamail.MimeMessageHelper;
 import org.springframework.stereotype.Service;
 import reactor.core.publisher.Mono;
-import reactor.core.scheduler.Schedulers;
+import reactor.core.scheduler.Scheduler;
 
 import java.io.UnsupportedEncodingException;
 import java.util.Random;
@@ -21,10 +21,12 @@
 public class MailSend {
     private final AuthCodeRepository codeRepository;
     private final JavaMailSender mailSender;
+    private final Scheduler scheduler;
 
-    public MailSend(AuthCodeRepository authCodeRepository, JavaMailSender javaMailSender) {
+    public MailSend(AuthCodeRepository authCodeRepository, JavaMailSender javaMailSender, Scheduler scheduler) {
         this.codeRepository = authCodeRepository;
         this.mailSender = javaMailSender;
+        this.scheduler = scheduler;
     }
 
     @Value("${admin.mail}")
@@ -36,8 +38,8 @@ public Mono<Void> execute(AuthCodeRequest request) {
         String authCode = getRandomCode();
         String mail = request.mail();
 
-        Mono<Void> sendMailMono = sendMail(mail, authCode).subscribeOn(Schedulers.boundedElastic());
-        Mono<Void> saveAuthCodeMono = saveAuthCode(mail, authCode).subscribeOn(Schedulers.boundedElastic());
+        Mono<Void> sendMailMono = sendMail(mail, authCode).subscribeOn(scheduler);
+        Mono<Void> saveAuthCodeMono = saveAuthCode(mail, authCode).subscribeOn(scheduler);
 
         return Mono.when(sendMailMono,
                 saveAuthCodeMono);