Only add some methods to router handler

MouseHatGames · Dec 4, 2020 · 1f5385d · 1f5385d
1 parent d4e5226
commit 1f5385d
Show file tree

Hide file tree

Showing 3 changed files with 16 additions and 7 deletions.
diff --git a/router/handler.go b/router/handler.go
@@ -25,7 +25,7 @@ func (e *HandlerError) Unwrap() error {
 	return e.err
 }
 
-func newHandler(h interface{}) *handler {
+func newHandler(h interface{}, methods map[string]bool) *handler {
 	handtype := reflect.TypeOf(h)
 	handval := reflect.ValueOf(h)
 	name := reflect.Indirect(handval).Type().Name()
@@ -35,6 +35,10 @@ func newHandler(h interface{}) *handler {
 	for i := 0; i < handtype.NumMethod(); i++ {
 		m := handtype.Method(i)
 
+		if methods != nil && !methods[m.Name] {
+			continue
+		}
+
 		e := getEndpoint(m)
 		if e == nil {
 			continue

diff --git a/router/router.go b/router/router.go
@@ -15,7 +15,7 @@ var ErrMalformedPath = errors.New("malformed request path")
 var ErrEndpointNotFound = errors.New("endpoint not found")
 
 type Router interface {
-	AddHandler(h interface{})
+	AddHandler(h interface{}, methods []string)
 	Handle(path string, data []byte) ([]byte, error)
 }
 
@@ -33,8 +33,13 @@ func NewRouter(cod codec.Codec, log logger.Logger) Router {
 	}
 }
 
-func (s *router) AddHandler(h interface{}) {
-	hdl := newHandler(h)
+func (s *router) AddHandler(h interface{}, methods []string) {
+	metmap := make(map[string]bool, len(methods))
+	for _, m := range methods {
+		metmap[m] = true
+	}
+
+	hdl := newHandler(h, metmap)
 	s.handlers[hdl.Name] = hdl
 
 	for k := range hdl.Endpoints {

diff --git a/server/server.go b/server/server.go
@@ -14,7 +14,7 @@ import (
 
 type Server interface {
 	Start() error
-	AddHandler(h interface{})
+	AddHandler(h interface{}, methods ...string)
 	Publish(ctx context.Context, topic string, data interface{}) error
 }
 
@@ -45,8 +45,8 @@ func (s *server) Start() error {
 	return nil
 }
 
-func (s *server) AddHandler(h interface{}) {
-	s.opts.Router.AddHandler(h)
+func (s *server) AddHandler(h interface{}, methods ...string) {
+	s.opts.Router.AddHandler(h, methods)
 }
 
 func (s *server) handle(soc transport.Socket) {