Merge branch 'master' of github.com:sofyan48/pubsub-router

Author: sofyan48

router.go

@@ -1,8 +1,10 @@
 package pubsubrouter
 
 import (
+	"runtime/debug"
 	"sync"
 
+	"github.com/google/martian/log"
 	"github.com/sofyan48/pubsub-router/pkg/client"
 )
 
@@ -28,10 +30,19 @@ func (r *Router) Handle(routes string, h Handler) *Router {
 
 func (r *Router) HandleMessage(m *Message) error {
 	path := m.Payload.Attributes[client.MessageAttributeNameRoute]
+	defer func() {
+		if err := recover(); err != nil {
+			log.Errorf("panic recovered: %v | stack : %v", err, string(debug.Stack()))
+		}
+	}()
 	h, okRoute := r.handlers[path]
 	if okRoute {
+		err := h.HandleMessage(m)
+		if err != nil {
+			m.Payload.Nack()
+			return err
+		}
 		m.Payload.Ack()
-		return h.HandleMessage(m)
 	}
 	// if you need reporting please contrib this error handling
 	// return errors.New("Route Not Any Match")