added 5 minute timoute on ocr reader

Kirari04 · Jan 22, 2024 · 7cf5feb · 7cf5feb
1 parent aa6add1
commit 7cf5feb
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/services/Encoder.go b/services/Encoder.go
@@ -5,6 +5,7 @@ import (
 	"ch/kirari04/videocms/helpers"
 	"ch/kirari04/videocms/inits"
 	"ch/kirari04/videocms/models"
+	"context"
 	"crypto/sha256"
 	"fmt"
 	"log"
@@ -712,8 +713,12 @@ func prepocessPgs(encodingTask models.Subtitle, absFolderOutput string, absFileI
 		return fmt.Errorf("Error happend while opening pgs subtitle: %v", err.Error())
 	}
 
+	ctx, cancel := context.WithTimeout(context.Background(), time.Minute*5)
+	defer cancel()
+
 	client := req.C()
 	res, err := client.R().
+		SetContext(ctx).
 		SetFileReader("file", "subtitle.sup", pgsFile).
 		Post(config.ENV.PluginPgsServer)
 	if err != nil {