server.go

package server

import (
	"fmt"
	"sync"

	as "github.com/godevsig/adaptiveservice"
)

type statMgr struct {
	sync.RWMutex
	clients     map[string]struct{}
	subscribers map[chan string]struct{}
	sessionNum  int32
	counter     int64
}

const (
	// Publisher is the service(s) publisher
	Publisher = "example"
	// ServiceEcho is the echo service
	ServiceEcho = "echo.v1.0"
)

// Run runs the server.
func Run(opts []as.Option) {
	s := as.NewServer(opts...).SetPublisher(Publisher)

	mgr := &statMgr{
		clients:     make(map[string]struct{}),
		subscribers: make(map[chan string]struct{}),
	}
	if err := s.Publish(ServiceEcho,
		echoKnownMsgs,
		as.OnNewStreamFunc(mgr.onNewStream),
		as.OnStreamCloseFunc(mgr.onStreamClose),
		as.OnConnectFunc(mgr.onConnect),
		as.OnDisconnectFunc(mgr.onDisconnect),
	); err != nil {
		fmt.Println(err)
		return
	}

	if err := s.Serve(); err != nil { // ctrl+c to exit
		fmt.Println(err)
	}
	fmt.Printf("echo server has served %d requests\n", mgr.counter)
}