增加请求发送以及响应体读取
This commit is contained in:
parent
091571128e
commit
24fe4c9adc
27
rpc.go
27
rpc.go
@ -8,8 +8,11 @@
|
|||||||
package rpc
|
package rpc
|
||||||
|
|
||||||
import (
|
import (
|
||||||
|
"bytes"
|
||||||
|
"encoding/json"
|
||||||
"errors"
|
"errors"
|
||||||
"fmt"
|
"fmt"
|
||||||
|
"io"
|
||||||
"net/http"
|
"net/http"
|
||||||
"strings"
|
"strings"
|
||||||
"sync"
|
"sync"
|
||||||
@ -261,12 +264,21 @@ func (r *request) Send(ctx *gin.Context, serviceFlag string, apiFlag string, par
|
|||||||
err error
|
err error
|
||||||
fullURL string
|
fullURL string
|
||||||
client *httpclient.HttpClient
|
client *httpclient.HttpClient
|
||||||
|
body []byte
|
||||||
|
response *httpclient.Response
|
||||||
|
responseBody []byte
|
||||||
)
|
)
|
||||||
if serviceConfig, apiConfig, err = r.GetServiceApi(serviceFlag, apiFlag); nil != err {
|
if serviceConfig, apiConfig, err = r.GetServiceApi(serviceFlag, apiFlag); nil != err {
|
||||||
return err
|
return err
|
||||||
}
|
}
|
||||||
// 完整的请求地址
|
// 完整的请求地址
|
||||||
fullURL = r.getFullURL(serviceConfig, apiConfig, parameter)
|
fullURL, body = r.getFullURLAndBody(serviceConfig, apiConfig, parameter)
|
||||||
|
if response, err = client.Do(apiConfig.Method, fullURL, apiConfig.Header, bytes.NewReader(body)); nil != err {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
if responseBody, err = io.ReadAll(response.Body); nil != err {
|
||||||
|
return err
|
||||||
|
}
|
||||||
return nil
|
return nil
|
||||||
}
|
}
|
||||||
|
|
||||||
@ -295,12 +307,13 @@ func (r *request) GetHttpClient(header map[string]string, timeout ApiTimeout) *h
|
|||||||
// Author : go_developer@163.com<白茶清欢>
|
// Author : go_developer@163.com<白茶清欢>
|
||||||
//
|
//
|
||||||
// Date : 17:23 2022/6/30
|
// Date : 17:23 2022/6/30
|
||||||
func (r *request) getFullURL(serviceConfig *Service, apiConfig *Api, parameter map[string]interface{}) string {
|
func (r *request) getFullURLAndBody(serviceConfig *Service, apiConfig *Api, parameter map[string]interface{}) (string, []byte) {
|
||||||
fullURL := strings.ReplaceAll(serviceConfig.Domain+"/"+apiConfig.URI, "//", "/")
|
fullURL := strings.ReplaceAll(serviceConfig.Domain+"/"+apiConfig.URI, "//", "/")
|
||||||
for name, val := range parameter {
|
for name, val := range parameter {
|
||||||
fullURL = strings.ReplaceAll(fullURL, "{{"+name+"}}", fmt.Sprintf("%v", val))
|
fullURL = strings.ReplaceAll(fullURL, "{{"+name+"}}", fmt.Sprintf("%v", val))
|
||||||
}
|
}
|
||||||
parameterPair := make([]string, 0)
|
parameterPair := make([]string, 0)
|
||||||
|
var body []byte
|
||||||
switch strings.ToUpper(apiConfig.Method) {
|
switch strings.ToUpper(apiConfig.Method) {
|
||||||
case http.MethodGet:
|
case http.MethodGet:
|
||||||
fallthrough
|
fallthrough
|
||||||
@ -318,10 +331,16 @@ func (r *request) getFullURL(serviceConfig *Service, apiConfig *Api, parameter m
|
|||||||
_ = util.ConvertAssign(&valStr, val)
|
_ = util.ConvertAssign(&valStr, val)
|
||||||
parameterPair = append(parameterPair, fmt.Sprintf("%v=%v", name, valStr))
|
parameterPair = append(parameterPair, fmt.Sprintf("%v=%v", name, valStr))
|
||||||
}
|
}
|
||||||
|
case http.MethodPost:
|
||||||
|
fallthrough
|
||||||
|
case http.MethodPut:
|
||||||
|
fallthrough
|
||||||
|
case http.MethodDelete:
|
||||||
|
body, _ = json.Marshal(parameter)
|
||||||
}
|
}
|
||||||
query := strings.Join(parameterPair, "&")
|
query := strings.Join(parameterPair, "&")
|
||||||
if len(query) == 0 {
|
if len(query) == 0 {
|
||||||
return strings.ReplaceAll(fullURL, "//", "/")
|
return strings.ReplaceAll(fullURL, "//", "/"), body
|
||||||
}
|
}
|
||||||
return strings.ReplaceAll(fullURL, "//", "/") + "?" + query
|
return strings.ReplaceAll(fullURL, "//", "/") + "?" + query, body
|
||||||
}
|
}
|
||||||
|
Loading…
Reference in New Issue
Block a user