Refactor event system
This commit is contained in:
+1
-1
@@ -15,7 +15,7 @@ func Connect(addr string) (ret *proto.Conn, err error) {
|
|||||||
return
|
return
|
||||||
}
|
}
|
||||||
ret.SetState(proto_base.Handshaking)
|
ret.SetState(proto_base.Handshaking)
|
||||||
ret.CompressionThreshold = -1
|
ret.SetCompressionThreshold(-1)
|
||||||
ret.Actor = proto_base.Client
|
ret.Actor = proto_base.Client
|
||||||
ret.Conn, err = net.Dial("tcp", addr)
|
ret.Conn, err = net.Dial("tcp", addr)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
|
|||||||
+1
-1
@@ -80,7 +80,7 @@ func (s *LoginClient) OnStart(_ event.Start) (err error) {
|
|||||||
}
|
}
|
||||||
|
|
||||||
func (s *LoginClient) OnCompress(compress *v1_21_8.LoginToClientPacketCompress) (err error) {
|
func (s *LoginClient) OnCompress(compress *v1_21_8.LoginToClientPacketCompress) (err error) {
|
||||||
s.CompressionThreshold = compress.Threshold
|
s.SetCompressionThreshold(compress.Threshold)
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|||||||
+26
-1
@@ -2,16 +2,27 @@
|
|||||||
package event
|
package event
|
||||||
|
|
||||||
import (
|
import (
|
||||||
|
"context"
|
||||||
"errors"
|
"errors"
|
||||||
"fmt"
|
"fmt"
|
||||||
"log/slog"
|
"log/slog"
|
||||||
"reflect"
|
"reflect"
|
||||||
"strings"
|
"strings"
|
||||||
|
"time"
|
||||||
|
)
|
||||||
|
|
||||||
|
const (
|
||||||
|
LoopsPerSecond = 100 // i dunno seems reasonable
|
||||||
|
TimePerLoop = time.Second / LoopsPerSecond
|
||||||
)
|
)
|
||||||
|
|
||||||
type Loop struct {
|
type Loop struct {
|
||||||
handlerFunctions map[reflect.Type][]reflect.Value
|
handlerFunctions map[reflect.Type][]reflect.Value
|
||||||
Handlers []any
|
Handlers []any
|
||||||
|
|
||||||
|
Ctx context.Context
|
||||||
|
ErrChan chan error
|
||||||
|
cancel context.CancelFunc
|
||||||
}
|
}
|
||||||
|
|
||||||
type Unhandled struct {
|
type Unhandled struct {
|
||||||
@@ -107,15 +118,18 @@ func (l *Loop) Fire(event any) (err error) {
|
|||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
|
||||||
func (l *Loop) Start() (err error) {
|
func (l *Loop) startLoop() (err error) {
|
||||||
_ = *l // exit early on nil loop
|
_ = *l // exit early on nil loop
|
||||||
l.UpdateHandlerFunctions()
|
l.UpdateHandlerFunctions()
|
||||||
|
|
||||||
err = l.Fire(Start{})
|
err = l.Fire(Start{})
|
||||||
for err == nil {
|
for err == nil {
|
||||||
|
tickStartTime := time.Now()
|
||||||
err = l.Fire(Tick{})
|
err = l.Fire(Tick{})
|
||||||
if err != nil {
|
if err != nil {
|
||||||
break
|
break
|
||||||
}
|
}
|
||||||
|
time.Sleep(TimePerLoop - time.Since(tickStartTime))
|
||||||
}
|
}
|
||||||
if errors.Is(err, HandlerDoneErr{}) {
|
if errors.Is(err, HandlerDoneErr{}) {
|
||||||
err = errors.Unwrap(err)
|
err = errors.Unwrap(err)
|
||||||
@@ -126,3 +140,14 @@ func (l *Loop) Start() (err error) {
|
|||||||
}
|
}
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
|
||||||
|
func (l *Loop) Start() (err error) {
|
||||||
|
l.ErrChan = make(chan error)
|
||||||
|
l.Ctx, l.cancel = context.WithCancel(context.Background())
|
||||||
|
go func() {
|
||||||
|
l.ErrChan <- l.startLoop()
|
||||||
|
}()
|
||||||
|
err = <-l.ErrChan
|
||||||
|
l.cancel()
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|||||||
+42
-2
@@ -16,6 +16,7 @@ import (
|
|||||||
var BadPacketTypeError = errors.New("bad packet type")
|
var BadPacketTypeError = errors.New("bad packet type")
|
||||||
var BadPacketIdError = errors.New("bad packet id")
|
var BadPacketIdError = errors.New("bad packet id")
|
||||||
var PacketNotFullyDecodedError = errors.New("packet not fully decoded")
|
var PacketNotFullyDecodedError = errors.New("packet not fully decoded")
|
||||||
|
var ErrContextDone = errors.New("context done")
|
||||||
|
|
||||||
// UnCodablePacket represents a packet that could not be decoded.
|
// UnCodablePacket represents a packet that could not be decoded.
|
||||||
type UnCodablePacket struct {
|
type UnCodablePacket struct {
|
||||||
@@ -42,6 +43,25 @@ type Conn struct {
|
|||||||
Actor proto_base.Actor
|
Actor proto_base.Actor
|
||||||
Version string
|
Version string
|
||||||
ProtocolVersion int32
|
ProtocolVersion int32
|
||||||
|
|
||||||
|
receiveRoutine bool
|
||||||
|
packetCh chan proto_base.EncodeDecodeAble
|
||||||
|
}
|
||||||
|
|
||||||
|
func (c *Conn) ReceiveJob() {
|
||||||
|
for {
|
||||||
|
select {
|
||||||
|
case <-c.Ctx.Done():
|
||||||
|
return
|
||||||
|
default:
|
||||||
|
packet, err := c.Receive()
|
||||||
|
if err != nil {
|
||||||
|
c.ErrChan <- err
|
||||||
|
return
|
||||||
|
}
|
||||||
|
c.packetCh <- packet
|
||||||
|
}
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
func (c *Conn) SetState(state proto_base.State) {
|
func (c *Conn) SetState(state proto_base.State) {
|
||||||
@@ -62,10 +82,17 @@ func (c *Conn) SetVersion(version string) (err error) {
|
|||||||
if err != nil {
|
if err != nil {
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
c.ProtocolVersion = int32(versionData.Version)
|
c.ProtocolVersion = versionData.Version
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
|
||||||
|
func (c *Conn) ActivateReceiveRoutine() {
|
||||||
|
c.receiveRoutine = true
|
||||||
|
c.packetCh = make(chan proto_base.EncodeDecodeAble, 100) // i dunno how many to buffer 100 seems reasonable
|
||||||
|
go c.ReceiveJob()
|
||||||
|
|
||||||
|
}
|
||||||
|
|
||||||
func (c *Conn) Send(packet proto_base.EncodeDecodeAble) (err error) {
|
func (c *Conn) Send(packet proto_base.EncodeDecodeAble) (err error) {
|
||||||
slog.Debug("send", "actor", c.Actor, "packet", fmt.Sprintf("%#v", packet))
|
slog.Debug("send", "actor", c.Actor, "packet", fmt.Sprintf("%#v", packet))
|
||||||
switch packet := packet.(type) {
|
switch packet := packet.(type) {
|
||||||
@@ -120,12 +147,25 @@ func (c *Conn) Receive() (packet proto_base.EncodeDecodeAble, err error) {
|
|||||||
}
|
}
|
||||||
|
|
||||||
func (c *Conn) OnTick(_ event.Tick) (err error) {
|
func (c *Conn) OnTick(_ event.Tick) (err error) {
|
||||||
packet, err := c.Receive()
|
var packet proto_base.EncodeDecodeAble
|
||||||
|
if c.receiveRoutine {
|
||||||
|
select {
|
||||||
|
case packet = <-c.packetCh:
|
||||||
|
slog.Debug("receive", "actor", c.Actor, "packet", fmt.Sprintf("%#v", packet))
|
||||||
|
err = c.Loop.Fire(packet)
|
||||||
|
case <-c.Ctx.Done():
|
||||||
|
err = ErrContextDone
|
||||||
|
return
|
||||||
|
default:
|
||||||
|
}
|
||||||
|
} else {
|
||||||
|
packet, err = c.Receive()
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
slog.Debug("receive", "actor", c.Actor, "packet", fmt.Sprintf("%#v", packet))
|
slog.Debug("receive", "actor", c.Actor, "packet", fmt.Sprintf("%#v", packet))
|
||||||
err = c.Loop.Fire(packet)
|
err = c.Loop.Fire(packet)
|
||||||
|
}
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|||||||
+19
-4
@@ -8,6 +8,7 @@ import (
|
|||||||
"errors"
|
"errors"
|
||||||
"io"
|
"io"
|
||||||
"net"
|
"net"
|
||||||
|
"sync"
|
||||||
|
|
||||||
"github.com/admin-else/strom/crypto"
|
"github.com/admin-else/strom/crypto"
|
||||||
"github.com/admin-else/strom/proto_base"
|
"github.com/admin-else/strom/proto_base"
|
||||||
@@ -17,15 +18,29 @@ type RawConn struct {
|
|||||||
net.Conn
|
net.Conn
|
||||||
R io.Reader
|
R io.Reader
|
||||||
W io.Writer
|
W io.Writer
|
||||||
CompressionThreshold int32
|
compressionThreshold int32
|
||||||
|
CompressionThresholdMut sync.RWMutex
|
||||||
|
}
|
||||||
|
|
||||||
|
func (c *RawConn) SetCompressionThreshold(threshold int32) {
|
||||||
|
c.CompressionThresholdMut.Lock()
|
||||||
|
defer c.CompressionThresholdMut.Unlock()
|
||||||
|
c.compressionThreshold = threshold
|
||||||
|
}
|
||||||
|
|
||||||
|
func (c *RawConn) GetCompressionThreshold() int32 {
|
||||||
|
c.CompressionThresholdMut.RLock()
|
||||||
|
defer c.CompressionThresholdMut.RUnlock()
|
||||||
|
return c.compressionThreshold
|
||||||
}
|
}
|
||||||
|
|
||||||
func (c *RawConn) SendRaw(rawPacketBytes []byte) (err error) {
|
func (c *RawConn) SendRaw(rawPacketBytes []byte) (err error) {
|
||||||
//fmt.Println("send:", rawPacketBytes)
|
//fmt.Println("send:", rawPacketBytes)
|
||||||
var packetBytes []byte
|
var packetBytes []byte
|
||||||
if c.CompressionThreshold > 0 {
|
threshold := c.GetCompressionThreshold()
|
||||||
|
if threshold > 0 {
|
||||||
packetBuffer := bytes.NewBuffer(nil)
|
packetBuffer := bytes.NewBuffer(nil)
|
||||||
if int32(len(rawPacketBytes)) >= c.CompressionThreshold {
|
if int32(len(rawPacketBytes)) >= threshold {
|
||||||
err = proto_base.EncodeVarInt(packetBuffer, int32(len(rawPacketBytes)))
|
err = proto_base.EncodeVarInt(packetBuffer, int32(len(rawPacketBytes)))
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return
|
return
|
||||||
@@ -80,7 +95,7 @@ func (c *RawConn) ReceiveRaw() (packetBytes []byte, err error) {
|
|||||||
return
|
return
|
||||||
}
|
}
|
||||||
rawPacketBuffer := bytes.NewBuffer(rawPacketBytes)
|
rawPacketBuffer := bytes.NewBuffer(rawPacketBytes)
|
||||||
if c.CompressionThreshold > 0 {
|
if c.GetCompressionThreshold() > 0 {
|
||||||
var packetLen int32
|
var packetLen int32
|
||||||
packetLen, err = proto_base.DecodeVarInt(rawPacketBuffer)
|
packetLen, err = proto_base.DecodeVarInt(rawPacketBuffer)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
|
|||||||
@@ -90,10 +90,6 @@ func (l *LoginServer) SetCompressionThreshold(threshold int32) (err error) {
|
|||||||
|
|
||||||
func (l *LoginServer) OnLoginStart(packet *v1_21_8.LoginToServerPacketLoginStart) (err error) {
|
func (l *LoginServer) OnLoginStart(packet *v1_21_8.LoginToServerPacketLoginStart) (err error) {
|
||||||
l.Requested = NameAndUUID{packet.Username, packet.PlayerUUID}
|
l.Requested = NameAndUUID{packet.Username, packet.PlayerUUID}
|
||||||
//err = l.SetCompressionThreshold(l.CompressionThreshold)
|
|
||||||
//if err != nil {
|
|
||||||
// return
|
|
||||||
//}
|
|
||||||
if l.Given == nil {
|
if l.Given == nil {
|
||||||
l.Given = &l.Requested
|
l.Given = &l.Requested
|
||||||
}
|
}
|
||||||
@@ -124,7 +120,7 @@ func (l *LoginServer) OnCycle(_ event.Tick) (err error) {
|
|||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
|
||||||
func (s *LoginServer) OnClose(_ event.Close) (err error) {
|
func (l *LoginServer) OnClose(_ event.Close) (err error) {
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|||||||
+1
-1
@@ -14,7 +14,7 @@ func Servee(c net.Conn) (ret *proto.Conn) {
|
|||||||
ret = &proto.Conn{}
|
ret = &proto.Conn{}
|
||||||
_ = ret.SetVersion("1.21.8") // cant fail
|
_ = ret.SetVersion("1.21.8") // cant fail
|
||||||
ret.SetState(proto_base.Handshaking)
|
ret.SetState(proto_base.Handshaking)
|
||||||
ret.CompressionThreshold = -1
|
ret.SetCompressionThreshold(-1)
|
||||||
ret.Actor = proto_base.Servee
|
ret.Actor = proto_base.Servee
|
||||||
ret.Conn = c
|
ret.Conn = c
|
||||||
ret.R = c
|
ret.R = c
|
||||||
|
|||||||
Reference in New Issue
Block a user