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.
78 lines
2.1 KiB
Go
78 lines
2.1 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"
|
|
)
|
|
|
|
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]"`
|
|
// TODO: enable it in v1.2.4
|
|
// MaxRetries int `json:",default=0,range=[0:]"`
|
|
}
|
|
|
|
// 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"`
|
|
// TODO: enable it in v1.2.4
|
|
// Retry bool `json:",optional"` // grpc auto retry
|
|
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()
|
|
}
|
|
|
|
// HasCredential checks if there is a credential in config.
|
|
func (cc RpcClientConf) HasCredential() bool {
|
|
return len(cc.App) > 0 && len(cc.Token) > 0
|
|
}
|