From 491fd33eabdc55efdb4309b9d07c2ab071fa213f Mon Sep 17 00:00:00 2001 From: wangpengfei <274878379@qq.com> Date: 星期三, 02 八月 2023 20:14:58 +0800 Subject: [PATCH] fix --- model/contact.go | 14 ++++++++++++-- 1 files changed, 12 insertions(+), 2 deletions(-) diff --git a/model/contact.go b/model/contact.go index 96a355b..abadcea 100644 --- a/model/contact.go +++ b/model/contact.go @@ -112,14 +112,20 @@ return record, err } -func (slf *ContactSearch) FindAll() ([]*ContactDetail, error) { +func (slf *ContactSearch) FindAll() ([]*ContactDetail, int64, error) { var db = slf.build() var records = make([]*ContactDetail, 0) + var total int64 + if err := db.Count(&total).Error; err != nil { + return records, total, err + } if slf.PageNum > 0 && slf.PageSize > 0 { db = db.Limit(slf.PageSize).Offset((slf.PageNum - 1) * slf.PageSize) } + + err := db.Preload("FollowRecord").Preload("Client").Preload("Country").Preload("Province").Preload("City").Preload("Region").Find(&records).Error - return records, err + return records, total, err } func (slf *ContactSearch) SetId(id int) *ContactSearch { @@ -163,3 +169,7 @@ slf.OrderBy = order return slf } +func (slf *ContactSearch) SetIds(ids []int) *ContactSearch { + slf.Orm = slf.Orm.Where("id in (?)", ids) + return slf +} -- Gitblit v1.8.0