Skip to content

Commit cab11ae

Browse files
committed
fix: CtfChallengeHasCtfChallengeCategoryList 에 Entity를 add 하는 메서드 추가
issue: #471 pr: #472
1 parent bebaf38 commit cab11ae

File tree

2 files changed

+10
-7
lines changed

2 files changed

+10
-7
lines changed

src/main/java/keeper/project/homepage/ctf/entity/CtfChallengeEntity.java

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -60,6 +60,7 @@ public class CtfChallengeEntity {
6060
@Builder.Default
6161
@OneToMany(mappedBy = "challenge", cascade = CascadeType.REMOVE)
6262
List<CtfChallengeHasCtfChallengeCategoryEntity> ctfChallengeHasCtfChallengeCategoryList = new ArrayList<>();
63+
6364
@Column(nullable = false)
6465
@Setter
6566
Long score;
@@ -89,4 +90,8 @@ public class CtfChallengeEntity {
8990
@PrimaryKeyJoinColumn
9091
@Setter
9192
CtfDynamicChallengeInfoEntity dynamicChallengeInfoEntity;
93+
94+
public void addCtfChallengeHasCtfChallengeCategory(CtfChallengeHasCtfChallengeCategoryEntity ctfChallengeHasCtfChallengeCategoryEntity) {
95+
this.getCtfChallengeHasCtfChallengeCategoryList().add(ctfChallengeHasCtfChallengeCategoryEntity);
96+
}
9297
}

src/main/java/keeper/project/homepage/ctf/service/CtfAdminService.java

Lines changed: 5 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -5,9 +5,7 @@
55
import static keeper.project.homepage.util.service.CtfUtilService.VIRTUAL_TEAM_ID;
66

77
import java.nio.file.AccessDeniedException;
8-
import java.util.ArrayList;
98
import java.util.List;
10-
import java.util.stream.Collectors;
119
import javax.servlet.http.HttpServletRequest;
1210
import keeper.project.homepage.ctf.dto.CtfChallengeAdminDto;
1311
import keeper.project.homepage.ctf.dto.CtfChallengeCategoryDto;
@@ -338,17 +336,17 @@ private CtfChallengeEntity createChallengeEntityWithFileEntity(
338336

339337
private void setChallengeCategory(CtfChallengeEntity challenge,
340338
CtfChallengeAdminDto challengeAdminDto) {
341-
List<CtfChallengeCategoryEntity> ctfChallengeCategoryEntityList = challengeAdminDto.getCategory()
339+
List<CtfChallengeCategoryEntity> ctfChallengeCategoryEntityList = challengeAdminDto.getCategories()
342340
.stream()
343341
.map(CtfChallengeCategoryDto::toEntity).toList();
344342

345343
for (CtfChallengeCategoryEntity ctfChallengeCategory : ctfChallengeCategoryEntityList) {
346-
CtfChallengeHasCtfChallengeCategoryEntity save = ctfChallengeHasCtfChallengeCategoryRepository.save(
347-
CtfChallengeHasCtfChallengeCategoryEntity.builder()
344+
challenge.addCtfChallengeHasCtfChallengeCategory(ctfChallengeHasCtfChallengeCategoryRepository
345+
.save(CtfChallengeHasCtfChallengeCategoryEntity
346+
.builder()
348347
.challenge(challenge)
349348
.category(ctfChallengeCategory)
350-
.build());
351-
challenge.getCtfChallengeHasCtfChallengeCategoryList().add(save);
349+
.build()));
352350
}
353351
}
354352

0 commit comments

Comments
 (0)