fix: add server_host force listen to ipv4 (#566)

Author: fatelei

.env.example

@@ -1,3 +1,4 @@
+SERVER_HOST=0.0.0.0
 SERVER_PORT=5002
 SERVER_KEY=lYkiYYT6owG+71oLerGzA7GXCgOT++6ovaezWAjpCjf+Sjc3ZtU+qUEi
 GIN_MODE=release

internal/server/http_server.go

@@ -70,7 +70,7 @@ func (app *App) server(config *app.Config) func() {
 	app.pprofGroup(pprofGroup, config)
 
 	srv := &http.Server{
-		Addr:    fmt.Sprintf(":%d", config.ServerPort),
+		Addr:    fmt.Sprintf("%s:%d", config.ServerHost, config.ServerPort),
 		Handler: engine,
 	}
 

internal/server/http_server_test.go

@@ -0,0 +1,86 @@
+package server
+
+import (
+	"net/http"
+	"strconv"
+	"testing"
+	"time"
+
+	"github.com/langgenius/dify-plugin-daemon/internal/types/app"
+	"github.com/langgenius/dify-plugin-daemon/pkg/utils/network"
+	"github.com/langgenius/dify-plugin-daemon/pkg/utils/routine"
+)
+
+func TestServerHostBinding(t *testing.T) {
+	tests := []struct {
+		name           string
+		host           string
+		connectToHost  string
+		wantStatusCode int
+	}{
+		{
+			name:           "default host 0.0.0.0",
+			host:           "0.0.0.0",
+			connectToHost:  "127.0.0.1",
+			wantStatusCode: http.StatusOK,
+		},
+		{
+			name:           "localhost",
+			host:           "127.0.0.1",
+			connectToHost:  "127.0.0.1",
+			wantStatusCode: http.StatusOK,
+		},
+	}
+
+	for _, tt := range tests {
+		t.Run(tt.name, func(t *testing.T) {
+			port, err := network.GetRandomPort()
+			if err != nil {
+				t.Errorf("failed to get random port: %s", err.Error())
+				return
+			}
+
+			config := &app.Config{
+				ServerPort:          port,
+				ServerHost:          tt.host,
+				ServerKey:           "test-key",
+				HealthApiLogEnabled: true,
+				RoutinePoolSize:     100,
+			}
+			config.SetDefault()
+
+			routine.InitPool(config.RoutinePoolSize)
+
+			appInstance := &App{}
+			cancel := appInstance.server(config)
+
+			if cancel == nil {
+				t.Errorf("failed to start server")
+				return
+			}
+			defer cancel()
+
+			time.Sleep(100 * time.Millisecond)
+
+			client := &http.Client{Timeout: 5 * time.Second}
+			url := "http://" + tt.connectToHost + ":" + strconv.Itoa(int(config.ServerPort)) + "/health/check"
+
+			req, err := http.NewRequest("GET", url, nil)
+			if err != nil {
+				t.Errorf("failed to create request: %s", err.Error())
+				return
+			}
+
+			resp, err := client.Do(req)
+			if err != nil {
+				t.Errorf("failed to send request: %s", err.Error())
+				return
+			}
+			defer resp.Body.Close()
+
+			if resp.StatusCode != tt.wantStatusCode {
+				t.Errorf("expected status %d, got %d", tt.wantStatusCode, resp.StatusCode)
+			}
+		})
+	}
+}

internal/types/app/config.go

@@ -17,6 +17,7 @@ const (
 
 type Config struct {
 	// server
+	ServerHost string `envconfig:"SERVER_HOST"`
 	ServerPort uint16 `envconfig:"SERVER_PORT" validate:"required"`
 	ServerKey  string `envconfig:"SERVER_KEY" validate:"required"`
 

internal/types/app/default.go

@@ -10,6 +10,7 @@ func (config *Config) SetDefault() {
 	case DB_TYPE_OCEANBASE, DB_TYPE_SEEKDB:
 		config.DBType = DB_TYPE_MYSQL
 	}
+	setDefaultString(&config.ServerHost, "0.0.0.0")
 	setDefaultInt(&config.ServerPort, 5002)
 	setDefaultInt(&config.RoutinePoolSize, 10000)
 	setDefaultInt(&config.LifetimeCollectionGCInterval, 60)

internal/types/app/default_test.go

@@ -0,0 +1,39 @@
+package app
+
+import (
+	"testing"
+)
+
+func TestServerHostDefault(t *testing.T) {
+	tests := []struct {
+		name          string
+		inputHost     string
+		expectedHost  string
+	}{
+		{
+			name:         "empty host should default to 0.0.0.0",
+			inputHost:    "",
+			expectedHost: "0.0.0.0",
+		},
+		{
+			name:         "custom host should be preserved",
+			inputHost:    "127.0.0.1",
+			expectedHost: "127.0.0.1",
+		},
+	}
+
+	for _, tt := range tests {
+		t.Run(tt.name, func(t *testing.T) {
+			config := &Config{
+				ServerHost: tt.inputHost,
+				ServerPort: 5002,
+				ServerKey:  "test-key",
+			}
+			config.SetDefault()
+
+			if config.ServerHost != tt.expectedHost {
+				t.Errorf("expected ServerHost %s, got %s", tt.expectedHost, config.ServerHost)
+			}
+		})
+	}
+}