micro/examples/server/main.go

79 lines
1.7 KiB
Go
Raw Normal View History

2015-01-14 02:31:27 +03:00
package main
import (
2015-01-31 18:49:21 +03:00
log "github.com/golang/glog"
2015-11-20 19:17:33 +03:00
"github.com/micro/go-micro/cmd"
"github.com/micro/go-micro/examples/server/handler"
"github.com/micro/go-micro/examples/server/subscriber"
"github.com/micro/go-micro/server"
2015-12-02 14:54:36 +03:00
"golang.org/x/net/context"
2015-01-14 02:31:27 +03:00
)
2015-12-02 14:54:36 +03:00
func logWrapper(fn server.HandlerFunc) server.HandlerFunc {
2015-12-02 23:56:50 +03:00
return func(ctx context.Context, req server.Request, rsp interface{}) error {
log.Infof("[Log Wrapper] Before serving request method: %v", req.Method())
2015-12-02 14:54:36 +03:00
err := fn(ctx, req, rsp)
log.Infof("[Log Wrapper] After serving request")
return err
}
}
2015-12-02 23:00:06 +03:00
func logSubWrapper(fn server.SubscriberFunc) server.SubscriberFunc {
2015-12-02 23:56:50 +03:00
return func(ctx context.Context, req server.Publication) error {
log.Infof("[Log Sub Wrapper] Before serving publication topic: %v", req.Topic())
2015-12-02 23:00:06 +03:00
err := fn(ctx, req)
log.Infof("[Log Sub Wrapper] After serving publication")
return err
}
}
2015-01-14 02:31:27 +03:00
func main() {
// optionally setup command line usage
cmd.Init()
md := server.DefaultOptions().Metadata
2015-12-09 18:06:31 +03:00
md["datacenter"] = "local"
2015-12-02 14:54:36 +03:00
server.DefaultServer = server.NewServer(
server.WrapHandler(logWrapper),
2015-12-02 23:00:06 +03:00
server.WrapSubscriber(logSubWrapper),
2015-12-09 18:06:31 +03:00
server.Metadata(md),
2015-12-02 14:54:36 +03:00
)
2015-11-27 03:17:36 +03:00
2015-01-14 02:31:27 +03:00
// Initialise Server
2015-05-27 00:39:48 +03:00
server.Init(
server.Name("go.micro.srv.example"),
)
2015-01-14 02:31:27 +03:00
// Register Handlers
server.Handle(
server.NewHandler(
2015-01-14 02:31:27 +03:00
new(handler.Example),
),
)
// Register Subscribers
if err := server.Subscribe(
server.NewSubscriber(
"topic.go.micro.srv.example",
new(subscriber.Example),
),
); err != nil {
log.Fatal(err)
}
if err := server.Subscribe(
server.NewSubscriber(
"topic.go.micro.srv.example",
subscriber.Handler,
),
); err != nil {
log.Fatal(err)
}
2015-01-14 02:31:27 +03:00
// Run server
if err := server.Run(); err != nil {
log.Fatal(err)
}
}