fix: Modify creating README Logic

readme-generator · Oct 12, 2022 · e111731 · e111731
1 parent 601580b
commit e111731
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 9 deletions.
diff --git a/src/main/java/kr/markdown/alreadyme/service/AiService.java b/src/main/java/kr/markdown/alreadyme/service/AiService.java
@@ -23,12 +23,13 @@ public class AiService {
     @Value("${ai-server.host}")
     private String aiServerHost;
 
-    public String getReadmeText(String requestJsonData, String githubOriginalUrl) throws IOException {
+    public void postReadmeText(Long id, String githubOriginalUrl, String requestJsonData) throws IOException {
 
         //requestJsonData
         ObjectMapper objectMapper = new ObjectMapper();
         ObjectNode objectNode = objectMapper.createObjectNode();
 
+        objectNode.put("id", id);
         objectNode.put("githubOriginalUrl", githubOriginalUrl);
 
         ObjectNode jsonDataNode = (ObjectNode) new ObjectMapper().readTree(requestJsonData);
@@ -40,8 +41,6 @@ public String getReadmeText(String requestJsonData, String githubOriginalUrl) th
         httpPost.setHeader("Content-type", "application/json");
 
         httpPost.setEntity(new StringEntity(objectNode.toString(), "UTF-8"));
-        HttpResponse httpResponse = httpClient.execute(httpPost);
-
-        return EntityUtils.toString(httpResponse.getEntity());
+        httpClient.execute(httpPost);
     }
 }
diff --git a/src/main/java/kr/markdown/alreadyme/service/AppService.java b/src/main/java/kr/markdown/alreadyme/service/AppService.java
@@ -72,17 +72,18 @@ public ReadmeItem create(Create createDto) throws Exception {
         JGitUtil.close(git);
         FileUtils.deleteDirectory(new File(git.getRepository().getDirectory().getParentFile().getPath()));;
 
-        //Get readmeText by ai-server
-        String readmeText = aiService.getReadmeText(requestJsonData, gitUrl);
-
         //Create ReadmeItem
         ReadmeItem readmeItem = ReadmeItem.builder()
                 .githubOriginalUrl(gitUrl)
-                .readmeText(readmeText)
                 .createdTime(LocalDateTime.now())
                 .build();
+        ReadmeItem newReadmeItem = readmeItemRepository.save(readmeItem);
+        newReadmeItem = findReadmeItemThrowException(newReadmeItem.getId());
+
+        //POST call readmeText by ai-server
+        aiService.postReadmeText(newReadmeItem.getId(), gitUrl, requestJsonData);
 
-        return readmeItemRepository.save(readmeItem);
+        return readmeItemRepository.save(newReadmeItem);
     }
 
     @Transactional