初始化提交
This commit is contained in:
395
router/minecraft/server.pb.go
Normal file
395
router/minecraft/server.pb.go
Normal file
@ -0,0 +1,395 @@
|
||||
// Code generated by protoc-gen-go. DO NOT EDIT.
|
||||
// versions:
|
||||
// protoc-gen-go v1.36.6
|
||||
// protoc v5.29.3
|
||||
// source: server.proto
|
||||
|
||||
package minecraft
|
||||
|
||||
import (
|
||||
protoreflect "google.golang.org/protobuf/reflect/protoreflect"
|
||||
protoimpl "google.golang.org/protobuf/runtime/protoimpl"
|
||||
reflect "reflect"
|
||||
sync "sync"
|
||||
unsafe "unsafe"
|
||||
)
|
||||
|
||||
const (
|
||||
// Verify that this generated code is sufficiently up-to-date.
|
||||
_ = protoimpl.EnforceVersion(20 - protoimpl.MinVersion)
|
||||
// Verify that runtime/protoimpl is sufficiently up-to-date.
|
||||
_ = protoimpl.EnforceVersion(protoimpl.MaxVersion - 20)
|
||||
)
|
||||
|
||||
type ServerPacketType int32
|
||||
|
||||
const (
|
||||
ServerPacketType_PING ServerPacketType = 0
|
||||
ServerPacketType_PONG ServerPacketType = 1
|
||||
ServerPacketType_NEW ServerPacketType = 2
|
||||
ServerPacketType_DELETE ServerPacketType = 3
|
||||
ServerPacketType_CONNECT ServerPacketType = 4
|
||||
)
|
||||
|
||||
// Enum value maps for ServerPacketType.
|
||||
var (
|
||||
ServerPacketType_name = map[int32]string{
|
||||
0: "PING",
|
||||
1: "PONG",
|
||||
2: "NEW",
|
||||
3: "DELETE",
|
||||
4: "CONNECT",
|
||||
}
|
||||
ServerPacketType_value = map[string]int32{
|
||||
"PING": 0,
|
||||
"PONG": 1,
|
||||
"NEW": 2,
|
||||
"DELETE": 3,
|
||||
"CONNECT": 4,
|
||||
}
|
||||
)
|
||||
|
||||
func (x ServerPacketType) Enum() *ServerPacketType {
|
||||
p := new(ServerPacketType)
|
||||
*p = x
|
||||
return p
|
||||
}
|
||||
|
||||
func (x ServerPacketType) String() string {
|
||||
return protoimpl.X.EnumStringOf(x.Descriptor(), protoreflect.EnumNumber(x))
|
||||
}
|
||||
|
||||
func (ServerPacketType) Descriptor() protoreflect.EnumDescriptor {
|
||||
return file_server_proto_enumTypes[0].Descriptor()
|
||||
}
|
||||
|
||||
func (ServerPacketType) Type() protoreflect.EnumType {
|
||||
return &file_server_proto_enumTypes[0]
|
||||
}
|
||||
|
||||
func (x ServerPacketType) Number() protoreflect.EnumNumber {
|
||||
return protoreflect.EnumNumber(x)
|
||||
}
|
||||
|
||||
// Deprecated: Use ServerPacketType.Descriptor instead.
|
||||
func (ServerPacketType) EnumDescriptor() ([]byte, []int) {
|
||||
return file_server_proto_rawDescGZIP(), []int{0}
|
||||
}
|
||||
|
||||
type ServerPacket struct {
|
||||
state protoimpl.MessageState `protogen:"open.v1"`
|
||||
Typ ServerPacketType `protobuf:"varint,1,opt,name=typ,proto3,enum=ServerPacketType" json:"typ,omitempty"`
|
||||
Payload []byte `protobuf:"bytes,2,opt,name=payload,proto3,oneof" json:"payload,omitempty"`
|
||||
unknownFields protoimpl.UnknownFields
|
||||
sizeCache protoimpl.SizeCache
|
||||
}
|
||||
|
||||
func (x *ServerPacket) Reset() {
|
||||
*x = ServerPacket{}
|
||||
mi := &file_server_proto_msgTypes[0]
|
||||
ms := protoimpl.X.MessageStateOf(protoimpl.Pointer(x))
|
||||
ms.StoreMessageInfo(mi)
|
||||
}
|
||||
|
||||
func (x *ServerPacket) String() string {
|
||||
return protoimpl.X.MessageStringOf(x)
|
||||
}
|
||||
|
||||
func (*ServerPacket) ProtoMessage() {}
|
||||
|
||||
func (x *ServerPacket) ProtoReflect() protoreflect.Message {
|
||||
mi := &file_server_proto_msgTypes[0]
|
||||
if x != nil {
|
||||
ms := protoimpl.X.MessageStateOf(protoimpl.Pointer(x))
|
||||
if ms.LoadMessageInfo() == nil {
|
||||
ms.StoreMessageInfo(mi)
|
||||
}
|
||||
return ms
|
||||
}
|
||||
return mi.MessageOf(x)
|
||||
}
|
||||
|
||||
// Deprecated: Use ServerPacket.ProtoReflect.Descriptor instead.
|
||||
func (*ServerPacket) Descriptor() ([]byte, []int) {
|
||||
return file_server_proto_rawDescGZIP(), []int{0}
|
||||
}
|
||||
|
||||
func (x *ServerPacket) GetTyp() ServerPacketType {
|
||||
if x != nil {
|
||||
return x.Typ
|
||||
}
|
||||
return ServerPacketType_PING
|
||||
}
|
||||
|
||||
func (x *ServerPacket) GetPayload() []byte {
|
||||
if x != nil {
|
||||
return x.Payload
|
||||
}
|
||||
return nil
|
||||
}
|
||||
|
||||
type ServerNewPacket struct {
|
||||
state protoimpl.MessageState `protogen:"open.v1"`
|
||||
ServerId string `protobuf:"bytes,1,opt,name=serverId,proto3" json:"serverId,omitempty"`
|
||||
Name string `protobuf:"bytes,2,opt,name=name,proto3" json:"name,omitempty"`
|
||||
Host string `protobuf:"bytes,3,opt,name=host,proto3" json:"host,omitempty"`
|
||||
Port int32 `protobuf:"varint,4,opt,name=port,proto3" json:"port,omitempty"`
|
||||
Motd string `protobuf:"bytes,5,opt,name=motd,proto3" json:"motd,omitempty"`
|
||||
Group string `protobuf:"bytes,6,opt,name=group,proto3" json:"group,omitempty"`
|
||||
unknownFields protoimpl.UnknownFields
|
||||
sizeCache protoimpl.SizeCache
|
||||
}
|
||||
|
||||
func (x *ServerNewPacket) Reset() {
|
||||
*x = ServerNewPacket{}
|
||||
mi := &file_server_proto_msgTypes[1]
|
||||
ms := protoimpl.X.MessageStateOf(protoimpl.Pointer(x))
|
||||
ms.StoreMessageInfo(mi)
|
||||
}
|
||||
|
||||
func (x *ServerNewPacket) String() string {
|
||||
return protoimpl.X.MessageStringOf(x)
|
||||
}
|
||||
|
||||
func (*ServerNewPacket) ProtoMessage() {}
|
||||
|
||||
func (x *ServerNewPacket) ProtoReflect() protoreflect.Message {
|
||||
mi := &file_server_proto_msgTypes[1]
|
||||
if x != nil {
|
||||
ms := protoimpl.X.MessageStateOf(protoimpl.Pointer(x))
|
||||
if ms.LoadMessageInfo() == nil {
|
||||
ms.StoreMessageInfo(mi)
|
||||
}
|
||||
return ms
|
||||
}
|
||||
return mi.MessageOf(x)
|
||||
}
|
||||
|
||||
// Deprecated: Use ServerNewPacket.ProtoReflect.Descriptor instead.
|
||||
func (*ServerNewPacket) Descriptor() ([]byte, []int) {
|
||||
return file_server_proto_rawDescGZIP(), []int{1}
|
||||
}
|
||||
|
||||
func (x *ServerNewPacket) GetServerId() string {
|
||||
if x != nil {
|
||||
return x.ServerId
|
||||
}
|
||||
return ""
|
||||
}
|
||||
|
||||
func (x *ServerNewPacket) GetName() string {
|
||||
if x != nil {
|
||||
return x.Name
|
||||
}
|
||||
return ""
|
||||
}
|
||||
|
||||
func (x *ServerNewPacket) GetHost() string {
|
||||
if x != nil {
|
||||
return x.Host
|
||||
}
|
||||
return ""
|
||||
}
|
||||
|
||||
func (x *ServerNewPacket) GetPort() int32 {
|
||||
if x != nil {
|
||||
return x.Port
|
||||
}
|
||||
return 0
|
||||
}
|
||||
|
||||
func (x *ServerNewPacket) GetMotd() string {
|
||||
if x != nil {
|
||||
return x.Motd
|
||||
}
|
||||
return ""
|
||||
}
|
||||
|
||||
func (x *ServerNewPacket) GetGroup() string {
|
||||
if x != nil {
|
||||
return x.Group
|
||||
}
|
||||
return ""
|
||||
}
|
||||
|
||||
type ServerDeletePacket struct {
|
||||
state protoimpl.MessageState `protogen:"open.v1"`
|
||||
ServerId string `protobuf:"bytes,1,opt,name=serverId,proto3" json:"serverId,omitempty"`
|
||||
unknownFields protoimpl.UnknownFields
|
||||
sizeCache protoimpl.SizeCache
|
||||
}
|
||||
|
||||
func (x *ServerDeletePacket) Reset() {
|
||||
*x = ServerDeletePacket{}
|
||||
mi := &file_server_proto_msgTypes[2]
|
||||
ms := protoimpl.X.MessageStateOf(protoimpl.Pointer(x))
|
||||
ms.StoreMessageInfo(mi)
|
||||
}
|
||||
|
||||
func (x *ServerDeletePacket) String() string {
|
||||
return protoimpl.X.MessageStringOf(x)
|
||||
}
|
||||
|
||||
func (*ServerDeletePacket) ProtoMessage() {}
|
||||
|
||||
func (x *ServerDeletePacket) ProtoReflect() protoreflect.Message {
|
||||
mi := &file_server_proto_msgTypes[2]
|
||||
if x != nil {
|
||||
ms := protoimpl.X.MessageStateOf(protoimpl.Pointer(x))
|
||||
if ms.LoadMessageInfo() == nil {
|
||||
ms.StoreMessageInfo(mi)
|
||||
}
|
||||
return ms
|
||||
}
|
||||
return mi.MessageOf(x)
|
||||
}
|
||||
|
||||
// Deprecated: Use ServerDeletePacket.ProtoReflect.Descriptor instead.
|
||||
func (*ServerDeletePacket) Descriptor() ([]byte, []int) {
|
||||
return file_server_proto_rawDescGZIP(), []int{2}
|
||||
}
|
||||
|
||||
func (x *ServerDeletePacket) GetServerId() string {
|
||||
if x != nil {
|
||||
return x.ServerId
|
||||
}
|
||||
return ""
|
||||
}
|
||||
|
||||
type ServerConnectPacket struct {
|
||||
state protoimpl.MessageState `protogen:"open.v1"`
|
||||
ServerId string `protobuf:"bytes,1,opt,name=serverId,proto3" json:"serverId,omitempty"`
|
||||
Player string `protobuf:"bytes,2,opt,name=player,proto3" json:"player,omitempty"`
|
||||
unknownFields protoimpl.UnknownFields
|
||||
sizeCache protoimpl.SizeCache
|
||||
}
|
||||
|
||||
func (x *ServerConnectPacket) Reset() {
|
||||
*x = ServerConnectPacket{}
|
||||
mi := &file_server_proto_msgTypes[3]
|
||||
ms := protoimpl.X.MessageStateOf(protoimpl.Pointer(x))
|
||||
ms.StoreMessageInfo(mi)
|
||||
}
|
||||
|
||||
func (x *ServerConnectPacket) String() string {
|
||||
return protoimpl.X.MessageStringOf(x)
|
||||
}
|
||||
|
||||
func (*ServerConnectPacket) ProtoMessage() {}
|
||||
|
||||
func (x *ServerConnectPacket) ProtoReflect() protoreflect.Message {
|
||||
mi := &file_server_proto_msgTypes[3]
|
||||
if x != nil {
|
||||
ms := protoimpl.X.MessageStateOf(protoimpl.Pointer(x))
|
||||
if ms.LoadMessageInfo() == nil {
|
||||
ms.StoreMessageInfo(mi)
|
||||
}
|
||||
return ms
|
||||
}
|
||||
return mi.MessageOf(x)
|
||||
}
|
||||
|
||||
// Deprecated: Use ServerConnectPacket.ProtoReflect.Descriptor instead.
|
||||
func (*ServerConnectPacket) Descriptor() ([]byte, []int) {
|
||||
return file_server_proto_rawDescGZIP(), []int{3}
|
||||
}
|
||||
|
||||
func (x *ServerConnectPacket) GetServerId() string {
|
||||
if x != nil {
|
||||
return x.ServerId
|
||||
}
|
||||
return ""
|
||||
}
|
||||
|
||||
func (x *ServerConnectPacket) GetPlayer() string {
|
||||
if x != nil {
|
||||
return x.Player
|
||||
}
|
||||
return ""
|
||||
}
|
||||
|
||||
var File_server_proto protoreflect.FileDescriptor
|
||||
|
||||
const file_server_proto_rawDesc = "" +
|
||||
"\n" +
|
||||
"\fserver.proto\"^\n" +
|
||||
"\fServerPacket\x12#\n" +
|
||||
"\x03typ\x18\x01 \x01(\x0e2\x11.ServerPacketTypeR\x03typ\x12\x1d\n" +
|
||||
"\apayload\x18\x02 \x01(\fH\x00R\apayload\x88\x01\x01B\n" +
|
||||
"\n" +
|
||||
"\b_payload\"\x93\x01\n" +
|
||||
"\x0fServerNewPacket\x12\x1a\n" +
|
||||
"\bserverId\x18\x01 \x01(\tR\bserverId\x12\x12\n" +
|
||||
"\x04name\x18\x02 \x01(\tR\x04name\x12\x12\n" +
|
||||
"\x04host\x18\x03 \x01(\tR\x04host\x12\x12\n" +
|
||||
"\x04port\x18\x04 \x01(\x05R\x04port\x12\x12\n" +
|
||||
"\x04motd\x18\x05 \x01(\tR\x04motd\x12\x14\n" +
|
||||
"\x05group\x18\x06 \x01(\tR\x05group\"0\n" +
|
||||
"\x12ServerDeletePacket\x12\x1a\n" +
|
||||
"\bserverId\x18\x01 \x01(\tR\bserverId\"I\n" +
|
||||
"\x13ServerConnectPacket\x12\x1a\n" +
|
||||
"\bserverId\x18\x01 \x01(\tR\bserverId\x12\x16\n" +
|
||||
"\x06player\x18\x02 \x01(\tR\x06player*H\n" +
|
||||
"\x10ServerPacketType\x12\b\n" +
|
||||
"\x04PING\x10\x00\x12\b\n" +
|
||||
"\x04PONG\x10\x01\x12\a\n" +
|
||||
"\x03NEW\x10\x02\x12\n" +
|
||||
"\n" +
|
||||
"\x06DELETE\x10\x03\x12\v\n" +
|
||||
"\aCONNECT\x10\x04B\x12Z\x10router/minecraftb\x06proto3"
|
||||
|
||||
var (
|
||||
file_server_proto_rawDescOnce sync.Once
|
||||
file_server_proto_rawDescData []byte
|
||||
)
|
||||
|
||||
func file_server_proto_rawDescGZIP() []byte {
|
||||
file_server_proto_rawDescOnce.Do(func() {
|
||||
file_server_proto_rawDescData = protoimpl.X.CompressGZIP(unsafe.Slice(unsafe.StringData(file_server_proto_rawDesc), len(file_server_proto_rawDesc)))
|
||||
})
|
||||
return file_server_proto_rawDescData
|
||||
}
|
||||
|
||||
var file_server_proto_enumTypes = make([]protoimpl.EnumInfo, 1)
|
||||
var file_server_proto_msgTypes = make([]protoimpl.MessageInfo, 4)
|
||||
var file_server_proto_goTypes = []any{
|
||||
(ServerPacketType)(0), // 0: ServerPacketType
|
||||
(*ServerPacket)(nil), // 1: ServerPacket
|
||||
(*ServerNewPacket)(nil), // 2: ServerNewPacket
|
||||
(*ServerDeletePacket)(nil), // 3: ServerDeletePacket
|
||||
(*ServerConnectPacket)(nil), // 4: ServerConnectPacket
|
||||
}
|
||||
var file_server_proto_depIdxs = []int32{
|
||||
0, // 0: ServerPacket.typ:type_name -> ServerPacketType
|
||||
1, // [1:1] is the sub-list for method output_type
|
||||
1, // [1:1] is the sub-list for method input_type
|
||||
1, // [1:1] is the sub-list for extension type_name
|
||||
1, // [1:1] is the sub-list for extension extendee
|
||||
0, // [0:1] is the sub-list for field type_name
|
||||
}
|
||||
|
||||
func init() { file_server_proto_init() }
|
||||
func file_server_proto_init() {
|
||||
if File_server_proto != nil {
|
||||
return
|
||||
}
|
||||
file_server_proto_msgTypes[0].OneofWrappers = []any{}
|
||||
type x struct{}
|
||||
out := protoimpl.TypeBuilder{
|
||||
File: protoimpl.DescBuilder{
|
||||
GoPackagePath: reflect.TypeOf(x{}).PkgPath(),
|
||||
RawDescriptor: unsafe.Slice(unsafe.StringData(file_server_proto_rawDesc), len(file_server_proto_rawDesc)),
|
||||
NumEnums: 1,
|
||||
NumMessages: 4,
|
||||
NumExtensions: 0,
|
||||
NumServices: 0,
|
||||
},
|
||||
GoTypes: file_server_proto_goTypes,
|
||||
DependencyIndexes: file_server_proto_depIdxs,
|
||||
EnumInfos: file_server_proto_enumTypes,
|
||||
MessageInfos: file_server_proto_msgTypes,
|
||||
}.Build()
|
||||
File_server_proto = out.File
|
||||
file_server_proto_goTypes = nil
|
||||
file_server_proto_depIdxs = nil
|
||||
}
|
||||
365
router/minecraft/servers.go
Normal file
365
router/minecraft/servers.go
Normal file
@ -0,0 +1,365 @@
|
||||
package minecraft
|
||||
|
||||
import (
|
||||
"coreapp/util"
|
||||
"coreapp/util/ws"
|
||||
"github.com/gin-gonic/gin"
|
||||
"github.com/gorilla/websocket"
|
||||
"google.golang.org/protobuf/proto"
|
||||
"math/rand"
|
||||
"net/http"
|
||||
"strings"
|
||||
"sync"
|
||||
"time"
|
||||
)
|
||||
|
||||
type ServerType string
|
||||
|
||||
const (
|
||||
ServerTypeUnknown ServerType = ""
|
||||
ServerTypeLobby ServerType = "LOBBY"
|
||||
ServerTypeRoom ServerType = "ROOM"
|
||||
ServerTypeLimbo ServerType = "LIMBO"
|
||||
ServerTypeSystem ServerType = "SYSTEM"
|
||||
)
|
||||
|
||||
func (typ ServerType) Prefix() string {
|
||||
switch typ {
|
||||
case ServerTypeLobby:
|
||||
return "L"
|
||||
case ServerTypeRoom:
|
||||
return "mini"
|
||||
case ServerTypeLimbo:
|
||||
return ""
|
||||
case ServerTypeSystem:
|
||||
return "S"
|
||||
case ServerTypeUnknown:
|
||||
default:
|
||||
return "U"
|
||||
}
|
||||
return "U"
|
||||
}
|
||||
|
||||
type Server struct {
|
||||
Id string `json:"id"`
|
||||
Name string `json:"name"`
|
||||
Group string `json:"group"`
|
||||
Host string `json:"host"`
|
||||
Port int `json:"port"`
|
||||
Type ServerType `json:"type"`
|
||||
Motd string `json:"motd"`
|
||||
LastHB time.Time
|
||||
}
|
||||
|
||||
var servers = sync.Map{}
|
||||
var serverProxyConn *ws.Conn
|
||||
var serverProxyLastHB time.Time
|
||||
|
||||
func ServerMiddleware() gin.HandlerFunc {
|
||||
return func(c *gin.Context) {
|
||||
if strings.HasSuffix(c.Request.URL.Path, "ws") || (c.Request.Method == "POST" && !strings.HasSuffix(c.Request.URL.Path, "/register")) {
|
||||
if c.GetHeader("Authorization") != "NEXTCRAFT" {
|
||||
c.AbortWithStatusJSON(401, gin.H{
|
||||
"code": 1000,
|
||||
"message": "Who are you?",
|
||||
})
|
||||
return
|
||||
}
|
||||
serverId := c.Query("id")
|
||||
if serverId == "" {
|
||||
c.AbortWithStatusJSON(400, gin.H{
|
||||
"code": -1,
|
||||
"message": "请提供服务器ID",
|
||||
})
|
||||
return
|
||||
}
|
||||
if serverId != "PROXY" {
|
||||
if _, ok := servers.Load(serverId); !ok {
|
||||
c.AbortWithStatusJSON(404, gin.H{
|
||||
"code": -1,
|
||||
"message": "服务器不存在",
|
||||
})
|
||||
return
|
||||
}
|
||||
}
|
||||
c.Set("serverId", serverId)
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
func ServerList(c *gin.Context) {
|
||||
serverList := make(map[string]*Server)
|
||||
|
||||
// 遍历 sync.Map
|
||||
servers.Range(func(key, value interface{}) bool {
|
||||
serverList[key.(string)] = value.(*Server)
|
||||
return true // 继续遍历
|
||||
})
|
||||
|
||||
c.JSON(200, gin.H{
|
||||
"code": 0,
|
||||
"message": "success",
|
||||
"data": serverList,
|
||||
})
|
||||
}
|
||||
|
||||
type ServerRegisterData struct {
|
||||
Name string `json:"name"`
|
||||
Host string `json:"host"`
|
||||
Port int `json:"port"`
|
||||
Type ServerType `json:"type"`
|
||||
Group string `json:"group"`
|
||||
Motd string `json:"motd"`
|
||||
}
|
||||
|
||||
func ServerRegister(c *gin.Context) {
|
||||
data := &ServerRegisterData{}
|
||||
if err := c.ShouldBindJSON(data); err != nil {
|
||||
c.JSON(400, gin.H{
|
||||
"code": -1,
|
||||
"message": "请提供正确的注册信息",
|
||||
})
|
||||
return
|
||||
}
|
||||
|
||||
if data.Name == "" {
|
||||
data.Name = "AUTO"
|
||||
}
|
||||
|
||||
if data.Type == "" {
|
||||
c.JSON(400, gin.H{
|
||||
"code": -1,
|
||||
"message": "请提供正确的服务器类型{LOBBY, ROOM, SYSTEM, LIMBO, UNKNOWN}",
|
||||
})
|
||||
return
|
||||
}
|
||||
|
||||
if data.Host == "" || data.Port == 0 {
|
||||
c.JSON(400, gin.H{
|
||||
"code": -1,
|
||||
"message": "请提供正确的服务器IP和端口",
|
||||
})
|
||||
return
|
||||
}
|
||||
|
||||
if data.Group == "" {
|
||||
data.Group = "default"
|
||||
}
|
||||
|
||||
server := &Server{
|
||||
Id: generateServerId(data.Type.Prefix()),
|
||||
Host: data.Host,
|
||||
Port: data.Port,
|
||||
Type: data.Type,
|
||||
Group: data.Group,
|
||||
LastHB: time.Now(),
|
||||
}
|
||||
|
||||
if data.Name == "AUTO" {
|
||||
server.Name = server.Id
|
||||
} else {
|
||||
server.Name = data.Name
|
||||
}
|
||||
|
||||
servers.Store(server.Id, server) // 使用 Store 替代原来的 map 写入
|
||||
|
||||
newPkt := &ServerNewPacket{
|
||||
ServerId: server.Id,
|
||||
Name: server.Name,
|
||||
Host: server.Host,
|
||||
Port: int32(server.Port),
|
||||
Group: server.Group,
|
||||
Motd: server.Motd,
|
||||
}
|
||||
|
||||
if serverProxyConn != nil {
|
||||
buf, err := serverWrapPacket(ServerPacketType_NEW, newPkt)
|
||||
if err != nil {
|
||||
panic(err)
|
||||
}
|
||||
serverProxyConn.WriteChan <- ws.Binary(buf)
|
||||
}
|
||||
|
||||
c.JSON(200, gin.H{
|
||||
"code": 0,
|
||||
"message": "success",
|
||||
"data": server,
|
||||
})
|
||||
}
|
||||
|
||||
func ServerUnregister(c *gin.Context) {
|
||||
serverId := c.GetString("serverId")
|
||||
deletePkt := &ServerDeletePacket{
|
||||
ServerId: serverId,
|
||||
}
|
||||
if serverProxyConn != nil {
|
||||
buf, err := serverWrapPacket(ServerPacketType_DELETE, deletePkt)
|
||||
if err != nil {
|
||||
panic(err)
|
||||
}
|
||||
serverProxyConn.WriteChan <- ws.Binary(buf)
|
||||
}
|
||||
servers.Delete(serverId) // 直接删除
|
||||
c.JSON(200, gin.H{
|
||||
"code": 0,
|
||||
"message": "success",
|
||||
})
|
||||
}
|
||||
|
||||
// ServerMessage 函数用于处理服务器消息
|
||||
func ServerMessage(c *gin.Context) {
|
||||
conn, err := serverUpgrader.Upgrade(c.Writer, c.Request, nil)
|
||||
if err != nil {
|
||||
c.JSON(http.StatusInternalServerError, gin.H{
|
||||
"code": -666,
|
||||
"message": "WebSocket升级失败",
|
||||
})
|
||||
panic(c.Error(err))
|
||||
return
|
||||
}
|
||||
wrapped := ws.Wrap(conn)
|
||||
go wrapped.ReadLoop()
|
||||
go wrapped.WriteLoop()
|
||||
if c.GetString("serverId") == "PROXY" {
|
||||
serverProxyConn = wrapped
|
||||
serverProxyLastHB = time.Now()
|
||||
go serverProxyReadLoop(wrapped)
|
||||
} else {
|
||||
go serverReadLoop(wrapped, c.GetString("serverId"))
|
||||
}
|
||||
}
|
||||
|
||||
func serverReadLoop(conn *ws.Conn, serverId string) {
|
||||
for {
|
||||
// 使用 Load 读取，避免并发问题
|
||||
serverVal, ok := servers.Load(serverId)
|
||||
if !ok {
|
||||
conn.Close()
|
||||
return
|
||||
}
|
||||
server := serverVal.(*Server)
|
||||
|
||||
select {
|
||||
case msg := <-conn.ReadChan:
|
||||
if msg == nil {
|
||||
return
|
||||
}
|
||||
pkt := &ServerPacket{}
|
||||
err := proto.Unmarshal(msg, pkt)
|
||||
if err != nil {
|
||||
panic(err)
|
||||
}
|
||||
switch pkt.GetTyp() {
|
||||
case ServerPacketType_PING:
|
||||
// 更新 LastHB（由于 Server 本身是结构体指针，可以直接修改）
|
||||
server.LastHB = time.Now()
|
||||
|
||||
resp := &ServerPacket{
|
||||
Typ: ServerPacketType_PONG,
|
||||
}
|
||||
buf, err := proto.Marshal(resp)
|
||||
if err != nil {
|
||||
panic(err)
|
||||
}
|
||||
conn.WriteChan <- ws.Binary(buf)
|
||||
case ServerPacketType_CONNECT:
|
||||
if serverProxyConn != nil {
|
||||
serverProxyConn.WriteChan <- ws.Binary(msg)
|
||||
}
|
||||
}
|
||||
default:
|
||||
if conn.IsClosed() {
|
||||
return
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
func serverProxyReadLoop(conn *ws.Conn) {
|
||||
defer func() {
|
||||
serverProxyConn = nil
|
||||
}()
|
||||
for {
|
||||
select {
|
||||
case msg := <-conn.ReadChan:
|
||||
if msg == nil {
|
||||
return
|
||||
}
|
||||
pkt := &ServerPacket{}
|
||||
err := proto.Unmarshal(msg, pkt)
|
||||
if err != nil {
|
||||
panic(err)
|
||||
}
|
||||
switch pkt.GetTyp() {
|
||||
case ServerPacketType_PING:
|
||||
serverProxyLastHB = time.Now()
|
||||
resp := &ServerPacket{
|
||||
Typ: ServerPacketType_PONG,
|
||||
}
|
||||
buf, err := proto.Marshal(resp)
|
||||
if err != nil {
|
||||
panic(err)
|
||||
}
|
||||
conn.WriteChan <- ws.Binary(buf)
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
var serverUpgrader = websocket.Upgrader{
|
||||
ReadBufferSize: 1024,
|
||||
WriteBufferSize: 1024,
|
||||
}
|
||||
|
||||
func generateServerId(prefix string) string {
|
||||
return string(prefix) + util.RandStringNumber(rand.Intn(2)+1) + strings.ToUpper(util.RandStringAlphabet(rand.Intn(2)+1))
|
||||
}
|
||||
|
||||
var serverHBTicker = time.NewTicker(time.Second * 5)
|
||||
|
||||
func init() {
|
||||
go func() {
|
||||
for range serverHBTicker.C {
|
||||
// 遍历所有服务器，检查心跳
|
||||
servers.Range(func(key, value interface{}) bool {
|
||||
server := value.(*Server)
|
||||
if time.Since(server.LastHB) > time.Second*10 {
|
||||
deletePkt := &ServerDeletePacket{
|
||||
ServerId: server.Id,
|
||||
}
|
||||
if serverProxyConn != nil {
|
||||
buf, err := serverWrapPacket(ServerPacketType_DELETE, deletePkt)
|
||||
if err != nil {
|
||||
panic(err)
|
||||
}
|
||||
serverProxyConn.WriteChan <- ws.Binary(buf)
|
||||
}
|
||||
servers.Delete(key) // 超时则删除
|
||||
}
|
||||
return true // 继续遍历
|
||||
})
|
||||
if time.Since(serverProxyLastHB) > time.Second*10 && serverProxyConn != nil {
|
||||
serverProxyConn.Close()
|
||||
}
|
||||
}
|
||||
}()
|
||||
}
|
||||
|
||||
func serverWrapPacket(typ ServerPacketType, payload proto.Message) ([]byte, error) {
|
||||
pkt := &ServerPacket{
|
||||
Typ: typ,
|
||||
Payload: nil,
|
||||
}
|
||||
if payload != nil {
|
||||
buf, err := proto.Marshal(payload)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
pkt.Payload = buf
|
||||
}
|
||||
buf, err := proto.Marshal(pkt)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
return buf, nil
|
||||
}
|
||||
20
router/router.go
Normal file
20
router/router.go
Normal file
@ -0,0 +1,20 @@
|
||||
package router
|
||||
|
||||
import (
|
||||
"coreapp/router/minecraft"
|
||||
"github.com/gin-gonic/gin"
|
||||
)
|
||||
|
||||
func Route(server *gin.Engine) {
|
||||
rootGroup := server.Group("/")
|
||||
|
||||
minecraftGroup := rootGroup.Group("/minecraft")
|
||||
{
|
||||
servers := minecraftGroup.Group("/servers")
|
||||
servers.Use(minecraft.ServerMiddleware())
|
||||
servers.GET("/list", minecraft.ServerList)
|
||||
servers.POST("/register", minecraft.ServerRegister)
|
||||
servers.POST("/unregister", minecraft.ServerUnregister)
|
||||
servers.GET("/ws", minecraft.ServerMessage)
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user