update

app/ecpm/ecpm.go

@@ -14,7 +14,7 @@ import (
 	"google.golang.org/grpc/reflection"
 )
 
-const ServiceName = "douyin_ecpm"
+const ServiceName = "ecpm"
 
 func main() {
 	flag.Parse()

cmd/etcd_config.go

@@ -0,0 +1,26 @@
+package main
+
+import (
+	"fmt"
+	"gitea.youtukeji.com.cn/youtu/youtu_grpc/pkg/config"
+	"os"
+	"strings"
+)
+
+// 连接etcd服务器并将./etc目录下的.json文件推送到etcd中
+func main() {
+	dir, err := os.ReadDir("./etc")
+	if err != nil {
+		panic(err)
+	}
+
+	for _, i := range dir {
+		fmt.Println(i.Name())
+		b, err := os.ReadFile("./etc/" + i.Name())
+		if err != nil {
+			panic(err)
+		}
+		serverName, _ := strings.CutSuffix(i.Name(), ".json")
+		err = config.SetConfig(string(b), serverName)
+	}
+}

etc/auth.json

@@ -0,0 +1,18 @@
+{
+  "RpcServerConf": {
+    "ListenOn": "0.0.0.0:8081",
+    "Etcd": {
+      "Hosts": [
+        "192.168.0.47:2379"
+      ],
+      "Key": "auth.rpc"
+    }
+  },
+  "Mode": "dev",
+  "Mysql": "root:youtu!0113@tcp(192.168.0.47:3306)/ecpm?charset=utf8mb4&parseTime=True&loc=Local",
+  "Redis": [
+    {
+      "host": "192.168.0.47:6379"
+    }
+  ]
+}

etc/ecpm.json

@@ -0,0 +1,19 @@
+{
+  "RpcServerConf": {
+    "ListenOn": "0.0.0.0:8083",
+    "Etcd": {
+      "Hosts": [
+        "192.168.0.47:2379"
+      ],
+      "Key": "ecpm.rpc"
+    }
+  },
+  "Mode": "dev",
+  "Mysql": "root:youtu!0113@tcp(192.168.0.47:3306)/ecpm?charset=utf8mb4&parseTime=True&loc=Local",
+  "Redis": [
+    {
+      "host": "192.168.0.47:6379",
+      "password": ""
+    }
+  ]
+}

etc/ranking.json

@@ -0,0 +1,19 @@
+{
+  "RpcServerConf": {
+    "ListenOn": "0.0.0.0:8082",
+    "Etcd": {
+      "Hosts": [
+        "192.168.0.47:2379"
+      ],
+      "Key": "ranking.rpc"
+    }
+  },
+  "Mode": "dev",
+  "Mysql": "root:youtu!0113@tcp(192.168.0.47:3306)/ecpm?charset=utf8mb4&parseTime=True&loc=Local",
+  "Redis": [
+    {
+      "host": "192.168.0.47:6379",
+      "password": ""
+    }
+  ]
+}

etc/user.json

@@ -0,0 +1,20 @@
+{
+  "RpcServerConf": {
+    "ListenOn": "0.0.0.0:8080",
+    "Etcd": {
+      "Hosts": [
+        "192.168.0.47:2379"
+      ],
+      "Key": "user.rpc"
+    },
+    "Auth": false
+  },
+  "Mode": "dev",
+  "Mysql": "root:youtu!0113@tcp(192.168.0.47:3306)/ecpm?charset=utf8mb4&parseTime=True&loc=Local",
+  "Redis": [
+    {
+      "host": "192.168.0.47:6379",
+      "password": ""
+    }
+  ]
+}

pkg/config/get_config.go

@@ -1,9 +1,12 @@
 package config
 
 import (
+	"context"
+	"fmt"
 	"github.com/spf13/viper"
 	_ "github.com/spf13/viper/remote"
 	"github.com/zeromicro/go-zero/zrpc"
+	clientv3 "go.etcd.io/etcd/client/v3"
 )
 
 const (
@@ -11,7 +14,7 @@ const (
 	TypeEtcd = "etcd"
 	TypeEnv  = "env"
 
-	EtcdPrefix  = "/youtu/"
+	Prefix      = "/youtu/config/" // 配置文件前缀
 	EtcdAddrKey = "ETCD_ADDR"
 )
 
@@ -39,7 +42,8 @@ func init() {
 }
 
 func GetConfig(c *Config, serverName string) (err error) {
-	err = viper.AddRemoteProvider("etcd3", viper.GetString(EtcdAddrKey), EtcdPrefix+serverName+".rpc")
+	err = viper.AddRemoteProvider("etcd3", viper.GetString(EtcdAddrKey), Prefix+serverName+".rpc")
+	fmt.Println(Prefix + serverName + ".rpc")
 	if err != nil {
 		return
 	}
@@ -56,3 +60,12 @@ func GetConfig(c *Config, serverName string) (err error) {
 	c.RpcServerConf.ServiceConf.Mode = c.Mode
 	return
 }
+
+func SetConfig(b, serverName string) (err error) {
+	etcdCli, err := clientv3.NewFromURL(viper.GetString(EtcdAddrKey))
+	if err != nil {
+		return
+	}
+	_, err = etcdCli.Put(context.Background(), Prefix+serverName+".rpc", b)
+	return
+}