昔念
cb87fda0d1
- 在 rpc 包中添加了 g.Log() 调用，用于记录 RPC 服务器启动和错误信息 - 重构了 controller 包，将 LogicClient 从 Controller 结构体中分离出来 - 优化了 login 控制器中的 Login 方法，增加了新客户端的创建和数据发送逻辑 - 简化了 systemtime 控制器中的 SystemTimeInfo 方法 - 更新了 server.go 中的 Start 函数，使用 new(controller.LogicClient) 替代 controller.Maincontroller
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 uint16) (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 uint16) {
|
|
//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(serverid, uint16(port), new(controller.LogicClient))
|
|
//TODO 待实现掉线重新连接login
|
|
controller.Maincontroller.RPCClient = *t //将RPC赋值Start
|
|
controller.Maincontroller.Port = uint16(port) //赋值服务器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()
|
|
}
|
|
|
|
}
|