nyne/nysoure
1
0
Fork 0
mirror of https://github.com/wgh136/nysoure.git synced 2025-09-27 04:17:23 +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

20
server/dao/collection.go
View File

@@ -80,7 +80,11 @@ func AddResourceToCollection(collectionID uint, resourceID uint) error {
collection := &model.Collection{}
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{
@@ -91,6 +95,10 @@ func AddResourceToCollection(collectionID uint, resourceID uint) error {
return err
}
if err := tx.Model(collection).UpdateColumn("resources_count", gorm.Expr("resources_count + ?", 1)).Error; err != nil {
return err
}
return nil
})
}
@@ -100,7 +108,11 @@ func RemoveResourceFromCollection(collectionID uint, resourceID uint) error {
collection := &model.Collection{}
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{
@@ -111,6 +123,10 @@ func RemoveResourceFromCollection(collectionID uint, resourceID uint) error {
return err
}
if err := tx.Model(collection).UpdateColumn("resources_count", gorm.Expr("resources_count - ?", 1)).Error; err != nil {
return err
}
return nil
})
}

35
server/model/collection.go
View File

@@ -4,28 +4,31 @@ import "gorm.io/gorm"
type Collection struct {
gorm.Model
Title string `gorm:"not null"`
Article string `gorm:"not null"`
UserID uint `gorm:"not null"`
User User `gorm:"foreignKey:UserID;references:ID"`
Images []Image `gorm:"many2many:collection_images;"`
Resources []Resource `gorm:"many2many:collection_resources;"`
Title string `gorm:"not null"`
Article string `gorm:"not null"`
UserID uint `gorm:"not null"`
User User `gorm:"foreignKey:UserID;references:ID"`
ResourcesCount int `gorm:"default:0"`
Images []Image `gorm:"many2many:collection_images;"`
Resources []Resource `gorm:"many2many:collection_resources;"`
}
type CollectionView struct {
ID uint `json:"id"`
Title string `json:"title"`
Article string `json:"article"`
User UserView `json:"user"`
Images []Image `json:"images"`
ID uint `json:"id"`
Title string `json:"title"`
Article string `json:"article"`
User UserView `json:"user"`
ResourcesCount int `json:"resources_count"`
Images []Image `json:"images"`
}
func (c Collection) ToView() *CollectionView {
return &CollectionView{
ID: c.ID,
Title: c.Title,
Article: c.Article,
User: c.User.ToView(),
Images: c.Images,
ID: c.ID,
Title: c.Title,
Article: c.Article,
User: c.User.ToView(),
ResourcesCount: c.ResourcesCount,
Images: c.Images,
}
}