104 lines
1.9 KiB
Go
104 lines
1.9 KiB
Go
package server
|
|
|
|
import (
|
|
"os"
|
|
"os/signal"
|
|
"syscall"
|
|
|
|
log "github.com/golang/glog"
|
|
"github.com/pborman/uuid"
|
|
)
|
|
|
|
type Server interface {
|
|
Config() options
|
|
Init(...Option)
|
|
Handle(Handler) error
|
|
NewHandler(interface{}) Handler
|
|
NewSubscriber(string, interface{}) Subscriber
|
|
Subscribe(Subscriber) error
|
|
Register() error
|
|
Deregister() error
|
|
Start() error
|
|
Stop() error
|
|
}
|
|
|
|
type Option func(*options)
|
|
|
|
var (
|
|
DefaultAddress = ":0"
|
|
DefaultName = "go-server"
|
|
DefaultVersion = "1.0.0"
|
|
DefaultId = uuid.NewUUID().String()
|
|
DefaultServer Server = newRpcServer()
|
|
)
|
|
|
|
func Config() options {
|
|
return DefaultServer.Config()
|
|
}
|
|
|
|
func Init(opt ...Option) {
|
|
if DefaultServer == nil {
|
|
DefaultServer = newRpcServer(opt...)
|
|
}
|
|
DefaultServer.Init(opt...)
|
|
}
|
|
|
|
func NewServer(opt ...Option) Server {
|
|
return newRpcServer(opt...)
|
|
}
|
|
|
|
func NewSubscriber(topic string, h interface{}) Subscriber {
|
|
return DefaultServer.NewSubscriber(topic, h)
|
|
}
|
|
|
|
func NewHandler(h interface{}) Handler {
|
|
return DefaultServer.NewHandler(h)
|
|
}
|
|
|
|
func Handle(h Handler) error {
|
|
return DefaultServer.Handle(h)
|
|
}
|
|
|
|
func Subscribe(s Subscriber) error {
|
|
return DefaultServer.Subscribe(s)
|
|
}
|
|
|
|
func Register() error {
|
|
return DefaultServer.Register()
|
|
}
|
|
|
|
func Deregister() error {
|
|
return DefaultServer.Deregister()
|
|
}
|
|
|
|
func Run() error {
|
|
if err := Start(); err != nil {
|
|
return err
|
|
}
|
|
|
|
if err := DefaultServer.Register(); err != nil {
|
|
return err
|
|
}
|
|
|
|
ch := make(chan os.Signal, 1)
|
|
signal.Notify(ch, syscall.SIGTERM, syscall.SIGINT, syscall.SIGKILL)
|
|
log.Infof("Received signal %s", <-ch)
|
|
|
|
if err := DefaultServer.Deregister(); err != nil {
|
|
return err
|
|
}
|
|
|
|
return Stop()
|
|
}
|
|
|
|
func Start() error {
|
|
config := DefaultServer.Config()
|
|
log.Infof("Starting server %s id %s", config.Name(), config.Id())
|
|
return DefaultServer.Start()
|
|
}
|
|
|
|
func Stop() error {
|
|
log.Infof("Stopping server")
|
|
return DefaultServer.Stop()
|
|
}
|