micro/server/server.go

186 lines
4.2 KiB
Go
Raw Normal View History

2015-11-26 15:21:00 +03:00
/*
Server represents a server instance in go-micro which handles synchronous
requests via handlers and asynchronous requests via subscribers that
2015-11-26 03:17:05 +03:00
register with a broker.
2015-11-26 15:21:00 +03:00
The server combines the all the packages in go-micro to create a whole unit
used for building applications including discovery, client/server communication
2015-11-26 03:17:05 +03:00
and pub/sub.
2015-12-03 04:05:16 +03:00
import "github.com/micro/go-micro/server"
type Greeter struct {}
func (g *Greeter) Hello(ctx context.Context, req *greeter.Request, rsp *greeter.Response) error {
rsp.Msg = "Hello " + req.Name
return nil
}
s := server.NewServer()
s.Handle(
s.NewHandler(&Greeter{}),
)
s.Start()
2015-11-26 03:17:05 +03:00
*/
2015-01-14 02:31:27 +03:00
package server
import (
"os"
"os/signal"
"syscall"
2015-01-31 18:49:21 +03:00
log "github.com/golang/glog"
2015-08-26 14:15:37 +03:00
"github.com/pborman/uuid"
2015-12-17 23:37:35 +03:00
"golang.org/x/net/context"
2015-01-14 02:31:27 +03:00
)
type Server interface {
2015-05-27 00:39:48 +03:00
Config() options
Init(...Option)
Handle(Handler) error
NewHandler(interface{}) Handler
NewSubscriber(string, interface{}) Subscriber
Subscribe(Subscriber) error
Register() error
Deregister() error
2015-01-14 02:31:27 +03:00
Start() error
Stop() error
}
2015-12-02 23:56:50 +03:00
type Publication interface {
Topic() string
Message() interface{}
ContentType() string
}
type Request interface {
Service() string
Method() string
ContentType() string
Request() interface{}
2015-12-17 23:37:35 +03:00
// indicates whether the request will be streamed
2015-12-02 23:56:50 +03:00
Stream() bool
}
2015-12-17 23:37:35 +03:00
// Streamer represents a stream established with a client.
// A stream can be bidirectional which is indicated by the request.
// The last error will be left in Error().
// EOF indicated end of the stream.
type Streamer interface {
Context() context.Context
Request() Request
Send(interface{}) error
Recv(interface{}) error
Error() error
Close() error
}
2015-05-21 21:28:57 +03:00
type Option func(*options)
2015-05-21 21:24:57 +03:00
2015-01-14 02:31:27 +03:00
var (
2015-05-27 00:39:48 +03:00
DefaultAddress = ":0"
DefaultName = "go-server"
DefaultVersion = "1.0.0"
2015-05-27 00:39:48 +03:00
DefaultId = uuid.NewUUID().String()
DefaultServer Server = newRpcServer()
2015-01-14 02:31:27 +03:00
)
2015-11-26 03:13:17 +03:00
// Returns config options for the default service
2015-05-27 00:39:48 +03:00
func Config() options {
return DefaultServer.Config()
}
2015-11-26 03:13:17 +03:00
// Initialises the default server with options passed in
2015-05-27 00:39:48 +03:00
func Init(opt ...Option) {
2015-01-14 02:31:27 +03:00
if DefaultServer == nil {
2015-05-27 00:39:48 +03:00
DefaultServer = newRpcServer(opt...)
2015-01-14 02:31:27 +03:00
}
2015-05-27 00:39:48 +03:00
DefaultServer.Init(opt...)
2015-01-14 02:31:27 +03:00
}
2015-11-26 03:13:17 +03:00
// Returns a new server with options passed in
2015-05-27 00:39:48 +03:00
func NewServer(opt ...Option) Server {
return newRpcServer(opt...)
2015-05-23 19:40:53 +03:00
}
2015-11-26 03:13:17 +03:00
// Creates a new subscriber interface with the given topic
// and handler using the default server
func NewSubscriber(topic string, h interface{}) Subscriber {
return DefaultServer.NewSubscriber(topic, h)
}
2015-11-26 03:13:17 +03:00
// Creates a new handler interface using the default server
2015-12-01 21:41:43 +03:00
// Handlers are required to be a public object with public
// methods. Call to a service method such as Foo.Bar expects
// the type:
//
2015-12-01 21:45:52 +03:00
// type Foo struct {}
// func (f *Foo) Bar(ctx, req, rsp) error {
// return nil
// }
2015-12-01 21:41:43 +03:00
//
func NewHandler(h interface{}) Handler {
return DefaultServer.NewHandler(h)
}
2015-11-26 03:13:17 +03:00
// Registers a handler interface with the default server to
// handle inbound requests
func Handle(h Handler) error {
return DefaultServer.Handle(h)
2015-01-14 02:31:27 +03:00
}
2015-11-26 03:13:17 +03:00
// Registers a subscriber interface with the default server
// which subscribes to specified topic with the broker
func Subscribe(s Subscriber) error {
return DefaultServer.Subscribe(s)
}
2015-11-26 03:13:17 +03:00
// Registers the default server with the discovery system
func Register() error {
return DefaultServer.Register()
2015-01-14 02:31:27 +03:00
}
2015-11-26 03:13:17 +03:00
// Deregisters the default server from the discovery system
func Deregister() error {
return DefaultServer.Deregister()
2015-01-14 02:31:27 +03:00
}
2015-11-26 03:13:17 +03:00
// Blocking run starts the default server and waits for a kill
// signal before exiting. Also registers/deregisters the server
2015-01-14 02:31:27 +03:00
func Run() error {
if err := Start(); err != nil {
return err
}
if err := DefaultServer.Register(); err != nil {
return err
2015-05-21 21:24:57 +03:00
}
2015-01-14 02:31:27 +03:00
ch := make(chan os.Signal, 1)
signal.Notify(ch, syscall.SIGTERM, syscall.SIGINT, syscall.SIGKILL)
2015-01-31 18:49:21 +03:00
log.Infof("Received signal %s", <-ch)
2015-01-14 02:31:27 +03:00
if err := DefaultServer.Deregister(); err != nil {
return err
}
2015-01-14 02:31:27 +03:00
return Stop()
}
2015-11-26 03:13:17 +03:00
// Starts the default server
2015-01-14 02:31:27 +03:00
func Start() error {
2015-05-27 00:39:48 +03:00
config := DefaultServer.Config()
log.Infof("Starting server %s id %s", config.Name(), config.Id())
2015-01-14 02:31:27 +03:00
return DefaultServer.Start()
}
2015-11-26 03:13:17 +03:00
// Stops the default server
2015-01-14 02:31:27 +03:00
func Stop() error {
2015-01-31 18:49:21 +03:00
log.Infof("Stopping server")
2015-01-14 02:31:27 +03:00
return DefaultServer.Stop()
}