<feat> 完善grpc插件

pku-hit · Dec 25, 2019 · 7d21545 · 7d21545
1 parent 431574c
commit 7d21545
Showing 1 changed file with 8 additions and 4 deletions.
diff --git a/grpc-server-plugin.go b/grpc-server-plugin.go
@@ -15,24 +15,28 @@ import (
 var Server = struct {
 	RunMode    string `yaml:"runmode"`
 	GRPCPort   string `yaml:"grpcport"`
-	ConsulHost string `yaml:"ConsulHost"`
-	ConsulPort string `yaml:"ConsulPort"`
 	ServerName string `yaml:"servername"`
 }{}
 
+var Registry = struct {
+	Type string `yaml:"type"`
+	Host string `yaml:"host"`
+}{}
+
 var Service micro.Service
 
 func init() {
 	config, err := ioutil.ReadFile("config/server.yml")
+	registyConfig, err := ioutil.ReadFile("config/registry.yml")
 	if err != nil {
 		fmt.Print(err)
 	}
 	yaml.Unmarshal(config, &Server)
+	yaml.Unmarshal(registyConfig, &Registry)
 
-	consulAddress := Server.ConsulHost + ":" + Server.ConsulPort
 	reg := consul.NewRegistry(func(op *registry.Options) {
 		op.Addrs = []string{
-			consulAddress,
+			Registry.Host,
 		}
 	})