增加URL解析以及单元测试

This commit is contained in:
白茶清欢 2021-07-24 21:37:28 +08:00
parent 9ae2ae32fa
commit 8c67426f0f
2 changed files with 105 additions and 0 deletions

62
gin/request/util.go Normal file
View File

@ -0,0 +1,62 @@
// Package request...
//
// Description : 解析请求
//
// Author : go_developer@163.com<白茶清欢>
//
// Date : 2021-07-24 8:53 下午
package request
import (
"net/url"
"strings"
"github.com/gin-gonic/gin"
)
// ParseGetRequestURI 解析URL请求参数
//
// Author : go_developer@163.com<白茶清欢>
//
// Date : 9:03 下午 2021/7/24
func ParseGetRequestURI(url *url.URL) map[string][]string {
url.RawQuery = strings.TrimSpace(url.RawQuery)
parseResult := make(map[string][]string)
if len(url.RawQuery) == 0 {
return parseResult
}
requestURIArr := strings.Split(url.RawQuery, "&")
for _, kvPair := range requestURIArr {
kvPair = strings.TrimSpace(kvPair)
if len(kvPair) == 0 {
continue
}
kvArr := strings.Split(kvPair, "=")
_, keyExist := parseResult[kvArr[0]]
if len(kvArr) == 1 {
// 识别到 ?a=1&b=2&c 这种格式,默认补齐成 ?a=1&b=2&c=
if keyExist {
parseResult[kvPair] = append(parseResult[kvPair], "")
} else {
parseResult[kvPair] = []string{""}
}
continue
}
// 兼容参数值中包含=的情况
if keyExist {
parseResult[kvArr[0]] = append(parseResult[kvArr[0]], strings.Join(kvArr[1:], "="))
} else {
parseResult[kvArr[0]] = []string{strings.Join(kvArr[1:], "=")}
}
}
return parseResult
}
// ParseGetRequestBody 解析get请求的body
//
// Author : zhangdeman001@ke.com<张德满>
//
// Date : 9:07 下午 2021/7/24
func ParseGetRequestBody(ctx *gin.Context) {
}

43
gin/request/util_test.go Normal file
View File

@ -0,0 +1,43 @@
// Package request...
//
// Description : request...
//
// Author : go_developer@163.com<张德满>
//
// Date : 2021-07-24 9:21 下午
package request
import (
"net/url"
"reflect"
"testing"
)
// TestParseGetRequestURI 测试解析请求uri
//
// Author : go_developer@163.com<白茶清欢>
//
// Date : 9:28 下午 2021/7/24
func TestParseGetRequestURI(t *testing.T) {
type args struct {
url *url.URL
}
tests := []struct {
name string
args args
want map[string][]string
}{
{"不存在任何参数", args{url: &url.URL{RawQuery: ""}}, make(map[string][]string)},
{"正常参数", args{url: &url.URL{RawQuery: "name=zhang&age=10"}}, map[string][]string{"name": {"zhang"}, "age": {"10"}}},
{"多个相同参数", args{url: &url.URL{RawQuery: "name=zhang&age=10&name=wang"}}, map[string][]string{"name": {"zhang", "wang"}, "age": {"10"}}},
{"末尾存在&符", args{url: &url.URL{RawQuery: "name=zhang&age=10&name=wang&"}}, map[string][]string{"name": {"zhang", "wang"}, "age": {"10"}}},
{"缺失=号", args{url: &url.URL{RawQuery: "name=zhang&age=10&name=wang&test"}}, map[string][]string{"name": {"zhang", "wang"}, "age": {"10"}, "test": {""}}},
}
for _, tt := range tests {
t.Run(tt.name, func(t *testing.T) {
if got := ParseGetRequestURI(tt.args.url); !reflect.DeepEqual(got, tt.want) {
t.Errorf("ParseGetRequestURI() = %v, want %v", got, tt.want)
}
})
}
}