You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
73 lines
1.5 KiB
Go
73 lines
1.5 KiB
Go
package internal
|
|
|
|
import (
|
|
"os"
|
|
"strings"
|
|
|
|
"github.com/zeromicro/go-zero/core/discov"
|
|
"github.com/zeromicro/go-zero/core/netx"
|
|
)
|
|
|
|
const (
|
|
allEths = "0.0.0.0"
|
|
envPodIp = "POD_IP"
|
|
)
|
|
|
|
// NewRpcPubServer returns a Server.
|
|
func NewRpcPubServer(etcd discov.EtcdConf, listenOn string, opts ...ServerOption) (Server, error) {
|
|
registerEtcd := func() error {
|
|
pubListenOn := figureOutListenOn(listenOn)
|
|
var pubOpts []discov.PubOption
|
|
if etcd.HasAccount() {
|
|
pubOpts = append(pubOpts, discov.WithPubEtcdAccount(etcd.User, etcd.Pass))
|
|
}
|
|
if etcd.HasTLS() {
|
|
pubOpts = append(pubOpts, discov.WithPubEtcdTLS(etcd.CertFile, etcd.CertKeyFile,
|
|
etcd.CACertFile, etcd.InsecureSkipVerify))
|
|
}
|
|
pubClient := discov.NewPublisher(etcd.Hosts, etcd.Key, pubListenOn, pubOpts...)
|
|
return pubClient.KeepAlive()
|
|
}
|
|
server := keepAliveServer{
|
|
registerEtcd: registerEtcd,
|
|
Server: NewRpcServer(listenOn, opts...),
|
|
}
|
|
|
|
return server, nil
|
|
}
|
|
|
|
type keepAliveServer struct {
|
|
registerEtcd func() error
|
|
Server
|
|
}
|
|
|
|
func (s keepAliveServer) Start(fn RegisterFn) error {
|
|
if err := s.registerEtcd(); err != nil {
|
|
return err
|
|
}
|
|
|
|
return s.Server.Start(fn)
|
|
}
|
|
|
|
func figureOutListenOn(listenOn string) string {
|
|
fields := strings.Split(listenOn, ":")
|
|
if len(fields) == 0 {
|
|
return listenOn
|
|
}
|
|
|
|
host := fields[0]
|
|
if len(host) > 0 && host != allEths {
|
|
return listenOn
|
|
}
|
|
|
|
ip := os.Getenv(envPodIp)
|
|
if len(ip) == 0 {
|
|
ip = netx.InternalIp()
|
|
}
|
|
if len(ip) == 0 {
|
|
return listenOn
|
|
}
|
|
|
|
return strings.Join(append([]string{ip}, fields[1:]...), ":")
|
|
}
|