fix: aws timeout

langgenius · Aug 23, 2024 · bbc84bc · bbc84bc
1 parent 32d0160
commit bbc84bc
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 6 deletions.
diff --git a/cmd/server/main.go b/cmd/server/main.go
@@ -1,8 +1,6 @@
 package main
 
 import (
-	"time"
-
 	"github.com/joho/godotenv"
 	"github.com/kelseyhightower/envconfig"
 	"github.com/langgenius/dify-plugin-daemon/internal/server"
@@ -43,7 +41,7 @@ func setDefault(config *app.Config) {
 	setDefaultInt(&config.PluginRemoteInstallServerEventLoopNums, 8)
 	setDefaultInt(&config.PluginRemoteInstallingMaxConn, 128)
 	setDefaultInt(&config.MaxPluginPackageSize, 52428800)
-	setDefaultInt(&config.MaxAWSLambdaTransactionTimeout, time.Second*150)
+	setDefaultInt(&config.MaxAWSLambdaTransactionTimeout, 150)
 	setDefaultBool(&config.PluginRemoteInstallingEnabled, true)
 	setDefaultBool(&config.PluginWebhookEnabled, true)
 	setDefaultString(&config.DBSslMode, "disable")

diff --git a/internal/server/http_server.go b/internal/server/http_server.go
@@ -4,6 +4,7 @@ import (
 	"context"
 	"fmt"
 	"net/http"
+	"time"
 
 	"github.com/gin-gonic/gin"
 	"github.com/langgenius/dify-plugin-daemon/internal/core/plugin_daemon/backwards_invocation/transaction"
@@ -76,7 +77,9 @@ func (app *App) webhookGroup(group *gin.RouterGroup, config *app.Config) {
 
 func (appRef *App) awsLambdaTransactionGroup(group *gin.RouterGroup, config *app.Config) {
 	if config.Platform == app.PLATFORM_AWS_LAMBDA {
-		appRef.aws_transaction_handler = transaction.NewAWSTransactionHandler(config.MaxAWSLambdaTransactionTimeout)
+		appRef.aws_transaction_handler = transaction.NewAWSTransactionHandler(
+			time.Duration(config.MaxAWSLambdaTransactionTimeout) * time.Second,
+		)
 		group.POST(
 			"/transaction",
 			appRef.RedirectAWSLambdaTransaction,

diff --git a/internal/types/app/config.go b/internal/types/app/config.go
@@ -2,7 +2,6 @@ package app
 
 import (
 	"fmt"
-	"time"
 
 	"github.com/go-playground/validator/v10"
 )
@@ -51,7 +50,7 @@ type Config struct {
 
 	MaxPluginPackageSize int64 `envconfig:"MAX_PLUGIN_PACKAGE_SIZE" validate:"required"`
 
-	MaxAWSLambdaTransactionTimeout time.Duration `envconfig:"MAX_AWS_LAMBDA_TRANSACTION_TIMEOUT"`
+	MaxAWSLambdaTransactionTimeout int `envconfig:"MAX_AWS_LAMBDA_TRANSACTION_TIMEOUT"`
 }
 
 func (c *Config) Validate() error {