nyne/nysoure
1
0
Fork 0
mirror of https://github.com/wgh136/nysoure.git synced 2025-09-27 12:17:24 +00:00
Code Issues Packages Projects Releases Wiki Activity

Add resources_count to collection.

Browse Source
This commit is contained in:
nyne
2025-07-31 14:08:38 +08:00
parent 40fb3a93b6
commit fd7c3797ea
3 changed files with 39 additions and 19 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
frontend/src/network/models.ts
View File

@@ -197,5 +197,6 @@ export interface Collection {
title: string; title: string;
article: string; article: string;
user: User; user: User;
resources_count: number;
images: Image[]; images: Image[];
} }

20
server/dao/collection.go
View File

@@ -80,7 +80,11 @@ func AddResourceToCollection(collectionID uint, resourceID uint) error {
collection := &model.Collection{} collection := &model.Collection{}
if err := tx.Where("id = ?", collectionID).First(collection).Error; err != nil { if err := tx.Where("id = ?", collectionID).First(collection).Error; err != nil {
return err return model.NewRequestError("Invalid collection ID")
}
if err := tx.Model(&model.Resource{}).Where("id = ?", resourceID).First(&model.Resource{}).Error; err != nil {
return model.NewRequestError("Invalid resource ID")
} }
if err := tx.Model(collection).Association("Resources").Append(&model.Resource{ if err := tx.Model(collection).Association("Resources").Append(&model.Resource{
@@ -91,6 +95,10 @@ func AddResourceToCollection(collectionID uint, resourceID uint) error {
return err return err
} }
if err := tx.Model(collection).UpdateColumn("resources_count", gorm.Expr("resources_count + ?", 1)).Error; err != nil {
return err
}
return nil return nil
}) })
} }
@@ -100,7 +108,11 @@ func RemoveResourceFromCollection(collectionID uint, resourceID uint) error {
collection := &model.Collection{} collection := &model.Collection{}
if err := tx.Where("id = ?", collectionID).First(collection).Error; err != nil { if err := tx.Where("id = ?", collectionID).First(collection).Error; err != nil {
return err return model.NewRequestError("Invalid collection ID")
}
if err := tx.Model(&model.Resource{}).Where("id = ?", resourceID).First(&model.Resource{}).Error; err != nil {
return model.NewRequestError("Invalid resource ID")
} }
if err := tx.Model(collection).Association("Resources").Delete(&model.Resource{ if err := tx.Model(collection).Association("Resources").Delete(&model.Resource{
@@ -111,6 +123,10 @@ func RemoveResourceFromCollection(collectionID uint, resourceID uint) error {
return err return err
} }
if err := tx.Model(collection).UpdateColumn("resources_count", gorm.Expr("resources_count - ?", 1)).Error; err != nil {
return err
}
return nil return nil
}) })
} }

3
server/model/collection.go
View File

@@ -8,6 +8,7 @@ type Collection struct {
Article string `gorm:"not null"` Article string `gorm:"not null"`
UserID uint `gorm:"not null"` UserID uint `gorm:"not null"`
User User `gorm:"foreignKey:UserID;references:ID"` User User `gorm:"foreignKey:UserID;references:ID"`
ResourcesCount int `gorm:"default:0"`
Images []Image `gorm:"many2many:collection_images;"` Images []Image `gorm:"many2many:collection_images;"`
Resources []Resource `gorm:"many2many:collection_resources;"` Resources []Resource `gorm:"many2many:collection_resources;"`
} }
@@ -17,6 +18,7 @@ type CollectionView struct {
Title string `json:"title"` Title string `json:"title"`
Article string `json:"article"` Article string `json:"article"`
User UserView `json:"user"` User UserView `json:"user"`
ResourcesCount int `json:"resources_count"`
Images []Image `json:"images"` Images []Image `json:"images"`
} }
@@ -26,6 +28,7 @@ func (c Collection) ToView() *CollectionView {
Title: c.Title, Title: c.Title,
Article: c.Article, Article: c.Article,
User: c.User.ToView(), User: c.User.ToView(),
ResourcesCount: c.ResourcesCount,
Images: c.Images, Images: c.Images,
} }
} }