From a89f1830a47c737d10abde760748500df5dcce46 Mon Sep 17 00:00:00 2001
From: Vasiliy Tolstov <v.tolstov@unistack.org>
Date: Mon, 28 Nov 2022 12:29:53 +0300
Subject: [PATCH] add options

Signed-off-by: Vasiliy Tolstov <v.tolstov@unistack.org>
---
 wrapper.go | 72 ++++++++++++++++++++++++++++++++++++++++--------------
 1 file changed, 54 insertions(+), 18 deletions(-)

diff --git a/wrapper.go b/wrapper.go
index 2793239..9b37155 100644
--- a/wrapper.go
+++ b/wrapper.go
@@ -4,22 +4,60 @@ import (
 	"context"
 	"fmt"
 
+	"go.unistack.org/micro/v3/errors"
 	"go.unistack.org/micro/v3/server"
 )
 
-type wrapper struct {
-	serverHandlerFunc    func(context.Context, server.Request, interface{}, error) error
-	serverSubscriberFunc func(context.Context, server.Message, error) error
-	/*
-		clientCallFunc       func(context.Context, string, client.Request, interface{}, client.CallOptions, error) error
-		clientClient         func(client.Client, error) error
-	*/
+var (
+	_ server.HandlerWrapper    = NewServerHandlerWrapper()
+	_ server.SubscriberWrapper = NewServerSubscriberWrapper()
+)
+
+func NewOptions(opts ...Option) Options {
+	options := Options{
+		ServerHandlerFn:    DefaultServerHandlerFn,
+		ServerSubscriberFn: DefaultServerSubscriberFn,
+	}
+	for _, o := range opts {
+		o(&options)
+	}
+	return options
 }
 
-func NewServerHandlerWrapper(fn func(context.Context, server.Request, interface{}, error) error) server.HandlerWrapper {
-	handler := &wrapper{
-		serverHandlerFunc: fn,
+type Options struct {
+	ServerHandlerFn    func(context.Context, server.Request, interface{}, error) error
+	ServerSubscriberFn func(context.Context, server.Message, error) error
+}
+
+type Option func(*Options)
+
+func ServerHandlerFn(fn func(context.Context, server.Request, interface{}, error) error) Option {
+	return func(o *Options) {
+		o.ServerHandlerFn = fn
 	}
+}
+
+func ServerSubscriberFn(fn func(context.Context, server.Message, error) error) Option {
+	return func(o *Options) {
+		o.ServerSubscriberFn = fn
+	}
+}
+
+var (
+	DefaultServerHandlerFn = func(ctx context.Context, req server.Request, rsp interface{}, err error) error {
+		return errors.BadRequest("", "%v", err)
+	}
+	DefaultServerSubscriberFn = func(ctx context.Context, req server.Message, err error) error {
+		return errors.BadRequest("", "%v", err)
+	}
+)
+
+type wrapper struct {
+	opts Options
+}
+
+func NewServerHandlerWrapper(opts ...Option) server.HandlerWrapper {
+	handler := &wrapper{opts: NewOptions(opts...)}
 	return handler.HandlerFunc
 }
 
@@ -31,9 +69,9 @@ func (w *wrapper) HandlerFunc(fn server.HandlerFunc) server.HandlerFunc {
 			case nil:
 				return
 			case error:
-				err = w.serverHandlerFunc(ctx, req, rsp, verr)
+				err = w.opts.ServerHandlerFn(ctx, req, rsp, verr)
 			default:
-				err = w.serverHandlerFunc(ctx, req, rsp, fmt.Errorf("%v", r))
+				err = w.opts.ServerHandlerFn(ctx, req, rsp, fmt.Errorf("%v", r))
 			}
 		}()
 		err = fn(ctx, req, rsp)
@@ -41,10 +79,8 @@ func (w *wrapper) HandlerFunc(fn server.HandlerFunc) server.HandlerFunc {
 	}
 }
 
-func NewServerSubscriberWrapper(fn func(context.Context, server.Message, error) error) server.SubscriberWrapper {
-	handler := &wrapper{
-		serverSubscriberFunc: fn,
-	}
+func NewServerSubscriberWrapper(opts ...Option) server.SubscriberWrapper {
+	handler := &wrapper{opts: NewOptions(opts...)}
 	return handler.SubscriberFunc
 }
 
@@ -56,9 +92,9 @@ func (w *wrapper) SubscriberFunc(fn server.SubscriberFunc) server.SubscriberFunc
 			case nil:
 				return
 			case error:
-				err = w.serverSubscriberFunc(ctx, msg, verr)
+				err = w.opts.ServerSubscriberFn(ctx, msg, verr)
 			default:
-				err = w.serverSubscriberFunc(ctx, msg, fmt.Errorf("%v", r))
+				err = w.opts.ServerSubscriberFn(ctx, msg, fmt.Errorf("%v", r))
 			}
 		}()
 		err = fn(ctx, msg)