golang 项目 gorm 列表条件过滤查询和分页

最新推荐文章于 2024-09-12 08:32:51 发布

chu_yubo

最新推荐文章于 2024-09-12 08:32:51 发布

阅读量1.8k

点赞数

分类专栏： Go orm 文章标签： golang 开发语言

本文链接：https://blog.csdn.net/chu_yubo/article/details/129228219

版权

Go 同时被 2 个专栏收录

5 篇文章 0 订阅

订阅专栏

orm

2 篇文章 0 订阅

订阅专栏

文章介绍了如何使用GORMORM在Go语言中编写通用的分页查询和条件过滤方法，包括基于`scopes`的分页函数`Paginate`、排序函数`Order`、字符串过滤`StringFilter`以及数组过滤`ArrayFilter`，允许进行`LIKE`、`IN`、`NOTIN`等操作，方便在项目中对数据库查询进行抽象和复用。

摘要由CSDN通过智能技术生成

在我们编写的项目的时候，列表总是有各种条件进行查询，不过一般都是and 查询，这里编写了一个稍微通用些的方法利用scopes，返回列表查询数据；

创建orm文件：这里目录结构我这边设计是存在名叫ormModels 文件夹中，新建base.go文件

package ormModels

import (
	"github.com/spf13/cast"
	"gorm.io/gorm"

	"reflect"
)
// 这里是分页也相关 传入参数 page和 size, 以及对应 默认分页配置 s eg:map[string]interface{}{"page" : 1, "size" : 20}, 如果传入为 map[string]interface{}{},默认设定 page 1, size 20
func Paginate(pageParam interface{}, sizeParam interface{}, s map[string]interface{}) func(db *gorm.DB) *gorm.DB {
	return func(db *gorm.DB) *gorm.DB {
		page := cast.ToInt(pageParam)
		size := cast.ToInt(sizeParam)

		if page == 0 {
			if s["page"] != nil {
				page = s["page"].(int)
			} else {
				page = 1
			}
		}

		if size == 0 {
			if s["size"] != nil {
				size = s["size"].(int)
			} else {
				size = 20
			}
		}
		offset := (page - 1) * size
		return db.Offset(offset).Limit(size)
	}
}

func Order(sort string) func(db *gorm.DB) *gorm.DB {
	return func(db *gorm.DB) *gorm.DB {
		returnDb := db
		if len(sort) == 0 {
			sort = "id desc"
		}
		return returnDb.Order(sort)
	}
}

//进行快速条件过滤
func StringFilter(key string, value string, operator string) func(db *gorm.DB) *gorm.DB {
	return func(db *gorm.DB) *gorm.DB {
		returnDb := db
		if value != "" {
			switch operator {
			case "like":
				returnDb = returnDb.Where(key+" Like ? ", "%"+value+"%")
			case "=", ">=", ">", "<=", "<":
				returnDb = returnDb.Where(key+" "+operator+" "+"?", value)
			}

		}
		return returnDb
	}
}
// 进行wehre in 或者 where not in
func ArrayFilter(key string, value interface{}, operator string) func(db *gorm.DB) *gorm.DB {
	return func(db *gorm.DB) *gorm.DB {
		returnDb := db
		if reflect.ValueOf(value).Len() != 0 {
			whereMap := map[string]interface{}{
				key: value,
			}
			switch operator {
			case "in":
				returnDb = returnDb.Where(whereMap)
			case "not in":
				returnDb = returnDb.Not(whereMap)
			}
		}
		return returnDb
	}
}

之后创建的所有模型都可以进行使用分页和 where 查询使用方法：

builder.Scopes(ormModels.Paginate(params.Page, params.Size, map[string]interface{}{})).
		Find(&list)

	builder := ormModels.GetDb().Model(&ormModels.Test{}).
		Scopes(ormModels.StringFilter("state", params.State, "=")).
		Scopes(ormModels.StringFilter("tax_no", params.TaxNo, "like"))

	builder := ormModels.GetDb().Model(&ormModels.Test{}).
		Scopes(ormModels.ArrayFilter("serial_no", params.SerialNos, "in")).

golang 项目 gorm 列表 条件过滤查询 和 分页

golang 项目 gorm 列表条件过滤查询和分页