@ -1,32 +1,42 @@
package controller
import (
"Open_IM/pkg/common/constant"
"Open_IM/pkg/common/db/cache"
"Open_IM/pkg/common/db/relation"
relationTb "Open_IM/pkg/common/db/table/relation"
"Open_IM/pkg/utils"
"context"
"encoding/json"
"gorm.io/gorm"
)
type ConversationInterface interface {
//GetUserIDExistConversation 获取拥有该会话的的用户ID列表
GetUserIDExistConversation ( ctx context . Context , userIDList [ ] string , conversationID string ) ( [ ] string , error )
//UpdateUserConversationFiled 更新用户该会话的属性信息
UpdateUsersConversationFiled ( ctx context . Context , U serIDList [ ] string , conversationID string , args map [ string ] interface { } ) error
UpdateUsersConversationFiled ( ctx context . Context , u serIDList [ ] string , conversationID string , args map [ string ] interface { } ) error
//CreateConversation 创建一批新的会话
CreateConversation ( ctx context . Context , conversations [ ] * relationTb . ConversationModel ) error
//SyncPeerUserPrivateConversation 同步对端私聊会话内部保证事务操作
SyncPeerUserPrivateConversationTx ( ctx context . Context , conversation * relationTb . ConversationModel ) error
//FindConversations 根据会话ID获取某个用户的多个会话
FindConversations ( ctx context . Context , ownerUserID string , conversationID [ ] string ) ( [ ] * relationTb . ConversationModel , error )
FindConversations ( ctx context . Context , ownerUserID string , conversationID s [ ] string ) ( [ ] * relationTb . ConversationModel , error )
//GetUserAllConversation 获取一个用户在服务器上所有的会话
GetUserAllConversation ( ctx context . Context , ownerUserID string ) ( [ ] * relationTb . ConversationModel , error )
//SetUserConversations 设置用户多个会话属性，如果会话不存在则创建，否则更新,内部保证原子性
SetUserConversations ( ctx context . Context , ownerUserID string , conversations [ ] * relationTb . ConversationModel ) error
//SetUsersConversationFiledTx 设置多个用户会话关于某个字段的更新操作，如果会话不存在则创建，否则更新，内部保证事务操作
SetUsersConversationFiledTx ( ctx context . Context , userIDList [ ] string , conversation * relationTb . ConversationModel , filedMap map [ string ] interface { } ) error
}
type ConversationController struct {
database ConversationDataBaseInterface
}
func ( c * ConversationController ) SetUsersConversationFiledTx ( ctx context . Context , userIDList [ ] string , conversation * relationTb . ConversationModel , filedMap map [ string ] interface { } ) error {
return c . database . SetUsersConversationFiledTx ( ctx , userIDList , conversation , filedMap )
}
func NewConversationController ( database ConversationDataBaseInterface ) * ConversationController {
return & ConversationController { database : database }
}
@ -36,26 +46,26 @@ func (c *ConversationController) GetUserIDExistConversation(ctx context.Context,
}
func ( c ConversationController ) UpdateUsersConversationFiled ( ctx context . Context , UserIDList [ ] string , conversationID string , args map [ string ] interface { } ) error {
panic ( "implement me" )
return c . database . UpdateUsersConversationFiled ( ctx , UserIDList , conversationID , args )
}
func ( c ConversationController ) CreateConversation ( ctx context . Context , conversations [ ] * relationTb . ConversationModel ) error {
panic ( "implement me" )
return c . database . CreateConversation ( ctx , conversations )
}
func ( c ConversationController ) SyncPeerUserPrivateConversationTx ( ctx context . Context , conversation * relationTb . ConversationModel ) error {
panic ( "implement me" )
return c . database . SyncPeerUserPrivateConversationTx ( ctx , conversation )
}
func ( c ConversationController ) FindConversations ( ctx context . Context , ownerUserID string , conversationID [ ] string ) ( [ ] * relationTb . ConversationModel , error ) {
panic ( "implement me" )
func ( c ConversationController ) FindConversations ( ctx context . Context , ownerUserID string , conversationID s [ ] string ) ( [ ] * relationTb . ConversationModel , error ) {
return c . database . FindConversations ( ctx , ownerUserID , conversationIDs )
}
func ( c ConversationController ) GetUserAllConversation ( ctx context . Context , ownerUserID string ) ( [ ] * relationTb . ConversationModel , error ) {
panic ( "implement me" )
return c . database . GetUserAllConversation ( ctx , ownerUserID )
}
func ( c ConversationController ) SetUserConversations ( ctx context . Context , ownerUserID string , conversations [ ] * relationTb . ConversationModel ) error {
panic ( "implement me" )
return c . database . SetUserConversations ( ctx , ownerUserID , conversations )
}
var _ ConversationInterface = ( * ConversationController ) ( nil )
@ -70,19 +80,69 @@ type ConversationDataBaseInterface interface {
//SyncPeerUserPrivateConversation 同步对端私聊会话内部保证事务操作
SyncPeerUserPrivateConversationTx ( ctx context . Context , conversation * relationTb . ConversationModel ) error
//FindConversations 根据会话ID获取某个用户的多个会话
FindConversations ( ctx context . Context , ownerUserID string , conversationID [ ] string ) ( [ ] * relationTb . ConversationModel , error )
FindConversations ( ctx context . Context , ownerUserID string , conversationID s [ ] string ) ( [ ] * relationTb . ConversationModel , error )
//GetUserAllConversation 获取一个用户在服务器上所有的会话
GetUserAllConversation ( ctx context . Context , ownerUserID string ) ( [ ] * relationTb . ConversationModel , error )
//SetUserConversations 设置用户多个会话属性，如果会话不存在则创建，否则更新,内部保证原子性
SetUserConversations ( ctx context . Context , ownerUserID string , conversations [ ] * relationTb . ConversationModel ) error
//SetUsersConversationFiledTx 设置多个用户会话关于某个字段的更新操作，如果会话不存在则创建，否则更新，内部保证事务操作
SetUsersConversationFiledTx ( ctx context . Context , userIDList [ ] string , conversation * relationTb . ConversationModel , filedMap map [ string ] interface { } ) error
}
var _ ConversationDataBaseInterface = ( * ConversationDataBase ) ( nil )
type ConversationDataBase struct {
db relation . Conversation
cache cache . ConversationCache
conversationDB relation . Conversation
cache cache . ConversationCache
db * gorm . DB
}
func ( c ConversationDataBase ) SetUsersConversationFiledTx ( ctx context . Context , userIDList [ ] string , conversation * relationTb . ConversationModel , filedMap map [ string ] interface { } ) error {
return c . db . Transaction ( func ( tx * gorm . DB ) error {
haveUserID , err := c . conversationDB . FindUserID ( ctx , userIDList , conversation . ConversationID , tx )
if err != nil {
return err
}
if len ( haveUserID ) > 0 {
err = c . conversationDB . UpdateByMap ( ctx , haveUserID , conversation . ConversationID , filedMap , tx )
if err != nil {
return err
}
}
NotUserID := utils . DifferenceString ( haveUserID , userIDList )
var cList [ ] * relationTb . ConversationModel
for _ , v := range NotUserID {
temp := new ( relationTb . ConversationModel )
if err := utils . CopyStructFields ( temp , conversation ) ; err != nil {
return err
}
temp . OwnerUserID = v
cList = append ( cList , temp )
}
err = c . conversationDB . Create ( ctx , cList )
if err != nil {
return err
}
if len ( NotUserID ) > 0 {
err = c . cache . DelUsersConversationIDs ( ctx , NotUserID )
if err != nil {
return err
}
}
err = c . cache . DelUsersConversation ( ctx , haveUserID , conversation . ConversationID )
if err != nil {
return err
}
return nil
} )
}
func NewConversationDataBase ( db relation . Conversation , cache cache . ConversationCache ) * ConversationDataBase {
return & ConversationDataBase { conversationDB : db , cache : cache }
}
func ( c ConversationDataBase ) GetUserIDExistConversation ( ctx context . Context , userIDList [ ] string , conversationID string ) ( [ ] string , error ) {
panic ( "implement me" )
}
func ( c ConversationDataBase ) UpdateUsersConversationFiled ( ctx context . Context , UserIDList [ ] string , conversationID string , args map [ string ] interface { } ) error {
@ -94,26 +154,155 @@ func (c ConversationDataBase) CreateConversation(ctx context.Context, conversati
}
func ( c ConversationDataBase ) SyncPeerUserPrivateConversationTx ( ctx context . Context , conversation * relationTb . ConversationModel ) error {
panic ( "implement me" )
return c . db . Transaction ( func ( tx * gorm . DB ) error {
userIDList := [ ] string { conversation . OwnerUserID , conversation . UserID }
haveUserID , err := c . conversationDB . FindUserID ( ctx , userIDList , conversation . ConversationID , tx )
if err != nil {
return err
}
filedMap := map [ string ] interface { } { "is_private_chat" : conversation . IsPrivateChat }
if len ( haveUserID ) > 0 {
err = c . conversationDB . UpdateByMap ( ctx , haveUserID , conversation . ConversationID , filedMap , tx )
if err != nil {
return err
}
}
NotUserID := utils . DifferenceString ( haveUserID , userIDList )
var cList [ ] * relationTb . ConversationModel
for _ , v := range NotUserID {
temp := new ( relationTb . ConversationModel )
if v == conversation . UserID {
temp . OwnerUserID = conversation . UserID
temp . ConversationID = utils . GetConversationIDBySessionType ( conversation . OwnerUserID , constant . SingleChatType )
temp . ConversationType = constant . SingleChatType
temp . UserID = conversation . OwnerUserID
temp . IsPrivateChat = conversation . IsPrivateChat
} else {
if err := utils . CopyStructFields ( temp , conversation ) ; err != nil {
return err
}
temp . OwnerUserID = v
}
cList = append ( cList , temp )
}
if len ( NotUserID ) > 0 {
err = c . conversationDB . Create ( ctx , cList )
if err != nil {
return err
}
}
err = c . cache . DelUsersConversationIDs ( ctx , NotUserID )
if err != nil {
return err
}
err = c . cache . DelUsersConversation ( ctx , haveUserID , conversation . ConversationID )
if err != nil {
return err
}
return nil
} )
}
func ( c ConversationDataBase ) FindConversations ( ctx context . Context , ownerUserID string , conversationID [ ] string ) ( [ ] * relationTb . ConversationModel , error ) {
panic ( "implement me" )
func ( c ConversationDataBase ) FindConversations ( ctx context . Context , ownerUserID string , conversationIDs [ ] string ) ( [ ] * relationTb . ConversationModel , error ) {
getConversation := func ( ) ( string , error ) {
conversationList , err := c . conversationDB . Find ( ctx , ownerUserID , conversationIDs )
if err != nil {
return "" , utils . Wrap ( err , "get failed" )
}
bytes , err := json . Marshal ( conversationList )
if err != nil {
return "" , utils . Wrap ( err , "Marshal failed" )
}
return string ( bytes ) , nil
}
return c . cache . GetConversations ( ctx , ownerUserID , conversationIDs , getConversation )
}
func ( c ConversationDataBase ) GetConversation ( ctx context . Context , ownerUserID string , conversationID string ) ( * relationTb . ConversationModel , error ) {
getConversation := func ( ) ( string , error ) {
conversationList , err := c . conversationDB . Take ( ctx , ownerUserID , conversationID )
if err != nil {
return "" , utils . Wrap ( err , "get failed" )
}
bytes , err := json . Marshal ( conversationList )
if err != nil {
return "" , utils . Wrap ( err , "Marshal failed" )
}
return string ( bytes ) , nil
}
return c . cache . GetConversation ( ctx , ownerUserID , conversationID , getConversation )
}
func ( c ConversationDataBase ) GetUserAllConversation ( ctx context . Context , ownerUserID string ) ( [ ] * relationTb . ConversationModel , error ) {
panic ( "implement me" )
getConversationIDList := func ( ) ( string , error ) {
conversationIDList , err := c . conversationDB . FindUserIDAllConversationID ( ctx , ownerUserID )
if err != nil {
return "" , utils . Wrap ( err , "getConversationIDList failed" )
}
bytes , err := json . Marshal ( conversationIDList )
if err != nil {
return "" , utils . Wrap ( err , "" )
}
return string ( bytes ) , nil
}
conversationIDList , err := c . cache . GetUserConversationIDs ( ctx , ownerUserID , getConversationIDList )
if err != nil {
return nil , err
}
var conversations [ ] * relationTb . ConversationModel
for _ , conversationID := range conversationIDList {
conversation , tErr := c . GetConversation ( ctx , ownerUserID , conversationID )
if tErr != nil {
return nil , utils . Wrap ( tErr , "GetConversation failed" )
}
conversations = append ( conversations , conversation )
}
return conversations , nil
}
func ( c ConversationDataBase ) SetUserConversations ( ctx context . Context , ownerUserID string , conversations [ ] * relationTb . ConversationModel ) error {
panic ( "implement me" )
}
return c . db . Transaction ( func ( tx * gorm . DB ) error {
var conversationIDList [ ] string
for _ , conversation := range conversations {
conversationIDList = append ( conversationIDList , conversation . ConversationID )
}
haveConversations , err := c . conversationDB . Find ( ctx , ownerUserID , conversationIDList , tx )
if err != nil {
return err
}
if len ( haveConversations ) > 0 {
err = c . conversationDB . Update ( ctx , conversations , tx )
if err != nil {
return err
}
}
var haveConversationID [ ] string
for _ , conversation := range haveConversations {
haveConversationID = append ( haveConversationID , conversation . ConversationID )
}
func NewConversationDataBase ( db relation . Conversation , cache cache . ConversationCache ) * ConversationDataBase {
return & ConversationDataBase { db : db , cache : cache }
NotConversationID := utils . DifferenceString ( haveConversationID , conversationIDList )
var NotConversations [ ] * relationTb . ConversationModel
for _ , conversation := range conversations {
if ! utils . IsContain ( conversation . ConversationID , haveConversationID ) {
NotConversations = append ( NotConversations , conversation )
}
}
if len ( NotConversations ) > 0 {
err = c . conversationDB . Create ( ctx , NotConversations )
if err != nil {
return err
}
}
err = c . cache . DelUsersConversationIDs ( ctx , NotConversationID )
if err != nil {
return err
}
err = c . cache . DelUserConversations ( ctx , ownerUserID , haveConversationID )
if err != nil {
return err
}
return nil
} )
}
//func NewConversationController(db *gorm.DB, rdb redis.UniversalClient) ConversationInterface {
// groupController := &ConversationController{database: newGroupDatabase(db, rdb, mgoClient)}
// return groupController
//}
Gordon
2 years ago