mirror of https://gitee.com/answerdev/answer.git
109 lines
3.7 KiB
Go
109 lines
3.7 KiB
Go
package notification
|
|
|
|
import (
|
|
"context"
|
|
"time"
|
|
|
|
"github.com/answerdev/answer/internal/base/data"
|
|
"github.com/answerdev/answer/internal/base/pager"
|
|
"github.com/answerdev/answer/internal/base/reason"
|
|
"github.com/answerdev/answer/internal/entity"
|
|
"github.com/answerdev/answer/internal/schema"
|
|
notficationcommon "github.com/answerdev/answer/internal/service/notification_common"
|
|
"github.com/answerdev/answer/pkg/uid"
|
|
"github.com/segmentfault/pacman/errors"
|
|
)
|
|
|
|
// notificationRepo notification repository
|
|
type notificationRepo struct {
|
|
data *data.Data
|
|
}
|
|
|
|
// NewNotificationRepo new repository
|
|
func NewNotificationRepo(data *data.Data) notficationcommon.NotificationRepo {
|
|
return ¬ificationRepo{
|
|
data: data,
|
|
}
|
|
}
|
|
|
|
// AddNotification add notification
|
|
func (nr *notificationRepo) AddNotification(ctx context.Context, notification *entity.Notification) (err error) {
|
|
notification.ObjectID = uid.DeShortID(notification.ObjectID)
|
|
_, err = nr.data.DB.Insert(notification)
|
|
if err != nil {
|
|
return errors.InternalServer(reason.DatabaseError).WithError(err).WithStack()
|
|
}
|
|
return
|
|
}
|
|
|
|
func (nr *notificationRepo) UpdateNotificationContent(ctx context.Context, notification *entity.Notification) (err error) {
|
|
now := time.Now()
|
|
notification.UpdatedAt = now
|
|
notification.ObjectID = uid.DeShortID(notification.ObjectID)
|
|
_, err = nr.data.DB.Where("id =?", notification.ID).Cols("content", "updated_at").Update(notification)
|
|
if err != nil {
|
|
return errors.InternalServer(reason.DatabaseError).WithError(err).WithStack()
|
|
}
|
|
return
|
|
}
|
|
|
|
func (nr *notificationRepo) ClearUnRead(ctx context.Context, userID string, notificationType int) (err error) {
|
|
info := &entity.Notification{}
|
|
info.IsRead = schema.NotificationRead
|
|
_, err = nr.data.DB.Where("user_id =?", userID).And("type =?", notificationType).Cols("is_read").Update(info)
|
|
if err != nil {
|
|
return errors.InternalServer(reason.DatabaseError).WithError(err).WithStack()
|
|
}
|
|
return
|
|
}
|
|
|
|
func (nr *notificationRepo) ClearIDUnRead(ctx context.Context, userID string, id string) (err error) {
|
|
info := &entity.Notification{}
|
|
info.IsRead = schema.NotificationRead
|
|
_, err = nr.data.DB.Where("user_id =?", userID).And("id =?", id).Cols("is_read").Update(info)
|
|
if err != nil {
|
|
return errors.InternalServer(reason.DatabaseError).WithError(err).WithStack()
|
|
}
|
|
return
|
|
}
|
|
|
|
func (nr *notificationRepo) GetById(ctx context.Context, id string) (*entity.Notification, bool, error) {
|
|
info := &entity.Notification{}
|
|
exist, err := nr.data.DB.Where("id = ? ", id).Get(info)
|
|
if err != nil {
|
|
err = errors.InternalServer(reason.DatabaseError).WithError(err).WithStack()
|
|
return info, false, err
|
|
}
|
|
return info, exist, nil
|
|
}
|
|
|
|
func (nr *notificationRepo) GetByUserIdObjectIdTypeId(ctx context.Context, userID, objectID string, notificationType int) (*entity.Notification, bool, error) {
|
|
info := &entity.Notification{}
|
|
exist, err := nr.data.DB.Where("user_id = ? ", userID).And("object_id = ?", objectID).And("type = ?", notificationType).Get(info)
|
|
if err != nil {
|
|
err = errors.InternalServer(reason.DatabaseError).WithError(err).WithStack()
|
|
return info, false, err
|
|
}
|
|
return info, exist, nil
|
|
}
|
|
|
|
func (nr *notificationRepo) GetNotificationPage(ctx context.Context, searchCond *schema.NotificationSearch) (
|
|
notificationList []*entity.Notification, total int64, err error) {
|
|
notificationList = make([]*entity.Notification, 0)
|
|
if searchCond.UserID == "" {
|
|
return notificationList, 0, nil
|
|
}
|
|
|
|
session := nr.data.DB.NewSession()
|
|
session = session.Desc("updated_at")
|
|
cond := &entity.Notification{
|
|
UserID: searchCond.UserID,
|
|
Type: searchCond.Type,
|
|
}
|
|
total, err = pager.Help(searchCond.Page, searchCond.PageSize, ¬ificationList, cond, session)
|
|
if err != nil {
|
|
err = errors.InternalServer(reason.DatabaseError).WithError(err).WithStack()
|
|
}
|
|
return
|
|
}
|