server.go

package plex

import (
	"net"
	"sync"

	"github.com/swxctx/plex/plog"
)

var (
	server *plexServer
)

// plexServer
type plexServer struct {
	// config
	cfg *Config
	// tcp listen
	listener net.Listener
	// conn store cache
	store *connStore
	// tcp auth func
	authFunc func(body string) (bool, string)
	// client online or offline callback
	onlineStatusSubscribe func(isOnline bool, uid string)
}

// NewServer new plex server
func NewServer(config *Config, fn ...func(body string) (bool, string)) {
	// reload config
	cfg := reloadConfig(config)

	plog.Infof("reload config success.")

	// new server
	server = &plexServer{
		cfg:   cfg,
		store: newConnStore(cfg.MaxConnection),
	}

	// auth handler
	if len(fn) > 0 {
		server.authFunc = fn[0]
	}
	plog.Infof("new plex server success.")
}

// Start plex server start
func Start() {
	plog.Infof("--- server start begin ---")

	var (
		wg sync.WaitGroup
	)
	wg.Add(2)

	// start http
	go func() {
		defer wg.Done()
		server.startHttpServer()
	}()

	// start tcp
	go func() {
		defer wg.Done()
		server.startTcpServer()
	}()

	wg.Wait()
	plog.Infof("--- server start end ---")
}

// SetAuthFunc func set outer auth func
func SetAuthFunc(fn func(body string) (bool, string)) {
	server.authFunc = fn
}

// SetOnlineStatusSubscribe func client online status change callback
func SetOnlineStatusSubscribe(fn func(isOnline bool, uid string)) {
	server.onlineStatusSubscribe = fn
}