From 53ed7728f5310a2f49ae0a362412378d18cc2c62 Mon Sep 17 00:00:00 2001
From: withchao <993506633@qq.com>
Date: Thu, 23 Mar 2023 14:54:40 +0800
Subject: [PATCH] ctx

---
 pkg/common/db/ormutil/utils.go | 11 ++++++-----
 1 file changed, 6 insertions(+), 5 deletions(-)

diff --git a/pkg/common/db/ormutil/utils.go b/pkg/common/db/ormutil/utils.go
index c64938f6c..4ef2a37ea 100644
--- a/pkg/common/db/ormutil/utils.go
+++ b/pkg/common/db/ormutil/utils.go
@@ -1,18 +1,19 @@
 package ormutil
 
 import (
-	"github.com/OpenIMSDK/Open-IM-Server/pkg/utils"
+	"github.com/OpenIMSDK/Open-IM-Server/pkg/errs"
 	"gorm.io/gorm"
 )
 
 func GormPage[E any](db *gorm.DB, pageNumber, showNumber int32) (uint32, []*E, error) {
 	var count int64
-	if err := db.Count(&count).Error; err != nil {
-		return 0, nil, utils.Wrap(err, "")
+	var model E
+	if err := db.Model(&model).Count(&count).Error; err != nil {
+		return 0, nil, errs.Wrap(err)
 	}
 	var es []*E
 	if err := db.Limit(int(showNumber)).Offset(int(pageNumber * showNumber)).Find(&es).Error; err != nil {
-		return 0, nil, utils.Wrap(err, "")
+		return 0, nil, errs.Wrap(err)
 	}
 	return uint32(count), es, nil
 }
@@ -46,7 +47,7 @@ func MapCount(db *gorm.DB, field string) (map[string]uint32, error) {
 		Count uint32 `gorm:"column:count"`
 	}
 	if err := db.Select(field + " as id, count(1) as count").Group(field).Find(&items).Error; err != nil {
-		return nil, err
+		return nil, errs.Wrap(err)
 	}
 	m := make(map[string]uint32)
 	for _, item := range items {