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