优化

internal/data/media.go

@@ -73,12 +73,6 @@ func (r *mediaRepo) FindMediaByHash(ctx context.Context, userID uint, hash strin
 	return mediaFromPO(po), nil
 }
 func (r *mediaRepo) ListMedia(ctx context.Context, filter biz.MediaFilter) ([]*biz.MediaFile, int64, error) {
-	if filter.Page < 1 {
-		filter.Page = 1
-	}
-	if filter.PageSize < 1 {
-		filter.PageSize = 10
-	}
 	db := r.data.gormDB.WithContext(ctx).Model(&mediaPO{})
 	if filter.Keyword != "" {
 		db = db.Where("name LIKE ?", "%"+filter.Keyword+"%")
@@ -110,7 +104,19 @@ func (r *mediaRepo) ListMedia(ctx context.Context, filter biz.MediaFilter) ([]*b
 	if filter.Desc {
 		orderKey += " desc"
 	}
-	if err := applyPagination(db.Order(orderKey), filter.Page, filter.PageSize, 100).Find(&pos).Error; err != nil {
+	limit := filter.PageSize
+	if limit > 100 {
+		limit = 100
+	}
+	offset := 0
+	if limit > 0 {
+		page := filter.Page
+		if page <= 0 {
+			page = 1
+		}
+		offset = limit * (page - 1)
+	}
+	if err := db.Limit(limit).Offset(offset).Order(orderKey).Find(&pos).Error; err != nil {
 		return nil, 0, err
 	}
 	out := make([]*biz.MediaFile, 0, len(pos))
@@ -141,9 +147,6 @@ func (r *mediaRepo) CreateMediaBatch(ctx context.Context, items []*biz.MediaFile
 	for _, v := range items {
 		pos = append(pos, mediaToPO(v))
 	}
-	if len(pos) == 0 {
-		return nil
-	}
 	return r.data.gormDB.WithContext(ctx).Create(&pos).Error
 }
 func (r *mediaRepo) SaveCategory(ctx context.Context, v *biz.AttachmentCategory) error {
@@ -178,7 +181,7 @@ func (r *mediaRepo) DeleteCategory(ctx context.Context, id uint) error {
 }
 func (r *mediaRepo) ListCategories(ctx context.Context) ([]*biz.AttachmentCategory, error) {
 	var pos []categoryPO
-	if err := r.data.gormDB.WithContext(ctx).Order("id").Find(&pos).Error; err != nil {
+	if err := r.data.gormDB.WithContext(ctx).Find(&pos).Error; err != nil {
 		return nil, err
 	}
 	byID := map[uint]*biz.AttachmentCategory{}