Merge pull request #77 from Prompt-Mate/feat/73/rewrite-result-finalize

feat: return rewriteResultId on rewrite preview

Author: park-hoyeon

src/main/java/com/tave/PromptMate/controller/RewriteController.java

@@ -53,8 +53,10 @@ public ResponseEntity<RewritePreviewResponse> preview(
         AiRewriteResult result = rewriteRunner.run(req.prompt());
         System.out.println("RewriteRunner bean = " + rewriteRunner.getClass().getName());
 
+        Long rewriteResultId = rewriteResultService.createDraft(userId, req.prompt(), result);
 
         return ResponseEntity.ok(new RewritePreviewResponse(
+                rewriteResultId,
                 result.rewrittenContent(),
                 result.latencyMs(),
                 result.modelName(),

src/main/java/com/tave/PromptMate/domain/RewriteResult.java

@@ -12,8 +12,11 @@ public class RewriteResult extends BaseTimeEntity{
     @Id @GeneratedValue(strategy = GenerationType.IDENTITY)
     private Long id;
 
-    @ManyToOne(fetch = FetchType.LAZY, optional = false)
-    @JoinColumn(name = "prompt_id")
+    @Column(name = "user_id", nullable = false)
+    private Long userId;
+
+    @ManyToOne(fetch = FetchType.LAZY, optional = true)
+    @JoinColumn(name="prompt_id", nullable = true)
     private Prompt prompt;
 
     @Column(nullable = false, length = 50)

src/main/java/com/tave/PromptMate/dto/rewrite/RewritePreviewResponse.java

@@ -1,6 +1,7 @@
 package com.tave.PromptMate.dto.rewrite;
 
 public record RewritePreviewResponse(
+        Long rewriteResultId,
         String rewrittenPrompt,
         Long latencyMs,
         String modelName,

src/main/java/com/tave/PromptMate/service/RewriteResultService.java

@@ -1,8 +1,10 @@
 package com.tave.PromptMate.service;
 
 import com.tave.PromptMate.common.NotFoundException;
+import com.tave.PromptMate.common.RewriteRunner;
 import com.tave.PromptMate.domain.Prompt;
 import com.tave.PromptMate.domain.RewriteResult;
+import com.tave.PromptMate.dto.rewrite.AiRewriteResult;
 import com.tave.PromptMate.dto.rewrite.CreateRewriteRequest;
 import com.tave.PromptMate.dto.rewrite.RewriteMapper;
 import com.tave.PromptMate.dto.rewrite.RewriteResponse;
@@ -11,8 +13,6 @@
 import lombok.RequiredArgsConstructor;
 import org.springframework.stereotype.Service;
 import org.springframework.transaction.annotation.Transactional;
-import com.tave.PromptMate.common.RewriteRunner;
-
 
 import java.util.List;
 
@@ -25,6 +25,27 @@ public class RewriteResultService {
     private final PromptRepository promptRepository;
     private final RewriteRunner rewriteRunner;
 
+
+    public Long createDraft(Long userId, String beforePrompt, AiRewriteResult ai) {
+
+        RewriteResult entity = RewriteResult.builder()
+                .userId(userId)
+                .prompt(null)
+                .modelName(ai.modelName())
+                .inputTokens(ai.inputTokens())
+                .outputTokens(ai.outputTokens())
+                .latencyMs(ai.latencyMs())
+                .content(ai.rewrittenContent())
+                .build();
+
+        rewriteResultRepository.save(entity);
+        return entity.getId();
+    }
+
+
+
+
+
     public RewriteResponse createAuto(Long promptId) {
 
         Prompt prompt = promptRepository.findById(promptId)