149 lines
5.3 KiB
Go
149 lines
5.3 KiB
Go
package server
|
||
|
||
import (
|
||
"github.com/kercylan98/minotaur/utils/hash"
|
||
"github.com/kercylan98/minotaur/utils/log"
|
||
"github.com/pkg/errors"
|
||
"go.uber.org/zap"
|
||
"google.golang.org/grpc"
|
||
"runtime/debug"
|
||
)
|
||
|
||
const (
|
||
// WebsocketMessageTypeText 表示文本数据消息。文本消息负载被解释为 UTF-8 编码的文本数据
|
||
WebsocketMessageTypeText = 1
|
||
// WebsocketMessageTypeBinary 表示二进制数据消息
|
||
WebsocketMessageTypeBinary = 2
|
||
// WebsocketMessageTypeClose 表示关闭控制消息。可选消息负载包含数字代码和文本。使用 FormatCloseMessage 函数来格式化关闭消息负载
|
||
WebsocketMessageTypeClose = 8
|
||
// WebsocketMessageTypePing 表示 ping 控制消息。可选的消息负载是 UTF-8 编码的文本
|
||
WebsocketMessageTypePing = 9
|
||
// WebsocketMessageTypePong 表示一个 pong 控制消息。可选的消息负载是 UTF-8 编码的文本
|
||
WebsocketMessageTypePong = 10
|
||
)
|
||
|
||
type Option func(srv *Server)
|
||
type CrossRegisterHandle func(server *Server) error
|
||
|
||
// WithCross 通过跨服的方式创建服务器
|
||
// - CrossQueue: 跨服队列是用于接收和发送跨服消息的队列接口
|
||
func WithCross(serverId int64, queues ...CrossQueue) Option {
|
||
return func(srv *Server) {
|
||
srv.id = &serverId
|
||
srv.RegStartFinishEvent(func(srv *Server) {
|
||
srv.cross = new(cross)
|
||
if err := srv.cross.Run(srv, queues...); err != nil {
|
||
srv.PushMessage(MessageTypeError, errors.WithMessage(err, string(debug.Stack())), MessageErrorActionShutdown)
|
||
return
|
||
}
|
||
log.Info("Server", zap.Int64("CrossID", serverId))
|
||
})
|
||
}
|
||
}
|
||
|
||
// WithConnectPacketDiversion 通过连接数据包消息分流的方式创建服务器
|
||
// - 连接消息分流后数据包消息将会从其他消息类型中独立出来，并且由多个消息管道及协程进行处理
|
||
// - 默认不会进行消息分流
|
||
// - 需要注意并发编程
|
||
func WithConnectPacketDiversion(diversionNumber, channelSize int) Option {
|
||
return func(srv *Server) {
|
||
if srv.network == NetworkHttp || srv.network == NetworkGRPC {
|
||
log.Warn("WithConnectPacketDiversion", zap.String("Network", string(srv.network)), zap.Error(ErrOnlySupportSocket))
|
||
return
|
||
}
|
||
srv.diversionMessageChannels = make([]chan *message, diversionNumber)
|
||
srv.diversionConsistency = hash.NewConsistency(3)
|
||
for i := 0; i < diversionNumber; i++ {
|
||
srv.diversionMessageChannels[i] = make(chan *message, channelSize)
|
||
srv.diversionConsistency.AddNode(i + 1)
|
||
}
|
||
}
|
||
}
|
||
|
||
// WithTLS 通过安全传输层协议TLS创建服务器
|
||
// - 支持：Http、Websocket
|
||
func WithTLS(certFile, keyFile string) Option {
|
||
return func(srv *Server) {
|
||
switch srv.network {
|
||
case NetworkHttp, NetworkWebsocket, NetworkTCP, NetworkTCP4, NetworkTCP6:
|
||
srv.certFile = certFile
|
||
srv.keyFile = keyFile
|
||
}
|
||
}
|
||
}
|
||
|
||
// WithGRPCServerOptions 通过GRPC的可选项创建GRPC服务器
|
||
func WithGRPCServerOptions(options ...grpc.ServerOption) Option {
|
||
return func(srv *Server) {
|
||
if srv.network != NetworkGRPC {
|
||
return
|
||
}
|
||
srv.grpcServer = grpc.NewServer(options...)
|
||
}
|
||
}
|
||
|
||
// WithProd 通过生产模式运行服务器
|
||
func WithProd() Option {
|
||
return func(srv *Server) {
|
||
srv.prod = true
|
||
}
|
||
}
|
||
|
||
// WithWebsocketWriteMessageType 设置客户端写入的Websocket消息类型
|
||
// - 默认： WebsocketMessageTypeBinary
|
||
func WithWebsocketWriteMessageType(messageType int) Option {
|
||
return func(srv *Server) {
|
||
switch messageType {
|
||
case WebsocketMessageTypeText, WebsocketMessageTypeBinary, WebsocketMessageTypeClose, WebsocketMessageTypePing, WebsocketMessageTypePong:
|
||
srv.websocketWriteMessageType = messageType
|
||
default:
|
||
log.Warn("WithWebsocketWriteMessageType", zap.Int("MessageType", messageType), zap.Error(ErrWebsocketMessageTypeException))
|
||
}
|
||
}
|
||
}
|
||
|
||
// WithWebsocketMessageType 设置仅支持特定类型的Websocket消息
|
||
func WithWebsocketMessageType(messageTypes ...int) Option {
|
||
return func(srv *Server) {
|
||
if srv.network != NetworkWebsocket {
|
||
log.Warn("WitchWebsocketMessageType", zap.String("Network", string(srv.network)), zap.Error(ErrNotWebsocketUseMessageType))
|
||
return
|
||
}
|
||
var supports = make(map[int]bool)
|
||
for _, messageType := range messageTypes {
|
||
switch messageType {
|
||
case WebsocketMessageTypeText, WebsocketMessageTypeBinary, WebsocketMessageTypeClose, WebsocketMessageTypePing, WebsocketMessageTypePong:
|
||
supports[messageType] = true
|
||
default:
|
||
log.Warn("WitchWebsocketMessageType", zap.Int("MessageType", messageType), zap.Error(ErrWebsocketMessageTypeException))
|
||
}
|
||
}
|
||
srv.supportMessageTypes = supports
|
||
}
|
||
}
|
||
|
||
// WithMessageBufferSize 通过特定的消息缓冲池大小运行服务器
|
||
// - 默认大小为 4096 * 1024
|
||
// - 消息数量超出这个值的时候，消息处理将会造成更大的开销（频繁创建新的结构体），同时服务器将输出警告内容
|
||
func WithMessageBufferSize(size int) Option {
|
||
return func(srv *Server) {
|
||
if size <= 0 {
|
||
return
|
||
}
|
||
srv.messagePoolSize = size
|
||
}
|
||
}
|
||
|
||
// WithMultiCore 通过特定核心数量运行服务器，默认为单核
|
||
// - count > 1 的情况下，将会有对应数量的 goroutine 来处理消息
|
||
// - 注意：HTTP和GRPC网络模式下不会生效
|
||
func WithMultiCore(count int) Option {
|
||
return func(srv *Server) {
|
||
srv.core = count
|
||
if srv.core < 1 {
|
||
log.Warn("WithMultiCore", zap.Int("count", count), zap.String("tips", "wrong core count configuration, corrected to 1, currently in single-core mode"))
|
||
srv.core = 1
|
||
}
|
||
}
|
||
}
|