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.
94 lines
2.4 KiB
Go
94 lines
2.4 KiB
Go
package zrpc
|
|
|
|
import (
|
|
"github.com/tal-tech/go-zero/core/discov"
|
|
"github.com/tal-tech/go-zero/core/service"
|
|
"github.com/tal-tech/go-zero/core/stores/redis"
|
|
"github.com/tal-tech/go-zero/zrpc/resolver"
|
|
)
|
|
|
|
type (
|
|
// A RpcServerConf is a rpc server config.
|
|
RpcServerConf struct {
|
|
service.ServiceConf
|
|
ListenOn string
|
|
Etcd discov.EtcdConf `json:",optional"`
|
|
Auth bool `json:",optional"`
|
|
Redis redis.RedisKeyConf `json:",optional"`
|
|
StrictControl bool `json:",optional"`
|
|
// setting 0 means no timeout
|
|
Timeout int64 `json:",default=2000"`
|
|
CpuThreshold int64 `json:",default=900,range=[0:1000]"`
|
|
}
|
|
|
|
// A RpcClientConf is a rpc client config.
|
|
RpcClientConf struct {
|
|
Etcd discov.EtcdConf `json:",optional"`
|
|
Endpoints []string `json:",optional"`
|
|
Target string `json:",optional"`
|
|
App string `json:",optional"`
|
|
Token string `json:",optional"`
|
|
NonBlock bool `json:",optional"`
|
|
Timeout int64 `json:",default=2000"`
|
|
}
|
|
)
|
|
|
|
// NewDirectClientConf returns a RpcClientConf.
|
|
func NewDirectClientConf(endpoints []string, app, token string) RpcClientConf {
|
|
return RpcClientConf{
|
|
Endpoints: endpoints,
|
|
App: app,
|
|
Token: token,
|
|
}
|
|
}
|
|
|
|
// NewEtcdClientConf returns a RpcClientConf.
|
|
func NewEtcdClientConf(hosts []string, key, app, token string) RpcClientConf {
|
|
return RpcClientConf{
|
|
Etcd: discov.EtcdConf{
|
|
Hosts: hosts,
|
|
Key: key,
|
|
},
|
|
App: app,
|
|
Token: token,
|
|
}
|
|
}
|
|
|
|
// HasEtcd checks if there is etcd settings in config.
|
|
func (sc RpcServerConf) HasEtcd() bool {
|
|
return len(sc.Etcd.Hosts) > 0 && len(sc.Etcd.Key) > 0
|
|
}
|
|
|
|
// Validate validates the config.
|
|
func (sc RpcServerConf) Validate() error {
|
|
if !sc.Auth {
|
|
return nil
|
|
}
|
|
|
|
return sc.Redis.Validate()
|
|
}
|
|
|
|
// BuildTarget builds the rpc target from the given config.
|
|
func (cc RpcClientConf) BuildTarget() (string, error) {
|
|
if len(cc.Endpoints) > 0 {
|
|
return resolver.BuildDirectTarget(cc.Endpoints), nil
|
|
} else if len(cc.Target) > 0 {
|
|
return cc.Target, nil
|
|
}
|
|
|
|
if err := cc.Etcd.Validate(); err != nil {
|
|
return "", err
|
|
}
|
|
|
|
if cc.Etcd.HasAccount() {
|
|
discov.RegisterAccount(cc.Etcd.Hosts, cc.Etcd.User, cc.Etcd.Pass)
|
|
}
|
|
|
|
return resolver.BuildDiscovTarget(cc.Etcd.Hosts, cc.Etcd.Key), nil
|
|
}
|
|
|
|
// HasCredential checks if there is a credential in config.
|
|
func (cc RpcClientConf) HasCredential() bool {
|
|
return len(cc.App) > 0 && len(cc.Token) > 0
|
|
}
|