mirror of https://github.com/rocboss/paopao-ce
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
114 lines
2.9 KiB
114 lines
2.9 KiB
// Copyright 2022 ROC. All rights reserved.
|
|
// Use of this source code is governed by a MIT style
|
|
// license that can be found in the LICENSE file.
|
|
|
|
package jinzhu
|
|
|
|
import (
|
|
"time"
|
|
|
|
"github.com/rocboss/paopao-ce/internal/core"
|
|
"github.com/rocboss/paopao-ce/internal/core/cs"
|
|
"github.com/rocboss/paopao-ce/internal/dao/jinzhu/dbr"
|
|
"gorm.io/gorm"
|
|
)
|
|
|
|
type tweetMetricSrvA struct {
|
|
db *gorm.DB
|
|
}
|
|
|
|
type commentMetricSrvA struct {
|
|
db *gorm.DB
|
|
}
|
|
|
|
type userMetricSrvA struct {
|
|
db *gorm.DB
|
|
}
|
|
|
|
func (s *tweetMetricSrvA) UpdateTweetMetric(metric *cs.TweetMetric) error {
|
|
return s.db.Transaction(func(tx *gorm.DB) (err error) {
|
|
postMetric := &dbr.PostMetric{PostId: metric.PostId}
|
|
db := s.db.Model(postMetric).Where("post_id=?", metric.PostId)
|
|
db.First(postMetric)
|
|
postMetric.RankScore = metric.RankScore(postMetric.MotivationFactor)
|
|
err = db.Save(postMetric).Error
|
|
return
|
|
})
|
|
}
|
|
|
|
func (s *tweetMetricSrvA) AddTweetMetric(postId int64) (err error) {
|
|
_, err = (&dbr.PostMetric{PostId: postId}).Create(s.db)
|
|
return
|
|
}
|
|
|
|
func (s *tweetMetricSrvA) DeleteTweetMetric(postId int64) (err error) {
|
|
return (&dbr.PostMetric{PostId: postId}).Delete(s.db)
|
|
}
|
|
|
|
func (s *commentMetricSrvA) UpdateCommentMetric(metric *cs.CommentMetric) error {
|
|
return s.db.Transaction(func(tx *gorm.DB) (err error) {
|
|
commentMetric := &dbr.CommentMetric{CommentId: metric.CommentId}
|
|
db := s.db.Model(commentMetric).Where("comment_id=?", metric.CommentId)
|
|
db.First(commentMetric)
|
|
commentMetric.RankScore = metric.RankScore(commentMetric.MotivationFactor)
|
|
err = db.Save(commentMetric).Error
|
|
return
|
|
})
|
|
}
|
|
|
|
func (s *commentMetricSrvA) AddCommentMetric(commentId int64) (err error) {
|
|
_, err = (&dbr.CommentMetric{CommentId: commentId}).Create(s.db)
|
|
return
|
|
}
|
|
|
|
func (s *commentMetricSrvA) DeleteCommentMetric(commentId int64) (err error) {
|
|
return (&dbr.CommentMetric{CommentId: commentId}).Delete(s.db)
|
|
}
|
|
|
|
func (s *userMetricSrvA) UpdateUserMetric(userId int64, action uint8) (err error) {
|
|
metric := &dbr.UserMetric{}
|
|
db := s.db.Model(metric)
|
|
if err = db.Where("user_id=?", userId).First(metric).Error; err != nil {
|
|
metric = &dbr.UserMetric{
|
|
UserId: userId,
|
|
}
|
|
}
|
|
metric.LatestTrendsOn = time.Now().Unix()
|
|
switch action {
|
|
case cs.MetricActionCreateTweet:
|
|
metric.TweetsCount++
|
|
case cs.MetricActionDeleteTweet:
|
|
if metric.TweetsCount > 0 {
|
|
metric.TweetsCount--
|
|
}
|
|
}
|
|
return db.Save(metric).Error
|
|
}
|
|
|
|
func (s *userMetricSrvA) AddUserMetric(userId int64) (err error) {
|
|
_, err = (&dbr.UserMetric{UserId: userId}).Create(s.db)
|
|
return
|
|
}
|
|
|
|
func (s *userMetricSrvA) DeleteUserMetric(userId int64) (err error) {
|
|
return (&dbr.UserMetric{UserId: userId}).Delete(s.db)
|
|
}
|
|
|
|
func newTweetMetricServentA(db *gorm.DB) core.TweetMetricServantA {
|
|
return &tweetMetricSrvA{
|
|
db: db,
|
|
}
|
|
}
|
|
|
|
func newCommentMetricServentA(db *gorm.DB) core.CommentMetricServantA {
|
|
return &commentMetricSrvA{
|
|
db: db,
|
|
}
|
|
}
|
|
|
|
func newUserMetricServentA(db *gorm.DB) core.UserMetricServantA {
|
|
return &userMetricSrvA{
|
|
db: db,
|
|
}
|
|
}
|