lkm/stream/stream_server.go

package stream

import (
	"github.com/lkmio/lkm/log"
	"github.com/lkmio/transport"
	"net"
)

type SessionHandler[T any] interface {
	OnNewSession(conn net.Conn) T

	OnCloseSession(session T)
}

type StreamServer[T any] struct {
	SourceType SourceType
	Handler    SessionHandler[T]
}

func (s *StreamServer[T]) OnConnected(conn net.Conn) []byte {
	log.Sugar.Debugf("%s连接 conn: %s", s.SourceType.String(), conn.RemoteAddr().String())
	if s.Handler != nil {
		conn.(*transport.Conn).Data = s.Handler.OnNewSession(conn)
	}
	return nil
}

func (s *StreamServer[T]) OnPacket(conn net.Conn, data []byte) []byte {
	if AppConfig.Debug {
		DumpStream2File(s.SourceType, conn, data)
	}

	return nil
}

func (s *StreamServer[T]) OnDisConnected(conn net.Conn, err error) {
	log.Sugar.Debugf("%s断开连接 conn:%s", s.SourceType.String(), conn.RemoteAddr().String())

	t := conn.(*transport.Conn)
	if s.Handler != nil && t.Data != nil {
		s.Handler.OnCloseSession(t.Data.(T))
		t.Data = nil
	}
}