处理服务的默认值配置

This commit is contained in:
白茶清欢 2022-06-30 18:14:43 +08:00
parent 7cb9e6b02c
commit f766b4c7e3
2 changed files with 15 additions and 0 deletions

View File

@ -73,4 +73,10 @@ const (
DefaultConnectTimeout = 100
// DefaultReadTimeout 默认读取超时
DefaultReadTimeout = 300
// DefaultCodeField 默认状态码字段
DefaultCodeField = "code"
// DefaultMessageField 默认描述字段
DefaultMessageField = "message"
// DefaultDataField 默认数据字段
DefaultDataField = "data"
)

9
rpc.go
View File

@ -73,6 +73,15 @@ func (r *request) AddService(serviceInfo *Service) error {
if nil == serviceInfo.SuccessHttpCodeList {
serviceInfo.SuccessHttpCodeList = []int{http.StatusOK}
}
if len(serviceInfo.CodeField) == 0 {
serviceInfo.CodeField = DefaultCodeField
}
if len(serviceInfo.DataField) == 0 {
serviceInfo.DataField = DefaultDataField
}
if len(serviceInfo.MessageField) == 0 {
serviceInfo.MessageField = DefaultMessageField
}
r.lock.Lock()
defer r.lock.Unlock()
if _, exist := r.serviceTable[serviceInfo.Flag]; exist {