|
|
// Copyright 2023 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 sakila
|
|
|
|
|
|
import (
|
|
|
"time"
|
|
|
|
|
|
"github.com/alimy/tryst/cfg"
|
|
|
"github.com/bitbus/sqlx"
|
|
|
"github.com/bitbus/sqlx/db"
|
|
|
"github.com/rocboss/paopao-ce/internal/core"
|
|
|
"github.com/rocboss/paopao-ce/internal/core/cs"
|
|
|
"github.com/rocboss/paopao-ce/internal/core/ms"
|
|
|
"github.com/rocboss/paopao-ce/internal/dao/sakila/auto/cc"
|
|
|
"github.com/rocboss/paopao-ce/internal/dao/sakila/auto/pgc"
|
|
|
)
|
|
|
|
|
|
var (
|
|
|
_ core.MessageService = (*messageSrv)(nil)
|
|
|
)
|
|
|
|
|
|
type messageSrv struct {
|
|
|
*sqlxSrv
|
|
|
q *cc.Message
|
|
|
}
|
|
|
|
|
|
func (s *messageSrv) CreateMessage(r *ms.Message) (*ms.Message, error) {
|
|
|
r.Model = &ms.Model{
|
|
|
CreatedOn: time.Now().Unix(),
|
|
|
}
|
|
|
res, err := s.q.CreateMessage.Exec(r)
|
|
|
if err != nil {
|
|
|
return nil, err
|
|
|
}
|
|
|
r.ID, err = res.LastInsertId()
|
|
|
if err != nil {
|
|
|
return nil, err
|
|
|
}
|
|
|
return r, nil
|
|
|
}
|
|
|
|
|
|
func (s *messageSrv) GetUnreadCount(userID int64) (res int64, err error) {
|
|
|
err = s.q.GetUnreadCount.Get(&res, userID)
|
|
|
return
|
|
|
}
|
|
|
|
|
|
func (s *messageSrv) GetMessageByID(id int64) (*ms.Message, error) {
|
|
|
return db.Get[ms.Message](s.q.GetMessageById, id)
|
|
|
}
|
|
|
|
|
|
func (s *messageSrv) ReadMessage(r *ms.Message) (err error) {
|
|
|
_, err = s.q.ReadMessage.Exec(time.Now().Unix(), r.ID)
|
|
|
return
|
|
|
}
|
|
|
|
|
|
func (s *messageSrv) ReadAllMessage(userId int64) (err error) {
|
|
|
_, err = s.q.ReadAllMessage.Exec(time.Now().Unix(), userId)
|
|
|
return
|
|
|
}
|
|
|
|
|
|
func (s *messageSrv) GetMessages(userId int64, style cs.MessageStyle, limit, offset int) (res []*ms.MessageFormated, total int64, err error) {
|
|
|
var messages []*ms.Message
|
|
|
// 1动态，2评论，3回复，4私信，5好友申请，99系统通知
|
|
|
switch style {
|
|
|
case cs.StyleMsgSystem:
|
|
|
if err = s.q.CountSystemMessages.Get(&total, userId); err == nil && total > 0 {
|
|
|
err = s.q.GetSystemMessages.Select(&messages, userId, limit, offset)
|
|
|
}
|
|
|
case cs.StyleMsgWhisper:
|
|
|
if err = s.q.CountWhisperMessages.Get(&total, userId, userId); err == nil && total > 0 {
|
|
|
err = s.q.GetWhisperMessages.Select(&messages, userId, userId, limit, offset)
|
|
|
}
|
|
|
case cs.StyleMsgRequesting:
|
|
|
if err = s.q.CountRequestingMessages.Get(&total, userId); err == nil && total > 0 {
|
|
|
err = s.q.GetRequestingMessages.Select(&messages, userId, limit, offset)
|
|
|
}
|
|
|
case cs.StyleMsgUnread:
|
|
|
if err = s.q.CountUnreadMessages.Get(&total, userId); err == nil && total > 0 {
|
|
|
err = s.q.GetUnreadMessages.Select(&messages, userId, limit, offset)
|
|
|
}
|
|
|
case cs.StyleMsgAll:
|
|
|
fallthrough
|
|
|
default:
|
|
|
if err = s.q.CountAllMessages.Get(&total, userId, userId); err == nil && total > 0 {
|
|
|
err = s.q.GetAllMessages.Select(&messages, userId, userId, limit, offset)
|
|
|
}
|
|
|
}
|
|
|
if err == nil {
|
|
|
for _, message := range messages {
|
|
|
res = append(res, message.Format())
|
|
|
}
|
|
|
}
|
|
|
return
|
|
|
|
|
|
}
|
|
|
|
|
|
func newMessageService(db *sqlx.DB) (s core.MessageService) {
|
|
|
ms := &messageSrv{
|
|
|
sqlxSrv: newSqlxSrv(db),
|
|
|
q: ccBuild(db, cc.BuildMessage),
|
|
|
}
|
|
|
s = ms
|
|
|
if cfg.Any("PostgreSQL", "PgSQL", "Postgres") {
|
|
|
s = &pgcMessageSrv{
|
|
|
messageSrv: ms,
|
|
|
p: pgcBuild(db, pgc.BuildMessage),
|
|
|
}
|
|
|
}
|
|
|
return
|
|
|
}
|
