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.
216 lines
8.4 KiB
216 lines
8.4 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 mgo
|
|
|
|
import (
|
|
"context"
|
|
"github.com/openimsdk/open-im-server/v3/pkg/common/storage/database"
|
|
"github.com/openimsdk/open-im-server/v3/pkg/common/storage/model"
|
|
"time"
|
|
|
|
"github.com/openimsdk/protocol/constant"
|
|
"github.com/openimsdk/tools/db/mongoutil"
|
|
"github.com/openimsdk/tools/db/pagination"
|
|
"github.com/openimsdk/tools/errs"
|
|
"go.mongodb.org/mongo-driver/bson"
|
|
"go.mongodb.org/mongo-driver/mongo"
|
|
"go.mongodb.org/mongo-driver/mongo/options"
|
|
)
|
|
|
|
func NewConversationMongo(db *mongo.Database) (*ConversationMgo, error) {
|
|
coll := db.Collection(database.ConversationName)
|
|
_, err := coll.Indexes().CreateOne(context.Background(), mongo.IndexModel{
|
|
Keys: bson.D{
|
|
{Key: "owner_user_id", Value: 1},
|
|
{Key: "conversation_id", Value: 1},
|
|
},
|
|
Options: options.Index().SetUnique(true),
|
|
})
|
|
if err != nil {
|
|
return nil, errs.Wrap(err)
|
|
}
|
|
version, err := NewVersionLog(db.Collection(database.ConversationVersionName))
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return &ConversationMgo{version: version, coll: coll}, nil
|
|
}
|
|
|
|
type ConversationMgo struct {
|
|
version database.VersionLog
|
|
coll *mongo.Collection
|
|
}
|
|
|
|
func (c *ConversationMgo) Create(ctx context.Context, conversations []*model.Conversation) (err error) {
|
|
return mongoutil.IncrVersion(func() error {
|
|
return mongoutil.InsertMany(ctx, c.coll, conversations)
|
|
}, func() error {
|
|
userConversation := make(map[string][]string)
|
|
for _, conversation := range conversations {
|
|
userConversation[conversation.OwnerUserID] = append(userConversation[conversation.OwnerUserID], conversation.ConversationID)
|
|
}
|
|
for userID, conversationIDs := range userConversation {
|
|
if err := c.version.IncrVersion(ctx, userID, conversationIDs, model.VersionStateInsert); err != nil {
|
|
return err
|
|
}
|
|
}
|
|
return nil
|
|
})
|
|
}
|
|
|
|
func (c *ConversationMgo) UpdateByMap(ctx context.Context, userIDs []string, conversationID string, args map[string]any) (int64, error) {
|
|
if len(args) == 0 || len(userIDs) == 0 {
|
|
return 0, nil
|
|
}
|
|
filter := bson.M{
|
|
"conversation_id": conversationID,
|
|
"owner_user_id": bson.M{"$in": userIDs},
|
|
}
|
|
var rows int64
|
|
err := mongoutil.IncrVersion(func() error {
|
|
res, err := mongoutil.UpdateMany(ctx, c.coll, filter, bson.M{"$set": args})
|
|
if err != nil {
|
|
return err
|
|
}
|
|
rows = res.ModifiedCount
|
|
return nil
|
|
}, func() error {
|
|
for _, userID := range userIDs {
|
|
if err := c.version.IncrVersion(ctx, userID, []string{conversationID}, model.VersionStateUpdate); err != nil {
|
|
return err
|
|
}
|
|
}
|
|
return nil
|
|
})
|
|
if err != nil {
|
|
return 0, err
|
|
}
|
|
return rows, nil
|
|
}
|
|
|
|
func (c *ConversationMgo) Update(ctx context.Context, conversation *model.Conversation) (err error) {
|
|
return mongoutil.IncrVersion(func() error {
|
|
return mongoutil.UpdateOne(ctx, c.coll, bson.M{"owner_user_id": conversation.OwnerUserID, "conversation_id": conversation.ConversationID}, bson.M{"$set": conversation}, true)
|
|
}, func() error {
|
|
return c.version.IncrVersion(ctx, conversation.OwnerUserID, []string{conversation.ConversationID}, model.VersionStateUpdate)
|
|
})
|
|
}
|
|
|
|
func (c *ConversationMgo) Find(ctx context.Context, ownerUserID string, conversationIDs []string) (conversations []*model.Conversation, err error) {
|
|
return mongoutil.Find[*model.Conversation](ctx, c.coll, bson.M{"owner_user_id": ownerUserID, "conversation_id": bson.M{"$in": conversationIDs}})
|
|
}
|
|
|
|
func (c *ConversationMgo) FindUserID(ctx context.Context, userIDs []string, conversationIDs []string) ([]string, error) {
|
|
return mongoutil.Find[string](
|
|
ctx,
|
|
c.coll,
|
|
bson.M{"owner_user_id": bson.M{"$in": userIDs}, "conversation_id": bson.M{"$in": conversationIDs}},
|
|
options.Find().SetProjection(bson.M{"_id": 0, "owner_user_id": 1}),
|
|
)
|
|
}
|
|
func (c *ConversationMgo) FindUserIDAllConversationID(ctx context.Context, userID string) ([]string, error) {
|
|
return mongoutil.Find[string](ctx, c.coll, bson.M{"owner_user_id": userID}, options.Find().SetProjection(bson.M{"_id": 0, "conversation_id": 1}))
|
|
}
|
|
|
|
func (c *ConversationMgo) Take(ctx context.Context, userID, conversationID string) (conversation *model.Conversation, err error) {
|
|
return mongoutil.FindOne[*model.Conversation](ctx, c.coll, bson.M{"owner_user_id": userID, "conversation_id": conversationID})
|
|
}
|
|
|
|
func (c *ConversationMgo) FindConversationID(ctx context.Context, userID string, conversationIDs []string) (existConversationID []string, err error) {
|
|
return mongoutil.Find[string](ctx, c.coll, bson.M{"owner_user_id": userID, "conversation_id": bson.M{"$in": conversationIDs}}, options.Find().SetProjection(bson.M{"_id": 0, "conversation_id": 1}))
|
|
}
|
|
|
|
func (c *ConversationMgo) FindUserIDAllConversations(ctx context.Context, userID string) (conversations []*model.Conversation, err error) {
|
|
return mongoutil.Find[*model.Conversation](ctx, c.coll, bson.M{"owner_user_id": userID})
|
|
}
|
|
|
|
func (c *ConversationMgo) FindRecvMsgUserIDs(ctx context.Context, conversationID string, recvOpts []int) ([]string, error) {
|
|
var filter any
|
|
if len(recvOpts) == 0 {
|
|
filter = bson.M{"conversation_id": conversationID}
|
|
} else {
|
|
filter = bson.M{"conversation_id": conversationID, "recv_msg_opt": bson.M{"$in": recvOpts}}
|
|
}
|
|
return mongoutil.Find[string](ctx, c.coll, filter, options.Find().SetProjection(bson.M{"_id": 0, "owner_user_id": 1}))
|
|
}
|
|
|
|
func (c *ConversationMgo) GetUserRecvMsgOpt(ctx context.Context, ownerUserID, conversationID string) (opt int, err error) {
|
|
return mongoutil.FindOne[int](ctx, c.coll, bson.M{"owner_user_id": ownerUserID, "conversation_id": conversationID}, options.FindOne().SetProjection(bson.M{"recv_msg_opt": 1}))
|
|
}
|
|
|
|
func (c *ConversationMgo) GetAllConversationIDs(ctx context.Context) ([]string, error) {
|
|
return mongoutil.Aggregate[string](ctx, c.coll, []bson.M{
|
|
{"$group": bson.M{"_id": "$conversation_id"}},
|
|
{"$project": bson.M{"_id": 0, "conversation_id": "$_id"}},
|
|
})
|
|
}
|
|
|
|
func (c *ConversationMgo) GetAllConversationIDsNumber(ctx context.Context) (int64, error) {
|
|
counts, err := mongoutil.Aggregate[int64](ctx, c.coll, []bson.M{
|
|
{"$group": bson.M{"_id": "$conversation_id"}},
|
|
{"$group": bson.M{"_id": nil, "count": bson.M{"$sum": 1}}},
|
|
{"$project": bson.M{"_id": 0}},
|
|
})
|
|
if err != nil {
|
|
return 0, err
|
|
}
|
|
if len(counts) == 0 {
|
|
return 0, nil
|
|
}
|
|
return counts[0], nil
|
|
}
|
|
|
|
func (c *ConversationMgo) PageConversationIDs(ctx context.Context, pagination pagination.Pagination) (conversationIDs []string, err error) {
|
|
return mongoutil.FindPageOnly[string](ctx, c.coll, bson.M{}, pagination, options.Find().SetProjection(bson.M{"conversation_id": 1}))
|
|
}
|
|
|
|
func (c *ConversationMgo) GetConversationsByConversationID(ctx context.Context, conversationIDs []string) ([]*model.Conversation, error) {
|
|
return mongoutil.Find[*model.Conversation](ctx, c.coll, bson.M{"conversation_id": bson.M{"$in": conversationIDs}})
|
|
}
|
|
|
|
func (c *ConversationMgo) GetConversationIDsNeedDestruct(ctx context.Context) ([]*model.Conversation, error) {
|
|
// "is_msg_destruct = 1 && msg_destruct_time != 0 && (UNIX_TIMESTAMP(NOW()) > (msg_destruct_time + UNIX_TIMESTAMP(latest_msg_destruct_time)) || latest_msg_destruct_time is NULL)"
|
|
return mongoutil.Find[*model.Conversation](ctx, c.coll, bson.M{
|
|
"is_msg_destruct": 1,
|
|
"msg_destruct_time": bson.M{"$ne": 0},
|
|
"$or": []bson.M{
|
|
{
|
|
"$expr": bson.M{
|
|
"$gt": []any{
|
|
time.Now(),
|
|
bson.M{"$add": []any{"$msg_destruct_time", "$latest_msg_destruct_time"}},
|
|
},
|
|
},
|
|
},
|
|
{
|
|
"latest_msg_destruct_time": nil,
|
|
},
|
|
},
|
|
})
|
|
}
|
|
|
|
func (c *ConversationMgo) GetConversationNotReceiveMessageUserIDs(ctx context.Context, conversationID string) ([]string, error) {
|
|
return mongoutil.Find[string](
|
|
ctx,
|
|
c.coll,
|
|
bson.M{"conversation_id": conversationID, "recv_msg_opt": bson.M{"$ne": constant.ReceiveMessage}},
|
|
options.Find().SetProjection(bson.M{"_id": 0, "owner_user_id": 1}),
|
|
)
|
|
}
|
|
|
|
func (c *ConversationMgo) FindConversationUserVersion(ctx context.Context, userID string, version uint, limit int) (*model.VersionLog, error) {
|
|
return c.version.FindChangeLog(ctx, userID, version, limit)
|
|
}
|