Add statistics.

server/dao/comment.go

@@ -1,15 +1,30 @@
 package dao
 
-import "nysoure/server/model"
+import (
+	"gorm.io/gorm"
+	"nysoure/server/model"
+)
 
 func CreateComment(content string, userID uint, resourceID uint) (model.Comment, error) {
-	c := model.Comment{
-		Content:    content,
-		UserID:     userID,
-		ResourceID: resourceID,
+	var comment model.Comment
+	err := db.Transaction(func(tx *gorm.DB) error {
+		comment = model.Comment{
+			Content:    content,
+			UserID:     userID,
+			ResourceID: resourceID,
+		}
+		if err := tx.Create(&comment).Error; err != nil {
+			return err
+		}
+		if err := tx.Model(&model.User{}).Where("id = ?", userID).Update("comments_count", gorm.Expr("comments_count + 1")).Error; err != nil {
+			return err
+		}
+		return nil
+	})
+	if err != nil {
+		return model.Comment{}, err
 	}
-	err := db.Save(&c).Error
-	return c, err
+	return comment, nil
 }
 
 func GetCommentByResourceID(resourceID uint, page, pageSize int) ([]model.Comment, int, error) {

server/dao/resource.go

@@ -9,8 +9,17 @@ import (
 )
 
 func CreateResource(r model.Resource) (model.Resource, error) {
-	// Create a new resource in the database
-	if err := db.Create(&r).Error; err != nil {
+	err := db.Transaction(func(tx *gorm.DB) error {
+		err := tx.Create(&r).Error
+		if err != nil {
+			return err
+		}
+		if err := tx.Model(&model.User{}).Where("id = ?", r.UserID).Update("uploads_count", gorm.Expr("uploads_count + ?", 1)).Error; err != nil {
+			return err
+		}
+		return nil
+	})
+	if err != nil {
 		return model.Resource{}, err
 	}
 	return r, nil
@@ -55,13 +64,22 @@ func UpdateResource(r model.Resource) error {
 }
 
 func DeleteResource(id uint) error {
-	// Delete a resource from the database
-	r := model.Resource{}
-	r.ID = id
-	if err := db.Delete(&r).Error; err != nil && !errors.Is(err, gorm.ErrRecordNotFound) {
-		return err
-	}
-	return nil
+	return db.Transaction(func(tx *gorm.DB) error {
+		var r model.Resource
+		if err := tx.Where("id = ?", id).First(&r).Error; err != nil {
+			if errors.Is(err, gorm.ErrRecordNotFound) {
+				return model.NewNotFoundError("Resource not found")
+			}
+			return err
+		}
+		if err := tx.Model(&model.User{}).Where("id = ?", r.UserID).Update("uploads_count", gorm.Expr("uploads_count - ?", 1)).Error; err != nil {
+			return err
+		}
+		if err := tx.Delete(&r).Error; err != nil {
+			return err
+		}
+		return nil
+	})
 }
 
 func Search(query string, page, pageSize int) ([]model.Resource, int, error) {
@@ -178,3 +196,17 @@ func ExistsResource(id uint) (bool, error) {
 	}
 	return true, nil
 }
+
+func AddResourceViewCount(id uint) error {
+	if err := db.Model(&model.Resource{}).Where("id = ?", id).Update("views", gorm.Expr("views + ?", 1)).Error; err != nil {
+		return err
+	}
+	return nil
+}
+
+func AddResourceDownloadCount(id uint) error {
+	if err := db.Model(&model.Resource{}).Where("id = ?", id).Update("downloads", gorm.Expr("downloads + ?", 1)).Error; err != nil {
+		return err
+	}
+	return nil
+}

server/dao/user.go

@@ -61,6 +61,9 @@ func GetUserByUsername(username string) (model.User, error) {
 func GetUserByID(id uint) (model.User, error) {
 	var user model.User
 	if err := db.First(&user, id).Error; err != nil {
+		if errors.Is(err, gorm.ErrRecordNotFound) {
+			return user, model.NewNotFoundError("User not found")
+		}
 		return user, err
 	}
 	return user, nil