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.
63 lines
2.1 KiB
63 lines
2.1 KiB
// Copyright © 2023 OpenIM. All rights reserved.
|
|
//
|
|
// Licensed under the Apache License, Version 2.0 (the "License");
|
|
// you may not use this file except in compliance with the License.
|
|
// You may obtain a copy of the License at
|
|
//
|
|
// http://www.apache.org/licenses/LICENSE-2.0
|
|
//
|
|
// Unless required by applicable law or agreed to in writing, software
|
|
// distributed under the License is distributed on an "AS IS" BASIS,
|
|
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
// See the License for the specific language governing permissions and
|
|
// limitations under the License.
|
|
|
|
package relation
|
|
|
|
import (
|
|
"github.com/golang/protobuf/jsonpb"
|
|
"github.com/jinzhu/copier"
|
|
"google.golang.org/protobuf/proto"
|
|
"gorm.io/gorm"
|
|
|
|
"github.com/OpenIMSDK/Open-IM-Server/pkg/common/constant"
|
|
"github.com/OpenIMSDK/Open-IM-Server/pkg/common/db/table/relation"
|
|
pbMsg "github.com/OpenIMSDK/Open-IM-Server/pkg/proto/msg"
|
|
sdkws "github.com/OpenIMSDK/Open-IM-Server/pkg/proto/sdkws"
|
|
"github.com/OpenIMSDK/Open-IM-Server/pkg/utils"
|
|
)
|
|
|
|
type ChatLogGorm struct {
|
|
*MetaDB
|
|
}
|
|
|
|
func NewChatLogGorm(db *gorm.DB) relation.ChatLogModelInterface {
|
|
return &ChatLogGorm{NewMetaDB(db, &relation.ChatLogModel{})}
|
|
}
|
|
|
|
func (c *ChatLogGorm) Create(msg *pbMsg.MsgDataToMQ) error {
|
|
chatLog := new(relation.ChatLogModel)
|
|
copier.Copy(chatLog, msg.MsgData)
|
|
switch msg.MsgData.SessionType {
|
|
case constant.GroupChatType, constant.SuperGroupChatType:
|
|
chatLog.RecvID = msg.MsgData.GroupID
|
|
case constant.SingleChatType:
|
|
chatLog.RecvID = msg.MsgData.RecvID
|
|
}
|
|
if msg.MsgData.ContentType >= constant.NotificationBegin && msg.MsgData.ContentType <= constant.NotificationEnd {
|
|
var tips sdkws.TipsComm
|
|
_ = proto.Unmarshal(msg.MsgData.Content, &tips)
|
|
marshaler := jsonpb.Marshaler{
|
|
OrigName: true,
|
|
EnumsAsInts: false,
|
|
EmitDefaults: false,
|
|
}
|
|
chatLog.Content, _ = marshaler.MarshalToString(&tips)
|
|
} else {
|
|
chatLog.Content = string(msg.MsgData.Content)
|
|
}
|
|
chatLog.CreateTime = utils.UnixMillSecondToTime(msg.MsgData.CreateTime)
|
|
chatLog.SendTime = utils.UnixMillSecondToTime(msg.MsgData.SendTime)
|
|
return c.DB.Create(chatLog).Error
|
|
}
|