tank/code/tool/util/util_params.go

package util

import (
	"fmt"
	"net/http"
	"strconv"
	"strings"
	"time"
)

// param is required. when missing, panic error.
func ExtractRequestString(request *http.Request, key string) string {
	str := request.FormValue(key)
	if str == "" {
		panic(fmt.Sprintf("%s is required", key))
	} else {
		return str
	}
}

// param is required. when missing, panic error.
func ExtractRequestStringWithErrorHint(request *http.Request, key string, errorHint string) string {
	str := request.FormValue(key)
	if str == "" {
		panic(errorHint)
	} else {
		return str
	}
}

// split by comma , param is required. when missing, panic error.
func ExtractRequestArray(request *http.Request, key string) []string {
	str := request.FormValue(key)
	if str == "" {
		panic(fmt.Sprintf("%s is required", key))
	} else {
		arr := strings.Split(str, ",")
		if len(arr) == 0 {
			panic(fmt.Sprintf("%s cannot be empty", key))
		}
		return arr
	}
}

// param is required. when missing, panic error.
func ExtractRequestInt64(request *http.Request, key string) int64 {
	keyStr := request.FormValue(key)

	var num int64 = 0
	if keyStr == "" {
		panic(fmt.Sprintf("%s is required", key))
	} else {
		intVal, err := strconv.Atoi(keyStr)
		if err != nil {
			panic(err)
		}
		num = int64(intVal)
		return num
	}
}

// param is required. when missing, panic error.
func ExtractRequestInt64WithErrorHint(request *http.Request, key string, errorHint string) int64 {
	keyStr := request.FormValue(key)

	var num int64 = 0
	if keyStr == "" {
		panic(errorHint)
	} else {
		intVal, err := strconv.Atoi(keyStr)
		if err != nil {
			panic(err)
		}
		num = int64(intVal)
		return num
	}
}

// param is required. when missing, panic error.
func ExtractRequestOptionalInt(request *http.Request, key string, defaultValue int) int {
	str := request.FormValue(key)
	if str == "" {
		return defaultValue
	} else {
		intVal, err := strconv.Atoi(str)
		if err != nil {
			panic(err)
		}
		return intVal
	}
}

// param is required. when missing, panic error.
func ExtractRequestOptionalString(request *http.Request, key string, defaultValue string) string {
	str := request.FormValue(key)
	if str == "" {
		return defaultValue
	} else {
		return str
	}
}

// param is required. when missing, panic error.
func ExtractRequestBool(request *http.Request, key string) bool {
	str := request.FormValue(key)
	if str == "true" {
		return true
	} else {
		return false
	}
}

// param is required. when missing, panic error.
func ExtractRequestTime(request *http.Request, key string) time.Time {
	str := request.FormValue(key)
	return ConvertDateTimeStringToTime(str)
}

// param is optional. when missing, use default.
func ExtractRequestOptionalBool(request *http.Request, key string, defaultValue bool) bool {
	str := request.FormValue(key)
	if str == "" {
		return defaultValue
	} else {
		return str == "true"
	}
}