TnT-Server/src/main/java/com/tnt/member/application/WithdrawService.java at c938d48fcb8d0eebea6d70fe68f8fd27eb82624c · YAPP-Github/TnT-Server · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
package com.tnt.member.application;

import static com.tnt.member.domain.MemberType.TRAINEE;
import static com.tnt.member.domain.MemberType.TRAINER;

import java.util.List;

import org.springframework.stereotype.Service;
import org.springframework.transaction.annotation.Transactional;

import com.tnt.common.error.exception.NotFoundException;
import com.tnt.gateway.application.SessionService;
import com.tnt.member.application.repository.MemberRepository;
import com.tnt.member.domain.Member;
import com.tnt.member.dto.WithdrawDto;
import com.tnt.pt.application.PtService;
import com.tnt.pt.application.repository.PtLessonRepository;
import com.tnt.pt.application.repository.PtTrainerTraineeRepository;
import com.tnt.pt.domain.PtLesson;
import com.tnt.pt.domain.PtTrainerTrainee;
import com.tnt.trainee.application.DietService;
import com.tnt.trainee.application.TraineeService;
import com.tnt.trainee.application.repository.DietRepository;
import com.tnt.trainee.application.repository.TraineeRepository;
import com.tnt.trainee.domain.Diet;
import com.tnt.trainee.domain.Trainee;
import com.tnt.trainer.application.TrainerService;
import com.tnt.trainer.application.repository.TrainerRepository;
import com.tnt.trainer.domain.Trainer;

import lombok.RequiredArgsConstructor;

@Service
@RequiredArgsConstructor
public class WithdrawService {

	private final SessionService sessionService;
	private final MemberService memberService;
	private final TrainerService trainerService;
	private final TraineeService traineeService;
	private final DietService dietService;
	private final PtService ptService;
	private final MemberRepository memberRepository;
	private final TrainerRepository trainerRepository;
	private final TraineeRepository traineeRepository;
	private final PtLessonRepository ptLessonRepository;
	private final PtTrainerTraineeRepository ptTrainerTraineeRepository;
	private final DietRepository dietRepository;

	@Transactional
	public WithdrawDto withdraw(Long memberId) {
		Member member = memberService.getByMemberId(memberId);

		deleteMemberData(member);

		sessionService.removeSession(String.valueOf(memberId));

		return new WithdrawDto(member.getSocialId(), member.getSocialType(), member.getProfileImageUrl());
	}

	private void deleteMemberData(Member member) {
		if (member.getMemberType() == TRAINER) {
			Trainer trainer = trainerService.getByMemberId(member.getId());

			if (ptService.isPtTrainerTraineeExistWithTrainerId(trainer.getId())) {
				List<PtTrainerTrainee> ptTrainerTrainees = ptService.getAllPtTrainerTraineeWithTrainerId(
					trainer.getId());

				List<PtLesson> ptLessons = ptTrainerTrainees.stream()
					.map(ptService::getPtLessonWithPtTrainerTrainee)
					.flatMap(List::stream)
					.toList();

				ptLessons.forEach(PtLesson::softDelete);
				ptTrainerTrainees.forEach(PtTrainerTrainee::softDelete);

				ptLessonRepository.saveAll(ptLessons);
				ptTrainerTraineeRepository.saveAll(ptTrainerTrainees);
			}

			trainer.softDelete();
			trainerRepository.save(trainer);
		}

		if (member.getMemberType() == TRAINEE) {
			Trainee trainee = traineeService.getByMemberId(member.getId());
			List<Diet> diets = dietService.getAllByTraineeId(trainee.getId());

			if (ptService.isPtTrainerTraineeExistWithTraineeId(trainee.getId())) {
				try {
					PtTrainerTrainee ptTrainerTrainee = ptService.getPtTrainerTraineeWithTraineeId(trainee.getId());
					List<PtLesson> ptLessons = ptService.getPtLessonWithPtTrainerTrainee(ptTrainerTrainee);

					ptLessons.forEach(PtLesson::softDelete);
					ptTrainerTrainee.softDelete();

					ptLessonRepository.saveAll(ptLessons);
					ptTrainerTraineeRepository.save(ptTrainerTrainee);
				} catch (NotFoundException e) {
					// Do nothing
				}
			}

			diets.forEach(Diet::softDelete);
			dietRepository.saveAll(diets);

			trainee.softDelete();
			traineeRepository.save(trainee);
		}

		member.softDelete();
		memberRepository.save(member);
	}
}