fix: startWithListener register etcd

Author: meizzliu

zrpc/internal/rpcpubserver.go

@@ -1,6 +1,7 @@
 package internal
 
 import (
+	"net"
 	"os"
 	"strings"
 
@@ -53,6 +54,14 @@ func (s keepAliveServer) Start(fn RegisterFn) error {
 	return s.Server.Start(fn)
 }
 
+func (s keepAliveServer) StartWithListener(listener net.Listener, fn RegisterFn) error {
+	if err := s.registerEtcd(); err != nil {
+		return err
+	}
+
+	return s.Server.StartWithListener(listener, fn)
+}
+
 func figureOutListenOn(listenOn string) string {
 	fields := strings.Split(listenOn, ":")
 	if len(fields) == 0 {

zrpc/internal/rpcpubserver_test.go

@@ -18,6 +18,10 @@ func TestNewRpcPubServer(t *testing.T) {
 	assert.NotPanics(t, func() {
 		s.Start(nil)
 	})
+
+	assert.NotPanics(t, func() {
+		s.StartWithListener(nil, nil)
+	})
 }
 
 func TestFigureOutListenOn(t *testing.T) {