7
0
gaore-common-sdk-go/sdk/requests/request.go

316 lines
6.8 KiB
Go
Raw Normal View History

2020-08-04 00:51:19 +08:00
package requests
import (
2020-08-06 10:35:24 +08:00
"errors"
"fmt"
2020-08-06 15:41:02 +08:00
"golib.gaore.com/GaoreGo/gaore-common-sdk-go/sdk/utils"
2020-08-05 12:14:49 +08:00
"io"
2020-08-06 12:15:51 +08:00
"math/cmplx"
2020-08-06 10:35:24 +08:00
"reflect"
2020-08-06 12:15:51 +08:00
"strconv"
2020-08-04 00:51:19 +08:00
"time"
)
const (
2024-05-30 18:40:57 +08:00
RPC = "RPC"
ROA = "ROA"
STREAM = "STREAM"
2020-08-04 00:51:19 +08:00
HTTP = "HTTP"
HTTPS = "HTTPS"
2020-08-06 10:35:24 +08:00
JSON = "JSON"
XML = "XML"
2020-08-04 00:51:19 +08:00
DefaultHttpPort = "80"
GET = "GET"
PUT = "PUT"
POST = "POST"
DELETE = "DELETE"
PATCH = "PATCH"
HEAD = "HEAD"
OPTIONS = "OPTIONS"
Json = "application/json"
Xml = "application/xml"
Raw = "application/octet-stream"
Form = "application/x-www-form-urlencoded"
Header = "Header"
Query = "Query"
Body = "Body"
Path = "Path"
TEST = "TEST"
PRE = "PRE"
RELEASE = "RELEASE"
2020-08-04 00:51:19 +08:00
HeaderSeparator = "\n"
)
type Host struct {
Default string
Func func(string) string
}
2020-08-06 15:41:02 +08:00
var debug utils.Debug
func init() {
debug = utils.Init("request")
}
2020-08-04 00:51:19 +08:00
type AcsRequest interface {
GetReadTimeout() time.Duration
GetConnectTimeout() time.Duration
SetReadTimeout(readTimeOut time.Duration)
SetConnectTimeout(connectTimeOut time.Duration)
SetHTTPSInsecure(isInsecure bool)
GetHTTPSInsecure() *bool
GetQueryParams() map[string]string
GetFormParams() map[string]string
GetMethod() string
2020-08-04 11:22:37 +08:00
GetScheme() string
GetDomain() Host
SetDomain(host Host)
2020-08-04 12:23:34 +08:00
GetActionName() string
2020-08-05 17:01:10 +08:00
GetAcceptFormat() string
GetAccept() string
2020-08-05 12:14:49 +08:00
GetHeaders() map[string]string
GetStyle() string
InitWithApiInfo(domain Host, version, urlPath string)
GetEnv() string
SetEnv(string)
2020-08-04 11:22:37 +08:00
BuildUrl() string
BuildQueries() string
SetScheme(scheme string)
2020-08-06 10:35:24 +08:00
SetContent(content []byte)
2020-08-04 00:51:19 +08:00
SetStringToSign(stringToSign string)
2020-08-06 12:15:51 +08:00
GetStringToSign() string
2020-08-05 12:14:49 +08:00
GetBodyReader() io.Reader
AddHeaderParam(key, value string)
2020-08-05 12:14:49 +08:00
addQueryParam(key, value string)
addFormParam(key, value string)
2020-08-04 00:51:19 +08:00
}
type baseRequest struct {
Scheme string
Method string
2020-08-04 11:22:37 +08:00
Port string
Domain Host
2020-08-04 00:51:19 +08:00
From string
ReadTimeout time.Duration
ConnectTimeout time.Duration
isInsecure *bool
Env string
2020-08-04 00:51:19 +08:00
AcceptFormat string
2020-08-04 12:23:34 +08:00
actionName string
2020-08-04 00:51:19 +08:00
userAgent map[string]string
product string
version string
QueryParams map[string]string
Headers map[string]string
FormParams map[string]string
Content []byte
queries string
stringToSign string
}
func (request *baseRequest) GetEnv() string {
return request.Env
}
func (request *baseRequest) SetEnv(e string) {
request.Env = e
}
2020-08-06 12:15:51 +08:00
func (request *baseRequest) GetStringToSign() string {
return request.stringToSign
}
2020-08-06 10:35:24 +08:00
func (request *baseRequest) SetContent(content []byte) {
request.Content = content
}
2020-08-05 17:01:10 +08:00
func (request *baseRequest) GetAcceptFormat() string {
return request.AcceptFormat
}
func (request *baseRequest) GetAccept() string {
switch request.GetAcceptFormat() {
case JSON:
return Json
case XML:
return Xml
}
return ""
}
2020-08-05 12:14:49 +08:00
func (request *baseRequest) GetHeaders() map[string]string {
return request.Headers
}
2020-08-04 12:23:34 +08:00
func (request *baseRequest) GetActionName() string {
return request.actionName
}
2020-08-04 11:22:37 +08:00
func (request *baseRequest) SetScheme(scheme string) {
request.Scheme = scheme
}
func (request *baseRequest) SetDomain(host Host) {
2020-08-04 11:22:37 +08:00
request.Domain = host
}
func (request *baseRequest) GetScheme() string {
return request.Scheme
}
func (request *baseRequest) GetDomain() Host {
2020-08-04 11:22:37 +08:00
return request.Domain
}
2020-08-04 00:51:19 +08:00
func (request *baseRequest) GetMethod() string {
return request.Method
}
func (request *baseRequest) GetFormParams() map[string]string {
return request.FormParams
}
func (request *baseRequest) GetQueryParams() map[string]string {
return request.QueryParams
}
func (request *baseRequest) SetHTTPSInsecure(isInsecure bool) {
request.isInsecure = &isInsecure
}
func (request *baseRequest) GetHTTPSInsecure() *bool {
return request.isInsecure
}
func (request *baseRequest) GetReadTimeout() time.Duration {
return request.ReadTimeout
}
func (request *baseRequest) GetConnectTimeout() time.Duration {
return request.ConnectTimeout
}
func (request *baseRequest) SetReadTimeout(readTimeOut time.Duration) {
request.ReadTimeout = readTimeOut
}
func (request *baseRequest) SetConnectTimeout(connectTimeOut time.Duration) {
request.ConnectTimeout = connectTimeOut
}
func (request *baseRequest) SetStringToSign(stringToSign string) {
request.stringToSign = stringToSign
}
func (request *baseRequest) AddHeaderParam(key, val string) {
2020-08-05 12:14:49 +08:00
request.Headers[key] = val
}
func (request *baseRequest) addQueryParam(key, val string) {
request.QueryParams[key] = val
}
func (request *baseRequest) addFormParam(key, val string) {
request.FormParams[key] = val
}
2020-08-04 00:51:19 +08:00
func defaultBaseRequest() (request *baseRequest) {
request = &baseRequest{
2020-08-05 17:01:10 +08:00
Scheme: HTTP,
2020-08-06 10:35:24 +08:00
AcceptFormat: JSON,
2020-08-04 00:51:19 +08:00
Method: GET,
QueryParams: make(map[string]string),
Headers: map[string]string{
"Gr-Sdk-Client": "golang/1.14",
"Gr-Sdk-Invoke-Type": "normal",
2020-08-06 10:35:24 +08:00
"Accept-Encoding": Json,
2020-08-04 00:51:19 +08:00
},
FormParams: make(map[string]string),
}
return
}
2020-08-04 11:22:37 +08:00
func InitParam(request AcsRequest) (err error) {
2020-08-06 10:35:24 +08:00
reflectValue := reflect.ValueOf(request).Elem()
2020-08-06 12:15:51 +08:00
err = flatRepeatedList(reflectValue, request, "")
return nil
}
func flatRepeatedList(reflectValue reflect.Value, request AcsRequest, position string) (err error) {
2020-08-06 10:35:24 +08:00
reflectType := reflectValue.Type()
for i := 0; i < reflectType.NumField(); i++ {
field := reflectType.Field(i)
name, isContiansNameTag := field.Tag.Lookup("field")
2020-08-06 12:15:51 +08:00
fieldPosition := position
if fieldPosition == "" {
fieldPosition, _ = field.Tag.Lookup("position")
}
fieldDefault, _ := field.Tag.Lookup("default")
2020-08-06 15:41:02 +08:00
debug("%s %s %s", name, fieldPosition, fieldDefault)
2020-08-06 10:35:24 +08:00
if isContiansNameTag {
2020-08-06 12:15:51 +08:00
var value string
switch field.Type.Kind() {
case reflect.Int, reflect.Int8, reflect.Int16, reflect.Int32, reflect.Int64:
value = strconv.FormatInt(reflectValue.Field(i).Int(), 10)
case reflect.Uintptr, reflect.Uint, reflect.Uint8, reflect.Uint16, reflect.Uint32, reflect.Uint64:
value = strconv.FormatUint(reflectValue.Field(i).Uint(), 10)
case reflect.Float32, reflect.Float64:
value = strconv.FormatFloat(reflectValue.Field(i).Float(), 'E', -1, 64)
case reflect.Bool:
value = strconv.FormatBool(reflectValue.Field(i).Bool())
case reflect.Complex64, reflect.Complex128:
value = fmt.Sprint(cmplx.Sqrt(reflectValue.Field(i).Complex()))
default:
value = reflectValue.Field(i).String()
}
if len(value) == 0 {
value = fieldDefault
}
2020-08-06 16:04:44 +08:00
if value == "0" && fieldDefault != "" && fieldDefault != "0" {
value = fieldDefault
}
2020-08-06 12:15:51 +08:00
err = addParam(request, fieldPosition, name, value)
2020-08-06 10:35:24 +08:00
}
}
2020-08-06 12:15:51 +08:00
return
2020-08-04 11:22:37 +08:00
}
2020-08-06 10:35:24 +08:00
func addParam(request AcsRequest, position, key, value string) (err error) {
if len(value) > 0 {
switch position {
case Header:
request.AddHeaderParam(key, value)
2020-08-06 10:35:24 +08:00
case Query:
request.addQueryParam(key, value)
case Body:
request.addFormParam(key, value)
default:
errmsg := fmt.Sprintf("unsupport positions add param `%s`", position)
err = errors.New(errmsg)
}
}
return
}