kercylan98
71 lines
1.7 KiB
Go
71 lines
1.7 KiB
Go
package corss
|
|
|
|
import (
|
|
"encoding/json"
|
|
"fmt"
|
|
"github.com/kercylan98/minotaur/server"
|
|
"github.com/kercylan98/minotaur/utils/log"
|
|
"github.com/kercylan98/minotaur/utils/synchronization"
|
|
"github.com/nats-io/nats.go"
|
|
"go.uber.org/zap"
|
|
)
|
|
|
|
func NewNats(url string, options ...nats.Option) *Nats {
|
|
return &Nats{
|
|
url: url,
|
|
subject: "MINOTAUR_CROSS",
|
|
options: options,
|
|
messagePool: synchronization.NewPool[*Message](1024*100, func() *Message {
|
|
return new(Message)
|
|
}, func(data *Message) {
|
|
data.ServerId = 0
|
|
data.Packet = nil
|
|
}),
|
|
}
|
|
}
|
|
|
|
func NewNatsWithSubject(url, subject string, options ...nats.Option) *Nats {
|
|
n := NewNats(url, options...)
|
|
n.subject = subject
|
|
return n
|
|
}
|
|
|
|
type Nats struct {
|
|
conn *nats.Conn
|
|
url string
|
|
subject string
|
|
realSubject string
|
|
options []nats.Option
|
|
messagePool *synchronization.Pool[*Message]
|
|
}
|
|
|
|
func (slf *Nats) Init(server *server.Server, packetHandle func(serverId int64, packet []byte)) (err error) {
|
|
slf.realSubject = fmt.Sprintf("%s_%d", slf.subject, server.GetID())
|
|
slf.conn, err = nats.Connect(slf.url, slf.options...)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
_, err = slf.conn.Subscribe(slf.realSubject, func(msg *nats.Msg) {
|
|
message := slf.messagePool.Get()
|
|
defer slf.messagePool.Release(message)
|
|
if err := json.Unmarshal(msg.Data, &message); err != nil {
|
|
log.Error("Cross.Nats", zap.Error(err))
|
|
return
|
|
}
|
|
packetHandle(message.ServerId, message.Packet)
|
|
})
|
|
return err
|
|
}
|
|
|
|
func (slf *Nats) PushMessage(serverId int64, packet []byte) error {
|
|
message := slf.messagePool.Get()
|
|
defer slf.messagePool.Release(message)
|
|
message.ServerId = serverId
|
|
message.Packet = packet
|
|
data, err := json.Marshal(message)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
return slf.conn.Publish(slf.realSubject, data)
|
|
}
|