package lucky

import (
	"github.com/helloh2o/lucky/log"
	"github.com/xtaci/kcp-go"
	"net"
	"runtime/debug"
	"sync"
)

type kcpServer struct {
	mu        sync.Mutex
	addr      string
	ln        net.Listener
	processor Processor
}

// NewKcpServer return a *kcpServer
func NewKcpServer(addr string, processor Processor) (s *kcpServer, err error) {
	ts := new(kcpServer)
	ts.addr = addr
	ts.ln, err = kcp.ListenWithOptions(addr, nil, 10, 3)
	if processor == nil {
		panic("processor must be set.")
	}
	ts.processor = processor
	if err != nil {
		return nil, err
	}
	return ts, err
}

// Run kcp server
func (s *kcpServer) Run() error {
	log.Release("Starting kcp server on %s", s.addr)
	for {
		conn, err := s.ln.Accept()
		if err != nil {
			return err
		}
		go s.Handle(conn)
	}
}

// Handle goroutine handle connection
func (s *kcpServer) Handle(conn net.Conn) {
	defer func() {
		if r := recover(); r != nil {
			log.Error("PANIC %v TCP handle, stack %s", r, string(debug.Stack()))
		}
	}()
	var ic IConnection
	// 可靠的UDP协议, like tcp
	ic = NewKcpConn(conn, s.processor)
	ic.ReadMsg()
}