func TestToDBNameGenerateFriendlyName(t *testing.T) {
var maps = map[string]string{
"": "",
"ThisIsATest": "this_is_a_test",
"PFAndESI": "pf_and_esi",
"AbcAndJkl": "abc_and_jkl",
"EmployeeID": "employee_id",
"SKU_ID": "sku_id",
"HTTPAndSMTP": "http_and_smtp",
"HTTPServerHandlerForURLID": "http_server_handler_for_url_id",
"UUID": "uuid",
"HTTPURL": "http_url",
"HTTP_URL": "http_url",
"ThisIsActuallyATestSoWeMayBeAbleToUseThisCodeInGormPackageAlsoIdCanBeUsedAtTheEndAsID": "this_is_actually_a_test_so_we_may_be_able_to_use_this_code_in_gorm_package_also_id_can_be_used_at_the_end_as_id",
}
for key, value := range maps {
if gorm.ToDBName(key) != value {
t.Errorf("%v ToDBName should equal %v, but got %v", key, value, gorm.ToDBName(key))
}
}
}
// ToParamString replaces spaces and separates words (by uppercase letters) with
// underscores in a string, also downcase it
// e.g. ToParamString -> to_param_string, To ParamString -> to_param_string
func ToParamString(str string) string {
return gorm.ToDBName(strings.Replace(str, " ", "_", -1))
}
//Page Query will check
func PageQuery(c *gin.Context, v interface{}) error {
// TODO: However check v must be *[]
c.Request.ParseForm()
page, _ := strconv.Atoi(c.Request.Form.Get("page"))
c.Request.Form.Del("page")
if page < 1 {
page = 1
}
count, _ := strconv.Atoi(c.Request.Form.Get("count"))
c.Request.Form.Del("count")
if count < 1 {
count = 20
}
if count > 2000 {
//Return Error
return errors.New("InvaildCount")
}
s := (page - 1) * count
order := c.Request.Form.Get("order")
switch order {
case "asc":
case "desc":
default:
order = "asc"
}
t := reflect.ValueOf(v).Elem().Type().Elem()
tv := reflect.New(reflect.ValueOf(v).Elem().Type())
if t.Kind() == reflect.Ptr {
t = t.Elem()
tv = tv.Elem()
}
mdb := mariadb.Maria
cdb := mdb.Model(tv.Interface())
//Check Json Tag First
for i := 0; i < t.NumField(); i++ {
f := t.Field(i)
fk := f.Tag.Get("json")
if "" == fk {
fk = strings.ToLower(f.Name)
}
v, ok := c.Request.Form[fk]
if !ok {
continue
}
inVlaues := [](interface{}){}
inExprision := " "
for _, f := range v {
inVlaues = append(inVlaues, f)
inExprision += "?,"
}
inExprision = inExprision[0 : len(inExprision)-1]
k := gorm.ToDBName(f.Name)
if 1 == len(inVlaues) {
//for int/bool
inFormat := fmt.Sprintf("`%v` = ?", k)
switch f.Type.Kind() {
case reflect.String:
inFormat := fmt.Sprintf("`%v` like ?", k)
mdb = mdb.Where(inFormat, mysqlEscape(fmt.Sprintf("%%%v%%", inVlaues[0])))
cdb = cdb.Where(inFormat, mysqlEscape(fmt.Sprintf("%%%v%%", inVlaues[0])))
case reflect.Bool:
vb := true
if "false" == fmt.Sprint(inVlaues[0]) {
vb = false
}
inFormat := fmt.Sprintf("`%v` = ?", k)
mdb = mdb.Where(inFormat, vb)
cdb = cdb.Where(inFormat, vb)
case reflect.Int:
id, err := strconv.Atoi(fmt.Sprint(inVlaues[0]))
if nil != err {
return err
}
fmt.Println(inVlaues[0], id)
mdb = mdb.Where(inFormat, id)
cdb = cdb.Where(inFormat, id)
default:
inFormat := fmt.Sprintf("`%v` = ?", k)
mdb = mdb.Where(inFormat, fmt.Sprintf("%v", inVlaues[0]))
cdb = cdb.Where(inFormat, fmt.Sprintf("%v", inVlaues[0]))
}
} else {
inFormat := fmt.Sprintf("`%v` in ( %v )", k, inExprision)
mdb = mdb.Where(inFormat, inVlaues...)
cdb = cdb.Where(inFormat, inVlaues...)
}
}
total := 0
cdb.Count(&total)
if err := mdb.Limit(count).Offset(s).Order("id " + order).Find(v).Error; nil != err {
return err
}
last := total / count
if total%count > 0 {
last += 1
}
// Set Pageable Header
c.Writer.Header().Set("X-Total-Resource", fmt.Sprint(total))
c.Writer.Header().Set("X-Total-Page", fmt.Sprint(last))
c.Writer.Header().Set("X-Current-Page", fmt.Sprint(page))
// Set Stand Link for pageable
query := c.Request.URL.Query()
query.Set("count", fmt.Sprint(count))
query.Del("page")
url := config.Read("Schema") + "://" + c.Request.Host + c.Request.URL.Path
query.Set("page", fmt.Sprint(last))
rawQuery := fmt.Sprintf(`<%s?%s>; rel="%s"`, url, query.Encode(), "last")
c.Writer.Header().Set("Link", rawQuery)
query.Set("page", fmt.Sprint(1))
rawQuery = fmt.Sprintf(`<%s?%s>; rel="%s"`, url, query.Encode(), "first")
c.Writer.Header().Add("Link", rawQuery)
if page < last {
query.Set("page", fmt.Sprint(page+1))
rawQuery = fmt.Sprintf(`<%s?%s>; rel="%s"`, url, query.Encode(), "next")
c.Writer.Header().Add("Link", rawQuery)
}
if page > 1 {
query.Set("page", fmt.Sprint(page-1))
rawQuery = fmt.Sprintf(`<%s?%s>; rel="%s"`, url, query.Encode(), "prev")
c.Writer.Header().Add("Link", rawQuery)
}
if 0 == reflect.ValueOf(v).Elem().Len() {
reflect.ValueOf(v).Elem().Set(reflect.MakeSlice(reflect.SliceOf(t), 0, 0))
}
return nil
}