bingyi/gf
1
0
Fork 0
mirror of https://gitee.com/johng/gf synced 2026-06-06 02:25:47 +08:00
Code Issues Packages Projects Releases Wiki Activity

improve perameter parsing for ghttp.Server

Browse Source
This commit is contained in:
John
2019-09-18 23:20:45 +08:00
parent c2ad9f5fb9
commit 7d32be3b6c
22 changed files with 379 additions and 314 deletions
Download Patch File Download Diff File Expand all files Collapse all files

34
.example/database/gdb/mysql/gdb_update_field.go Normal file
View File

@ -0,0 +1,34 @@
package main
import (
"database/sql"
"github.com/gogf/gf/os/gfile"
"github.com/gogf/gf/encoding/gjson"
"github.com/gogf/gf/frame/g"
)
func main() {
db := g.DB()
table := "medicine_clinics_upload_yinchuan"
list, err := db.Table(table).All()
if err != nil && err != sql.ErrNoRows {
panic(err)
}
content := ""
for _, item := range list {
if j, err := gjson.DecodeToJson(item["upload_data"].String()); err != nil {
panic(err)
} else {
s, _ := j.ToJsonIndentString()
content += item["id"].String() + "\t" + item["medicine_clinic_id"].String() + "\t"
content += s
content += "\n\n"
//if _, err := db.Table(table).Data("data_decode", s).Where("id", item["id"].Int()).Update(); err != nil {
// panic(err)
//}
}
}
gfile.PutContents("/Users/john/Temp/medicine_clinics_upload_yinchuan.txt", content)
}

1
.example/database/gdb/mysql/gdb_value.go
View File

@ -8,7 +8,6 @@ import (
func main() {
db := g.DB()
// 开启调试模式，以便于记录所有执行的SQL
db.SetDebug(true)
r, e := db.Table("test").Where("id IN (?)", []interface{}{1, 2}).All()

16
.example/net/ghttp/server/form/form.go Normal file
View File

@ -0,0 +1,16 @@
package main
import (
"github.com/gogf/gf/frame/g"
"github.com/gogf/gf/net/ghttp"
)
func main() {
s := g.Server()
s.BindHandler("/", func(r *ghttp.Request) {
g.Dump(r.GetPostMap())
r.Response.WriteTpl("form.html")
})
s.SetPort(8199)
s.Run()
}

30
.example/net/ghttp/server/form/form.html Normal file
View File

@ -0,0 +1,30 @@
<html>
<head>
<title>form test</title>
</head>
<body>
<h1>form1</h1>
<form action="?" method="post" enctype="multipart/form-data">
<p><input type="text" name="input1" value="1"/></p>
<p><input type="text" name="input2" value="2"/></p>
<p><input type="text" name="array1" value="3"/></p>
<p><input type="text" name="array1" value="4"/></p>
<p><input type="text" name="array2[]" value="5"/></p>
<p><input type="text" name="array2[]" value="6"/></p>
<p><input type="text" name="map[a]" value="7"/></p>
<p><input type="text" name="map[b]" value="8"/></p>
<p><input type="password" name="password1" value="9"/></p>
<input type="submit" value="Submit" />
</form>
<h1>form2</h1>
<form action="?" method="post" enctype="application/x-www-form-urlencoded">
<p><input type="text" name="input[a]" value="1"/></p>
<p><input type="text" name="input[b]" value="2"/></p>
<p><input type="text" name="array" value="3"/></p>
<p><input type="text" name="array" value="4"/></p>
<p><input type="password" name="password2" value="5"/></p>
<input type="submit" value="Submit" />
</form>
</body>

7
.example/other/test.go
View File

@ -2,10 +2,15 @@ package main
import (
"fmt"
"reflect"
"github.com/gogf/gf/frame/g"
"github.com/gogf/gf/text/gstr"
)
func main() {
fmt.Println(gstr.CamelCase("userLoginLog.bak"))
m, _ := gstr.Parse("map[a]=1&map[b]=2")
g.Dump(m)
fmt.Println(reflect.TypeOf(m["map"].(map[string]interface{})["b"]))
}

6
go.sum
View File

@ -6,15 +6,21 @@ github.com/fatih/structs v1.1.0 h1:Q7juDM0QtcnhCpeyLGQKyg4TOIghuNXrkL32pHAUMxo=
github.com/fatih/structs v1.1.0/go.mod h1:9NiDSp5zOcgEDl+j00MP/WkGVPOlPRLejGD8Ga6PJ7M=
github.com/fsnotify/fsnotify v1.4.7 h1:IXs+QLmnXW2CcXuY+8Mzv/fWEsPGWxqefPtCP5CnV9I=
github.com/fsnotify/fsnotify v1.4.7/go.mod h1:jwhsz4b93w/PPRr/qN1Yymfu8t87LnFCMoQvtojpjFo=
github.com/gf-third/mysql v1.4.2 h1:f1M5CNFUG3WkE07UOomtu4o0n/KJKeuUUf5Nc9ZFXs4=
github.com/gf-third/mysql v1.4.2/go.mod h1:+dd90V663ppI2fV5uQ6+rHk0u8KCyU6FkG8Um8Cx3ms=
github.com/gf-third/yaml/v3 v3.0.0 h1:IMLH3JWFpNraTz5d6jzaNZFvWaVYAwhP9w1lhk9SFUs=
github.com/gf-third/yaml/v3 v3.0.0/go.mod h1:En6jd9ZtAhuCiVfRnTo8NYVgbiZvg/F26r8Tj+vsUy4=
github.com/gofrs/flock v0.7.1/go.mod h1:F1TvTiK9OcQqauNUHlbJvyl9Qa1QvF/gOUDKA14jxHU=
github.com/golang/protobuf v1.3.1/go.mod h1:6lQm79b+lXiMfvg/cZm0SGofjICqVBUtrP5yJMmIC1U=
github.com/gomodule/redigo v2.0.0+incompatible h1:K/R+8tc58AaqLkqG2Ol3Qk+DR/TlNuhuh457pBFPtt0=
github.com/gomodule/redigo v2.0.0+incompatible/go.mod h1:B4C85qUVwatsJoIUNIfCRsp7qO0iAmpGFZ4EELWSbC4=
github.com/gorilla/websocket v1.4.1 h1:q7AeDBpnBk8AogcD4DSag/Ukw/KV+YhzLj2bP5HvKCM=
github.com/gorilla/websocket v1.4.1/go.mod h1:YR8l580nyteQvAITg2hZ9XVh4b55+EU/adAjf1fMHhE=
github.com/grokify/html-strip-tags-go v0.0.0-20190916062342-6f856a90d556 h1:RS7ewakEriTcISIy+USQV9tE37SpWCblCiPE6QWxagk=
github.com/grokify/html-strip-tags-go v0.0.0-20190916062342-6f856a90d556/go.mod h1:Xk7G0nwBiIloTMbLddk4WWJOqi4i/JLhadLd0HUXO30=
github.com/mattn/go-runewidth v0.0.4 h1:2BvfKmzob6Bmd4YsL0zygOqfdFnK7GR4QL06Do4/p7Y=
github.com/mattn/go-runewidth v0.0.4/go.mod h1:LwmH8dsx7+W8Uxz3IHJYH5QSwggIsqBzpuz5H//U1FU=
github.com/olekukonko/tablewriter v0.0.1 h1:b3iUnf1v+ppJiOfNX4yxxqfWKMQPZR5yoh8urCTFX88=
github.com/olekukonko/tablewriter v0.0.1/go.mod h1:vsDQFd/mU46D+Z4whnwzcISnGGzXWMclvtLoiIKAKIo=
github.com/theckman/go-flock v0.7.1/go.mod h1:kjuth3y9VJ2aNlkNEO99G/8lp9fMIKaGyBmh84IBheM=
golang.org/x/crypto v0.0.0-20190308221718-c2843e01d9a2/go.mod h1:djNgcEr1/C05ACkg1iLfiJU5Ep61QUkGW8qpdssI0+w=

37
net/ghttp/ghttp_request.go
View File

@ -38,8 +38,11 @@ type Request struct {
hasServeHandler bool // 是否检索到服务函数
parsedGet bool // GET参数是否已经解析
parsedPost bool // POST参数是否已经解析
queryVars map[string][]string // GET参数
routerVars map[string][]string // 路由解析参数
parsedRaw bool // 原始参数是否已经解析
getMap map[string]interface{} // GET解析参数
postMap map[string]interface{} // POST解析参数
routerMap map[string]interface{} // 路由解析参数
rawVarMap map[string]interface{} // 原始数据参数
error error // 当前请求执行错误
exit bool // 是否退出当前请求流程执行
params map[string]interface{} // 开发者自定义参数(请求流程中有效)
@ -52,12 +55,12 @@ type Request struct {
// 创建一个Request对象
func newRequest(s *Server, r *http.Request, w http.ResponseWriter) *Request {
request := &Request{
routerVars: make(map[string][]string),
Id: s.servedCount.Add(1),
Server: s,
Request: r,
Response: newResponse(s, w),
EnterTime: gtime.Microsecond(),
routerMap: make(map[string]interface{}),
Id: s.servedCount.Add(1),
Server: s,
Request: r,
Response: newResponse(s, w),
EnterTime: gtime.Microsecond(),
}
// 会话处理
request.Cookie = GetCookie(request)
@ -82,8 +85,8 @@ func (r *Request) WebSocket() (*WebSocket, error) {
// 获得指定名称的参数字符串(Router/GET/POST)，同 GetRequestString
// 这是常用方法的简化别名
func (r *Request) Get(key string, def ...interface{}) string {
return r.GetRequestString(key, def...)
func (r *Request) Get(key string, def ...interface{}) interface{} {
return r.GetRequest(key, def...)
}
// 建议都用该参数替代参数获取
@ -113,6 +116,10 @@ func (r *Request) GetString(key string, def ...interface{}) string {
return r.GetRequestString(key, def...)
}
func (r *Request) GetBool(key string, def ...interface{}) bool {
return r.GetRequestBool(key, def...)
}
func (r *Request) GetInt(key string, def ...interface{}) int {
return r.GetRequestInt(key, def...)
}
@ -149,7 +156,7 @@ func (r *Request) GetInterfaces(key string, def ...interface{}) []interface{} {
return r.GetRequestInterfaces(key, def...)
}
func (r *Request) GetMap(def ...map[string]string) map[string]string {
func (r *Request) GetMap(def ...map[string]interface{}) map[string]interface{} {
return r.GetRequestMap(def...)
}
@ -236,14 +243,6 @@ func (r *Request) GetSessionId() string {
return id
}
// 生成随机的SESSIONID
func (r *Request) MakeSessionId() string {
id := gsession.NewSessionId()
r.Cookie.SetSessionId(id)
r.Response.Header().Set(r.Server.GetSessionIdName(), id)
return id
}
// 获得请求来源URL地址
func (r *Request) GetReferer() string {
return r.Header.Get("Referer")

2
net/ghttp/ghttp_request_middleware.go
View File

@ -38,7 +38,7 @@ func (m *Middleware) Next() {
}
// 路由参数赋值
for k, v := range item.values {
m.request.routerVars[k] = v
m.request.routerMap[k] = v
}
m.request.Router = item.handler.router
// 执行函数处理

15
net/ghttp/ghttp_request_params.go
View File

@ -17,14 +17,17 @@ func (r *Request) SetParam(key string, value interface{}) {
}
// 获取请求流程共享变量
func (r *Request) GetParam(key string, def ...interface{}) *gvar.Var {
func (r *Request) GetParam(key string, def ...interface{}) interface{} {
if r.params != nil {
if v, ok := r.params[key]; ok {
return gvar.New(v)
}
return r.params[key]
}
if len(def) > 0 {
return gvar.New(def[0])
return def[0]
}
return gvar.New(nil)
return nil
}
// 获取请求流程共享变量
func (r *Request) GetParamVar(key string, def ...interface{}) *gvar.Var {
return gvar.New(r.GetParam(key, def...))
}

155
net/ghttp/ghttp_request_post.go
View File

@ -7,156 +7,153 @@
package ghttp
import (
"strings"
"github.com/gogf/gf/encoding/gurl"
"github.com/gogf/gf/container/gvar"
"github.com/gogf/gf/internal/structs"
"github.com/gogf/gf/text/gstr"
"github.com/gogf/gf/util/gconv"
)
// 初始化POST请求参数
func (r *Request) initPost() {
if !r.parsedPost {
// MultiMedia表单请求解析允许最大使用内存：1GB
if r.ParseMultipartForm(1024*1024*1024) == nil {
r.parsedPost = true
r.parsedPost = true
if v := r.Header.Get("Content-Type"); v != "" && gstr.Contains(v, "multipart/") {
// multipart/form-data, multipart/mixed
// MultiMedia表单请求解析允许最大使用内存：1GB
r.ParseMultipartForm(1024 * 1024 * 1024)
if len(r.PostForm) > 0 {
// 重新组织数据格式，使用统一的数据Parse方式
params := ""
for name, values := range r.PostForm {
if len(values) == 1 {
if len(params) > 0 {
params += "&"
}
params += name + "=" + gurl.Encode(values[0])
} else {
if len(name) > 2 && name[len(name)-2:] == "[]" {
name = name[:len(name)-2]
for _, v := range values {
if len(params) > 0 {
params += "&"
}
params += name + "[]=" + gurl.Encode(v)
}
} else {
if len(params) > 0 {
params += "&"
}
params += name + "=" + gurl.Encode(values[len(values)-1])
}
}
}
r.postMap, _ = gstr.Parse(params)
}
} else if strings.EqualFold(r.Method, "POST") {
r.parsedRaw = true
if raw := r.GetRawString(); len(raw) > 0 {
r.postMap, _ = gstr.Parse(raw)
}
}
}
}
// 设置POST参数，仅在ghttp.Server内有效，**注意并发安全性**
func (r *Request) SetPost(key string, value string) {
// 设置当前请求的POST参数
func (r *Request) SetPost(key string, value interface{}) {
r.initPost()
r.PostForm[key] = []string{value}
r.postMap[key] = value
}
func (r *Request) AddPost(key string, value string) {
func (r *Request) GetPost(key string, def ...interface{}) interface{} {
r.initPost()
r.PostForm[key] = append(r.PostForm[key], value)
}
func (r *Request) GetPost(key string, def ...interface{}) []string {
r.initPost()
if v, ok := r.PostForm[key]; ok {
if v, ok := r.postMap[key]; ok {
return v
}
if len(def) > 0 {
return gconv.Strings(def[0])
return def[0]
}
return nil
}
func (r *Request) GetPostVar(key string, def ...interface{}) *gvar.Var {
return gvar.New(r.GetPostString(key, def...))
return gvar.New(r.GetPost(key, def...))
}
func (r *Request) GetPostString(key string, def ...interface{}) string {
value := r.GetPost(key, def...)
if value != nil && value[0] != "" {
return value[0]
}
return ""
return r.GetPostVar(key, def...).String()
}
func (r *Request) GetPostBool(key string, def ...interface{}) bool {
value := r.GetPostString(key, def...)
if value != "" {
return gconv.Bool(value)
}
return false
return r.GetPostVar(key, def...).Bool()
}
func (r *Request) GetPostInt(key string, def ...interface{}) int {
value := r.GetPostString(key, def...)
if value != "" {
return gconv.Int(value)
}
return 0
return r.GetPostVar(key, def...).Int()
}
func (r *Request) GetPostInts(key string, def ...interface{}) []int {
value := r.GetPost(key, def...)
if value != nil {
return gconv.Ints(value)
}
return nil
return r.GetPostVar(key, def...).Ints()
}
func (r *Request) GetPostUint(key string, def ...interface{}) uint {
value := r.GetPostString(key, def...)
if value != "" {
return gconv.Uint(value)
}
return 0
return r.GetPostVar(key, def...).Uint()
}
func (r *Request) GetPostFloat32(key string, def ...interface{}) float32 {
value := r.GetPostString(key, def...)
if value != "" {
return gconv.Float32(value)
}
return 0
return r.GetPostVar(key, def...).Float32()
}
func (r *Request) GetPostFloat64(key string, def ...interface{}) float64 {
value := r.GetPostString(key, def...)
if value != "" {
return gconv.Float64(value)
}
return 0
return r.GetPostVar(key, def...).Float64()
}
func (r *Request) GetPostFloats(key string, def ...interface{}) []float64 {
value := r.GetPost(key, def...)
if value != nil {
return gconv.Floats(value)
}
return nil
return r.GetPostVar(key, def...).Floats()
}
func (r *Request) GetPostArray(key string, def ...interface{}) []string {
return r.GetPost(key, def...)
return r.GetPostVar(key, def...).Strings()
}
func (r *Request) GetPostStrings(key string, def ...interface{}) []string {
return r.GetPost(key, def...)
return r.GetPostVar(key, def...).Strings()
}
func (r *Request) GetPostInterfaces(key string, def ...interface{}) []interface{} {
value := r.GetPost(key, def...)
if value != nil {
return gconv.Interfaces(value)
}
return nil
return r.GetPostVar(key, def...).Interfaces()
}
// 获取指定键名的关联数组，并且给定当指定键名不存在时的默认值
// 需要注意的是，如果其中一个字段为数组形式，那么只会返回第一个元素，如果需要获取全部的元素，请使用GetPostArray获取特定字段内容
func (r *Request) GetPostMap(def ...map[string]string) map[string]string {
// 获取指定键名的关联数组，并且给定当指定键名不存在时的默认值。
// 当不指定键值对关联数组时，默认获取POST方式提交的所有的提交键值对数据。
func (r *Request) GetPostMap(kvMap ...map[string]interface{}) map[string]interface{} {
r.initPost()
m := make(map[string]string)
for k, v := range r.PostForm {
m[k] = v[0]
}
if len(def) > 0 {
for k, v := range def[0] {
if _, ok := m[k]; !ok {
m[k] = v
if len(kvMap) > 0 {
m := make(map[string]interface{})
for k, defValue := range kvMap[0] {
if postValue, ok := r.postMap[k]; ok {
m[k] = postValue
} else {
m[k] = defValue
}
}
return m
} else {
return r.postMap
}
return m
}
// 将所有的request参数映射到struct属性上，参数object应当为一个struct对象的指针, mapping为非必需参数，自定义参数与属性的映射关系
func (r *Request) GetPostToStruct(pointer interface{}, mapping ...map[string]string) error {
r.initPost()
tagMap := structs.TagMapName(pointer, paramTagPriority, true)
if len(mapping) > 0 {
for k, v := range mapping[0] {
tagMap[k] = v
}
}
params := make(map[string]interface{})
for k, v := range r.GetPostMap() {
params[k] = v
}
return gconv.StructDeep(params, pointer, tagMap)
return gconv.StructDeep(r.postMap, pointer, tagMap)
}

129
net/ghttp/ghttp_request_query.go
View File

@ -9,6 +9,8 @@ package ghttp
import (
"strings"
"github.com/gogf/gf/text/gstr"
"github.com/gogf/gf/container/gvar"
"github.com/gogf/gf/internal/structs"
@ -18,156 +20,111 @@ import (
// 初始化GET请求参数
func (r *Request) initGet() {
if !r.parsedGet {
r.queryVars = r.URL.Query()
if strings.EqualFold(r.Method, "GET") {
r.parsedGet = true
if r.URL.RawQuery != "" {
r.getMap, _ = gstr.Parse(r.URL.RawQuery)
} else if strings.EqualFold(r.Method, "GET") {
r.parsedRaw = true
if raw := r.GetRawString(); len(raw) > 0 {
var array []string
for _, item := range strings.Split(raw, "&") {
array = strings.Split(item, "=")
r.queryVars[array[0]] = append(r.queryVars[array[0]], array[1])
}
r.getMap, _ = gstr.Parse(raw)
}
}
r.parsedGet = true
}
}
// 设置GET参数，仅在ghttp.Server内有效，**注意并发安全性**
func (r *Request) SetQuery(key string, value string) {
// 设置当前请求的GET参数
func (r *Request) SetQuery(key string, value interface{}) {
r.initGet()
r.queryVars[key] = []string{value}
}
// 添加GET参数，构成[]string
func (r *Request) AddQuery(key string, value string) {
r.initGet()
r.queryVars[key] = append(r.queryVars[key], value)
r.getMap[key] = value
}
// 获得指定名称的get参数列表
func (r *Request) GetQuery(key string, def ...interface{}) []string {
func (r *Request) GetQuery(key string, def ...interface{}) interface{} {
r.initGet()
if v, ok := r.queryVars[key]; ok {
if v, ok := r.getMap[key]; ok {
return v
}
if len(def) > 0 {
return gconv.Strings(def[0])
return def[0]
}
return nil
}
func (r *Request) GetQueryVar(key string, def ...interface{}) *gvar.Var {
return gvar.New(r.GetQueryString(key, def...))
return gvar.New(r.GetQuery(key, def...))
}
func (r *Request) GetQueryString(key string, def ...interface{}) string {
value := r.GetQuery(key, def...)
if value != nil && value[0] != "" {
return value[0]
}
return ""
return r.GetQueryVar(key, def...).String()
}
func (r *Request) GetQueryBool(key string, def ...interface{}) bool {
value := r.GetQueryString(key, def...)
if value != "" {
return gconv.Bool(value)
}
return false
return r.GetQueryVar(key, def...).Bool()
}
func (r *Request) GetQueryInt(key string, def ...interface{}) int {
value := r.GetQueryString(key, def...)
if value != "" {
return gconv.Int(value)
}
return 0
return r.GetQueryVar(key, def...).Int()
}
func (r *Request) GetQueryInts(key string, def ...interface{}) []int {
value := r.GetQuery(key, def...)
if value != nil {
return gconv.Ints(value)
}
return nil
return r.GetQueryVar(key, def...).Ints()
}
func (r *Request) GetQueryUint(key string, def ...interface{}) uint {
value := r.GetQueryString(key, def...)
if value != "" {
return gconv.Uint(value)
}
return 0
return r.GetQueryVar(key, def...).Uint()
}
func (r *Request) GetQueryFloat32(key string, def ...interface{}) float32 {
value := r.GetQueryString(key, def...)
if value != "" {
return gconv.Float32(value)
}
return 0
return r.GetQueryVar(key, def...).Float32()
}
func (r *Request) GetQueryFloat64(key string, def ...interface{}) float64 {
value := r.GetQueryString(key, def...)
if value != "" {
return gconv.Float64(value)
}
return 0
return r.GetQueryVar(key, def...).Float64()
}
func (r *Request) GetQueryFloats(key string, def ...interface{}) []float64 {
value := r.GetQuery(key, def...)
if value != nil {
return gconv.Floats(value)
}
return nil
return r.GetQueryVar(key, def...).Floats()
}
func (r *Request) GetQueryArray(key string, def ...interface{}) []string {
return r.GetQuery(key, def...)
return r.GetQueryVar(key, def...).Strings()
}
func (r *Request) GetQueryStrings(key string, def ...interface{}) []string {
return r.GetQuery(key, def...)
return r.GetQueryVar(key, def...).Strings()
}
func (r *Request) GetQueryInterfaces(key string, def ...interface{}) []interface{} {
value := r.GetQuery(key, def...)
if value != nil {
return gconv.Interfaces(value)
}
return nil
return r.GetQueryVar(key, def...).Interfaces()
}
// 获取指定键名的关联数组，并且给定当指定键名不存在时的默认值
func (r *Request) GetQueryMap(def ...map[string]string) map[string]string {
// 获取指定键名的关联数组，并且给定当指定键名不存在时的默认值。
// 当不指定键值对关联数组时，默认获取GET方式提交的所有的提交键值对数据。
func (r *Request) GetQueryMap(kvMap ...map[string]interface{}) map[string]interface{} {
r.initGet()
m := make(map[string]string)
for k, v := range r.queryVars {
m[k] = v[0]
}
if len(def) > 0 {
for k, v := range def[0] {
if _, ok := m[k]; !ok {
m[k] = v
if len(kvMap) > 0 {
m := make(map[string]interface{})
for k, defValue := range kvMap[0] {
if queryValue, ok := r.getMap[k]; ok {
m[k] = queryValue
} else {
m[k] = defValue
}
}
return m
} else {
return r.getMap
}
return m
}
// 将所有的get参数映射到struct属性上，参数object应当为一个struct对象的指针, mapping为非必需参数，自定义参数与属性的映射关系
func (r *Request) GetQueryToStruct(pointer interface{}, mapping ...map[string]string) error {
tagmap := structs.TagMapName(pointer, paramTagPriority, true)
r.initGet()
tagMap := structs.TagMapName(pointer, paramTagPriority, true)
if len(mapping) > 0 {
for k, v := range mapping[0] {
tagmap[k] = v
tagMap[k] = v
}
}
params := make(map[string]interface{})
for k, v := range r.GetQueryMap() {
params[k] = v
}
return gconv.StructDeep(params, pointer, tagmap)
return gconv.StructDeep(r.getMap, pointer, tagMap)
}

125
net/ghttp/ghttp_request_request.go
View File

@ -9,126 +9,112 @@ package ghttp
import (
"github.com/gogf/gf/container/gvar"
"github.com/gogf/gf/internal/structs"
"github.com/gogf/gf/text/gstr"
"github.com/gogf/gf/util/gconv"
)
// 初始化RAW请求参数
func (r *Request) initRaw() {
if !r.parsedRaw {
r.parsedRaw = true
if raw := r.GetRawString(); len(raw) > 0 {
r.rawVarMap, _ = gstr.Parse(raw)
}
}
}
// 获得router、post或者get提交的参数，如果有同名参数，那么按照router->get->post优先级进行覆盖
func (r *Request) GetRequest(key string, def ...interface{}) []string {
v := r.GetRouterArray(key)
func (r *Request) GetRequest(key string, def ...interface{}) interface{} {
v := r.GetRouterValue(key)
if v == nil {
v = r.GetQuery(key)
}
if v == nil {
v = r.GetPost(key)
}
if v != nil {
return v
}
r.initRaw()
v = r.rawVarMap[key]
if v == nil && len(def) > 0 {
return gconv.Strings(def[0])
return def[0]
}
return v
}
func (r *Request) GetRequestVar(key string, def ...interface{}) *gvar.Var {
value := r.GetRequest(key, def...)
if value != nil {
return gvar.New(value[0])
}
return gvar.New(nil)
return gvar.New(r.GetRequest(key, def...))
}
func (r *Request) GetRequestString(key string, def ...interface{}) string {
value := r.GetRequest(key, def...)
if value != nil && value[0] != "" {
return value[0]
}
return ""
return r.GetRequestVar(key, def...).String()
}
func (r *Request) GetRequestBool(key string, def ...interface{}) bool {
value := r.GetRequestString(key, def...)
if value != "" {
return gconv.Bool(value)
}
return false
return r.GetRequestVar(key, def...).Bool()
}
func (r *Request) GetRequestInt(key string, def ...interface{}) int {
value := r.GetRequestString(key, def...)
if value != "" {
return gconv.Int(value)
}
return 0
return r.GetRequestVar(key, def...).Int()
}
func (r *Request) GetRequestInts(key string, def ...interface{}) []int {
value := r.GetRequest(key, def...)
if value != nil {
return gconv.Ints(value)
}
return nil
return r.GetRequestVar(key, def...).Ints()
}
func (r *Request) GetRequestUint(key string, def ...interface{}) uint {
value := r.GetRequestString(key, def...)
if value != "" {
return gconv.Uint(value)
}
return 0
return r.GetRequestVar(key, def...).Uint()
}
func (r *Request) GetRequestFloat32(key string, def ...interface{}) float32 {
value := r.GetRequestString(key, def...)
if value != "" {
return gconv.Float32(value)
}
return 0
return r.GetRequestVar(key, def...).Float32()
}
func (r *Request) GetRequestFloat64(key string, def ...interface{}) float64 {
value := r.GetRequestString(key, def...)
if value != "" {
return gconv.Float64(value)
}
return 0
return r.GetRequestVar(key, def...).Float64()
}
func (r *Request) GetRequestFloats(key string, def ...interface{}) []float64 {
value := r.GetRequest(key, def...)
if value != nil {
return gconv.Floats(value)
}
return nil
return r.GetRequestVar(key, def...).Floats()
}
func (r *Request) GetRequestArray(key string, def ...interface{}) []string {
return r.GetRequest(key, def...)
return r.GetRequestVar(key, def...).Strings()
}
func (r *Request) GetRequestStrings(key string, def ...interface{}) []string {
return r.GetRequest(key, def...)
return r.GetRequestVar(key, def...).Strings()
}
func (r *Request) GetRequestInterfaces(key string, def ...interface{}) []interface{} {
value := r.GetRequest(key, def...)
if value != nil {
return gconv.Interfaces(value)
}
return nil
return r.GetRequestVar(key, def...).Interfaces()
}
// 获取指定键名的关联数组，并且给定当指定键名不存在时的默认值
// 需要注意的是，如果其中一个字段为数组形式，那么只会返回第一个元素，如果需要获取全部的元素，请使用GetRequestArray获取特定字段内容
func (r *Request) GetRequestMap(def ...map[string]string) map[string]string {
m := r.GetQueryMap()
if len(m) == 0 {
m = r.GetPostMap()
}
if len(def) > 0 {
for k, v := range def[0] {
if _, ok := m[k]; !ok {
m[k] = v
func (r *Request) GetRequestMap(kvMap ...map[string]interface{}) map[string]interface{} {
r.initRaw()
m := r.rawVarMap
if len(kvMap) > 0 {
m = make(map[string]interface{})
for k, defValue := range kvMap[0] {
if rawValue, ok := r.rawVarMap[k]; ok {
m[k] = rawValue
} else {
m[k] = defValue
}
}
}
if m == nil {
m = make(map[string]interface{})
}
for k, v := range r.GetPostMap(kvMap...) {
m[k] = v
}
for k, v := range r.GetQueryMap(kvMap...) {
m[k] = v
}
return m
}
@ -140,14 +126,5 @@ func (r *Request) GetRequestToStruct(pointer interface{}, mapping ...map[string]
tagMap[k] = v
}
}
params := make(map[string]interface{})
for k, v := range r.GetRequestMap() {
params[k] = v
}
if len(params) == 0 {
if j := r.GetJson(); j != nil {
params = j.ToMap()
}
}
return gconv.StructDeep(params, pointer, tagMap)
return gconv.StructDeep(r.GetRequestMap(), pointer, tagMap)
}

28
net/ghttp/ghttp_request_router.go
View File

@ -6,26 +6,24 @@
package ghttp
func (r *Request) SetRouterString(key, value string) {
r.routerVars[key] = []string{value}
}
import "github.com/gogf/gf/container/gvar"
func (r *Request) AddRouterString(key, value string) {
r.routerVars[key] = append(r.routerVars[key], value)
func (r *Request) SetRouterValue(key string, value interface{}) {
r.routerMap[key] = value
}
// 获得路由解析参数
func (r *Request) GetRouterString(key string) string {
if v := r.GetRouterArray(key); v != nil {
return v[0]
func (r *Request) GetRouterValue(key string, def ...interface{}) interface{} {
if r.routerMap != nil {
return r.routerMap[key]
}
return ""
}
// 获得路由解析参数
func (r *Request) GetRouterArray(key string) []string {
if v, ok := r.routerVars[key]; ok {
return v
if len(def) > 0 {
return def[0]
}
return nil
}
// 获得路由解析参数, gvar.Var
func (r *Request) GetRouterVar(key string, def ...interface{}) *gvar.Var {
return gvar.New(r.GetRouterValue(key, def...))
}

2
net/ghttp/ghttp_response.go
View File

@ -99,7 +99,7 @@ func (r *Response) WriteJsonP(content interface{}) error {
return err
} else {
//r.Header().Set("Content-Type", "application/json")
if callback := r.request.Get("callback"); callback != "" {
if callback := r.request.GetString("callback"); callback != "" {
buffer := []byte(callback)
buffer = append(buffer, byte('('))
buffer = append(buffer, b...)

4
net/ghttp/ghttp_server.go
View File

@ -75,8 +75,8 @@ type (
// 根据特定URL.Path解析后的路由检索结果项
handlerParsedItem struct {
handler *handlerItem // 路由注册项
values map[string][]string // 特定URL.Path的Router解析参数
handler *handlerItem // 路由注册项
values map[string]string // 特定URL.Path的Router解析参数
}
// 控制器服务函数反射信息

2
net/ghttp/ghttp_server_pprof.go
View File

@ -20,7 +20,7 @@ type utilPprof struct{}
func (p *utilPprof) Index(r *Request) {
profiles := runpprof.Profiles()
action := r.Get("action")
action := r.GetString("action")
data := map[string]interface{}{
"uri": strings.TrimRight(r.URL.Path, "/") + "/",
"profiles": profiles,

10
net/ghttp/ghttp_server_router_hook.go
View File

@ -34,20 +34,20 @@ func (s *Server) callHookHandler(hook string, r *Request) {
hookItems := r.getHookHandlers(hook)
if len(hookItems) > 0 {
// 备份原有的router变量
oldRouterVars := r.routerVars
oldRouterVars := r.routerMap
for _, item := range hookItems {
// hook方法不能更改serve方法的路由参数，其匹配的路由参数只能自己使用，
// 且在多个hook方法之间不能共享路由参数，单可以使用匹配的serve方法路由参数。
// 当前回调函数的路由参数只在当前回调函数下有效。
r.routerVars = make(map[string][]string)
r.routerMap = make(map[string]interface{})
if len(oldRouterVars) > 0 {
for k, v := range oldRouterVars {
r.routerVars[k] = v
r.routerMap[k] = v
}
}
if len(item.values) > 0 {
for k, v := range item.values {
r.routerVars[k] = v
r.routerMap[k] = v
}
}
// 不使用hook的router对象，保留路由注册服务的router对象，不能覆盖
@ -66,7 +66,7 @@ func (s *Server) callHookHandler(hook string, r *Request) {
}
}
// 恢复原有的router变量
r.routerVars = oldRouterVars
r.routerMap = oldRouterVars
}
}

10
net/ghttp/ghttp_server_router_serve.go
View File

@ -113,14 +113,10 @@ func (s *Server) searchHandlers(method, path, domain string) (parsedItems []*han
// 如果需要query匹配，那么需要重新正则解析URL
if len(item.router.RegNames) > 0 {
if len(match) > len(item.router.RegNames) {
parsedItem.values = make(map[string][]string)
// 如果存在存在同名路由参数名称，那么执行数组追加
parsedItem.values = make(map[string]string)
// 如果存在存在同名路由参数名称，那么执行覆盖
for i, name := range item.router.RegNames {
if _, ok := parsedItem.values[name]; ok {
parsedItem.values[name] = append(parsedItem.values[name], match[i+1])
} else {
parsedItem.values[name] = []string{match[i+1]}
}
parsedItem.values[name] = match[i+1]
}
}
}

7
net/ghttp/ghttp_unit_cookie_test.go
View File

@ -20,14 +20,13 @@ func Test_Cookie(t *testing.T) {
p := ports.PopRand()
s := g.Server(p)
s.BindHandler("/set", func(r *ghttp.Request) {
r.Cookie.Set(r.Get("k"), r.Get("v"))
r.Cookie.Set(r.GetString("k"), r.GetString("v"))
})
s.BindHandler("/get", func(r *ghttp.Request) {
//fmt.Println(r.Cookie.Map())
r.Response.Write(r.Cookie.Get(r.Get("k")))
r.Response.Write(r.Cookie.Get(r.GetString("k")))
})
s.BindHandler("/remove", func(r *ghttp.Request) {
r.Cookie.Remove(r.Get("k"))
r.Cookie.Remove(r.GetString("k"))
})
s.SetPort(p)
s.SetDumpRouteMap(false)

47
net/ghttp/ghttp_unit_param_test.go
View File

@ -48,6 +48,32 @@ func Test_Params_Basic(t *testing.T) {
r.Response.Write(r.GetQueryString("string"))
}
})
s.BindHandler("/put", func(r *ghttp.Request) {
if r.Get("slice") != nil {
r.Response.Write(r.Get("slice"))
}
if r.Get("bool") != nil {
r.Response.Write(r.GetBool("bool"))
}
if r.Get("float32") != nil {
r.Response.Write(r.GetFloat32("float32"))
}
if r.Get("float64") != nil {
r.Response.Write(r.GetFloat64("float64"))
}
if r.Get("int") != nil {
r.Response.Write(r.GetInt("int"))
}
if r.Get("uint") != nil {
r.Response.Write(r.GetUint("uint"))
}
if r.Get("string") != nil {
r.Response.Write(r.GetString("string"))
}
if r.Get("map") != nil {
r.Response.Write(r.GetMap()["map"].(map[string]interface{})["b"])
}
})
s.BindHandler("/post", func(r *ghttp.Request) {
if r.GetPost("slice") != nil {
r.Response.Write(r.GetPost("slice"))
@ -70,6 +96,9 @@ func Test_Params_Basic(t *testing.T) {
if r.GetPost("string") != nil {
r.Response.Write(r.GetPostString("string"))
}
if r.GetPost("map") != nil {
r.Response.Write(r.GetPostMap()["map"].(map[string]interface{})["b"])
}
})
s.BindHandler("/map", func(r *ghttp.Request) {
if m := r.GetQueryMap(); len(m) > 0 {
@ -132,7 +161,7 @@ func Test_Params_Basic(t *testing.T) {
client := ghttp.NewClient()
client.SetPrefix(fmt.Sprintf("http://127.0.0.1:%d", p))
// GET
gtest.Assert(client.GetContent("/get", "slice=1&slice=2"), `["1","2"]`)
gtest.Assert(client.GetContent("/get", "slice=1&slice=2"), `2`)
gtest.Assert(client.GetContent("/get", "bool=1"), `true`)
gtest.Assert(client.GetContent("/get", "bool=0"), `false`)
gtest.Assert(client.GetContent("/get", "float32=0.11"), `0.11`)
@ -143,8 +172,21 @@ func Test_Params_Basic(t *testing.T) {
gtest.Assert(client.GetContent("/get", "uint=9"), `9`)
gtest.Assert(client.GetContent("/get", "string=key"), `key`)
// PUT
gtest.Assert(client.PutContent("/put", "slice=1&slice=2"), `2`)
gtest.Assert(client.PutContent("/put", "bool=1"), `true`)
gtest.Assert(client.PutContent("/put", "bool=0"), `false`)
gtest.Assert(client.PutContent("/put", "float32=0.11"), `0.11`)
gtest.Assert(client.PutContent("/put", "float64=0.22"), `0.22`)
gtest.Assert(client.PutContent("/put", "int=-10000"), `-10000`)
gtest.Assert(client.PutContent("/put", "int=10000"), `10000`)
gtest.Assert(client.PutContent("/put", "uint=10000"), `10000`)
gtest.Assert(client.PutContent("/put", "uint=9"), `9`)
gtest.Assert(client.PutContent("/put", "string=key"), `key`)
gtest.Assert(client.PutContent("/put", "map[a]=1&map[b]=2"), `2`)
// POST
gtest.Assert(client.PostContent("/post", "slice=1&slice=2"), `["1","2"]`)
gtest.Assert(client.PostContent("/post", "slice=1&slice=2"), `2`)
gtest.Assert(client.PostContent("/post", "bool=1"), `true`)
gtest.Assert(client.PostContent("/post", "bool=0"), `false`)
gtest.Assert(client.PostContent("/post", "float32=0.11"), `0.11`)
@ -154,6 +196,7 @@ func Test_Params_Basic(t *testing.T) {
gtest.Assert(client.PostContent("/post", "uint=10000"), `10000`)
gtest.Assert(client.PostContent("/post", "uint=9"), `9`)
gtest.Assert(client.PostContent("/post", "string=key"), `key`)
gtest.Assert(client.PostContent("/post", "map[a]=1&map[b]=2"), `2`)
// Map
gtest.Assert(client.GetContent("/map", "id=1&name=john"), `john`)

18
net/ghttp/ghttp_unit_session_test.go
View File

@ -20,13 +20,13 @@ func Test_Session_Cookie(t *testing.T) {
p := ports.PopRand()
s := g.Server(p)
s.BindHandler("/set", func(r *ghttp.Request) {
r.Session.Set(r.Get("k"), r.Get("v"))
r.Session.Set(r.GetString("k"), r.GetString("v"))
})
s.BindHandler("/get", func(r *ghttp.Request) {
r.Response.Write(r.Session.Get(r.Get("k")))
r.Response.Write(r.Session.Get(r.GetString("k")))
})
s.BindHandler("/remove", func(r *ghttp.Request) {
r.Session.Remove(r.Get("k"))
r.Session.Remove(r.GetString("k"))
})
s.BindHandler("/clear", func(r *ghttp.Request) {
r.Session.Clear()
@ -68,13 +68,13 @@ func Test_Session_Header(t *testing.T) {
p := ports.PopRand()
s := g.Server(p)
s.BindHandler("/set", func(r *ghttp.Request) {
r.Session.Set(r.Get("k"), r.Get("v"))
r.Session.Set(r.GetString("k"), r.GetString("v"))
})
s.BindHandler("/get", func(r *ghttp.Request) {
r.Response.Write(r.Session.Get(r.Get("k")))
r.Response.Write(r.Session.Get(r.GetString("k")))
})
s.BindHandler("/remove", func(r *ghttp.Request) {
r.Session.Remove(r.Get("k"))
r.Session.Remove(r.GetString("k"))
})
s.BindHandler("/clear", func(r *ghttp.Request) {
r.Session.Clear()
@ -121,8 +121,8 @@ func Test_Session_StorageFile(t *testing.T) {
p := ports.PopRand()
s := g.Server(p)
s.BindHandler("/set", func(r *ghttp.Request) {
r.Session.Set(r.Get("k"), r.Get("v"))
r.Response.Write(r.Get("k"), "=", r.Get("v"))
r.Session.Set(r.GetString("k"), r.GetString("v"))
r.Response.Write(r.GetString("k"), "=", r.GetString("v"))
})
s.SetPort(p)
s.SetDumpRouteMap(false)
@ -146,7 +146,7 @@ func Test_Session_StorageFile(t *testing.T) {
p := ports.PopRand()
s := g.Server(p)
s.BindHandler("/get", func(r *ghttp.Request) {
r.Response.Write(r.Session.Get(r.Get("k")))
r.Response.Write(r.Session.Get(r.GetString("k")))
})
s.SetPort(p)
s.SetDumpRouteMap(false)

8
text/gstr/gstr_z_unit_parse_test.go
View File

@ -18,8 +18,14 @@ import (
func Test_Parse(t *testing.T) {
gtest.Case(t, func() {
// name overwrite
m, err := gstr.Parse("a=1&a=2")
gtest.Assert(err, nil)
gtest.Assert(m, g.Map{
"a": 2,
})
// slice
m, err := gstr.Parse("a[]=1&a[]=2")
m, err = gstr.Parse("a[]=1&a[]=2")
gtest.Assert(err, nil)
gtest.Assert(m, g.Map{
"a": g.Slice{"1", "2"},