Initial commit

qgrpc/config.go

@@ -0,0 +1,66 @@
+package qgrpc
+
+import (
+	"fmt"
+	"net"
+	"os"
+	"strings"
+)
+
+// 基础配置，公共字段
+type BasicConfig struct {
+	DialTimeout int      `toml:"dial_timeout"` // 连接超时时间
+	KeepAlive   int      `toml:"keep_alive"`   // 保活时间
+	EndPoints   []string `toml:"endpoints"`    // etcd的地址
+	Prefix      string   `toml:"prefix"`       // etcd服务的路径前缀，区分环境用
+}
+
+// 服务端配置
+type ServerConfig struct {
+	BasicConfig
+	Servers []string `toml:"servers"` // 注册的服务名称，可以是多个
+	Addr    string   `toml:"addr"`    // 服务地址
+}
+
+func (self *ServerConfig) GetAddr() string {
+	if strings.HasPrefix(self.Addr, ":") {
+		localIP := getLocalIP()
+		return fmt.Sprintf("%s%s", localIP, self.Addr)
+	}
+	return self.Addr
+}
+
+// 客户端配置
+type ClientConfig struct {
+	BasicConfig
+	WatchServers []string `toml:"watch_servers"` // 依赖的服务名称，多个服务用','分割；每个服务名后增加'|ip:port'表示默认连接的服务地址
+}
+
+func getHostName() string {
+	if h, err := os.Hostname(); err == nil {
+		return h
+	}
+	return getLocalIP()
+}
+
+//获取本机的内网Ip, 如果发现对方的ip 和自己的ip 相同，用127.0.0.1 替代
+func getLocalIP() string {
+	ifaces, _ := net.Interfaces()
+	for _, i := range ifaces {
+		addrs, _ := i.Addrs()
+		for _, addr := range addrs {
+			var ip net.IP
+			switch v := addr.(type) {
+			case *net.IPNet:
+				ip = v.IP
+			case *net.IPAddr:
+				ip = v.IP
+			}
+			ipAddr := ip.String()
+			if strings.HasPrefix(ipAddr, "172.") || strings.HasPrefix(ipAddr, "192.") || strings.HasPrefix(ipAddr, "10.") {
+				return ipAddr
+			}
+		}
+	}
+	return "127.0.0.1"
+}