gf/g/net/ghttp/ghttp_request_post.go

// Copyright 2017 gf Author(https://gitee.com/johng/gf). All Rights Reserved.
//
// This Source Code Form is subject to the terms of the MIT License.
// If a copy of the MIT was not distributed with this file,
// You can obtain one at https://gitee.com/johng/gf.

package ghttp

import (
    "gitee.com/johng/gf/g/util/gconv"
)

// 初始化POST请求参数
func (r *Request) initPost() {
    if !r.parsedPost.Val() {
        // 快速保存，尽量避免并发问题
        r.parsedPost.Set(true)
        // MultiMedia表单请求解析允许最大使用内存：1GB
        r.ParseMultipartForm(1024*1024*1024)
    }
}

// 设置POST参数，仅在ghttp.Server内有效，**注意并发安全性**
func (r *Request) SetPost(key string, value string) {
    r.initPost()
    r.PostForm[key] = []string{value}
}

func (r *Request) AddPost(key string, value string) {
    r.initPost()
    r.PostForm[key] = append(r.PostForm[key], value)
}

// 获得post参数
func (r *Request) GetPost(key string, def...[]string) []string {
    r.initPost()
    if v, ok := r.PostForm[key]; ok {
        return v
    }
    if len(def) > 0 {
        return def[0]
    }
    return nil
}

func (r *Request) GetPostString(key string, def ... string) string {
    value := r.GetPost(key)
    if value != nil && value[0] != "" {
        return value[0]
    }
    if len(def) > 0 {
        return def[0]
    }
    return ""
}

func (r *Request) GetPostBool(key string, def ... bool) bool {
    value := r.GetPostString(key)
    if value != "" {
        return gconv.Bool(value)
    }
    if len(def) > 0 {
        return def[0]
    }
    return false
}

func (r *Request) GetPostInt(key string, def ... int) int {
    value := r.GetPostString(key)
    if value != "" {
        return gconv.Int(value)
    }
    if len(def) > 0 {
        return def[0]
    }
    return 0
}

func (r *Request) GetPostInts(key string, def ... []int) []int {
    value := r.GetPost(key)
    if value != nil {
        return gconv.Ints(value)
    }
    if len(def) > 0 {
        return def[0]
    }
    return nil
}

func (r *Request) GetPostUint(key string, def ... uint) uint {
    value := r.GetPostString(key)
    if value != "" {
        return gconv.Uint(value)
    }
    if len(def) > 0 {
        return def[0]
    }
    return 0
}

func (r *Request) GetPostFloat32(key string, def ... float32) float32 {
    value := r.GetPostString(key)
    if value != "" {
        return gconv.Float32(value)
    }
    if len(def) > 0 {
        return def[0]
    }
    return 0
}

func (r *Request) GetPostFloat64(key string, def ... float64) float64 {
    value := r.GetPostString(key)
    if value != "" {
        return gconv.Float64(value)
    }
    if len(def) > 0 {
        return def[0]
    }
    return 0
}

func (r *Request) GetPostFloats(key string, def ... []float64) []float64 {
    value := r.GetPost(key)
    if value != nil {
        return gconv.Floats(value)
    }
    if len(def) > 0 {
        return def[0]
    }
    return nil
}

func (r *Request) GetPostArray(key string, def ... []string) []string {
    return r.GetPost(key, def...)
}

func (r *Request) GetPostStrings(key string, def ... []string) []string {
    return r.GetPost(key, def...)
}

func (r *Request) GetPostInterfaces(key string, def ... []interface{}) []interface{} {
    value := r.GetPost(key)
    if value != nil {
        return gconv.Interfaces(value)
    }
    if len(def) > 0 {
        return def[0]
    }
    return nil
}

// 获取指定键名的关联数组，并且给定当指定键名不存在时的默认值
// 需要注意的是，如果其中一个字段为数组形式，那么只会返回第一个元素，如果需要获取全部的元素，请使用GetPostArray获取特定字段内容
func (r *Request) GetPostMap(def...map[string]string) map[string]string {
    r.initPost()
    m := make(map[string]string)
    if len(def) == 0 {
        for k, v := range r.PostForm {
            m[k] = v[0]
        }
    } else {
        for k, v := range def[0] {
            if v2, ok := r.PostForm[k]; ok {
                m[k] = v2[0]
            } else {
                m[k] = v
            }
        }
    }
    return m
}

// 将所有的request参数映射到struct属性上，参数object应当为一个struct对象的指针, mapping为非必需参数，自定义参数与属性的映射关系
func (r *Request) GetPostToStruct(object interface{}, mapping...map[string]string) {
    tagmap := r.getStructParamsTagMap(object)
    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
    }
    gconv.Struct(params, object, tagmap)
}