feat: config center

icey-yu · Dec 27, 2024 · 1c1360e · 1c1360e
1 parent 76f3b69
commit 1c1360e
Showing 1 changed file with 15 additions and 2 deletions.
diff --git a/pkg/common/cmd/root.go b/pkg/common/cmd/root.go
@@ -152,9 +152,22 @@ func (r *RootCmd) updateConfigFromEtcd(opts *CmdOpts) error {
 	}
 
 	update := func(configFileName string, configStruct any) error {
+		ctx := context.TODO()
 		key := disetcd.BuildKey(configFileName)
-		etcdRes, err := r.etcdClient.Get(context.TODO(), key)
-		if err != nil || etcdRes.Count == 0 {
+		etcdRes, err := r.etcdClient.Get(ctx, key)
+		if err != nil {
+			log.ZWarn(ctx, "root cmd updateConfigFromEtcd, etcd Get err: %v", errs.Wrap(err))
+			return nil
+		}
+		if etcdRes.Count == 0 {
+			data, err := json.Marshal(configStruct)
+			if err != nil {
+				return errs.ErrArgs.WithDetail(err.Error()).Wrap()
+			}
+			_, err = r.etcdClient.Put(ctx, disetcd.BuildKey(configFileName), string(data))
+			if err != nil {
+				log.ZWarn(ctx, "root cmd updateConfigFromEtcd, etcd Put err: %v", errs.Wrap(err))
+			}
 			return nil
 		}
 		err = json.Unmarshal(etcdRes.Kvs[0].Value, configStruct)