nysoure/server/dao/activity.go

package dao

import (
	"errors"
	"gorm.io/gorm"
	"nysoure/server/model"
	"time"
)

func AddNewResourceActivity(userID, resourceID uint) error {
	activity := &model.Activity{
		UserID: userID,
		Type:   model.ActivityTypeNewResource,
		RefID:  resourceID,
	}
	return db.Create(activity).Error
}

func AddUpdateResourceActivity(userID, resourceID uint) error {
	var userLastActivity model.Activity
	if err := db.Model(&userLastActivity).Where("user_id = ?", userID).First(&userLastActivity).Error; err != nil {
		if !errors.Is(err, gorm.ErrRecordNotFound) {
			return err
		}
	}
	if userLastActivity.Type == model.ActivityTypeUpdateResource && userLastActivity.RefID == resourceID {
		if time.Since(userLastActivity.CreatedAt) < 10*time.Minute {
			// If the last activity is an update to the same resource within 10 minutes, skip creating a new activity
			return nil
		}
	}

	activity := &model.Activity{
		UserID: userID,
		Type:   model.ActivityTypeUpdateResource,
		RefID:  resourceID,
	}
	return db.Create(activity).Error
}

func AddNewCommentActivity(userID, commentID uint) error {
	activity := &model.Activity{
		UserID: userID,
		Type:   model.ActivityTypeNewComment,
		RefID:  commentID,
	}
	return db.Create(activity).Error
}

func AddNewFileActivity(userID, fileID uint) error {
	activity := &model.Activity{
		UserID: userID,
		Type:   model.ActivityTypeNewFile,
		RefID:  fileID,
	}
	return db.Create(activity).Error
}

func DeleteResourceActivity(resourceID uint) error {
	return db.Where("ref_id = ? AND (type = ? OR type = ?)", resourceID, model.ActivityTypeNewResource, model.ActivityTypeUpdateResource).Delete(&model.Activity{}).Error
}

func DeleteCommentActivity(commentID uint) error {
	return db.Where("ref_id = ? AND type = ?", commentID, model.ActivityTypeNewComment).Delete(&model.Activity{}).Error
}

func GetActivityList(offset, limit int) ([]model.Activity, int, error) {
	var activities []model.Activity
	var total int64

	if err := db.Model(&model.Activity{}).Count(&total).Error; err != nil {
		return nil, 0, err
	}

	if err := db.Offset(offset).Limit(limit).Order("id DESC").Find(&activities).Error; err != nil {
		return nil, 0, err
	}

	return activities, int(total), nil
}