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 statistics.

Browse Source
This commit is contained in:
nyne
2025-05-13 18:59:23 +08:00
parent f1345f9a0c
commit 12c1e0e413
8 changed files with 98 additions and 28 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
frontend/src/network/models.ts
View File

@@ -5,6 +5,8 @@ export interface User {
avatar_path: string;
is_admin: boolean;
can_upload: boolean;
uploads_count: number;
comments_count: number;
}
export interface UserWithToken extends User {
@@ -62,6 +64,8 @@ export interface ResourceDetails {
images: Image[];
files: RFile[];
author: User;
views: number;
downloads: number;
}
export interface Storage {

23
server/dao/comment.go
View File

@@ -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{
var comment model.Comment
err := db.Transaction(func(tx *gorm.DB) error {
comment = model.Comment{
Content: content,
UserID: userID,
ResourceID: resourceID,
}
err := db.Save(&c).Error
return c, err
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
}
return comment, nil
}
func GetCommentByResourceID(resourceID uint, page, pageSize int) ([]model.Comment, int, error) {

44
server/dao/resource.go
View File

@@ -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 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
}

3
server/dao/user.go
View File

@@ -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

6
server/model/resource.go
View File

@@ -16,6 +16,8 @@ type Resource struct {
Files []File `gorm:"foreignKey:ResourceID"`
UserID uint
User User
Views uint
Downloads uint
}
type ResourceView struct {
@@ -37,6 +39,8 @@ type ResourceDetailView struct {
Images []ImageView `json:"images"`
Files []FileView `json:"files"`
Author UserView `json:"author"`
Views uint `json:"views"`
Downloads uint `json:"downloads"`
}
func (r *Resource) ToView() ResourceView {
@@ -85,5 +89,7 @@ func (r *Resource) ToDetailView() ResourceDetailView {
Images: images,
Files: files,
Author: r.User.ToView(),
Views: r.Views,
Downloads: r.Downloads,
}
}

6
server/model/user.go
View File

@@ -13,6 +13,8 @@ type User struct {
IsAdmin bool
CanUpload bool
AvatarVersion int
UploadsCount int
CommentsCount int
Resources []Resource `gorm:"foreignKey:UserID"`
}
@@ -23,6 +25,8 @@ type UserView struct {
AvatarPath string `json:"avatar_path"`
IsAdmin bool `json:"is_admin"`
CanUpload bool `json:"can_upload"`
UploadsCount int `json:"uploads_count"`
CommentsCount int `json:"comments_count"`
}
type UserViewWithToken struct {
@@ -38,6 +42,8 @@ func (u User) ToView() UserView {
AvatarPath: fmt.Sprintf("/api/user/avatar/%d?v=%d", u.ID, u.AvatarVersion),
IsAdmin: u.IsAdmin,
CanUpload: u.CanUpload || u.IsAdmin,
UploadsCount: u.UploadsCount,
CommentsCount: u.CommentsCount,
}
}

3
server/service/file.go
View File

@@ -379,6 +379,7 @@ func DownloadFile(fid string) (string, string, error) {
if file.StorageID == nil {
if file.RedirectUrl != "" {
_ = dao.AddResourceDownloadCount(file.ResourceID)
return file.RedirectUrl, file.Filename, nil
}
return "", "", model.NewRequestError("file is not available")
@@ -396,5 +397,7 @@ func DownloadFile(fid string) (string, string, error) {
path, err := iStorage.Download(file.StorageKey, file.Filename)
_ = dao.AddResourceDownloadCount(file.ResourceID)
return path, file.Filename, err
}

1
server/service/resource.go
View File

@@ -56,6 +56,7 @@ func CreateResource(uid uint, params *ResourceCreateParams) (uint, error) {
func GetResource(id uint) (*model.ResourceDetailView, error) {
r, err := dao.GetResourceByID(id)
_ = dao.AddResourceViewCount(id)
if err != nil {
return nil, err
}