feat(zipper): router can be customized in NewZipper() (#667)

Author: woorui

cli/serve.go

@@ -22,6 +22,7 @@ import (
 
 	"github.com/spf13/cobra"
 	"github.com/yomorun/yomo"
+	"github.com/yomorun/yomo/core/router"
 	pkgconfig "github.com/yomorun/yomo/pkg/config"
 	"github.com/yomorun/yomo/pkg/log"
 	"github.com/yomorun/yomo/pkg/trace"
@@ -62,7 +63,7 @@ var serveCmd = &cobra.Command{
 			}
 		}
 
-		zipper, err := yomo.NewZipper(conf.Name, conf.Downstreams, options...)
+		zipper, err := yomo.NewZipper(conf.Name, router.Default(), conf.Downstreams, options...)
 		if err != nil {
 			log.FailureStatusEvent(os.Stdout, err.Error())
 			return

zipper.go

@@ -43,7 +43,7 @@ func RunZipper(ctx context.Context, configPath string) error {
 		}
 	}
 
-	zipper, err := NewZipper(conf.Name, conf.Downstreams, options...)
+	zipper, err := NewZipper(conf.Name, router.Default(), conf.Downstreams, options...)
 	if err != nil {
 		return err
 	}
@@ -53,7 +53,7 @@ func RunZipper(ctx context.Context, configPath string) error {
 }
 
 // NewZipper returns a zipper.
-func NewZipper(name string, meshConfig map[string]config.Downstream, options ...ZipperOption) (Zipper, error) {
+func NewZipper(name string, router router.Router, meshConfig map[string]config.Downstream, options ...ZipperOption) (Zipper, error) {
 	opts := &zipperOptions{}
 
 	for _, o := range options {
@@ -84,7 +84,7 @@ func NewZipper(name string, meshConfig map[string]config.Downstream, options ...
 		server.AddDownstreamServer(downstream)
 	}
 
-	server.ConfigRouter(router.Default())
+	server.ConfigRouter(router)
 
 	// watch signal.
 	go waitSignalForShutdownServer(server)

zipper_test.go

@@ -6,12 +6,14 @@ import (
 
 	"github.com/stretchr/testify/assert"
 	"github.com/yomorun/yomo/core"
+	"github.com/yomorun/yomo/core/router"
 	"github.com/yomorun/yomo/core/ylog"
 )
 
 func TestZipperRun(t *testing.T) {
 	zipper, err := NewZipper(
 		"zipper",
+		router.Default(),
 		nil,
 		// WithAuth("token", "<CREDENTIAL>"),
 		WithUpstreamOption(core.ClientOption(WithCredential("token:<CREDENTIAL>"))),