micro/server/server.go
2015-05-05 19:05:06 +01:00

98 lines
1.8 KiB
Go

package server
import (
"os"
"os/signal"
"strconv"
"strings"
"syscall"
"code.google.com/p/go-uuid/uuid"
log "github.com/golang/glog"
"github.com/myodc/go-micro/registry"
)
type Server interface {
Address() string
Init() error
NewReceiver(interface{}) Receiver
NewNamedReceiver(string, interface{}) Receiver
Register(Receiver) error
Start() error
Stop() error
}
var (
Address string
Name string
Id string
DefaultServer Server
)
func Init() error {
defer log.Flush()
if len(Name) == 0 {
Name = "go-server"
}
if len(Id) == 0 {
Id = Name + "-" + uuid.NewUUID().String()
}
if DefaultServer == nil {
DefaultServer = NewRpcServer(Address)
}
return DefaultServer.Init()
}
func NewReceiver(handler interface{}) Receiver {
return DefaultServer.NewReceiver(handler)
}
func NewNamedReceiver(path string, handler interface{}) Receiver {
return DefaultServer.NewNamedReceiver(path, handler)
}
func Register(r Receiver) error {
return DefaultServer.Register(r)
}
func Run() error {
if err := Start(); err != nil {
return err
}
// parse address for host, port
parts := strings.Split(DefaultServer.Address(), ":")
host := strings.Join(parts[:len(parts)-1], ":")
port, _ := strconv.Atoi(parts[len(parts)-1])
// register service
node := registry.NewNode(Id, host, port)
service := registry.NewService(Name, node)
log.Infof("Registering %s", node.Id())
registry.Register(service)
ch := make(chan os.Signal, 1)
signal.Notify(ch, syscall.SIGTERM, syscall.SIGINT, syscall.SIGKILL)
log.Infof("Received signal %s", <-ch)
log.Infof("Deregistering %s", node.Id())
registry.Deregister(service)
return Stop()
}
func Start() error {
log.Infof("Starting server %s id %s", Name, Id)
return DefaultServer.Start()
}
func Stop() error {
log.Infof("Stopping server")
return DefaultServer.Stop()
}