Add comment functionality.

server/dao/comment.go

@@ -0,0 +1,30 @@
+package dao
+
+import "nysoure/server/model"
+
+func CreateComment(content string, userID uint, resourceID uint) (model.Comment, error) {
+	c := model.Comment{
+		Content:    content,
+		UserID:     userID,
+		ResourceID: resourceID,
+	}
+	err := db.Save(&c).Error
+	return c, err
+}
+
+func GetCommentByResourceID(resourceID uint, page, pageSize int) ([]model.Comment, int, error) {
+	var comments []model.Comment
+	var total int64
+
+	if err := db.Model(&model.Comment{}).Where("resource_id = ?", resourceID).Count(&total).Error; err != nil {
+		return nil, 0, err
+	}
+
+	if err := db.Where("resource_id = ?", resourceID).Offset((page - 1) * pageSize).Limit(pageSize).Preload("User").Order("created_at DESC").Find(&comments).Error; err != nil {
+		return nil, 0, err
+	}
+
+	totalPages := (int(total) + pageSize - 1) / pageSize
+
+	return comments, totalPages, nil
+}

server/dao/db.go

@@ -15,7 +15,7 @@ func init() {
 		panic("failed to connect database")
 	}
 
-	_ = db.AutoMigrate(&model.User{}, &model.Resource{}, &model.Image{}, &model.Tag{}, &model.Storage{}, &model.File{}, &model.UploadingFile{}, &model.Statistic{})
+	_ = db.AutoMigrate(&model.User{}, &model.Resource{}, &model.Image{}, &model.Tag{}, &model.Storage{}, &model.File{}, &model.UploadingFile{}, &model.Statistic{}, &model.Comment{})
 }
 
 func GetDB() *gorm.DB {

server/dao/resource.go

@@ -167,3 +167,14 @@ func GetResourceByTag(tagID uint, page int, pageSize int) ([]model.Resource, int
 
 	return tag.Resources, totalPages, nil
 }
+
+func ExistsResource(id uint) (bool, error) {
+	var r model.Resource
+	if err := db.Model(&model.Resource{}).Where("id = ?", id).First(&r).Error; err != nil {
+		if errors.Is(err, gorm.ErrRecordNotFound) {
+			return false, nil
+		}
+		return false, err
+	}
+	return true, nil
+}

server/dao/user.go

@@ -39,6 +39,14 @@ func ExistsUser(username string) (bool, error) {
 	return count > 0, nil
 }
 
+func ExistsUserByID(id uint) (bool, error) {
+	var count int64
+	if err := db.Model(&model.User{}).Where("id = ?", id).Count(&count).Error; err != nil {
+		return false, err
+	}
+	return count > 0, nil
+}
+
 func GetUserByUsername(username string) (model.User, error) {
 	var user model.User
 	if err := db.Where("username = ?", username).First(&user).Error; err != nil {
@@ -76,17 +84,14 @@ func IsUserDataBaseEmpty() (bool, error) {
 	return false, nil
 }
 
-// 获取分页用户列表
 func ListUsers(page, pageSize int) ([]model.User, int64, error) {
 	var users []model.User
 	var total int64
 
-	// 获取总数
 	if err := db.Model(&model.User{}).Count(&total).Error; err != nil {
 		return nil, 0, err
 	}
 
-	// 分页获取用户
 	offset := (page - 1) * pageSize
 	if err := db.Offset(offset).Limit(pageSize).Order("id desc").Find(&users).Error; err != nil {
 		return nil, 0, err
@@ -95,17 +100,14 @@ func ListUsers(page, pageSize int) ([]model.User, int64, error) {
 	return users, total, nil
 }
 
-// 根据用户名搜索用户
 func SearchUsersByUsername(username string, page, pageSize int) ([]model.User, int64, error) {
 	var users []model.User
 	var total int64
 
-	// 获取符合条件的总数
 	if err := db.Model(&model.User{}).Where("username LIKE ?", "%"+username+"%").Count(&total).Error; err != nil {
 		return nil, 0, err
 	}
 
-	// 分页获取符合条件的用户
 	offset := (page - 1) * pageSize
 	if err := db.Where("username LIKE ?", "%"+username+"%").Offset(offset).Limit(pageSize).Order("id desc").Find(&users).Error; err != nil {
 		return nil, 0, err
@@ -114,7 +116,6 @@ func SearchUsersByUsername(username string, page, pageSize int) ([]model.User, i
 	return users, total, nil
 }
 
-// 删除用户
 func DeleteUser(id uint) error {
 	return db.Delete(&model.User{}, id).Error
 }