优化结构

internal/data/repository/media.go

@@ -3,10 +3,9 @@ package system
 import (
 	"context"
 	"errors"
+	"kra/internal/biz/system"
 	"time"
 
-	"kra/internal/biz"
-
 	"gorm.io/gorm"
 )
 
@@ -41,14 +40,14 @@ func (categoryPO) TableName() string { return "media_attachment_category" }
 
 type mediaRepo struct{ data Provider }
 
-func NewMediaRepo(data Provider) biz.MediaRepo { return &mediaRepo{data: data} }
-func mediaFromPO(v mediaPO) *biz.MediaFile {
-	return &biz.MediaFile{ID: v.ID, CreatedAt: v.CreatedAt, UpdatedAt: v.UpdatedAt, Name: v.Name, CategoryID: v.CategoryID, URL: v.URL, Tag: v.Tag, Key: v.Key, Size: v.Size, Mime: v.Mime, MD5: v.MD5, UserID: v.UserID}
+func NewMediaRepo(data Provider) system.MediaRepo { return &mediaRepo{data: data} }
+func mediaFromPO(v mediaPO) *system.MediaFile {
+	return &system.MediaFile{ID: v.ID, CreatedAt: v.CreatedAt, UpdatedAt: v.UpdatedAt, Name: v.Name, CategoryID: v.CategoryID, URL: v.URL, Tag: v.Tag, Key: v.Key, Size: v.Size, Mime: v.Mime, MD5: v.MD5, UserID: v.UserID}
 }
-func mediaToPO(v *biz.MediaFile) mediaPO {
+func mediaToPO(v *system.MediaFile) mediaPO {
 	return mediaPO{ID: v.ID, Name: v.Name, CategoryID: v.CategoryID, URL: v.URL, Tag: v.Tag, Key: v.Key, Size: v.Size, Mime: v.Mime, MD5: v.MD5, UserID: v.UserID}
 }
-func (r *mediaRepo) CreateMedia(ctx context.Context, v *biz.MediaFile) error {
+func (r *mediaRepo) CreateMedia(ctx context.Context, v *system.MediaFile) error {
 	po := mediaToPO(v)
 	if err := r.data.DB().WithContext(ctx).Create(&po).Error; err != nil {
 		return err
@@ -58,21 +57,21 @@ func (r *mediaRepo) CreateMedia(ctx context.Context, v *biz.MediaFile) error {
 	v.UpdatedAt = po.UpdatedAt
 	return nil
 }
-func (r *mediaRepo) FindMedia(ctx context.Context, id uint) (*biz.MediaFile, error) {
+func (r *mediaRepo) FindMedia(ctx context.Context, id uint) (*system.MediaFile, error) {
 	var po mediaPO
 	if err := r.data.DB().WithContext(ctx).First(&po, id).Error; err != nil {
 		return nil, err
 	}
 	return mediaFromPO(po), nil
 }
-func (r *mediaRepo) FindMediaByHash(ctx context.Context, userID uint, hash string) (*biz.MediaFile, error) {
+func (r *mediaRepo) FindMediaByHash(ctx context.Context, userID uint, hash string) (*system.MediaFile, error) {
 	var po mediaPO
 	if err := r.data.DB().WithContext(ctx).Where("user_id = ? AND md5 = ?", userID, hash).First(&po).Error; err != nil {
 		return nil, err
 	}
 	return mediaFromPO(po), nil
 }
-func (r *mediaRepo) ListMedia(ctx context.Context, filter biz.MediaFilter) ([]*biz.MediaFile, int64, error) {
+func (r *mediaRepo) ListMedia(ctx context.Context, filter system.MediaFilter) ([]*system.MediaFile, int64, error) {
 	db := r.data.DB().WithContext(ctx).Model(&mediaPO{})
 	if filter.Keyword != "" {
 		db = db.Where("name LIKE ?", "%"+filter.Keyword+"%")
@@ -119,7 +118,7 @@ func (r *mediaRepo) ListMedia(ctx context.Context, filter biz.MediaFilter) ([]*b
 	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))
+	out := make([]*system.MediaFile, 0, len(pos))
 	for _, po := range pos {
 		out = append(out, mediaFromPO(po))
 	}
@@ -142,14 +141,14 @@ func (r *mediaRepo) MediaKeyReferences(ctx context.Context, key string) (int64,
 	err := r.data.DB().WithContext(ctx).Model(&mediaPO{}).Where(map[string]any{"key": key}).Count(&count).Error
 	return count, err
 }
-func (r *mediaRepo) CreateMediaBatch(ctx context.Context, items []*biz.MediaFile) error {
+func (r *mediaRepo) CreateMediaBatch(ctx context.Context, items []*system.MediaFile) error {
 	pos := make([]mediaPO, 0, len(items))
 	for _, v := range items {
 		pos = append(pos, mediaToPO(v))
 	}
 	return r.data.DB().WithContext(ctx).Create(&pos).Error
 }
-func (r *mediaRepo) SaveCategory(ctx context.Context, v *biz.AttachmentCategory) error {
+func (r *mediaRepo) SaveCategory(ctx context.Context, v *system.AttachmentCategory) error {
 	db := r.data.DB().WithContext(ctx)
 	var count int64
 	query := db.Model(&categoryPO{}).Where("name = ? AND pid = ?", v.Name, v.ParentID)
@@ -179,17 +178,17 @@ func (r *mediaRepo) DeleteCategory(ctx context.Context, id uint) error {
 	}
 	return r.data.DB().WithContext(ctx).Unscoped().Delete(&categoryPO{}, id).Error
 }
-func (r *mediaRepo) ListCategories(ctx context.Context) ([]*biz.AttachmentCategory, error) {
+func (r *mediaRepo) ListCategories(ctx context.Context) ([]*system.AttachmentCategory, error) {
 	var pos []categoryPO
 	if err := r.data.DB().WithContext(ctx).Find(&pos).Error; err != nil {
 		return nil, err
 	}
-	byID := map[uint]*biz.AttachmentCategory{}
+	byID := map[uint]*system.AttachmentCategory{}
 	for _, po := range pos {
 		// Children remains nil for leaf nodes, which serializes as null.
-		byID[po.ID] = &biz.AttachmentCategory{ID: po.ID, CreatedAt: po.CreatedAt, UpdatedAt: po.UpdatedAt, Name: po.Name, ParentID: po.ParentID}
+		byID[po.ID] = &system.AttachmentCategory{ID: po.ID, CreatedAt: po.CreatedAt, UpdatedAt: po.UpdatedAt, Name: po.Name, ParentID: po.ParentID}
 	}
-	roots := []*biz.AttachmentCategory{}
+	roots := []*system.AttachmentCategory{}
 	for _, po := range pos {
 		v := byID[po.ID]
 		if parent := byID[po.ParentID]; parent != nil {