From c16e42fe9e3337617ba0635feaf1acdbd0daebb5 Mon Sep 17 00:00:00 2001
From: zcong1993 <zhangcong1992@gmail.com>
Date: Thu, 8 Aug 2024 11:48:08 +0800
Subject: [PATCH] chore: tweak

---
 zrpc/server.go      | 9 +++++----
 zrpc/server_test.go | 9 +++++++++
 2 files changed, 14 insertions(+), 4 deletions(-)

diff --git a/zrpc/server.go b/zrpc/server.go
index da2ecae4d1a7..813fc358d298 100644
--- a/zrpc/server.go
+++ b/zrpc/server.go
@@ -52,10 +52,8 @@ func NewServer(c RpcServerConf, register internal.RegisterFn) (*RpcServer, error
 	metrics.SetName(c.Name)
 	setupStreamInterceptors(server, c)
 	setupUnaryInterceptors(server, c, metrics)
-	if c.Auth {
-		if err = setupAuthInterceptors(server, c); err != nil {
-			return nil, err
-		}
+	if err = setupAuthInterceptors(server, c); err != nil {
+		return nil, err
 	}
 
 	rpcServer := &RpcServer{
@@ -112,6 +110,9 @@ func SetServerSlowThreshold(threshold time.Duration) {
 }
 
 func setupAuthInterceptors(svr internal.Server, c RpcServerConf) error {
+	if !c.Auth {
+		return nil
+	}
 	rds, err := redis.NewRedis(c.Redis.RedisConf)
 	if err != nil {
 		return err
diff --git a/zrpc/server_test.go b/zrpc/server_test.go
index 260ffdb1781a..e42e379a6cd1 100644
--- a/zrpc/server_test.go
+++ b/zrpc/server_test.go
@@ -272,6 +272,15 @@ func Test_setupStreamInterceptors(t *testing.T) {
 }
 
 func Test_setupAuthInterceptors(t *testing.T) {
+	t.Run("no need set auth", func(t *testing.T) {
+		s := &mockedServer{}
+		err := setupAuthInterceptors(s, RpcServerConf{
+			Auth:  false,
+			Redis: redis.RedisKeyConf{},
+		})
+		assert.NoError(t, err)
+	})
+
 	t.Run("redis error", func(t *testing.T) {
 		s := &mockedServer{}
 		err := setupAuthInterceptors(s, RpcServerConf{