昔念
33723334f9
- 在 Server 结构中添加 connected 字段，用于统计当前连接数 - 在 OnOpen 和 OnClose 事件处理函数中增加连接数的增减逻辑 - 添加 OnTick 方法，定期打印当前连接数 - 更新 ServerEvent 和 ServerOption 文件，集成新功能
103 lines
2.4 KiB
Go
103 lines
2.4 KiB
Go
package main
|
|
|
|
import (
|
|
"blazing/common/rpc"
|
|
"blazing/common/socket"
|
|
"blazing/common/socket/handler"
|
|
"blazing/cool"
|
|
"blazing/logic/controller"
|
|
"blazing/modules/blazing/service"
|
|
|
|
"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
|
|
}
|
|
|
|
// 如果id是0,那就是login server
|
|
func Start(serverid uint32) {
|
|
//ants.NewPool(100)
|
|
head := handler.NewTomeeHandler()
|
|
head.Callback = controller.Recv
|
|
if serverid != 0 { //logic服务器
|
|
// 确定端口
|
|
port, err := determinePort(serverid)
|
|
if err != nil {
|
|
log.Fatalf("Failed to determine port: %v", err)
|
|
}
|
|
|
|
// go func() {
|
|
t := rpc.StartClient(uint16(serverid), controller.Maincontroller)
|
|
|
|
controller.Maincontroller.RPCClient = *t //将RPC赋值Start
|
|
controller.Maincontroller.Port = serverid //赋值服务器ID
|
|
//}()
|
|
//go rpc.StartClient(uint16(serverid), &controller.Maincontroller)
|
|
|
|
service.NewLoginServiceService().SetServerID(serverid, gconv.Uint16(port), t)
|
|
socket.
|
|
NewServer(
|
|
socket.WithCORS(),
|
|
socket.WithPort(port),
|
|
|
|
socket.WithSocketHandler(head)).
|
|
Boot()
|
|
|
|
} else {
|
|
go rpc.StartServer()
|
|
|
|
//go rpcserver() //对login tcp启动
|
|
//ants.Submit(rpcserver)
|
|
socket.NewServer(
|
|
socket.WithCORS(),
|
|
socket.WithPort(defaultPort),
|
|
socket.WithSocketHandler(head)).Boot()
|
|
}
|
|
|
|
}
|