Merge pull request #107 from siwonKH/main

Add status webhook
Team-KHPT · Jul 12, 2023 · c5bc585 · c5bc585
2 parents 91d5352 + 2621f2e
commit c5bc585
Show file tree

Hide file tree

Showing 2 changed files with 15 additions and 4 deletions.
diff --git a/src/main/java/com/khpt/projectkim/controller/api/ChatRestController.java b/src/main/java/com/khpt/projectkim/controller/api/ChatRestController.java
@@ -65,6 +65,7 @@ public class ChatRestController {
     @PostConstruct
     public void init() {
         this.openAiService = new OpenAiService(token);
+        discordWebhookService.queueStatusLog("Server start up");
     }
 
     @PostMapping("/example")
@@ -274,10 +275,10 @@ public SseEmitter getChatEvents(HttpSession session, HttpServletResponse respons
                                             ChatMessageRole.SYSTEM.value(),
                                             "Below is job_code_table."
                                     ));
-                                    chatMessages.add(new ChatMessage(
-                                            ChatMessageRole.SYSTEM.value(),
-                                            CsvReader.getDetailedJobCode(user.getCategory())
-                                    ));
+//                                    chatMessages.add(new ChatMessage(
+//                                            ChatMessageRole.SYSTEM.value(),
+//                                            CsvReader.getDetailedJobCode(user.getCategory())
+//                                    ));
 
                                     chatMessages.add(accumulatedMessage);
                                     chatMessages.add(callResponse);

diff --git a/src/main/java/com/khpt/projectkim/service/DiscordWebhookService.java b/src/main/java/com/khpt/projectkim/service/DiscordWebhookService.java
@@ -68,6 +68,16 @@ public void queueErrorLog(String userId, String message, Throwable e) {
         errorLogQueue.add(embed);
     }
 
+    public void queueStatusLog(String message) {
+        DiscordWebhook.EmbedObject embed = new DiscordWebhook.EmbedObject()
+                .setTitle("김비서 상태")
+                .setColor(Color.LIGHT_GRAY)
+                .addField("message", message, false)
+                .setFooter(sdf.format(new Date()), "");
+
+        errorLogQueue.add(embed);
+    }
+
     public void queueChatLog(String userId, String userName, String message) {
         if (message.length() > 300) {
             message = message.substring(0, 300);