From af9ae855016f081800e4dd2c2a6a7b333b3d1872 Mon Sep 17 00:00:00 2001
From: MaxToby <apiy@qq.com>
Date: Mon, 30 Oct 2023 13:54:04 +0000
Subject: [PATCH] The NotLoadCacheAtStart parameter is added to nacos

---
 zrpc/registry/nacos/builder.go |  2 ++
 zrpc/registry/nacos/target.go  | 18 ++++++++++++++++++
 2 files changed, 20 insertions(+)

diff --git a/zrpc/registry/nacos/builder.go b/zrpc/registry/nacos/builder.go
index 4d2b7b3..bded878 100644
--- a/zrpc/registry/nacos/builder.go
+++ b/zrpc/registry/nacos/builder.go
@@ -46,6 +46,8 @@ func (b *builder) Build(url resolver.Target, conn resolver.ClientConn, opts reso
 		Username:    tgt.User,
 		Password:    tgt.Password,
 		TimeoutMs:   uint64(tgt.Timeout),
+		NotLoadCacheAtStart:  tgt.NotLoadCacheAtStart,
+		UpdateCacheWhenEmpty: tgt.UpdateCacheWhenEmpty,
 	}
 
 	if tgt.CacheDir != "" {
diff --git a/zrpc/registry/nacos/target.go b/zrpc/registry/nacos/target.go
index d90791f..443ac07 100644
--- a/zrpc/registry/nacos/target.go
+++ b/zrpc/registry/nacos/target.go
@@ -23,6 +23,8 @@ type target struct {
 	LogLevel    string        `key:",optional"`
 	LogDir      string        `key:",optional"`
 	CacheDir    string        `key:",optional"`
+	NotLoadCacheAtStart  bool `key:"notLoadCacheAtStart,optional"`
+	UpdateCacheWhenEmpty bool `key:"updateCacheWhenEmpty,optional"`
 }
 
 // parseURL with parameters
@@ -57,5 +59,21 @@ func parseURL(rawURL url.URL) (target, error) {
 	tgt.Addr = rawURL.Host
 	tgt.Service = strings.TrimLeft(rawURL.Path, "/")
 
+	if logLevel, exists := os.LookupEnv("NACOS_LOG_LEVEL"); exists {
+		tgt.LogLevel = logLevel
+	}
+
+	if logDir, exists := os.LookupEnv("NACOS_LOG_DIR"); exists {
+		tgt.LogDir = logDir
+	}
+
+	if notLoadCacheAtStart, exists := os.LookupEnv("NACOS_NOT_LOAD_CACHE_AT_START"); exists {
+		tgt.NotLoadCacheAtStart = notLoadCacheAtStart == "true"
+	}
+
+	if updateCacheWhenEmpty, exists := os.LookupEnv("NACOS_UPDATE_CACHE_WHEN_EMPTY"); exists {
+		tgt.UpdateCacheWhenEmpty = updateCacheWhenEmpty == "true"
+	}
+
 	return tgt, nil
 }