75 lines
1.6 KiB
Go
75 lines
1.6 KiB
Go
package main
|
|
|
|
import (
|
|
log "github.com/golang/glog"
|
|
"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"
|
|
"golang.org/x/net/context"
|
|
)
|
|
|
|
func logWrapper(fn server.HandlerFunc) server.HandlerFunc {
|
|
return func(ctx context.Context, req interface{}, rsp interface{}) error {
|
|
log.Infof("[Log Wrapper] Before serving request")
|
|
err := fn(ctx, req, rsp)
|
|
log.Infof("[Log Wrapper] After serving request")
|
|
return err
|
|
}
|
|
}
|
|
|
|
func logSubWrapper(fn server.SubscriberFunc) server.SubscriberFunc {
|
|
return func(ctx context.Context, req interface{}) error {
|
|
log.Infof("[Log Sub Wrapper] Before serving publication")
|
|
err := fn(ctx, req)
|
|
log.Infof("[Log Sub Wrapper] After serving publication")
|
|
return err
|
|
}
|
|
}
|
|
|
|
func main() {
|
|
// optionally setup command line usage
|
|
cmd.Init()
|
|
|
|
server.DefaultServer = server.NewServer(
|
|
server.WrapHandler(logWrapper),
|
|
server.WrapSubscriber(logSubWrapper),
|
|
)
|
|
|
|
// Initialise Server
|
|
server.Init(
|
|
server.Name("go.micro.srv.example"),
|
|
)
|
|
|
|
// Register Handlers
|
|
server.Handle(
|
|
server.NewHandler(
|
|
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)
|
|
}
|
|
|
|
// Run server
|
|
if err := server.Run(); err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
}
|