如何優(yōu)雅的使用 GORM 進(jìn)行分頁？
GORM 是 Go 中使用最廣泛的 ORM 包，但盡管如此，它缺少一些“基本”功能。其中一個缺失的功能就是分頁（Pagination）。
分頁是管理應(yīng)用程序中大型數(shù)據(jù)集的一個重要功能。它是一種限制和顯示數(shù)據(jù)庫中部分總數(shù)據(jù)的方法，這樣就不需要一次性檢索整個表，這樣可以極大的提高接口性能，降低超時失敗的概率。
雖然 GORM 提供了關(guān)于如何使用scopes 進(jìn)行分頁的文檔，但在靈活性和可用性方面仍有改進(jìn)的空間。在這里，我將介紹一種使用 GORM 的 Clauses 進(jìn)行分頁的替代方法。
使用 Scopes 的分頁
GORM 的官網(wǎng)文檔介紹了使用 Scopes 進(jìn)行分頁的代碼示例：
func Paginate(page, pageSize int) func(db *gorm.DB) *gorm.DB {
return func (db *gorm.DB) *gorm.DB {
// validate page and pageSize
...
offset := (page - 1) * pageSize
return db.Offset(offset).Limit(pageSize)
}
}
page := 0
pageSize := 10
db.Scopes(Paginate(page, pageSize)).Find(&users)因此，為了使用分頁功能，我們需要使用以下代碼 db.Scopes(Paginate(page, pageSize))…
使用 Clauses 的分頁
一種不同且可以說更優(yōu)雅的方法是使用 GORM Clauses，這種方法與使用 scope 略有不同，因為 Clauses 負(fù)責(zé)修改數(shù)據(jù)庫查詢，特別是 WHERE 子句。
讓我們先定義分頁結(jié)構(gòu)體：
type Pagination struct {
page int
pageSize int
}
func (p *Pagination) GetPage() int {
return p.page
}
func (p *Pagination) GetPageSize() int {
return p.pageSize
}然后，讓我們實現(xiàn)兩個接口，以便在 Gorm Clauses 函數(shù)中使用這個結(jié)構(gòu)體：
clause.Expression
gorm.StatementModifier所以結(jié)構(gòu)看起來是這樣的：
func (p *Pagination) ModifyStatement(stm *gorm.Statement) {
// We modify statement to add the pagination
db := stm.DB
stm.DB.Limit(p.size).Offset((p.page - 1) * p.pageSize)
}
func (p *Pagination) Build(_ clause.Builder) {
// The Build method is left empty since pagination does not require any additional SQL clauses.
}之后，可以按照以下方式使用分頁：
pagination := Pagination{
page: 0,
pageSize: 10,
}
db.Clauses(&pagination).Find(&users)為了使這種方法可重用并增強(qiáng)其功能，這里介紹一個小眾的包 PaGorminator[4]：一個利用此功能進(jìn)行分頁的庫，同時添加了其他功能，如無分頁請求和自動填充元數(shù)據(jù)，如總頁數(shù)和總計數(shù)。
使用方法如下：
// Add plugin
_ = db.Use(pagorminator.PaGormMinator{})
...
pageRequest, _ := pagorminator.PageRequest(0, 10)
db.Clauses(pageRequest).Find(&users)
// this will apply pagination, and also populate pageRequest with:
// - the total number of pages
// - total count
