From ef8303f11f7367b6f91985cf41f55d26820420f8 Mon Sep 17 00:00:00 2001
From: Asim <asim@chuhnk.me>
Date: Wed, 2 Dec 2015 20:00:06 +0000
Subject: [PATCH] add server publication wrapper example
---
examples/server/main.go | 10 ++++++++++
1 file changed, 10 insertions(+)
diff --git a/examples/server/main.go b/examples/server/main.go
index 7a718441..193d7eb4 100644
--- a/examples/server/main.go
+++ b/examples/server/main.go
@@ -18,12 +18,22 @@ func logWrapper(fn server.HandlerFunc) server.HandlerFunc {
}
}
+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