random server port

login546 · Aug 29, 2024 · ae770f4 · ae770f4
1 parent 071b6b4
commit ae770f4
Show file tree

Hide file tree

Showing 2 changed files with 22 additions and 3 deletions.
diff --git a/cmd/httpgo/main.go b/cmd/httpgo/main.go
@@ -57,10 +57,17 @@ func main() {
 		newdir := dir + "/" + *server + "/"
 		// 取随机字符串作为密码
 		Spasswd := utils.GenerateRandomString(10)
-		fmt.Printf("Serving：http://127.0.0.1:%d/%s.html\n", 6231, *server)
-		fmt.Printf("Serving：http://0.0.0.1:%d/%s.html\n", 6231, *server)
+
+		// 获取随机未占用端口
+		port, err := utils.GetRandomPort()
+		if err != nil {
+			log.Fatal(err)
+		}
+
+		fmt.Printf("Serving：http://127.0.0.1:%d/%s.html\n", port, *server)
+		fmt.Printf("Serving：http://0.0.0.1:%d/%s.html\n", port, *server)
 		fmt.Printf("UserInfo: admin/%s\n", Spasswd)
-		err := httpgo.ServeDirectoryWithAuth(newdir, "admin", Spasswd, 6231)
+		err = httpgo.ServeDirectoryWithAuth(newdir, "admin", Spasswd, port)
 		if err != nil {
 			log.Fatal(err)
 		}

diff --git a/pkg/utils/random.go b/pkg/utils/random.go
@@ -2,6 +2,7 @@ package utils
 
 import (
 	"math/rand"
+	"net"
 	"time"
 )
 
@@ -20,3 +21,14 @@ func GenerateRandomString(length int) string {
 
 	return string(b)
 }
+
+func GetRandomPort() (int, error) {
+	// Listen on a random port
+	addr, err := net.Listen("tcp", ":0")
+	if err != nil {
+		return 0, err
+	}
+	defer addr.Close()
+	port := addr.Addr().(*net.TCPAddr).Port
+	return port, nil
+}