feat: support auth account for etcd (#1174)

2021-10-31 09:05:38 +08:00
parent c1a8ccda11
commit 59b9687f31
15 changed files with 204 additions and 29 deletions
--- a/zrpc/client.go
+++ b/zrpc/client.go
@@ -4,6 +4,7 @@ import (
 	"log"
 	"time"

+	"github.com/tal-tech/go-zero/core/discov"
 	"github.com/tal-tech/go-zero/zrpc/internal"
 	"github.com/tal-tech/go-zero/zrpc/internal/auth"
 	"google.golang.org/grpc"
@@ -74,6 +75,10 @@ func NewClient(c RpcClientConf, options ...ClientOption) (Client, error) {
 			return nil, err
 		}

+		if c.Etcd.HasAccount() {
+			discov.RegisterAccount(c.Etcd.Hosts, c.Etcd.User, c.Etcd.Pass)
+		}
+
 		target = internal.BuildDiscovTarget(c.Etcd.Hosts, c.Etcd.Key)
 	}