Merge pull request #110 from Maple-pro/dev

fix(video): delete connection close
GuGoOrg · Aug 22, 2023 · cd94f1d · cd94f1d
2 parents 166a7bb + 26d2b41
commit cd94f1d
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 1 deletion.
diff --git 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
@@ -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