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.
81 lines
1.9 KiB
81 lines
1.9 KiB
// 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/jmoiron/sqlx"
|
|
"github.com/rocboss/paopao-ce/internal/core"
|
|
"github.com/rocboss/paopao-ce/internal/core/ms"
|
|
"github.com/rocboss/paopao-ce/internal/dao/jinzhu/dbr"
|
|
"github.com/rocboss/paopao-ce/internal/dao/sakila/yesql/cc"
|
|
)
|
|
|
|
var (
|
|
_ core.MessageService = (*messageSrv)(nil)
|
|
)
|
|
|
|
type messageSrv struct {
|
|
*sqlxSrv
|
|
q *cc.Message
|
|
}
|
|
|
|
func (s *messageSrv) CreateMessage(r *ms.Message) (*ms.Message, error) {
|
|
r.Model = &dbr.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)
|
|
return
|
|
}
|
|
|
|
func (s *messageSrv) GetMessageByID(id int64) (res *ms.Message, err error) {
|
|
err = s.q.GetMessageById.Get(res, id)
|
|
return
|
|
}
|
|
|
|
func (s *messageSrv) ReadMessage(r *ms.Message) (err error) {
|
|
_, err = s.q.ReadMessage.Exec(time.Now().Unix(), r.ID)
|
|
return
|
|
}
|
|
|
|
func (s *messageSrv) GetMessages(r *ms.ConditionsT, offset, limit int) ([]*ms.MessageFormated, error) {
|
|
var messages []*ms.Message
|
|
(*r)["limit"], (*r)["offset"] = limit, offset
|
|
if err := s.q.GetMessages.Select(&messages, r); err != nil {
|
|
return nil, err
|
|
}
|
|
mfs := make([]*dbr.MessageFormated, 0, len(messages))
|
|
for _, message := range messages {
|
|
mf := message.Format()
|
|
mfs = append(mfs, mf)
|
|
}
|
|
return mfs, nil
|
|
}
|
|
|
|
func (s *messageSrv) GetMessageCount(r *ms.ConditionsT) (res int64, err error) {
|
|
err = s.q.GetMessageCount.Get(&res, r)
|
|
return
|
|
}
|
|
|
|
func newMessageService(db *sqlx.DB) core.MessageService {
|
|
return &messageSrv{
|
|
sqlxSrv: newSqlxSrv(db),
|
|
q: mustBuild(db, cc.BuildMessage),
|
|
}
|
|
}
|