初始化项目

request.go

@@ -0,0 +1,218 @@
+package curlx
+
+import (
+	"bytes"
+	"encoding/json"
+	"encoding/xml"
+	"errors"
+	"io"
+	"net/http"
+	"net/url"
+	"strconv"
+	"strings"
+)
+
+/**
+ * 处理请求类型
+ */
+func (p *CurlParams) parseMethod() error {
+	if p.Method == "" {
+		return errors.New("请求类型不能为空")
+	}
+	return nil
+}
+
+/**
+ * 处理URL
+ */
+func (p *CurlParams) parseUrl() error {
+	_, err := url.Parse(p.Url)
+	if err != nil {
+		return err
+	}
+	return nil
+}
+
+/**
+ * 处理请求头Header
+ */
+func (p *CurlParams) parseHeaders(r *http.Request) {
+	if p.Headers != nil {
+		if r.Header.Get("User-Agent") == "" {
+			r.Header.Add("User-Agent", "Mozilla/5.0 (Windows NT 10.0; Win64; x64; rv:87.0) Gecko/20100101 Firefox/87.0 Send By Golang")
+		}
+		for k, v := range p.Headers {
+			if vv, ok := v.(string); ok {
+				r.Header.Set(k, vv)
+				continue
+			}
+			if vv, ok := v.([]string); ok {
+				for _, vvv := range vv {
+					r.Header.Add(k, vvv)
+				}
+			}
+		}
+	}
+}
+
+/**
+ * 处理请求参数
+ */
+func (p *CurlParams) parseParams() (str io.Reader, err error) {
+	err = nil
+
+	// 初始化(如未初始化)
+	if p.Headers == nil {
+		p.Headers = make(map[string]interface{})
+	}
+
+	if p.Params != nil {
+		if p.DataType == DataTypeJson {
+			// 判断是否存在
+			if _, ok := p.Headers["Content-Type"]; !ok {
+				p.Headers["Content-Type"] = "application/json"
+			}
+			strParam, ok := p.Params.(string)
+			if ok {
+				return bytes.NewReader([]byte(strParam)), nil
+			}
+			b, err := json.Marshal(p.Params)
+			if err == nil {
+				return bytes.NewReader(b), nil
+			}
+		} else if p.DataType == DataTypeXml {
+			if _, ok := p.Headers["Content-Type"]; !ok {
+				p.Headers["Content-Type"] = "application/xml"
+			}
+			var string_data string
+			if value, ok := p.Params.(string); ok {
+				string_data = string(value)
+			} else {
+				var by []byte
+				by, err = xml.Marshal(p.Params)
+				if err != nil {
+					return
+				}
+				string_data = string(by)
+			}
+			return strings.NewReader(string_data), nil
+			// switch p.Params.(type) {
+			// case map[string]string:
+			// 	// 请求参数转换成xml结构
+			// 	b, err := goutils.Map2XML(p.Params.(map[string]string))
+			// 	if err == nil {
+			// 		return bytes.NewBuffer(b)
+			// 	}
+			// default:
+			// 	b, err := xml.Marshal(p.Params)
+			// 	if err == nil {
+			// 		return bytes.NewBuffer(b)
+			// 	}
+			// }
+		} else if p.DataType == DataTypeText {
+			if _, ok := p.Headers["Content-Type"]; !ok {
+				p.Headers["Content-Type"] = "text/plain"
+			}
+
+			var string_data string
+			if value, ok := p.Params.(string); ok {
+				string_data = string(value)
+			} else {
+				err = errors.New("TEXT类型的参数仅支持字符串")
+				return
+			}
+
+			return strings.NewReader(string_data), nil
+		} else {
+			// FORM,""
+			if _, ok := p.Headers["Content-Type"]; !ok {
+				p.Headers["Content-Type"] = "application/x-www-form-urlencoded"
+			}
+
+			// 判断需要map[string]interface{}类型
+			paramValue, ok := p.Params.(map[string]interface{})
+			if !ok {
+				return strings.NewReader(""), errors.New("参数需map[string]interface{}")
+			}
+
+			values := url.Values{}
+			for k, v := range paramValue {
+				// 字符串
+				if v_string, ok := v.(string); ok {
+					values.Set(k, v_string)
+				}
+				// 字符串切片
+				if vv, ok := v.([]string); ok {
+					for _, vvv := range vv {
+						values.Add(k+"[]", vvv)
+					}
+				}
+				// int转string
+				if v_int, ok := v.(int); ok {
+					values.Set(k, strconv.Itoa(v_int))
+				}
+				// int64转string
+				if v_int64, ok := v.(int64); ok {
+					values.Set(k, strconv.FormatInt(v_int64, 10))
+				}
+				// float32转string
+				if v_float32, ok := v.(float32); ok {
+					values.Set(k, strconv.FormatFloat(float64(v_float32), 'f', -1, 32))
+				}
+				// float64转string
+				if v_float64, ok := v.(float64); ok {
+					values.Set(k, strconv.FormatFloat(v_float64, 'f', -1, 64))
+				}
+			}
+			return strings.NewReader(values.Encode()), nil
+		}
+
+	}
+	return
+}
+
+/**
+ * 处理Cookie
+ */
+func (p *CurlParams) parseCookies(r *http.Request) {
+	switch p.Cookies.(type) {
+	case string:
+		cookies := p.Cookies.(string)
+		r.Header.Add("Cookie", cookies)
+	case map[string]string:
+		cookies := p.Cookies.(map[string]string)
+		for k, v := range cookies {
+			r.AddCookie(&http.Cookie{
+				Name:  k,
+				Value: v,
+			})
+		}
+	case []*http.Cookie:
+		cookies := p.Cookies.([]*http.Cookie)
+		for _, cookie := range cookies {
+			r.AddCookie(cookie)
+		}
+	}
+}
+
+// func (r *Request) parseQuery() {
+// 	switch r.opts.Query.(type) {
+// 	case string:
+// 		str := r.opts.Query.(string)
+// 		r.req.URL.RawQuery = str
+// 	case map[string]interface{}:
+// 		q := r.req.URL.Query()
+// 		for k, v := range r.opts.Query.(map[string]interface{}) {
+// 			if vv, ok := v.(string); ok {
+// 				q.Set(k, vv)
+// 				continue
+// 			}
+// 			if vv, ok := v.([]string); ok {
+// 				for _, vvv := range vv {
+// 					q.Add(k, vvv)
+// 				}
+// 			}
+// 		}
+// 		r.req.URL.RawQuery = q.Encode()
+// 	}
+// }