115 lines
2.8 KiB
Go
115 lines
2.8 KiB
Go
package services
|
|
|
|
import (
|
|
"context"
|
|
"os"
|
|
"time"
|
|
|
|
"quyun/v2/app/errorx"
|
|
user_dto "quyun/v2/app/http/v1/dto"
|
|
"quyun/v2/app/jobs/args"
|
|
"quyun/v2/app/requests"
|
|
"quyun/v2/database/models"
|
|
"quyun/v2/providers/job"
|
|
)
|
|
|
|
// @provider
|
|
type notification struct {
|
|
job *job.Job
|
|
}
|
|
|
|
func (s *notification) List(ctx context.Context, tenantID, userID int64, page int, typeArg string) (*requests.Pager, error) {
|
|
tbl, q := models.NotificationQuery.QueryContext(ctx)
|
|
q = q.Where(tbl.UserID.Eq(userID))
|
|
if tenantID > 0 {
|
|
q = q.Where(tbl.TenantID.Eq(tenantID))
|
|
}
|
|
|
|
if typeArg != "" && typeArg != "all" {
|
|
q = q.Where(tbl.Type.Eq(typeArg))
|
|
}
|
|
|
|
q = q.Order(tbl.CreatedAt.Desc())
|
|
|
|
p := requests.Pagination{Page: int64(page), Limit: 20}
|
|
total, err := q.Count()
|
|
if err != nil {
|
|
return nil, errorx.ErrDatabaseError.WithCause(err)
|
|
}
|
|
|
|
list, err := q.Offset(int(p.Offset())).Limit(int(p.Limit)).Find()
|
|
if err != nil {
|
|
return nil, errorx.ErrDatabaseError.WithCause(err)
|
|
}
|
|
|
|
data := make([]user_dto.Notification, len(list))
|
|
for i, v := range list {
|
|
data[i] = user_dto.Notification{
|
|
ID: v.ID,
|
|
Type: v.Type,
|
|
Title: v.Title,
|
|
Content: v.Content,
|
|
Read: v.IsRead,
|
|
Time: v.CreatedAt.Format(time.RFC3339),
|
|
}
|
|
}
|
|
|
|
return &requests.Pager{
|
|
Pagination: requests.Pagination{Page: p.Page, Limit: p.Limit},
|
|
Total: total,
|
|
Items: data,
|
|
}, nil
|
|
}
|
|
|
|
func (s *notification) MarkRead(ctx context.Context, tenantID, userID, id int64) error {
|
|
tbl, q := models.NotificationQuery.QueryContext(ctx)
|
|
q = q.Where(tbl.ID.Eq(id), tbl.UserID.Eq(userID))
|
|
if tenantID > 0 {
|
|
q = q.Where(tbl.TenantID.Eq(tenantID))
|
|
}
|
|
_, err := q.UpdateSimple(tbl.IsRead.Value(true))
|
|
if err != nil {
|
|
return errorx.ErrDatabaseError.WithCause(err)
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func (s *notification) MarkAllRead(ctx context.Context, tenantID, userID int64) error {
|
|
tbl, q := models.NotificationQuery.QueryContext(ctx)
|
|
q = q.Where(tbl.UserID.Eq(userID), tbl.IsRead.Is(false))
|
|
if tenantID > 0 {
|
|
q = q.Where(tbl.TenantID.Eq(tenantID))
|
|
}
|
|
_, err := q.UpdateSimple(tbl.IsRead.Value(true))
|
|
if err != nil {
|
|
return errorx.ErrDatabaseError.WithCause(err)
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func (s *notification) Send(ctx context.Context, tenantID, userID int64, typ, title, content string) error {
|
|
arg := args.NotificationArgs{
|
|
TenantID: tenantID,
|
|
UserID: userID,
|
|
Type: typ,
|
|
Title: title,
|
|
Content: content,
|
|
}
|
|
// 测试环境下同步写入，避免异步任务未启动导致结果不确定。
|
|
if os.Getenv("JOB_INLINE") == "1" {
|
|
n := &models.Notification{
|
|
TenantID: tenantID,
|
|
UserID: userID,
|
|
Type: typ,
|
|
Title: title,
|
|
Content: content,
|
|
IsRead: false,
|
|
}
|
|
if err := models.NotificationQuery.WithContext(ctx).Create(n); err != nil {
|
|
return errorx.ErrDatabaseError.WithCause(err)
|
|
}
|
|
return nil
|
|
}
|
|
return s.job.Add(arg)
|
|
}
|