From 26d2b41c37a416b9b3f0811914353401801c8602 Mon Sep 17 00:00:00 2001
From: yangfeng <1719957182@qq.com>
Date: Tue, 22 Aug 2023 01:27:32 +0800
Subject: [PATCH] fix(video): delete connection close

- Add to requeue when ChatGPT response error;
- Add `init` method to rawvideo model
---
 src/models/rawvideo.go                 | 8 ++++++++
 src/services/publish/main.go           | 2 +-
 src/services/videoprocessor/summary.go | 8 ++++++++
 3 files changed, 17 insertions(+), 1 deletion(-)

diff --git a/src/models/rawvideo.go b/src/models/rawvideo.go
index 85413b5..5d82bf7 100644
--- a/src/models/rawvideo.go
+++ b/src/models/rawvideo.go
@@ -1,5 +1,7 @@
 package models
 
+import "GuGoTik/src/storage/database"
+
 type RawVideo struct {
 	ActorId   uint32
 	VideoId   uint32
@@ -7,3 +9,9 @@ type RawVideo struct {
 	FileName  string
 	CoverName string
 }
+
+func init() {
+	if err := database.Client.AutoMigrate(&RawVideo{}); err != nil {
+		panic(err)
+	}
+}
diff --git a/src/services/publish/main.go b/src/services/publish/main.go
index afbdb50..c411ea7 100644
--- a/src/services/publish/main.go
+++ b/src/services/publish/main.go
@@ -48,7 +48,7 @@ func main() {
 
 	var srv PublishServiceImpl
 	var probe healthImpl.ProbeImpl
-	defer CloseMQConn()
+	//defer CloseMQConn()
 	publish.RegisterPublishServiceServer(s, srv)
 	health.RegisterHealthServer(s, &probe)
 	if err := consul.RegisterConsul(config.PublishRpcServerName, config.PublishRpcServerPort); err != nil {
diff --git a/src/services/videoprocessor/summary.go b/src/services/videoprocessor/summary.go
index 79aaea5..99c3a98 100644
--- a/src/services/videoprocessor/summary.go
+++ b/src/services/videoprocessor/summary.go
@@ -113,6 +113,10 @@ func SummaryConsume(channel *amqp.Channel) {
 			}).Errorf("Failed to get summary of an audio from ChatGPT")
 			logging.SetSpanError(span, err)
 			summary = ""
+
+			errorHandler(d, true, logger, &span)
+			span.End()
+			continue
 		}
 
 		select {
@@ -124,6 +128,10 @@ func SummaryConsume(channel *amqp.Channel) {
 			}).Errorf("Failed to get keywords of an audio from ChatGPT")
 			logging.SetSpanError(span, err)
 			keywords = ""
+
+			errorHandler(d, true, logger, &span)
+			span.End()
+			continue
 		}
 
 		// Update summary information to database