gorm query with struct
package main
import (
"fmt"
"github.com/alingse/structquery"
"gorm.io/gorm"
)
// DB ----------------------------------------------------------
type UserModel struct {
gorm.Model
Name string `json:"name"`
Email string `json:"email"`
}
func (u UserModel) TableName() string {
return "users"
}
var DB = initGormDB()
var queryer = structquery.NewQueryer()
type UserModelQuery struct {
Name string `sq:"like"`
Email string `sq:"eq"`
}
func main() {
var q = UserModelQuery{
Name: "hello",
Email: "a@b",
}
db := DB.Model(&User{})
sql := db.ToSQL(func(tx *gorm.DB) *gorm.DB {
var users []User
tx, _ = queryer.Where(tx, q)
return tx.Find(&users)
})
// SELECT * FROM `users` WHERE (`name` LIKE "%hello%" AND `email` = "a@b")
fmt.Println(sql)
}
package tests_test
import (
"encoding/json"
"net/http"
"github.com/alingse/structquery"
"github.com/gorilla/schema"
"gorm.io/gorm"
)
// DB -------------------------------------------------------------------------
var GormDB *gorm.DB
type UserModel struct {
gorm.Model
Name string `json:"name"`
Email string `json:"email"`
}
func (u UserModel) TableName() string {
return "users"
}
// HTTP -----------------------------------------------------------------------
var queryDecoder = schema.NewDecoder()
func init() {
queryDecoder.IgnoreUnknownKeys(true)
queryDecoder.SetAliasTag("json")
}
// Query ----------------------------------------------------------------------
type UserQuery struct {
Name string `json:"name" sq:"like"`
Email string `json:"email" sq:"eq"`
}
var structQueryer = structquery.NewQueryer()
func GetUsers(w http.ResponseWriter, r *http.Request) {
var query UserQuery
// decode from URL
err := queryDecoder.Decode(&query, r.URL.Query())
if err != nil {
w.WriteHeader(http.StatusBadRequest)
return
}
db := GormDB.Model(&UserModel{})
// bind queryer
db, err = structQueryer.Where(db, query)
if err != nil {
w.WriteHeader(http.StatusBadRequest)
return
}
var users []UserModel
db.Find(&users)
w.WriteHeader(http.StatusOK)
_ = json.NewEncoder(w).Encode(users)
}
- support and/or with slice?
- add structQueryer.Parse return with interface
- more db test