55 lines
1.2 KiB
Go
55 lines
1.2 KiB
Go
|
package main
|
||
|
|
|
||
|
|
import (
|
||
|
|
"blazing/cool"
|
||
|
|
"fmt"
|
||
|
|
"log"
|
||
|
|
"math/rand"
|
||
|
|
"net"
|
||
|
|
"time"
|
||
|
|
|
||
|
|
"github.com/gogf/gf/v2/util/gconv"
|
||
|
|
)
|
||
|
|
|
||
|
|
const (
|
||
|
|
minRandomPort = 10000
|
||
|
|
maxRandomPort = 60000
|
||
|
|
maxPortRetryCount = 5
|
||
|
|
)
|
||
|
|
|
||
|
|
var defaultPort = gconv.Int(cool.Config.Port) //读入默认的端口
|
||
|
|
|
||
|
|
// determinePort 确定服务器使用的端口
|
||
|
|
func determinePort(serverid uint32) (int, error) {
|
||
|
|
rand.Seed(time.Now().UnixNano())
|
||
|
|
if serverid == 0 {
|
||
|
|
return defaultPort, nil
|
||
|
|
}
|
||
|
|
|
||
|
|
for i := 0; i < maxPortRetryCount; i++ {
|
||
|
|
port := generateRandomPort()
|
||
|
|
if isPortAvailable(port) {
|
||
|
|
return port, nil
|
||
|
|
}
|
||
|
|
log.Printf("Port %d is not available, retrying...", port)
|
||
|
|
}
|
||
|
|
|
||
|
|
return 0, fmt.Errorf("failed to find available port after %d attempts", maxPortRetryCount)
|
||
|
|
}
|
||
|
|
|
||
|
|
// generateRandomPort 生成指定范围内的随机端口
|
||
|
|
func generateRandomPort() int {
|
||
|
|
return minRandomPort + rand.Intn(maxRandomPort-minRandomPort)
|
||
|
|
}
|
||
|
|
|
||
|
|
// isPortAvailable 检查端口是否可用
|
||
|
|
func isPortAvailable(port int) bool {
|
||
|
|
address := fmt.Sprintf(":%d", port)
|
||
|
|
listener, err := net.Listen("tcp", address)
|
||
|
|
if err != nil {
|
||
|
|
return false
|
||
|
|
}
|
||
|
|
defer listener.Close()
|
||
|
|
return true
|
||
|
|
}
|