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"
|
||
}
|