67 lines
1.6 KiB
Go
67 lines
1.6 KiB
Go
|
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"
|
|||
|
|
}
|