如何通过查询参数动态添加 gorm 范围

学习Golang要努力，但是不要急！今天的这篇文章《如何通过查询参数动态添加 gorm 范围》将会介绍到等等知识点，如果你想深入学习Golang，可以关注我！我会持续更新相关文章的，希望对大家都能有所帮助！

我的挑战：根据给定的网址参数，我需要动态添加特定范围。实施它的最佳方法是什么？

package handler

import(
  "net/http"
  "gorm.io/gorm"
)

func scopea(age int) func(db *gorm.db) *gorm.db {
  return func(db *gorm.db) *gorm.db {
    return db.where("age > ?", age)
  }
}

func scopeb(cc string) func(db *gorm.db) *gorm.db {
  return func(db *gorm.db) *gorm.db {
    return db.where("country_code = ?", cc)
  }
}

func scopedefault(db *gorm.db) *gorm.db {
  return db.where("active = ?", true)
}

func myhandler(w http.responsewriter, r *http.request) {
  values := r.url.query()

  user := []model.user{}

  // howto add additional scope only if specific url parameter is given?

  var age int
  var cc string

  if values.has("country_code") {
    cc = values.get("country_code")
  }

  if values.has("age") {
    age = values.get("age")
  }

  if err := db.scopes(
    scopedefault,
    // scopea(age) only if parameter given,
    // scopeb(cc) only if parameter given,
  ).find(&user).error; err != nil {
    responderror(w, http.statusinternalservererror, err.error())
    return
  }

  respondjson(w, http.statusok, user)
}

我尝试看看 gorm scopes 的实现：

// impelementation of gorm scopes member-function
// https://github.com/go-gorm/gorm/blob/v1.22.5/chainable_api.go#l260
func (db *db) scopes(funcs ...func(*db) *db) (tx *db) {
  tx = db.getinstance()
  tx.statement.scopes = append(tx.statement.scopes, funcs...)
  return tx
}

但是因为 tx.statement.scopes 未导出，所以似乎不可能简单地附加我的 scopea 或 scopeb

// won't work
tx.Statement.scopes = append(tx.Statement.scopes, scopeA(age))

有人能把我推向正确的方向吗？预先感谢！

正确答案

像这样怎么样？

package main

import (
    "fmt"
    "net/http"
    "reflect"

    "github.com/gorilla/schema"
    "gorm.io/gorm"
)

type paramobj struct {
    age         *int    `json:"age" schema:"age" qrstr:"age > ?"`
    countrycode *string `json:"country_code" schema:"country_code" qrstr:"country_code = ?"`
}

func main() {

    http.handlefunc("/", func(w http.responsewriter, r *http.request) {
        var db *gorm.db // example open connection

        paramobj := new(paramobj)
        err := schema.newdecoder().decode(paramobj, r.url.query())
        if err != nil {
            panic(err)
        }

        fmt.printf("paramobj: %+v\n", paramobj)

        var temp interface{}
        db.scopes(customstructfilter(paramobj)...).find(&temp)

        w.writeheader(http.statusok)
    })

    http.listenandserve(":8080", nil)
}

func customstructfilter(v interface{}) []func(*gorm.db) *gorm.db {
    var arr = make([]func(*gorm.db) *gorm.db, 0)
    vl := reflect.valueof(v).elem()
    typ := reflect.typeof(v).elem()
    for idx := 0; idx < vl.numfield(); idx++ {
        if !vl.field(idx).isnil() {
            arr = append(arr, func(d *gorm.db) *gorm.db {
                return d.where(typ.field(idx).tag.get("qrstr"), vl.field(idx).interface())
            })
        }
    }
    return arr
}

通过使用 gorilla/schema（需要 schema 标签）将其解码为对象来接受查询参数，并为 customfilter 添加 qrstr 标签。

好吧，如果您在scopea()和scopeb()中添加验证检查，那就足够了。像这样的事情

// scopeA(r.URL.Query().Get("age"))
func scopeA(age string) func(db *gorm.DB) *gorm.DB {
    return func(db *gorm.DB) *gorm.DB {
        if age == "" {
            return db
        }
        if ageInt, err := strconv.Atoi(age); err == nil {
            return db.Where("age > ?", ageInt)
        } else {
            return db
        }
    }
}

今天关于《如何通过查询参数动态添加 gorm 范围》的内容就介绍到这里了，是不是学起来一目了然！想要了解更多关于的内容请关注golang学习网公众号！