Selaa lähdekoodia

fix(config): 修复 Etcd客户端初始化和清理逻辑

- 在 NewEtcdClientWithCleanup 函数中增加对 etcdClient 是否为 nil 的判断
- 如果 etcdClient 为 nil,返回空的 cleanup 函数,避免潜在的空指针异常
-优化 cleanup 函数的定义,提高代码可读性
lihf 2 kuukautta sitten
vanhempi
commit
ab873fe393
1 muutettua tiedostoa jossa 3 lisäystä ja 2 poistoa
  1. 3 2
      config/etcd_client.go

+ 3 - 2
config/etcd_client.go

@@ -37,8 +37,9 @@ func NewEtcdClient(startup *initutils.Startup) *clientv3.Client {
 
 func NewEtcdClientWithCleanup(startup *initutils.Startup) (*clientv3.Client, func()) {
 	etcdClient := NewEtcdClient(startup)
-	cleanup := func() {
-		etcdClient.Close()
+	if etcdClient == nil {
+		return nil, func() {}
 	}
+	cleanup := func() { etcdClient.Close() }
 	return etcdClient, cleanup
 }