기업정보 입력에 따른 AI 컨설팅 저장

README.md

@@ -0,0 +1 @@
+# Backend

build.gradle

@@ -30,6 +30,7 @@ dependencies {
 	implementation 'org.springframework.boot:spring-boot-starter-web'
 	compileOnly 'org.projectlombok:lombok'
 	runtimeOnly 'com.mysql:mysql-connector-j'
+	runtimeOnly 'org.postgresql:postgresql'
 	annotationProcessor 'org.projectlombok:lombok'
 	testImplementation 'org.springframework.boot:spring-boot-starter-test'
 	testImplementation 'org.springframework.security:spring-security-test'
@@ -48,4 +49,4 @@ dependencies {
 
 tasks.named('test') {
 	useJUnitPlatform()
-}
+}

src/main/java/com/swOnCampus/AIPlatform/domain/consulting/entity/Company.java

@@ -0,0 +1,59 @@
+package com.swOnCampus.AIPlatform.domain.consulting.entity;
+
+import com.swOnCampus.AIPlatform.domain.member.entity.Member;
+import com.swOnCampus.AIPlatform.global.entity.BaseEntity;
+import jakarta.persistence.CascadeType;
+import jakarta.persistence.Column;
+import jakarta.persistence.Entity;
+import jakarta.persistence.FetchType;
+import jakarta.persistence.GeneratedValue;
+import jakarta.persistence.GenerationType;
+import jakarta.persistence.Id;
+import jakarta.persistence.JoinColumn;
+import jakarta.persistence.ManyToOne;
+import jakarta.persistence.OneToOne;
+import jakarta.persistence.Table;
+import lombok.AccessLevel;
+import lombok.Builder;
+import lombok.Getter;
+import lombok.NoArgsConstructor;
+
+@Getter
+@NoArgsConstructor(access = AccessLevel.PROTECTED)
+@Entity
+@Table(name = "company")
+public class Company extends BaseEntity {
+
+    @Id
+    @GeneratedValue(strategy = GenerationType.IDENTITY)
+    private Long id;
+
+    @ManyToOne(fetch = FetchType.LAZY)
+    @JoinColumn(name = "member_id", nullable = false)
+    private Member member;
+
+    @Column(nullable = false)
+    private String name;
+
+    @Column(nullable = false)
+    private String companySize;
+
+    @Column(nullable = false)
+    private String industry;
+
+    @Column(nullable = false, columnDefinition = "TEXT")
+    private String painPoint;
+
+    @OneToOne(mappedBy = "company", cascade = CascadeType.ALL, orphanRemoval = true)
+    private Consulting consulting;
+
+    @Builder
+    public Company(Member member, String name, String companySize, String industry,
+        String painPoint) {
+        this.member = member;
+        this.name = name;
+        this.companySize = companySize;
+        this.industry = industry;
+        this.painPoint = painPoint;
+    }
+}

src/main/java/com/swOnCampus/AIPlatform/domain/consulting/entity/Consulting.java

@@ -0,0 +1,43 @@
+package com.swOnCampus.AIPlatform.domain.consulting.entity;
+
+import com.swOnCampus.AIPlatform.global.entity.BaseEntity;
+import jakarta.persistence.Column;
+import jakarta.persistence.Entity;
+import jakarta.persistence.GeneratedValue;
+import jakarta.persistence.GenerationType;
+import jakarta.persistence.Id;
+import jakarta.persistence.JoinColumn;
+import jakarta.persistence.OneToOne;
+import jakarta.persistence.Table;
+import lombok.AccessLevel;
+import lombok.Builder;
+import lombok.Getter;
+import lombok.NoArgsConstructor;
+
+@Getter
+@NoArgsConstructor(access = AccessLevel.PROTECTED)
+@Entity
+@Table(name = "consulting")
+public class Consulting extends BaseEntity {
+
+    @Id
+    @GeneratedValue(strategy = GenerationType.IDENTITY)
+    private Long id;
+
+    @OneToOne
+    @JoinColumn(name = "company_id", nullable = false)
+    private Company company; // 기업 입력 정보 (채팅방)
+
+    @Column(columnDefinition = "TEXT")
+    private String result;
+
+    @Column(columnDefinition = "TEXT")
+    private String summary;
+
+    @Builder
+    public Consulting(Company company, String result, String summary) {
+        this.company = company;
+        this.result = result;
+        this.summary = summary;
+    }
+}

src/main/java/com/swOnCampus/AIPlatform/domain/consulting/exception/ConsultingErrorCode.java

@@ -0,0 +1,16 @@
+package com.swOnCampus.AIPlatform.domain.consulting.exception;
+
+import com.swOnCampus.AIPlatform.global.exception.ErrorCode;
+import lombok.Getter;
+import lombok.RequiredArgsConstructor;
+import org.springframework.http.HttpStatus;
+
+@Getter
+@RequiredArgsConstructor
+public enum ConsultingErrorCode implements ErrorCode {
+    // 404
+    NOT_EXIST_CONSULTING(HttpStatus.NOT_FOUND, "해당 기업에 대한 컨설팅 결과가 존재하지 않습니다.");
+
+    private final HttpStatus httpStatus;
+    private final String message;
+}

src/main/java/com/swOnCampus/AIPlatform/domain/consulting/repository/CompanyRepository.java

@@ -0,0 +1,10 @@
+package com.swOnCampus.AIPlatform.domain.consulting.repository;
+
+import com.swOnCampus.AIPlatform.domain.consulting.entity.Company;
+import org.springframework.data.jpa.repository.JpaRepository;
+import org.springframework.stereotype.Repository;
+
+@Repository
+public interface CompanyRepository extends JpaRepository<Company, Long> {
+
+}

src/main/java/com/swOnCampus/AIPlatform/domain/consulting/repository/ConsultingRepository.java

@@ -0,0 +1,12 @@
+package com.swOnCampus.AIPlatform.domain.consulting.repository;
+
+import com.swOnCampus.AIPlatform.domain.consulting.entity.Consulting;
+import java.util.Optional;
+import org.springframework.data.jpa.repository.JpaRepository;
+import org.springframework.stereotype.Repository;
+
+@Repository
+public interface ConsultingRepository extends JpaRepository<Consulting, Long> {
+
+    Optional<Consulting> findByCompanyId(Long companyId);
+}

src/main/java/com/swOnCampus/AIPlatform/domain/consulting/service/CompanyService.java

@@ -0,0 +1,13 @@
+package com.swOnCampus.AIPlatform.domain.consulting.service;
+
+import com.swOnCampus.AIPlatform.domain.consulting.web.dto.request.CompanyInfoRequest;
+import com.swOnCampus.AIPlatform.domain.consulting.web.dto.response.ConsultingResponse;
+
+public interface CompanyService {
+
+    ConsultingResponse createOrGetConsulting(
+        Long memberId,
+        CompanyInfoRequest companyInfoRequest,
+        Long chatRoomId
+    );
+}

src/main/java/com/swOnCampus/AIPlatform/domain/consulting/service/CompanyServiceImpl.java

@@ -0,0 +1,73 @@
+package com.swOnCampus.AIPlatform.domain.consulting.service;
+
+import com.swOnCampus.AIPlatform.domain.consulting.entity.Company;
+import com.swOnCampus.AIPlatform.domain.consulting.entity.Consulting;
+import com.swOnCampus.AIPlatform.domain.consulting.exception.ConsultingErrorCode;
+import com.swOnCampus.AIPlatform.domain.consulting.repository.CompanyRepository;
+import com.swOnCampus.AIPlatform.domain.consulting.repository.ConsultingRepository;
+import com.swOnCampus.AIPlatform.domain.consulting.web.dto.ConsultingSave;
+import com.swOnCampus.AIPlatform.domain.consulting.web.dto.request.CompanyInfoRequest;
+import com.swOnCampus.AIPlatform.domain.consulting.web.dto.request.ConsultingRequest;
+import com.swOnCampus.AIPlatform.domain.consulting.web.dto.response.ConsultingResponse;
+import com.swOnCampus.AIPlatform.domain.member.entity.Member;
+import com.swOnCampus.AIPlatform.domain.member.repository.MemberRepository;
+import com.swOnCampus.AIPlatform.global.exception.CommonErrorCode;
+import com.swOnCampus.AIPlatform.global.exception.GlobalException;
+import lombok.RequiredArgsConstructor;
+import org.springframework.stereotype.Service;
+
+@Service
+@RequiredArgsConstructor
+public class CompanyServiceImpl implements CompanyService {
+
+    private final CompanyRepository companyRepository;
+    private final MemberRepository memberRepository;
+    private final ConsultingRepository consultingRepository;
+    private final ConsultingService consultingService;
+
+    @Override
+    public ConsultingResponse createOrGetConsulting(
+        Long memberId,
+        CompanyInfoRequest companyInfoRequest,
+        Long companyId // 채팅방 id
+    ) {
+        Member member = memberRepository.findById(memberId)
+            .orElseThrow(() -> new GlobalException(CommonErrorCode.NOT_EXIST_MEMBER));
+
+        if (companyId == null) {
+            companyId = createCompanyWithConsulting(member, companyInfoRequest);
+        }
+        Consulting consulting = consultingRepository.findByCompanyId(companyId)
+            .orElseThrow(() -> new GlobalException(ConsultingErrorCode.NOT_EXIST_CONSULTING));
+
+        return new ConsultingResponse(consulting.getSummary());
+    }
+
+    private Long createCompanyWithConsulting(Member member, CompanyInfoRequest companyInfoRequest) {
+        Company company = Company.builder()
+            .member(member)
+            .name(companyInfoRequest.name())
+            .companySize(companyInfoRequest.companySize())
+            .industry(companyInfoRequest.industry())
+            .painPoint(companyInfoRequest.painPoint())
+            .build();
+        companyRepository.save(company);
+
+        ConsultingRequest consultingRequest = new ConsultingRequest(
+            companyInfoRequest.industry(),
+            companyInfoRequest.companySize(),
+            companyInfoRequest.painPoint()
+        );
+
+        ConsultingSave result = consultingService.getConsultingResult(consultingRequest);
+
+        Consulting consulting = Consulting.builder()
+            .company(company)
+            .result(result.result())
+            .summary(result.summary())
+            .build();
+        consultingRepository.save(consulting);
+
+        return consulting.getCompany().getId();
+    }
+}

src/main/java/com/swOnCampus/AIPlatform/domain/consulting/service/ConsultingService.java

@@ -0,0 +1,9 @@
+package com.swOnCampus.AIPlatform.domain.consulting.service;
+
+import com.swOnCampus.AIPlatform.domain.consulting.web.dto.ConsultingSave;
+import com.swOnCampus.AIPlatform.domain.consulting.web.dto.request.ConsultingRequest;
+
+public interface ConsultingService {
+
+    ConsultingSave getConsultingResult(ConsultingRequest request);
+}

src/main/java/com/swOnCampus/AIPlatform/domain/consulting/service/ConsultingServiceImpl.java

@@ -0,0 +1,38 @@
+package com.swOnCampus.AIPlatform.domain.consulting.service;
+
+import com.swOnCampus.AIPlatform.domain.consulting.web.dto.ConsultingSave;
+import com.swOnCampus.AIPlatform.domain.consulting.web.dto.request.ConsultingRequest;
+import com.swOnCampus.AIPlatform.domain.consulting.web.dto.response.ConsultingAiResponse;
+import lombok.RequiredArgsConstructor;
+import org.springframework.beans.factory.annotation.Value;
+import org.springframework.stereotype.Service;
+import org.springframework.web.client.RestTemplate;
+import org.springframework.web.util.UriComponentsBuilder;
+
+@Service
+@RequiredArgsConstructor
+public class ConsultingServiceImpl implements ConsultingService {
+
+    private final RestTemplate restTemplate;
+
+    @Value("${ai.api.url}")
+    private String aiApiUrl;
+
+    @Override
+    public ConsultingSave getConsultingResult(ConsultingRequest request) {
+        ConsultingSave response = new ConsultingSave(
+            sendRequestToAi(request, false).result(),
+            sendRequestToAi(request, true).result()
+        );
+
+        return response;
+    }
+
+    private ConsultingAiResponse sendRequestToAi(ConsultingRequest request, boolean summary) {
+        String url = UriComponentsBuilder.fromHttpUrl(aiApiUrl + "/api/consulting")
+            .queryParam("summary", summary)
+            .toUriString();
+
+        return restTemplate.postForObject(url, request, ConsultingAiResponse.class);
+    }
+}

src/main/java/com/swOnCampus/AIPlatform/domain/consulting/web/controller/ConsultingController.java

@@ -0,0 +1,39 @@
+package com.swOnCampus.AIPlatform.domain.consulting.web.controller;
+
+import com.swOnCampus.AIPlatform.domain.consulting.service.CompanyService;
+import com.swOnCampus.AIPlatform.domain.consulting.web.dto.request.CompanyInfoRequest;
+import com.swOnCampus.AIPlatform.domain.consulting.web.dto.response.ConsultingResponse;
+import com.swOnCampus.AIPlatform.domain.member.entity.Member;
+import com.swOnCampus.AIPlatform.global.annotation.LoginMember;
+import com.swOnCampus.AIPlatform.global.response.ApiResponse;
+import lombok.RequiredArgsConstructor;
+import org.springframework.http.HttpStatus;
+import org.springframework.http.ResponseEntity;
+import org.springframework.web.bind.annotation.PostMapping;
+import org.springframework.web.bind.annotation.RequestBody;
+import org.springframework.web.bind.annotation.RequestMapping;
+import org.springframework.web.bind.annotation.RequestParam;
+import org.springframework.web.bind.annotation.RestController;
+
+@RequiredArgsConstructor
+@RestController
+@RequestMapping("/api/v1/consulting")
+public class ConsultingController {
+
+    private final CompanyService companyService;
+
+    @PostMapping()
+    public ResponseEntity<ApiResponse<?>> createConsulting(
+        @LoginMember Member member,
+        @RequestBody CompanyInfoRequest companyInfoRequest,
+        @RequestParam(required = false) Long companyId // 채팅방 id
+    ) {
+        ConsultingResponse result = companyService.createOrGetConsulting(
+            member.getMemberId(), companyInfoRequest, companyId);
+
+        ApiResponse<ConsultingResponse> response = ApiResponse.createSuccess(HttpStatus.OK.value(),
+            result, "컨설팅 결과가 생성되었습니다.");
+
+        return ResponseEntity.ok(response);
+    }
+}

src/main/java/com/swOnCampus/AIPlatform/domain/consulting/web/dto/ConsultingSave.java

@@ -0,0 +1,8 @@
+package com.swOnCampus.AIPlatform.domain.consulting.web.dto;
+
+public record ConsultingSave(
+    String result,
+    String summary
+) {
+
+}

src/main/java/com/swOnCampus/AIPlatform/domain/consulting/web/dto/request/CompanyInfoRequest.java

@@ -0,0 +1,15 @@
+package com.swOnCampus.AIPlatform.domain.consulting.web.dto.request;
+
+import jakarta.validation.constraints.NotBlank;
+
+public record CompanyInfoRequest(
+    @NotBlank(message="회사명을 입력해주세요.")
+    String name,
+    @NotBlank(message="기업규모를 입력해주세요.")
+    String companySize,
+    @NotBlank(message="산업 분야를 입력해주세요.")
+    String industry,
+    @NotBlank(message="Pain Point를 입력해주세요.")
+    String painPoint
+) {
+}

src/main/java/com/swOnCampus/AIPlatform/domain/consulting/web/dto/request/ConsultingRequest.java

@@ -0,0 +1,11 @@
+package com.swOnCampus.AIPlatform.domain.consulting.web.dto.request;
+
+import com.fasterxml.jackson.annotation.JsonProperty;
+
+public record ConsultingRequest(
+    @JsonProperty("industry") String industry,
+    @JsonProperty("company_size") String companySize,
+    @JsonProperty("pain_point") String painPoint
+) {
+
+}

src/main/java/com/swOnCampus/AIPlatform/domain/consulting/web/dto/response/ConsultingAiResponse.java

@@ -0,0 +1,7 @@
+package com.swOnCampus.AIPlatform.domain.consulting.web.dto.response;
+
+public record ConsultingAiResponse(
+    String result
+) {
+
+}

src/main/java/com/swOnCampus/AIPlatform/domain/consulting/web/dto/response/ConsultingResponse.java

@@ -0,0 +1,7 @@
+package com.swOnCampus.AIPlatform.domain.consulting.web.dto.response;
+
+public record ConsultingResponse(
+    String summary
+) {
+
+}