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

Add tag type and tag alias.

Browse Source
This commit is contained in:
nyne
2025-05-30 16:08:58 +08:00
parent 5e6d202acd
commit 5ef2816f98
12 changed files with 336 additions and 332 deletions
Download Patch File Download Diff File Expand all files Collapse all files

81
server/dao/resource.go
View File

@@ -180,20 +180,37 @@ func searchWithKeyword(keyword string) ([]model.Resource, error) {
}
if len([]rune(keyword)) < 20 {
var tag model.Tag
if err := db.Where("name = ?", keyword).First(&tag).Error; err != nil {
var err error
if tag, err = GetTagByName(keyword); err != nil {
if !errors.Is(err, gorm.ErrRecordNotFound) {
return nil, err
}
} else {
if err := db.Model(&tag).Preload("Resources").Find(&tag).Error; err != nil {
if tag.AliasOf != nil {
tag, err = GetTagByID(*tag.AliasOf)
if err != nil {
return nil, err
}
}
var tagIds []uint
tagIds = append(tagIds, tag.ID)
for _, alias := range tag.Aliases {
tagIds = append(tagIds, alias.ID)
}
var resources []model.Resource
subQuery := db.Table("resource_tags").
Select("resource_id").
Where("tag_id IN ?", tagIds).
Group("resource_id")
if err := db.Where("id IN (?)", subQuery).Select("id", "title", "alternative_titles").Preload("Tags").Find(&resources).Error; err != nil {
return nil, err
}
return tag.Resources, nil
return resources, nil
}
}
if len([]rune(keyword)) < 80 {
var resources []model.Resource
if err := db.Where("title LIKE ?", "%"+keyword+"%").Or("alternative_titles LIKE ?", "%"+keyword+"%").Find(&resources).Error; err != nil {
if err := db.Where("title LIKE ?", "%"+keyword+"%").Or("alternative_titles LIKE ?", "%"+keyword+"%").Select("id", "title", "alternative_titles").Preload("Tags").Find(&resources).Error; err != nil {
return nil, err
}
return resources, nil
@@ -202,23 +219,53 @@ func searchWithKeyword(keyword string) ([]model.Resource, error) {
}
func GetResourceByTag(tagID uint, page int, pageSize int) ([]model.Resource, int, error) {
var tag model.Tag
total := db.Model(&model.Tag{
Model: gorm.Model{
ID: tagID,
},
}).Association("Resources").Count()
if err := db.Model(&model.Tag{}).Where("id = ?", tagID).Preload("Resources", func(tx *gorm.DB) *gorm.DB {
return tx.Offset((page - 1) * pageSize).Limit(pageSize).Preload("Tags").Preload("User").Preload("Images").Order("created_at DESC")
}).First(&tag).Error; err != nil {
tag, err := GetTagByID(tagID)
if err != nil {
return nil, 0, err
}
totalPages := (int(total) + pageSize - 1) / pageSize
if tag.AliasOf != nil {
tag, err = GetTagByID(*tag.AliasOf)
if err != nil {
return nil, 0, err
}
}
return tag.Resources, totalPages, nil
var tagIds []uint
tagIds = append(tagIds, tag.ID)
for _, alias := range tag.Aliases {
tagIds = append(tagIds, alias.ID)
}
var resources []model.Resource
var total int64
subQuery := db.Table("resource_tags").
Select("resource_id").
Where("tag_id IN ?", tagIds).
Group("resource_id")
if err := db.Model(&model.Resource{}).
Where("id IN (?)", subQuery).
Count(&total).Error; err != nil {
return nil, 0, err
}
if err := db.Where("id IN (?)", subQuery).
Offset((page - 1) * pageSize).
Limit(pageSize).
Preload("User").
Preload("Images").
Preload("Tags").
Preload("Files").
Order("created_at DESC").
Find(&resources).Error; err != nil {
return nil, 0, err
}
totalPages := (total + int64(pageSize) - 1) / int64(pageSize)
return resources, int(totalPages), nil
}
func ExistsResource(id uint) (bool, error) {

22
server/dao/tag.go
View File

@@ -16,10 +16,14 @@ func CreateTag(tag string) (model.Tag, error) {
return t, nil
}
func SearchTag(keyword string) ([]model.Tag, error) {
func SearchTag(keyword string, mainTag bool) ([]model.Tag, error) {
// Search for a tag by its name in the database
var t []model.Tag
if err := db.Model(&model.Tag{}).Where("name Like ?", "%"+keyword+"%").Limit(10).Find(&t).Error; err != nil {
query := db.Model(&model.Tag{}).Where("name Like ?", "%"+keyword+"%")
if mainTag {
query = query.Where("alias_of IS NULL")
}
if err := query.Limit(10).Find(&t).Error; err != nil {
return nil, err
}
return t, nil
@@ -38,7 +42,7 @@ func DeleteTag(id uint) error {
func GetTagByID(id uint) (model.Tag, error) {
// Retrieve a tag by its ID from the database
var t model.Tag
if err := db.First(&t, id).Error; err != nil {
if err := db.Preload("Aliases").First(&t, id).Error; err != nil {
if errors.Is(err, gorm.ErrRecordNotFound) {
return model.Tag{}, model.NewNotFoundError("Tag not found")
}
@@ -50,7 +54,7 @@ func GetTagByID(id uint) (model.Tag, error) {
func GetTagByName(name string) (model.Tag, error) {
// Retrieve a tag by its name from the database
var t model.Tag
if err := db.Where("name = ?", name).First(&t).Error; err != nil {
if err := db.Preload("Aliases").Where("name = ?", name).First(&t).Error; err != nil {
if errors.Is(err, gorm.ErrRecordNotFound) {
return model.Tag{}, model.NewNotFoundError("Tag not found")
}
@@ -59,8 +63,14 @@ func GetTagByName(name string) (model.Tag, error) {
return t, nil
}
func SetTagDescription(id uint, description string) error {
if err := db.Model(model.Tag{}).Where("id = ?", id).Update("description", description).Error; err != nil {
func SetTagInfo(id uint, description string, aliasOf *uint, tagType string) error {
t := model.Tag{Model: gorm.Model{
ID: id,
}, Description: description, Type: tagType, AliasOf: aliasOf}
if err := db.Model(&t).Updates(t).Error; err != nil {
if errors.Is(err, gorm.ErrRecordNotFound) {
return model.NewNotFoundError("Tag not found")
}
return err
}
return nil