add chain.Chainable interface

2021-11-23 21:44:32 +08:00
parent ece51cb0b8
commit 7beb5bf6d7
18 changed files with 125 additions and 63 deletions
--- a/pkg/handler/forward/local/handler.go
+++ b/pkg/handler/forward/local/handler.go
@ -33,7 +33,6 @@ func NewHandler(opts ...handler.Option) handler.Handler {
 	}

 	return &forwardHandler{
-		chain:  options.Chain,
 		bypass: options.Bypass,
 		logger: options.Logger,
 	}
@ -43,6 +42,11 @@ func (h *forwardHandler) Init(md md.Metadata) (err error) {
 	return h.parseMetadata(md)
 }

+// implements chain.Chainable interface
+func (h *forwardHandler) WithChain(chain *chain.Chain) {
+	h.chain = chain
+}
+
 // Forward implements handler.Forwarder.
 func (h *forwardHandler) Forward(group *chain.NodeGroup) {
 	h.group = group