withchao
2 years ago
parent
df7bfc67b6
commit
768e6bc758
@ -1,119 +0,0 @@
|
|||||||
// 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 msgtransfer
|
|
||||||
|
|
||||||
import (
|
|
||||||
"context"
|
|
||||||
|
|
||||||
"github.com/OpenIMSDK/protocol/constant"
|
|
||||||
pbmsg "github.com/OpenIMSDK/protocol/msg"
|
|
||||||
"github.com/OpenIMSDK/tools/log"
|
|
||||||
"github.com/OpenIMSDK/tools/utils"
|
|
||||||
|
|
||||||
"github.com/openimsdk/open-im-server/v3/pkg/common/config"
|
|
||||||
"github.com/openimsdk/open-im-server/v3/pkg/common/db/controller"
|
|
||||||
kfk "github.com/openimsdk/open-im-server/v3/pkg/common/kafka"
|
|
||||||
|
|
||||||
"github.com/IBM/sarama"
|
|
||||||
"google.golang.org/protobuf/proto"
|
|
||||||
)
|
|
||||||
|
|
||||||
type PersistentConsumerHandler struct {
|
|
||||||
persistentConsumerGroup *kfk.MConsumerGroup
|
|
||||||
chatLogDatabase controller.ChatLogDatabase
|
|
||||||
}
|
|
||||||
|
|
||||||
func NewPersistentConsumerHandler(database controller.ChatLogDatabase) *PersistentConsumerHandler {
|
|
||||||
return &PersistentConsumerHandler{
|
|
||||||
persistentConsumerGroup: kfk.NewMConsumerGroup(&kfk.MConsumerGroupConfig{
|
|
||||||
KafkaVersion: sarama.V2_0_0_0,
|
|
||||||
OffsetsInitial: sarama.OffsetNewest, IsReturnErr: false,
|
|
||||||
}, []string{config.Config.Kafka.LatestMsgToRedis.Topic},
|
|
||||||
config.Config.Kafka.Addr, config.Config.Kafka.ConsumerGroupID.MsgToMySql),
|
|
||||||
chatLogDatabase: database,
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
func (pc *PersistentConsumerHandler) handleChatWs2Mysql(
|
|
||||||
ctx context.Context,
|
|
||||||
cMsg *sarama.ConsumerMessage,
|
|
||||||
msgKey string,
|
|
||||||
_ sarama.ConsumerGroupSession,
|
|
||||||
) {
|
|
||||||
msg := cMsg.Value
|
|
||||||
var tag bool
|
|
||||||
msgFromMQ := pbmsg.MsgDataToMQ{}
|
|
||||||
err := proto.Unmarshal(msg, &msgFromMQ)
|
|
||||||
if err != nil {
|
|
||||||
log.ZError(ctx, "msg_transfer Unmarshal msg err", err)
|
|
||||||
return
|
|
||||||
}
|
|
||||||
|
|
||||||
log.ZDebug(ctx, "handleChatWs2Mysql", "msg", msgFromMQ.MsgData)
|
|
||||||
// Control whether to store history messages (mysql)
|
|
||||||
isPersist := utils.GetSwitchFromOptions(msgFromMQ.MsgData.Options, constant.IsPersistent)
|
|
||||||
// Only process receiver data
|
|
||||||
if isPersist {
|
|
||||||
switch msgFromMQ.MsgData.SessionType {
|
|
||||||
case constant.SingleChatType, constant.NotificationChatType:
|
|
||||||
if msgKey == msgFromMQ.MsgData.RecvID {
|
|
||||||
tag = true
|
|
||||||
}
|
|
||||||
case constant.GroupChatType:
|
|
||||||
if msgKey == msgFromMQ.MsgData.SendID {
|
|
||||||
tag = true
|
|
||||||
}
|
|
||||||
case constant.SuperGroupChatType:
|
|
||||||
tag = true
|
|
||||||
}
|
|
||||||
if tag {
|
|
||||||
log.ZInfo(ctx, "msg_transfer msg persisting", "msg", string(msg))
|
|
||||||
if err = pc.chatLogDatabase.CreateChatLog(&msgFromMQ); err != nil {
|
|
||||||
log.ZError(ctx, "Message insert failed", err, "msg", msgFromMQ.String())
|
|
||||||
return
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
func (PersistentConsumerHandler) Setup(_ sarama.ConsumerGroupSession) error { return nil }
|
|
||||||
func (PersistentConsumerHandler) Cleanup(_ sarama.ConsumerGroupSession) error { return nil }
|
|
||||||
|
|
||||||
func (pc *PersistentConsumerHandler) ConsumeClaim(
|
|
||||||
sess sarama.ConsumerGroupSession,
|
|
||||||
claim sarama.ConsumerGroupClaim,
|
|
||||||
) error {
|
|
||||||
for msg := range claim.Messages() {
|
|
||||||
ctx := pc.persistentConsumerGroup.GetContextFromMsg(msg)
|
|
||||||
log.ZDebug(
|
|
||||||
ctx,
|
|
||||||
"kafka get info to mysql",
|
|
||||||
"msgTopic",
|
|
||||||
msg.Topic,
|
|
||||||
"msgPartition",
|
|
||||||
msg.Partition,
|
|
||||||
"msg",
|
|
||||||
string(msg.Value),
|
|
||||||
"key",
|
|
||||||
string(msg.Key),
|
|
||||||
)
|
|
||||||
if len(msg.Value) != 0 {
|
|
||||||
pc.handleChatWs2Mysql(ctx, msg, string(msg.Key), sess)
|
|
||||||
} else {
|
|
||||||
log.ZError(ctx, "msg get from kafka but is nil", nil, "key", msg.Key)
|
|
||||||
}
|
|
||||||
sess.MarkMessage(msg, "")
|
|
||||||
}
|
|
||||||
return nil
|
|
||||||
}
|
|
||||||
@ -1,37 +0,0 @@
|
|||||||
// 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 controller
|
|
||||||
|
|
||||||
import (
|
|
||||||
pbmsg "github.com/OpenIMSDK/protocol/msg"
|
|
||||||
|
|
||||||
relationtb "github.com/openimsdk/open-im-server/v3/pkg/common/db/table/relation"
|
|
||||||
)
|
|
||||||
|
|
||||||
type ChatLogDatabase interface {
|
|
||||||
CreateChatLog(msg *pbmsg.MsgDataToMQ) error
|
|
||||||
}
|
|
||||||
|
|
||||||
func NewChatLogDatabase(chatLogModelInterface relationtb.ChatLogModelInterface) ChatLogDatabase {
|
|
||||||
return &chatLogDatabase{chatLogModel: chatLogModelInterface}
|
|
||||||
}
|
|
||||||
|
|
||||||
type chatLogDatabase struct {
|
|
||||||
chatLogModel relationtb.ChatLogModelInterface
|
|
||||||
}
|
|
||||||
|
|
||||||
func (c *chatLogDatabase) CreateChatLog(msg *pbmsg.MsgDataToMQ) error {
|
|
||||||
return c.chatLogModel.Create(msg)
|
|
||||||
}
|
|
||||||
@ -0,0 +1,91 @@
|
|||||||
|
package mgo
|
||||||
|
|
||||||
|
import (
|
||||||
|
"context"
|
||||||
|
|
||||||
|
"github.com/OpenIMSDK/tools/mgoutil"
|
||||||
|
"github.com/OpenIMSDK/tools/pagination"
|
||||||
|
"go.mongodb.org/mongo-driver/bson"
|
||||||
|
"go.mongodb.org/mongo-driver/mongo"
|
||||||
|
"go.mongodb.org/mongo-driver/mongo/options"
|
||||||
|
|
||||||
|
"github.com/openimsdk/open-im-server/v3/pkg/common/db/table/relation"
|
||||||
|
)
|
||||||
|
|
||||||
|
func NewBlackMongo(db *mongo.Database) (relation.BlackModelInterface, error) {
|
||||||
|
coll := db.Collection("black")
|
||||||
|
_, err := coll.Indexes().CreateOne(context.Background(), mongo.IndexModel{
|
||||||
|
Keys: bson.D{
|
||||||
|
{Key: "owner_user_id", Value: 1},
|
||||||
|
{Key: "block_user_id", Value: 1},
|
||||||
|
},
|
||||||
|
Options: options.Index().SetUnique(true),
|
||||||
|
})
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
return &BlackMgo{coll: coll}, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
type BlackMgo struct {
|
||||||
|
coll *mongo.Collection
|
||||||
|
}
|
||||||
|
|
||||||
|
func (b *BlackMgo) blackFilter(ownerUserID, blockUserID string) bson.M {
|
||||||
|
return bson.M{
|
||||||
|
"owner_user_id": ownerUserID,
|
||||||
|
"block_user_id": blockUserID,
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func (b *BlackMgo) blacksFilter(blacks []*relation.BlackModel) bson.M {
|
||||||
|
if len(blacks) == 0 {
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
or := make(bson.A, 0, len(blacks))
|
||||||
|
for _, black := range blacks {
|
||||||
|
or = append(or, b.blackFilter(black.OwnerUserID, black.BlockUserID))
|
||||||
|
}
|
||||||
|
return bson.M{"$or": or}
|
||||||
|
}
|
||||||
|
|
||||||
|
func (b *BlackMgo) Create(ctx context.Context, blacks []*relation.BlackModel) (err error) {
|
||||||
|
return mgoutil.InsertMany(ctx, b.coll, blacks)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (b *BlackMgo) Delete(ctx context.Context, blacks []*relation.BlackModel) (err error) {
|
||||||
|
if len(blacks) == 0 {
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
return mgoutil.DeleteMany(ctx, b.coll, b.blacksFilter(blacks))
|
||||||
|
}
|
||||||
|
|
||||||
|
func (b *BlackMgo) UpdateByMap(ctx context.Context, ownerUserID, blockUserID string, args map[string]any) (err error) {
|
||||||
|
if len(args) == 0 {
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
return mgoutil.UpdateOne(ctx, b.coll, b.blackFilter(ownerUserID, blockUserID), bson.M{"$set": args}, false)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (b *BlackMgo) Find(ctx context.Context, blacks []*relation.BlackModel) (blackList []*relation.BlackModel, err error) {
|
||||||
|
return mgoutil.Find[*relation.BlackModel](ctx, b.coll, b.blacksFilter(blacks))
|
||||||
|
}
|
||||||
|
|
||||||
|
func (b *BlackMgo) Take(ctx context.Context, ownerUserID, blockUserID string) (black *relation.BlackModel, err error) {
|
||||||
|
return mgoutil.FindOne[*relation.BlackModel](ctx, b.coll, b.blackFilter(ownerUserID, blockUserID))
|
||||||
|
}
|
||||||
|
|
||||||
|
func (b *BlackMgo) FindOwnerBlacks(ctx context.Context, ownerUserID string, pagination pagination.Pagination) (total int64, blacks []*relation.BlackModel, err error) {
|
||||||
|
return mgoutil.FindPage[*relation.BlackModel](ctx, b.coll, bson.M{"owner_user_id": ownerUserID}, pagination)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (b *BlackMgo) FindOwnerBlackInfos(ctx context.Context, ownerUserID string, userIDs []string) (blacks []*relation.BlackModel, err error) {
|
||||||
|
if len(userIDs) == 0 {
|
||||||
|
return mgoutil.Find[*relation.BlackModel](ctx, b.coll, bson.M{"owner_user_id": ownerUserID})
|
||||||
|
}
|
||||||
|
return mgoutil.Find[*relation.BlackModel](ctx, b.coll, bson.M{"owner_user_id": ownerUserID, "block_user_id": bson.M{"$in": userIDs}})
|
||||||
|
}
|
||||||
|
|
||||||
|
func (b *BlackMgo) FindBlackUserIDs(ctx context.Context, ownerUserID string) (blackUserIDs []string, err error) {
|
||||||
|
return mgoutil.Find[string](ctx, b.coll, bson.M{"owner_user_id": ownerUserID}, options.Find().SetProjection(bson.M{"_id": 0, "block_user_id": 1}))
|
||||||
|
}
|
||||||
@ -0,0 +1,99 @@
|
|||||||
|
package mgo
|
||||||
|
|
||||||
|
import (
|
||||||
|
"context"
|
||||||
|
|
||||||
|
"github.com/OpenIMSDK/tools/mgoutil"
|
||||||
|
"github.com/OpenIMSDK/tools/pagination"
|
||||||
|
"go.mongodb.org/mongo-driver/mongo/options"
|
||||||
|
|
||||||
|
"go.mongodb.org/mongo-driver/bson"
|
||||||
|
"go.mongodb.org/mongo-driver/mongo"
|
||||||
|
|
||||||
|
"github.com/openimsdk/open-im-server/v3/pkg/common/db/table/relation"
|
||||||
|
)
|
||||||
|
|
||||||
|
func NewFriendRequestMongo(db *mongo.Database) (relation.FriendRequestModelInterface, error) {
|
||||||
|
coll := db.Collection("friend_request")
|
||||||
|
_, err := coll.Indexes().CreateOne(context.Background(), mongo.IndexModel{
|
||||||
|
Keys: bson.D{
|
||||||
|
{Key: "from_user_id", Value: 1},
|
||||||
|
{Key: "to_user_id", Value: 1},
|
||||||
|
},
|
||||||
|
Options: options.Index().SetUnique(true),
|
||||||
|
})
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
return &FriendRequestMgo{coll: coll}, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
type FriendRequestMgo struct {
|
||||||
|
coll *mongo.Collection
|
||||||
|
}
|
||||||
|
|
||||||
|
func (f *FriendRequestMgo) FindToUserID(ctx context.Context, toUserID string, pagination pagination.Pagination) (total int64, friendRequests []*relation.FriendRequestModel, err error) {
|
||||||
|
return mgoutil.FindPage[*relation.FriendRequestModel](ctx, f.coll, bson.M{"to_user_id": toUserID}, pagination)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (f *FriendRequestMgo) FindFromUserID(ctx context.Context, fromUserID string, pagination pagination.Pagination) (total int64, friendRequests []*relation.FriendRequestModel, err error) {
|
||||||
|
return mgoutil.FindPage[*relation.FriendRequestModel](ctx, f.coll, bson.M{"from_user_id": fromUserID}, pagination)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (f *FriendRequestMgo) FindBothFriendRequests(ctx context.Context, fromUserID, toUserID string) (friends []*relation.FriendRequestModel, err error) {
|
||||||
|
filter := bson.M{"$or": []bson.M{
|
||||||
|
{"from_user_id": fromUserID, "to_user_id": toUserID},
|
||||||
|
{"from_user_id": toUserID, "to_user_id": fromUserID},
|
||||||
|
}}
|
||||||
|
return mgoutil.Find[*relation.FriendRequestModel](ctx, f.coll, filter)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (f *FriendRequestMgo) Create(ctx context.Context, friendRequests []*relation.FriendRequestModel) error {
|
||||||
|
return mgoutil.InsertMany(ctx, f.coll, friendRequests)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (f *FriendRequestMgo) Delete(ctx context.Context, fromUserID, toUserID string) (err error) {
|
||||||
|
return mgoutil.DeleteOne(ctx, f.coll, bson.M{"from_user_id": fromUserID, "to_user_id": toUserID})
|
||||||
|
}
|
||||||
|
|
||||||
|
func (f *FriendRequestMgo) UpdateByMap(ctx context.Context, formUserID, toUserID string, args map[string]any) (err error) {
|
||||||
|
if len(args) == 0 {
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
return mgoutil.UpdateOne(ctx, f.coll, bson.M{"from_user_id": formUserID, "to_user_id": toUserID}, bson.M{"$set": args}, true)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (f *FriendRequestMgo) Update(ctx context.Context, friendRequest *relation.FriendRequestModel) (err error) {
|
||||||
|
updater := bson.M{}
|
||||||
|
if friendRequest.HandleResult != 0 {
|
||||||
|
updater["handle_result"] = friendRequest.HandleResult
|
||||||
|
}
|
||||||
|
if friendRequest.ReqMsg != "" {
|
||||||
|
updater["req_msg"] = friendRequest.ReqMsg
|
||||||
|
}
|
||||||
|
if friendRequest.HandlerUserID != "" {
|
||||||
|
updater["handler_user_id"] = friendRequest.HandlerUserID
|
||||||
|
}
|
||||||
|
if friendRequest.HandleMsg != "" {
|
||||||
|
updater["handle_msg"] = friendRequest.HandleMsg
|
||||||
|
}
|
||||||
|
if !friendRequest.HandleTime.IsZero() {
|
||||||
|
updater["handle_time"] = friendRequest.HandleTime
|
||||||
|
}
|
||||||
|
if friendRequest.Ex != "" {
|
||||||
|
updater["ex"] = friendRequest.Ex
|
||||||
|
}
|
||||||
|
if len(updater) == 0 {
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
filter := bson.M{"from_user_id": friendRequest.FromUserID, "to_user_id": friendRequest.ToUserID}
|
||||||
|
return mgoutil.UpdateOne(ctx, f.coll, filter, bson.M{"$set": updater}, true)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (f *FriendRequestMgo) Find(ctx context.Context, fromUserID, toUserID string) (friendRequest *relation.FriendRequestModel, err error) {
|
||||||
|
return mgoutil.FindOne[*relation.FriendRequestModel](ctx, f.coll, bson.M{"from_user_id": fromUserID, "to_user_id": toUserID})
|
||||||
|
}
|
||||||
|
|
||||||
|
func (f *FriendRequestMgo) Take(ctx context.Context, fromUserID, toUserID string) (friendRequest *relation.FriendRequestModel, err error) {
|
||||||
|
return f.Find(ctx, fromUserID, toUserID)
|
||||||
|
}
|
||||||
@ -0,0 +1,70 @@
|
|||||||
|
package mgo
|
||||||
|
|
||||||
|
import (
|
||||||
|
"context"
|
||||||
|
"time"
|
||||||
|
|
||||||
|
"github.com/OpenIMSDK/tools/mgoutil"
|
||||||
|
"github.com/OpenIMSDK/tools/pagination"
|
||||||
|
"go.mongodb.org/mongo-driver/bson"
|
||||||
|
"go.mongodb.org/mongo-driver/mongo"
|
||||||
|
"go.mongodb.org/mongo-driver/mongo/options"
|
||||||
|
|
||||||
|
"github.com/openimsdk/open-im-server/v3/pkg/common/db/table/relation"
|
||||||
|
)
|
||||||
|
|
||||||
|
func NewLogMongo(db *mongo.Database) (relation.LogInterface, error) {
|
||||||
|
coll := db.Collection("log")
|
||||||
|
_, err := coll.Indexes().CreateMany(context.Background(), []mongo.IndexModel{
|
||||||
|
{
|
||||||
|
Keys: bson.D{
|
||||||
|
{Key: "log_id", Value: 1},
|
||||||
|
},
|
||||||
|
Options: options.Index().SetUnique(true),
|
||||||
|
},
|
||||||
|
{
|
||||||
|
Keys: bson.D{
|
||||||
|
{Key: "user_id", Value: 1},
|
||||||
|
},
|
||||||
|
},
|
||||||
|
{
|
||||||
|
Keys: bson.D{
|
||||||
|
{Key: "create_time", Value: -1},
|
||||||
|
},
|
||||||
|
},
|
||||||
|
})
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
return &LogMgo{coll: coll}, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
type LogMgo struct {
|
||||||
|
coll *mongo.Collection
|
||||||
|
}
|
||||||
|
|
||||||
|
func (l *LogMgo) Create(ctx context.Context, log []*relation.LogModel) error {
|
||||||
|
return mgoutil.InsertMany(ctx, l.coll, log)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (l *LogMgo) Search(ctx context.Context, keyword string, start time.Time, end time.Time, pagination pagination.Pagination) (int64, []*relation.LogModel, error) {
|
||||||
|
filter := bson.M{"create_time": bson.M{"$gte": start, "$lte": end}}
|
||||||
|
if keyword != "" {
|
||||||
|
filter["user_id"] = bson.M{"$regex": keyword}
|
||||||
|
}
|
||||||
|
return mgoutil.FindPage[*relation.LogModel](ctx, l.coll, filter, pagination, options.Find().SetSort(bson.M{"create_time": -1}))
|
||||||
|
}
|
||||||
|
|
||||||
|
func (l *LogMgo) Delete(ctx context.Context, logID []string, userID string) error {
|
||||||
|
if userID == "" {
|
||||||
|
return mgoutil.DeleteMany(ctx, l.coll, bson.M{"log_id": bson.M{"$in": logID}})
|
||||||
|
}
|
||||||
|
return mgoutil.DeleteMany(ctx, l.coll, bson.M{"log_id": bson.M{"$in": logID}, "user_id": userID})
|
||||||
|
}
|
||||||
|
|
||||||
|
func (l *LogMgo) Get(ctx context.Context, logIDs []string, userID string) ([]*relation.LogModel, error) {
|
||||||
|
if userID == "" {
|
||||||
|
return mgoutil.Find[*relation.LogModel](ctx, l.coll, bson.M{"log_id": bson.M{"$in": logIDs}})
|
||||||
|
}
|
||||||
|
return mgoutil.Find[*relation.LogModel](ctx, l.coll, bson.M{"log_id": bson.M{"$in": logIDs}, "user_id": userID})
|
||||||
|
}
|
||||||
@ -1,48 +0,0 @@
|
|||||||
package newmgo
|
|
||||||
|
|
||||||
import (
|
|
||||||
"context"
|
|
||||||
|
|
||||||
"github.com/openimsdk/open-im-server/v3/pkg/common/db/newmgo/mgotool"
|
|
||||||
"github.com/openimsdk/open-im-server/v3/pkg/common/db/table/relation"
|
|
||||||
"go.mongodb.org/mongo-driver/bson"
|
|
||||||
"go.mongodb.org/mongo-driver/mongo"
|
|
||||||
)
|
|
||||||
|
|
||||||
type FriendRequestMgo struct {
|
|
||||||
coll *mongo.Collection
|
|
||||||
}
|
|
||||||
|
|
||||||
func NewFriendRequestMongo(db *mongo.Database) (relation.FriendRequestModelInterface, error) {
|
|
||||||
return &FriendRequestMgo{
|
|
||||||
coll: db.Collection(relation.FriendRequestModelCollectionName),
|
|
||||||
}, nil
|
|
||||||
}
|
|
||||||
|
|
||||||
func (f *FriendRequestMgo) Create(ctx context.Context, friendRequests []*relation.FriendRequestModel) error {
|
|
||||||
return mgotool.InsertMany(ctx, f.coll, friendRequests)
|
|
||||||
}
|
|
||||||
|
|
||||||
func (f *FriendRequestMgo) Delete(ctx context.Context, fromUserID, toUserID string) (err error) {
|
|
||||||
return mgotool.Delete[*relation.FriendRequestModel](ctx, f.coll, bson.M{"from_user_id": fromUserID, "to_user_id": toUserID})
|
|
||||||
}
|
|
||||||
|
|
||||||
func (f *FriendRequestMgo) UpdateByMap(ctx context.Context, formUserID, toUserID string, args map[string]any) (err error) {
|
|
||||||
if len(args) == 0 {
|
|
||||||
return nil
|
|
||||||
}
|
|
||||||
return mgotool.UpdateOne(ctx, f.coll, bson.M{"from_user_id": formUserID, "to_user_id": toUserID}, bson.M{"$set": args}, true)
|
|
||||||
}
|
|
||||||
|
|
||||||
func (f *FriendRequestMgo) Update(ctx context.Context, friendRequest *relation.FriendRequestModel) (err error) {
|
|
||||||
return mgotool.UpdateOne(ctx, f.coll, bson.M{"_id": friendRequest.ID}, bson.M{"$set": friendRequest}, true)
|
|
||||||
}
|
|
||||||
|
|
||||||
func (f *FriendRequestMgo) Find(ctx context.Context, fromUserID, toUserID string) (friendRequest *relation.FriendRequestModel, err error) {
|
|
||||||
return mgotool.FindOne[*relation.FriendRequestModel](ctx, f.coll, bson.M{"from_user_id": fromUserID, "to_user_id": toUserID})
|
|
||||||
}
|
|
||||||
|
|
||||||
func (f *FriendRequestMgo) Take(ctx context.Context, fromUserID, toUserID string) (friendRequest *relation.FriendRequestModel, err error) {
|
|
||||||
return f.Find(ctx, fromUserID, toUserID)
|
|
||||||
}
|
|
||||||
|
|
||||||
@ -1,49 +0,0 @@
|
|||||||
package newmgo
|
|
||||||
|
|
||||||
import (
|
|
||||||
"context"
|
|
||||||
"github.com/openimsdk/open-im-server/v3/pkg/common/db/newmgo/mgotool"
|
|
||||||
"github.com/openimsdk/open-im-server/v3/pkg/common/db/table/relation"
|
|
||||||
"github.com/openimsdk/open-im-server/v3/pkg/common/pagination"
|
|
||||||
"go.mongodb.org/mongo-driver/bson"
|
|
||||||
"go.mongodb.org/mongo-driver/mongo"
|
|
||||||
"go.mongodb.org/mongo-driver/mongo/options"
|
|
||||||
"time"
|
|
||||||
)
|
|
||||||
|
|
||||||
func NewLogMongo(db *mongo.Database) (relation.LogInterface, error) {
|
|
||||||
lm := &LogMgo{
|
|
||||||
coll: db.Collection("log"),
|
|
||||||
}
|
|
||||||
return lm, nil
|
|
||||||
}
|
|
||||||
|
|
||||||
type LogMgo struct {
|
|
||||||
coll *mongo.Collection
|
|
||||||
}
|
|
||||||
|
|
||||||
func (l *LogMgo) Create(ctx context.Context, log []*relation.Log) error {
|
|
||||||
return mgotool.InsertMany(ctx, l.coll, log)
|
|
||||||
}
|
|
||||||
|
|
||||||
func (l *LogMgo) Search(ctx context.Context, keyword string, start time.Time, end time.Time, pagination pagination.Pagination) (int64, []*relation.Log, error) {
|
|
||||||
filter := bson.M{"create_time": bson.M{"$gte": start, "$lte": end}}
|
|
||||||
if keyword != "" {
|
|
||||||
filter["user_id"] = bson.M{"$regex": keyword}
|
|
||||||
}
|
|
||||||
return mgotool.FindPage[*relation.Log](ctx, l.coll, filter, pagination, options.Find().SetSort(bson.M{"create_time": -1}))
|
|
||||||
}
|
|
||||||
|
|
||||||
func (l *LogMgo) Delete(ctx context.Context, logID []string, userID string) error {
|
|
||||||
if userID == "" {
|
|
||||||
return mgotool.DeleteMany(ctx, l.coll, bson.M{"log_id": bson.M{"$in": logID}})
|
|
||||||
}
|
|
||||||
return mgotool.DeleteMany(ctx, l.coll, bson.M{"log_id": bson.M{"$in": logID}, "user_id": userID})
|
|
||||||
}
|
|
||||||
|
|
||||||
func (l *LogMgo) Get(ctx context.Context, logIDs []string, userID string) ([]*relation.Log, error) {
|
|
||||||
if userID == "" {
|
|
||||||
return mgotool.Find[*relation.Log](ctx, l.coll, bson.M{"log_id": bson.M{"$in": logIDs}})
|
|
||||||
}
|
|
||||||
return mgotool.Find[*relation.Log](ctx, l.coll, bson.M{"log_id": bson.M{"$in": logIDs}, "user_id": userID})
|
|
||||||
}
|
|
||||||
@ -1,198 +0,0 @@
|
|||||||
package mgotool
|
|
||||||
|
|
||||||
import (
|
|
||||||
"context"
|
|
||||||
|
|
||||||
"github.com/OpenIMSDK/tools/errs"
|
|
||||||
"github.com/openimsdk/open-im-server/v3/pkg/common/pagination"
|
|
||||||
"go.mongodb.org/mongo-driver/mongo"
|
|
||||||
"go.mongodb.org/mongo-driver/mongo/options"
|
|
||||||
)
|
|
||||||
|
|
||||||
|
|
||||||
func basic[T any]() bool {
|
|
||||||
var t T
|
|
||||||
switch any(t).(type) {
|
|
||||||
case int:
|
|
||||||
case int8:
|
|
||||||
case int16:
|
|
||||||
case int32:
|
|
||||||
case int64:
|
|
||||||
case uint:
|
|
||||||
case uint8:
|
|
||||||
case uint16:
|
|
||||||
case uint32:
|
|
||||||
case uint64:
|
|
||||||
case float32:
|
|
||||||
case float64:
|
|
||||||
case string:
|
|
||||||
case []byte:
|
|
||||||
default:
|
|
||||||
return false
|
|
||||||
}
|
|
||||||
return true
|
|
||||||
}
|
|
||||||
|
|
||||||
func anes[T any](ts []T) []any {
|
|
||||||
val := make([]any, len(ts))
|
|
||||||
for i := range ts {
|
|
||||||
val[i] = ts[i]
|
|
||||||
}
|
|
||||||
return val
|
|
||||||
}
|
|
||||||
|
|
||||||
func findOptionToCountOption(opts []*options.FindOptions) *options.CountOptions {
|
|
||||||
countOpt := options.Count()
|
|
||||||
for _, opt := range opts {
|
|
||||||
if opt.Skip != nil {
|
|
||||||
countOpt.SetSkip(*opt.Skip)
|
|
||||||
}
|
|
||||||
if opt.Limit != nil {
|
|
||||||
countOpt.SetLimit(*opt.Limit)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
return countOpt
|
|
||||||
}
|
|
||||||
|
|
||||||
func InsertMany[T any](ctx context.Context, coll *mongo.Collection, val []T, opts ...*options.InsertManyOptions) error {
|
|
||||||
_, err := coll.InsertMany(ctx, anes(val), opts...)
|
|
||||||
if err != nil {
|
|
||||||
return errs.Wrap(err)
|
|
||||||
}
|
|
||||||
return nil
|
|
||||||
}
|
|
||||||
|
|
||||||
func UpdateOne(ctx context.Context, coll *mongo.Collection, filter any, update any, notMatchedErr bool, opts ...*options.UpdateOptions) error {
|
|
||||||
res, err := coll.UpdateOne(ctx, filter, update, opts...)
|
|
||||||
if err != nil {
|
|
||||||
return errs.Wrap(err)
|
|
||||||
}
|
|
||||||
if notMatchedErr && res.MatchedCount == 0 {
|
|
||||||
return errs.Wrap(mongo.ErrNoDocuments)
|
|
||||||
}
|
|
||||||
return nil
|
|
||||||
}
|
|
||||||
|
|
||||||
func UpdateMany(ctx context.Context, coll *mongo.Collection, filter any, update any, opts ...*options.UpdateOptions) (*mongo.UpdateResult, error) {
|
|
||||||
res, err := coll.UpdateMany(ctx, filter, update, opts...)
|
|
||||||
if err != nil {
|
|
||||||
return nil, errs.Wrap(err)
|
|
||||||
}
|
|
||||||
return res, nil
|
|
||||||
}
|
|
||||||
|
|
||||||
func Find[T any](ctx context.Context, coll *mongo.Collection, filter any, opts ...*options.FindOptions) ([]T, error) {
|
|
||||||
cur, err := coll.Find(ctx, filter, opts...)
|
|
||||||
if err != nil {
|
|
||||||
return nil, errs.Wrap(err)
|
|
||||||
}
|
|
||||||
defer cur.Close(ctx)
|
|
||||||
var res []T
|
|
||||||
if basic[T]() {
|
|
||||||
var temp []map[string]T
|
|
||||||
if err := cur.All(ctx, &temp); err != nil {
|
|
||||||
return nil, errs.Wrap(err)
|
|
||||||
}
|
|
||||||
res = make([]T, 0, len(temp))
|
|
||||||
for _, m := range temp {
|
|
||||||
if len(m) != 1 {
|
|
||||||
return nil, errs.ErrInternalServer.Wrap("mongo find result len(m) != 1")
|
|
||||||
}
|
|
||||||
for _, t := range m {
|
|
||||||
res = append(res, t)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
} else {
|
|
||||||
if err := cur.All(ctx, &res); err != nil {
|
|
||||||
return nil, errs.Wrap(err)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
return res, nil
|
|
||||||
}
|
|
||||||
|
|
||||||
func FindOne[T any](ctx context.Context, coll *mongo.Collection, filter any, opts ...*options.FindOneOptions) (res T, err error) {
|
|
||||||
cur := coll.FindOne(ctx, filter, opts...)
|
|
||||||
if err := cur.Err(); err != nil {
|
|
||||||
return res, errs.Wrap(err)
|
|
||||||
}
|
|
||||||
if err := cur.Decode(&res); err != nil {
|
|
||||||
return res, errs.Wrap(err)
|
|
||||||
}
|
|
||||||
return res, nil
|
|
||||||
}
|
|
||||||
|
|
||||||
func FindPage[T any](ctx context.Context, coll *mongo.Collection, filter any, pagination pagination.Pagination, opts ...*options.FindOptions) (int64, []T, error) {
|
|
||||||
count, err := Count(ctx, coll, filter, findOptionToCountOption(opts))
|
|
||||||
if err != nil {
|
|
||||||
return 0, nil, err
|
|
||||||
}
|
|
||||||
if count == 0 || pagination == nil {
|
|
||||||
return count, nil, nil
|
|
||||||
}
|
|
||||||
skip := int64(pagination.GetPageNumber()-1) * int64(pagination.GetShowNumber())
|
|
||||||
if skip < 0 || skip >= count || pagination.GetShowNumber() <= 0 {
|
|
||||||
return count, nil, nil
|
|
||||||
}
|
|
||||||
opt := options.Find().SetSkip(skip).SetLimit(int64(pagination.GetShowNumber()))
|
|
||||||
res, err := Find[T](ctx, coll, filter, append(opts, opt)...)
|
|
||||||
if err != nil {
|
|
||||||
return 0, nil, err
|
|
||||||
}
|
|
||||||
return count, res, nil
|
|
||||||
}
|
|
||||||
|
|
||||||
func FindPageOnly[T any](ctx context.Context, coll *mongo.Collection, filter any, pagination pagination.Pagination, opts ...*options.FindOptions) ([]T, error) {
|
|
||||||
skip := int64(pagination.GetPageNumber()-1) * int64(pagination.GetShowNumber())
|
|
||||||
if skip < 0 || pagination.GetShowNumber() <= 0 {
|
|
||||||
return nil, nil
|
|
||||||
}
|
|
||||||
opt := options.Find().SetSkip(skip).SetLimit(int64(pagination.GetShowNumber()))
|
|
||||||
return Find[T](ctx, coll, filter, append(opts, opt)...)
|
|
||||||
}
|
|
||||||
|
|
||||||
func Count(ctx context.Context, coll *mongo.Collection, filter any, opts ...*options.CountOptions) (int64, error) {
|
|
||||||
return coll.CountDocuments(ctx, filter, opts...)
|
|
||||||
}
|
|
||||||
|
|
||||||
func Exist(ctx context.Context, coll *mongo.Collection, filter any, opts ...*options.CountOptions) (bool, error) {
|
|
||||||
opts = append(opts, options.Count().SetLimit(1))
|
|
||||||
count, err := Count(ctx, coll, filter, opts...)
|
|
||||||
if err != nil {
|
|
||||||
return false, err
|
|
||||||
}
|
|
||||||
return count > 0, nil
|
|
||||||
}
|
|
||||||
|
|
||||||
func DeleteOne(ctx context.Context, coll *mongo.Collection, filter any, opts ...*options.DeleteOptions) error {
|
|
||||||
if _, err := coll.DeleteOne(ctx, filter, opts...); err != nil {
|
|
||||||
return errs.Wrap(err)
|
|
||||||
}
|
|
||||||
return nil
|
|
||||||
}
|
|
||||||
|
|
||||||
func DeleteMany(ctx context.Context, coll *mongo.Collection, filter any, opts ...*options.DeleteOptions) error {
|
|
||||||
if _, err := coll.DeleteMany(ctx, filter, opts...); err != nil {
|
|
||||||
return errs.Wrap(err)
|
|
||||||
}
|
|
||||||
return nil
|
|
||||||
}
|
|
||||||
|
|
||||||
// TODO
|
|
||||||
func Delete[T any](ctx context.Context, coll *mongo.Collection, filter any, opts ...*options.DeleteOptions) error {
|
|
||||||
if _, err := coll.DeleteMany(ctx, filter, opts...); err != nil {
|
|
||||||
return errs.Wrap(err)
|
|
||||||
}
|
|
||||||
return nil
|
|
||||||
}
|
|
||||||
|
|
||||||
func Aggregate[T any](ctx context.Context, coll *mongo.Collection, pipeline any, opts ...*options.AggregateOptions) ([]T, error) {
|
|
||||||
cur, err := coll.Aggregate(ctx, pipeline, opts...)
|
|
||||||
if err != nil {
|
|
||||||
return nil, err
|
|
||||||
}
|
|
||||||
var ts []T
|
|
||||||
if err := cur.All(ctx, &ts); err != nil {
|
|
||||||
return nil, err
|
|
||||||
}
|
|
||||||
return ts, nil
|
|
||||||
}
|
|
||||||
@ -1,111 +0,0 @@
|
|||||||
// 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 (
|
|
||||||
"context"
|
|
||||||
|
|
||||||
"github.com/OpenIMSDK/tools/errs"
|
|
||||||
|
|
||||||
"github.com/OpenIMSDK/tools/ormutil"
|
|
||||||
|
|
||||||
"gorm.io/gorm"
|
|
||||||
|
|
||||||
"github.com/OpenIMSDK/tools/utils"
|
|
||||||
|
|
||||||
"github.com/openimsdk/open-im-server/v3/pkg/common/db/table/relation"
|
|
||||||
)
|
|
||||||
|
|
||||||
type BlackGorm struct {
|
|
||||||
*MetaDB
|
|
||||||
}
|
|
||||||
|
|
||||||
func NewBlackGorm(db *gorm.DB) relation.BlackModelInterface {
|
|
||||||
return &BlackGorm{NewMetaDB(db, &relation.BlackModel{})}
|
|
||||||
}
|
|
||||||
|
|
||||||
func (b *BlackGorm) Create(ctx context.Context, blacks []*relation.BlackModel) (err error) {
|
|
||||||
return utils.Wrap(b.db(ctx).Create(&blacks).Error, "")
|
|
||||||
}
|
|
||||||
|
|
||||||
func (b *BlackGorm) Delete(ctx context.Context, blacks []*relation.BlackModel) (err error) {
|
|
||||||
return utils.Wrap(b.db(ctx).Delete(blacks).Error, "")
|
|
||||||
}
|
|
||||||
|
|
||||||
func (b *BlackGorm) UpdateByMap(
|
|
||||||
ctx context.Context,
|
|
||||||
ownerUserID, blockUserID string,
|
|
||||||
args map[string]any,
|
|
||||||
) (err error) {
|
|
||||||
return utils.Wrap(
|
|
||||||
b.db(ctx).Where("block_user_id = ? and block_user_id = ?", ownerUserID, blockUserID).Updates(args).Error,
|
|
||||||
"",
|
|
||||||
)
|
|
||||||
}
|
|
||||||
|
|
||||||
func (b *BlackGorm) Update(ctx context.Context, blacks []*relation.BlackModel) (err error) {
|
|
||||||
return utils.Wrap(b.db(ctx).Updates(&blacks).Error, "")
|
|
||||||
}
|
|
||||||
|
|
||||||
func (b *BlackGorm) Find(
|
|
||||||
ctx context.Context,
|
|
||||||
blacks []*relation.BlackModel,
|
|
||||||
) (blackList []*relation.BlackModel, err error) {
|
|
||||||
var where [][]any
|
|
||||||
for _, black := range blacks {
|
|
||||||
where = append(where, []any{black.OwnerUserID, black.BlockUserID})
|
|
||||||
}
|
|
||||||
return blackList, utils.Wrap(
|
|
||||||
b.db(ctx).Where("(owner_user_id, block_user_id) in ?", where).Find(&blackList).Error,
|
|
||||||
"",
|
|
||||||
)
|
|
||||||
}
|
|
||||||
|
|
||||||
func (b *BlackGorm) Take(ctx context.Context, ownerUserID, blockUserID string) (black *relation.BlackModel, err error) {
|
|
||||||
black = &relation.BlackModel{}
|
|
||||||
return black, utils.Wrap(
|
|
||||||
b.db(ctx).Where("owner_user_id = ? and block_user_id = ?", ownerUserID, blockUserID).Take(black).Error,
|
|
||||||
"",
|
|
||||||
)
|
|
||||||
}
|
|
||||||
|
|
||||||
func (b *BlackGorm) FindOwnerBlacks(
|
|
||||||
ctx context.Context,
|
|
||||||
ownerUserID string,
|
|
||||||
pageNumber, showNumber int32,
|
|
||||||
) (blacks []*relation.BlackModel, total int64, err error) {
|
|
||||||
err = b.db(ctx).Count(&total).Error
|
|
||||||
if err != nil {
|
|
||||||
return nil, 0, utils.Wrap(err, "")
|
|
||||||
}
|
|
||||||
totalUint32, blacks, err := ormutil.GormPage[relation.BlackModel](
|
|
||||||
b.db(ctx).Where("owner_user_id = ?", ownerUserID),
|
|
||||||
pageNumber,
|
|
||||||
showNumber,
|
|
||||||
)
|
|
||||||
total = int64(totalUint32)
|
|
||||||
return
|
|
||||||
}
|
|
||||||
|
|
||||||
func (b *BlackGorm) FindBlackUserIDs(ctx context.Context, ownerUserID string) (blackUserIDs []string, err error) {
|
|
||||||
return blackUserIDs, utils.Wrap(
|
|
||||||
b.db(ctx).Where("owner_user_id = ?", ownerUserID).Pluck("block_user_id", &blackUserIDs).Error,
|
|
||||||
"",
|
|
||||||
)
|
|
||||||
}
|
|
||||||
|
|
||||||
func (b *BlackGorm) FindOwnerBlackInfos(ctx context.Context, ownerUserID string, userIDs []string) (blacks []*relation.BlackModel, err error) {
|
|
||||||
return blacks, errs.Wrap(b.db(ctx).Where("owner_user_id = ? and block_user_id in ?", ownerUserID, userIDs).Find(&blacks).Error)
|
|
||||||
}
|
|
||||||
@ -1,63 +0,0 @@
|
|||||||
// 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/protocol/constant"
|
|
||||||
pbmsg "github.com/OpenIMSDK/protocol/msg"
|
|
||||||
sdkws "github.com/OpenIMSDK/protocol/sdkws"
|
|
||||||
"github.com/OpenIMSDK/tools/utils"
|
|
||||||
|
|
||||||
"github.com/openimsdk/open-im-server/v3/pkg/common/db/table/relation"
|
|
||||||
)
|
|
||||||
|
|
||||||
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
|
|
||||||
}
|
|
||||||
@ -1,244 +0,0 @@
|
|||||||
// 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 (
|
|
||||||
// "context"
|
|
||||||
//
|
|
||||||
// "github.com/OpenIMSDK/tools/errs"
|
|
||||||
// "gorm.io/gorm"
|
|
||||||
//
|
|
||||||
// "github.com/OpenIMSDK/protocol/constant"
|
|
||||||
// "github.com/OpenIMSDK/tools/utils"
|
|
||||||
//
|
|
||||||
// "github.com/openimsdk/open-im-server/v3/pkg/common/db/table/relation"
|
|
||||||
//)
|
|
||||||
//
|
|
||||||
//type ConversationGorm struct {
|
|
||||||
// *MetaDB
|
|
||||||
//}
|
|
||||||
//
|
|
||||||
//func NewConversationGorm(db *gorm.DB) relation.ConversationModelInterface {
|
|
||||||
// return &ConversationGorm{NewMetaDB(db, &relation.ConversationModel{})}
|
|
||||||
//}
|
|
||||||
//
|
|
||||||
//func (c *ConversationGorm) NewTx(tx any) relation.ConversationModelInterface {
|
|
||||||
// return &ConversationGorm{NewMetaDB(tx.(*gorm.DB), &relation.ConversationModel{})}
|
|
||||||
//}
|
|
||||||
//
|
|
||||||
//func (c *ConversationGorm) Create(ctx context.Context, conversations []*relation.ConversationModel) (err error) {
|
|
||||||
// return utils.Wrap(c.db(ctx).Create(&conversations).Error, "")
|
|
||||||
//}
|
|
||||||
//
|
|
||||||
//func (c *ConversationGorm) Delete(ctx context.Context, groupIDs []string) (err error) {
|
|
||||||
// return utils.Wrap(c.db(ctx).Where("group_id in (?)", groupIDs).Delete(&relation.ConversationModel{}).Error, "")
|
|
||||||
//}
|
|
||||||
//
|
|
||||||
//func (c *ConversationGorm) UpdateByMap(
|
|
||||||
// ctx context.Context,
|
|
||||||
// userIDList []string,
|
|
||||||
// conversationID string,
|
|
||||||
// args map[string]any,
|
|
||||||
//) (rows int64, err error) {
|
|
||||||
// result := c.db(ctx).Where("owner_user_id IN (?) and conversation_id=?", userIDList, conversationID).Updates(args)
|
|
||||||
// return result.RowsAffected, utils.Wrap(result.Error, "")
|
|
||||||
//}
|
|
||||||
//
|
|
||||||
//func (c *ConversationGorm) Update(ctx context.Context, conversation *relation.ConversationModel) (err error) {
|
|
||||||
// return utils.Wrap(
|
|
||||||
// c.db(ctx).
|
|
||||||
// Where("owner_user_id = ? and conversation_id = ?", conversation.OwnerUserID, conversation.ConversationID).
|
|
||||||
// Updates(conversation).
|
|
||||||
// Error,
|
|
||||||
// "",
|
|
||||||
// )
|
|
||||||
//}
|
|
||||||
//
|
|
||||||
//func (c *ConversationGorm) Find(
|
|
||||||
// ctx context.Context,
|
|
||||||
// ownerUserID string,
|
|
||||||
// conversationIDs []string,
|
|
||||||
//) (conversations []*relation.ConversationModel, err error) {
|
|
||||||
// err = utils.Wrap(
|
|
||||||
// c.db(ctx).
|
|
||||||
// Where("owner_user_id=? and conversation_id IN (?)", ownerUserID, conversationIDs).
|
|
||||||
// Find(&conversations).
|
|
||||||
// Error,
|
|
||||||
// "",
|
|
||||||
// )
|
|
||||||
// return conversations, err
|
|
||||||
//}
|
|
||||||
//
|
|
||||||
//func (c *ConversationGorm) Take(
|
|
||||||
// ctx context.Context,
|
|
||||||
// userID, conversationID string,
|
|
||||||
//) (conversation *relation.ConversationModel, err error) {
|
|
||||||
// cc := &relation.ConversationModel{}
|
|
||||||
// return cc, utils.Wrap(
|
|
||||||
// c.db(ctx).Where("conversation_id = ? And owner_user_id = ?", conversationID, userID).Take(cc).Error,
|
|
||||||
// "",
|
|
||||||
// )
|
|
||||||
//}
|
|
||||||
//
|
|
||||||
//func (c *ConversationGorm) FindUserID(
|
|
||||||
// ctx context.Context,
|
|
||||||
// userIDs []string,
|
|
||||||
// conversationIDs []string,
|
|
||||||
//) (existUserID []string, err error) {
|
|
||||||
// return existUserID, utils.Wrap(
|
|
||||||
// c.db(ctx).
|
|
||||||
// Where(" owner_user_id IN (?) and conversation_id in (?)", userIDs, conversationIDs).
|
|
||||||
// Pluck("owner_user_id", &existUserID).
|
|
||||||
// Error,
|
|
||||||
// "",
|
|
||||||
// )
|
|
||||||
//}
|
|
||||||
//
|
|
||||||
//func (c *ConversationGorm) FindConversationID(
|
|
||||||
// ctx context.Context,
|
|
||||||
// userID string,
|
|
||||||
// conversationIDList []string,
|
|
||||||
//) (existConversationID []string, err error) {
|
|
||||||
// return existConversationID, utils.Wrap(
|
|
||||||
// c.db(ctx).
|
|
||||||
// Where(" conversation_id IN (?) and owner_user_id=?", conversationIDList, userID).
|
|
||||||
// Pluck("conversation_id", &existConversationID).
|
|
||||||
// Error,
|
|
||||||
// "",
|
|
||||||
// )
|
|
||||||
//}
|
|
||||||
//
|
|
||||||
//func (c *ConversationGorm) FindUserIDAllConversationID(
|
|
||||||
// ctx context.Context,
|
|
||||||
// userID string,
|
|
||||||
//) (conversationIDList []string, err error) {
|
|
||||||
// return conversationIDList, utils.Wrap(
|
|
||||||
// c.db(ctx).Where("owner_user_id=?", userID).Pluck("conversation_id", &conversationIDList).Error,
|
|
||||||
// "",
|
|
||||||
// )
|
|
||||||
//}
|
|
||||||
//
|
|
||||||
//func (c *ConversationGorm) FindUserIDAllConversations(
|
|
||||||
// ctx context.Context,
|
|
||||||
// userID string,
|
|
||||||
//) (conversations []*relation.ConversationModel, err error) {
|
|
||||||
// return conversations, utils.Wrap(c.db(ctx).Where("owner_user_id=?", userID).Find(&conversations).Error, "")
|
|
||||||
//}
|
|
||||||
//
|
|
||||||
//func (c *ConversationGorm) FindRecvMsgNotNotifyUserIDs(
|
|
||||||
// ctx context.Context,
|
|
||||||
// groupID string,
|
|
||||||
//) (userIDs []string, err error) {
|
|
||||||
// return userIDs, utils.Wrap(
|
|
||||||
// c.db(ctx).
|
|
||||||
// Where("group_id = ? and recv_msg_opt = ?", groupID, constant.ReceiveNotNotifyMessage).
|
|
||||||
// Pluck("owner_user_id", &userIDs).
|
|
||||||
// Error,
|
|
||||||
// "",
|
|
||||||
// )
|
|
||||||
//}
|
|
||||||
//
|
|
||||||
//func (c *ConversationGorm) FindSuperGroupRecvMsgNotNotifyUserIDs(
|
|
||||||
// ctx context.Context,
|
|
||||||
// groupID string,
|
|
||||||
//) (userIDs []string, err error) {
|
|
||||||
// return userIDs, utils.Wrap(
|
|
||||||
// c.db(ctx).
|
|
||||||
// Where("group_id = ? and recv_msg_opt = ? and conversation_type = ?", groupID, constant.ReceiveNotNotifyMessage, constant.SuperGroupChatType).
|
|
||||||
// Pluck("owner_user_id", &userIDs).
|
|
||||||
// Error,
|
|
||||||
// "",
|
|
||||||
// )
|
|
||||||
//}
|
|
||||||
//
|
|
||||||
//func (c *ConversationGorm) GetUserRecvMsgOpt(
|
|
||||||
// ctx context.Context,
|
|
||||||
// ownerUserID, conversationID string,
|
|
||||||
//) (opt int, err error) {
|
|
||||||
// var conversation relation.ConversationModel
|
|
||||||
// return int(
|
|
||||||
// conversation.RecvMsgOpt,
|
|
||||||
// ), utils.Wrap(
|
|
||||||
// c.db(ctx).
|
|
||||||
// Where("conversation_id = ? And owner_user_id = ?", conversationID, ownerUserID).
|
|
||||||
// Select("recv_msg_opt").
|
|
||||||
// Find(&conversation).
|
|
||||||
// Error,
|
|
||||||
// "",
|
|
||||||
// )
|
|
||||||
//}
|
|
||||||
//
|
|
||||||
//func (c *ConversationGorm) GetAllConversationIDs(ctx context.Context) (conversationIDs []string, err error) {
|
|
||||||
// return conversationIDs, utils.Wrap(
|
|
||||||
// c.db(ctx).Distinct("conversation_id").Pluck("conversation_id", &conversationIDs).Error,
|
|
||||||
// "",
|
|
||||||
// )
|
|
||||||
//}
|
|
||||||
//
|
|
||||||
//func (c *ConversationGorm) GetAllConversationIDsNumber(ctx context.Context) (int64, error) {
|
|
||||||
// var num int64
|
|
||||||
// err := c.db(ctx).Select("COUNT(DISTINCT conversation_id)").Model(&relation.ConversationModel{}).Count(&num).Error
|
|
||||||
// return num, errs.Wrap(err)
|
|
||||||
//}
|
|
||||||
//
|
|
||||||
//func (c *ConversationGorm) PageConversationIDs(ctx context.Context, pageNumber, showNumber int32) (conversationIDs []string, err error) {
|
|
||||||
// err = c.db(ctx).Distinct("conversation_id").Limit(int(showNumber)).Offset(int((pageNumber-1)*showNumber)).Pluck("conversation_id", &conversationIDs).Error
|
|
||||||
// err = errs.Wrap(err)
|
|
||||||
// return
|
|
||||||
//}
|
|
||||||
//
|
|
||||||
//func (c *ConversationGorm) GetConversationsByConversationID(
|
|
||||||
// ctx context.Context,
|
|
||||||
// conversationIDs []string,
|
|
||||||
//) (conversations []*relation.ConversationModel, err error) {
|
|
||||||
// return conversations, utils.Wrap(
|
|
||||||
// c.db(ctx).Where("conversation_id IN (?)", conversationIDs).Find(&conversations).Error,
|
|
||||||
// "",
|
|
||||||
// )
|
|
||||||
//}
|
|
||||||
//
|
|
||||||
//func (c *ConversationGorm) GetConversationIDsNeedDestruct(
|
|
||||||
// ctx context.Context,
|
|
||||||
//) (conversations []*relation.ConversationModel, err error) {
|
|
||||||
// return conversations, utils.Wrap(
|
|
||||||
// c.db(ctx).
|
|
||||||
// Where("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)").
|
|
||||||
// Find(&conversations).
|
|
||||||
// Error,
|
|
||||||
// "",
|
|
||||||
// )
|
|
||||||
//}
|
|
||||||
//
|
|
||||||
//func (c *ConversationGorm) GetConversationRecvMsgOpt(ctx context.Context, userID string, conversationID string) (int32, error) {
|
|
||||||
// var recvMsgOpt int32
|
|
||||||
// return recvMsgOpt, errs.Wrap(
|
|
||||||
// c.db(ctx).
|
|
||||||
// Model(&relation.ConversationModel{}).
|
|
||||||
// Where("conversation_id = ? and owner_user_id in ?", conversationID, userID).
|
|
||||||
// Pluck("recv_msg_opt", &recvMsgOpt).
|
|
||||||
// Error,
|
|
||||||
// )
|
|
||||||
//}
|
|
||||||
//
|
|
||||||
//func (c *ConversationGorm) GetConversationNotReceiveMessageUserIDs(ctx context.Context, conversationID string) ([]string, error) {
|
|
||||||
// var userIDs []string
|
|
||||||
// return userIDs, errs.Wrap(
|
|
||||||
// c.db(ctx).
|
|
||||||
// Model(&relation.ConversationModel{}).
|
|
||||||
// Where("conversation_id = ? and recv_msg_opt <> ?", conversationID, constant.ReceiveMessage).
|
|
||||||
// Pluck("owner_user_id", &userIDs).Error,
|
|
||||||
// )
|
|
||||||
//}
|
|
||||||
@ -1,15 +0,0 @@
|
|||||||
// 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/openimsdk/open-im-server/v3/pkg/common/db/relation"
|
|
||||||
@ -1,154 +0,0 @@
|
|||||||
// 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
|
|
||||||
|
|
||||||
// type FriendRequestGorm struct {
|
|
||||||
// *MetaDB
|
|
||||||
// }
|
|
||||||
|
|
||||||
// func NewFriendRequestGorm(db *gorm.DB) relation.FriendRequestModelInterface {
|
|
||||||
// return &FriendRequestGorm{NewMetaDB(db, &relation.FriendRequestModel{})}
|
|
||||||
// }
|
|
||||||
|
|
||||||
// func (f *FriendRequestGorm) NewTx(tx any) relation.FriendRequestModelInterface {
|
|
||||||
// return &FriendRequestGorm{NewMetaDB(tx.(*gorm.DB), &relation.FriendRequestModel{})}
|
|
||||||
// }
|
|
||||||
|
|
||||||
// // 插入多条记录.
|
|
||||||
// func (f *FriendRequestGorm) Create(ctx context.Context, friendRequests []*relation.FriendRequestModel) (err error) {
|
|
||||||
// return utils.Wrap(f.db(ctx).Create(&friendRequests).Error, "")
|
|
||||||
// }
|
|
||||||
|
|
||||||
// // 删除记录.
|
|
||||||
// func (f *FriendRequestGorm) Delete(ctx context.Context, fromUserID, toUserID string) (err error) {
|
|
||||||
// return utils.Wrap(
|
|
||||||
// f.db(ctx).
|
|
||||||
// Where("from_user_id = ? AND to_user_id = ?", fromUserID, toUserID).
|
|
||||||
// Delete(&relation.FriendRequestModel{}).
|
|
||||||
// Error,
|
|
||||||
// "",
|
|
||||||
// )
|
|
||||||
// }
|
|
||||||
|
|
||||||
// // 更新零值.
|
|
||||||
// func (f *FriendRequestGorm) UpdateByMap(
|
|
||||||
// ctx context.Context,
|
|
||||||
// fromUserID string,
|
|
||||||
// toUserID string,
|
|
||||||
// args map[string]any,
|
|
||||||
// ) (err error) {
|
|
||||||
// return utils.Wrap(
|
|
||||||
// f.db(ctx).
|
|
||||||
// Model(&relation.FriendRequestModel{}).
|
|
||||||
// Where("from_user_id = ? AND to_user_id =?", fromUserID, toUserID).
|
|
||||||
// Updates(args).
|
|
||||||
// Error,
|
|
||||||
// "",
|
|
||||||
// )
|
|
||||||
// }
|
|
||||||
|
|
||||||
// // 更新记录 （非零值）.
|
|
||||||
// func (f *FriendRequestGorm) Update(ctx context.Context, friendRequest *relation.FriendRequestModel) (err error) {
|
|
||||||
// fr2 := *friendRequest
|
|
||||||
// fr2.FromUserID = ""
|
|
||||||
// fr2.ToUserID = ""
|
|
||||||
// return utils.Wrap(
|
|
||||||
// f.db(ctx).
|
|
||||||
// Where("from_user_id = ? AND to_user_id =?", friendRequest.FromUserID, friendRequest.ToUserID).
|
|
||||||
// Updates(fr2).
|
|
||||||
// Error,
|
|
||||||
// "",
|
|
||||||
// )
|
|
||||||
// }
|
|
||||||
|
|
||||||
// // 获取来指定用户的好友申请 未找到 不返回错误.
|
|
||||||
// func (f *FriendRequestGorm) Find(
|
|
||||||
// ctx context.Context,
|
|
||||||
// fromUserID, toUserID string,
|
|
||||||
// ) (friendRequest *relation.FriendRequestModel, err error) {
|
|
||||||
// friendRequest = &relation.FriendRequestModel{}
|
|
||||||
// err = utils.Wrap(
|
|
||||||
// f.db(ctx).Where("from_user_id = ? and to_user_id = ?", fromUserID, toUserID).Find(friendRequest).Error,
|
|
||||||
// "",
|
|
||||||
// )
|
|
||||||
// return friendRequest, err
|
|
||||||
// }
|
|
||||||
|
|
||||||
// func (f *FriendRequestGorm) Take(
|
|
||||||
// ctx context.Context,
|
|
||||||
// fromUserID, toUserID string,
|
|
||||||
// ) (friendRequest *relation.FriendRequestModel, err error) {
|
|
||||||
// friendRequest = &relation.FriendRequestModel{}
|
|
||||||
// err = utils.Wrap(
|
|
||||||
// f.db(ctx).Where("from_user_id = ? and to_user_id = ?", fromUserID, toUserID).Take(friendRequest).Error,
|
|
||||||
// "",
|
|
||||||
// )
|
|
||||||
// return friendRequest, err
|
|
||||||
// }
|
|
||||||
|
|
||||||
// // 获取toUserID收到的好友申请列表.
|
|
||||||
// func (f *FriendRequestGorm) FindToUserID(
|
|
||||||
// ctx context.Context,
|
|
||||||
// toUserID string,
|
|
||||||
// pageNumber, showNumber int32,
|
|
||||||
// ) (friendRequests []*relation.FriendRequestModel, total int64, err error) {
|
|
||||||
// err = f.db(ctx).Model(&relation.FriendRequestModel{}).Where("to_user_id = ? ", toUserID).Count(&total).Error
|
|
||||||
// if err != nil {
|
|
||||||
// return nil, 0, utils.Wrap(err, "")
|
|
||||||
// }
|
|
||||||
// err = utils.Wrap(
|
|
||||||
// f.db(ctx).
|
|
||||||
// Where("to_user_id = ? ", toUserID).
|
|
||||||
// Limit(int(showNumber)).
|
|
||||||
// Offset(int(pageNumber-1)*int(showNumber)).
|
|
||||||
// Find(&friendRequests).
|
|
||||||
// Error,
|
|
||||||
// "",
|
|
||||||
// )
|
|
||||||
// return
|
|
||||||
// }
|
|
||||||
|
|
||||||
// // 获取fromUserID发出去的好友申请列表.
|
|
||||||
// func (f *FriendRequestGorm) FindFromUserID(
|
|
||||||
// ctx context.Context,
|
|
||||||
// fromUserID string,
|
|
||||||
// pageNumber, showNumber int32,
|
|
||||||
// ) (friendRequests []*relation.FriendRequestModel, total int64, err error) {
|
|
||||||
// err = f.db(ctx).Model(&relation.FriendRequestModel{}).Where("from_user_id = ? ", fromUserID).Count(&total).Error
|
|
||||||
// if err != nil {
|
|
||||||
// return nil, 0, utils.Wrap(err, "")
|
|
||||||
// }
|
|
||||||
// err = utils.Wrap(
|
|
||||||
// f.db(ctx).
|
|
||||||
// Where("from_user_id = ? ", fromUserID).
|
|
||||||
// Limit(int(showNumber)).
|
|
||||||
// Offset(int(pageNumber-1)*int(showNumber)).
|
|
||||||
// Find(&friendRequests).
|
|
||||||
// Error,
|
|
||||||
// "",
|
|
||||||
// )
|
|
||||||
// return
|
|
||||||
// }
|
|
||||||
|
|
||||||
// func (f *FriendRequestGorm) FindBothFriendRequests(ctx context.Context, fromUserID, toUserID string) (friends []*relation.FriendRequestModel, err error) {
|
|
||||||
// err = utils.Wrap(
|
|
||||||
// f.db(ctx).
|
|
||||||
// Where("(from_user_id = ? AND to_user_id = ?) OR (from_user_id = ? AND to_user_id = ?)", fromUserID, toUserID, toUserID, fromUserID).
|
|
||||||
// Find(&friends).
|
|
||||||
// Error,
|
|
||||||
// "",
|
|
||||||
// )
|
|
||||||
// return
|
|
||||||
// }
|
|
||||||
@ -1,198 +0,0 @@
|
|||||||
// 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 (
|
|
||||||
// "context"
|
|
||||||
//
|
|
||||||
// "gorm.io/gorm"
|
|
||||||
//
|
|
||||||
// "github.com/OpenIMSDK/protocol/constant"
|
|
||||||
// "github.com/OpenIMSDK/tools/ormutil"
|
|
||||||
// "github.com/OpenIMSDK/tools/utils"
|
|
||||||
//
|
|
||||||
// "github.com/openimsdk/open-im-server/v3/pkg/common/db/table/relation"
|
|
||||||
//)
|
|
||||||
//
|
|
||||||
//var _ relation.GroupMemberModelInterface = (*GroupMemberGorm)(nil)
|
|
||||||
//
|
|
||||||
//type GroupMemberGorm struct {
|
|
||||||
// *MetaDB
|
|
||||||
//}
|
|
||||||
//
|
|
||||||
//func NewGroupMemberDB(db *gorm.DB) relation.GroupMemberModelInterface {
|
|
||||||
// return &GroupMemberGorm{NewMetaDB(db, &relation.GroupMemberModel{})}
|
|
||||||
//}
|
|
||||||
//
|
|
||||||
//func (g *GroupMemberGorm) NewTx(tx any) relation.GroupMemberModelInterface {
|
|
||||||
// return &GroupMemberGorm{NewMetaDB(tx.(*gorm.DB), &relation.GroupMemberModel{})}
|
|
||||||
//}
|
|
||||||
//
|
|
||||||
//func (g *GroupMemberGorm) Create(ctx context.Context, groupMemberList []*relation.GroupMemberModel) (err error) {
|
|
||||||
// return utils.Wrap(g.db(ctx).Create(&groupMemberList).Error, "")
|
|
||||||
//}
|
|
||||||
//
|
|
||||||
//func (g *GroupMemberGorm) Delete(ctx context.Context, groupID string, userIDs []string) (err error) {
|
|
||||||
// return utils.Wrap(
|
|
||||||
// g.db(ctx).Where("group_id = ? and user_id in (?)", groupID, userIDs).Delete(&relation.GroupMemberModel{}).Error,
|
|
||||||
// "",
|
|
||||||
// )
|
|
||||||
//}
|
|
||||||
//
|
|
||||||
//func (g *GroupMemberGorm) DeleteGroup(ctx context.Context, groupIDs []string) (err error) {
|
|
||||||
// return utils.Wrap(g.db(ctx).Where("group_id in (?)", groupIDs).Delete(&relation.GroupMemberModel{}).Error, "")
|
|
||||||
//}
|
|
||||||
//
|
|
||||||
//func (g *GroupMemberGorm) Update(ctx context.Context, groupID string, userID string, data map[string]any) (err error) {
|
|
||||||
// return utils.Wrap(g.db(ctx).Where("group_id = ? and user_id = ?", groupID, userID).Updates(data).Error, "")
|
|
||||||
//}
|
|
||||||
//
|
|
||||||
//func (g *GroupMemberGorm) UpdateRoleLevel(
|
|
||||||
// ctx context.Context,
|
|
||||||
// groupID string,
|
|
||||||
// userID string,
|
|
||||||
// roleLevel int32,
|
|
||||||
//) (rowsAffected int64, err error) {
|
|
||||||
// db := g.db(ctx).Where("group_id = ? and user_id = ?", groupID, userID).Updates(map[string]any{
|
|
||||||
// "role_level": roleLevel,
|
|
||||||
// })
|
|
||||||
// return db.RowsAffected, utils.Wrap(db.Error, "")
|
|
||||||
//}
|
|
||||||
//
|
|
||||||
//func (g *GroupMemberGorm) Find(
|
|
||||||
// ctx context.Context,
|
|
||||||
// groupIDs []string,
|
|
||||||
// userIDs []string,
|
|
||||||
// roleLevels []int32,
|
|
||||||
//) (groupMembers []*relation.GroupMemberModel, err error) {
|
|
||||||
// db := g.db(ctx)
|
|
||||||
// if len(groupIDs) > 0 {
|
|
||||||
// db = db.Where("group_id in (?)", groupIDs)
|
|
||||||
// }
|
|
||||||
// if len(userIDs) > 0 {
|
|
||||||
// db = db.Where("user_id in (?)", userIDs)
|
|
||||||
// }
|
|
||||||
// if len(roleLevels) > 0 {
|
|
||||||
// db = db.Where("role_level in (?)", roleLevels)
|
|
||||||
// }
|
|
||||||
// return groupMembers, utils.Wrap(db.Find(&groupMembers).Error, "")
|
|
||||||
//}
|
|
||||||
//
|
|
||||||
//func (g *GroupMemberGorm) Take(
|
|
||||||
// ctx context.Context,
|
|
||||||
// groupID string,
|
|
||||||
// userID string,
|
|
||||||
//) (groupMember *relation.GroupMemberModel, err error) {
|
|
||||||
// groupMember = &relation.GroupMemberModel{}
|
|
||||||
// return groupMember, utils.Wrap(
|
|
||||||
// g.db(ctx).Where("group_id = ? and user_id = ?", groupID, userID).Take(groupMember).Error,
|
|
||||||
// "",
|
|
||||||
// )
|
|
||||||
//}
|
|
||||||
//
|
|
||||||
//func (g *GroupMemberGorm) TakeOwner(
|
|
||||||
// ctx context.Context,
|
|
||||||
// groupID string,
|
|
||||||
//) (groupMember *relation.GroupMemberModel, err error) {
|
|
||||||
// groupMember = &relation.GroupMemberModel{}
|
|
||||||
// return groupMember, utils.Wrap(
|
|
||||||
// g.db(ctx).Where("group_id = ? and role_level = ?", groupID, constant.GroupOwner).Take(groupMember).Error,
|
|
||||||
// "",
|
|
||||||
// )
|
|
||||||
//}
|
|
||||||
//
|
|
||||||
//func (g *GroupMemberGorm) SearchMember(
|
|
||||||
// ctx context.Context,
|
|
||||||
// keyword string,
|
|
||||||
// groupIDs []string,
|
|
||||||
// userIDs []string,
|
|
||||||
// roleLevels []int32,
|
|
||||||
// pageNumber, showNumber int32,
|
|
||||||
//) (total uint32, groupList []*relation.GroupMemberModel, err error) {
|
|
||||||
// db := g.db(ctx)
|
|
||||||
// ormutil.GormIn(&db, "group_id", groupIDs)
|
|
||||||
// ormutil.GormIn(&db, "user_id", userIDs)
|
|
||||||
// ormutil.GormIn(&db, "role_level", roleLevels)
|
|
||||||
// return ormutil.GormSearch[relation.GroupMemberModel](db, []string{"nickname"}, keyword, pageNumber, showNumber)
|
|
||||||
//}
|
|
||||||
//
|
|
||||||
//func (g *GroupMemberGorm) MapGroupMemberNum(
|
|
||||||
// ctx context.Context,
|
|
||||||
// groupIDs []string,
|
|
||||||
//) (count map[string]uint32, err error) {
|
|
||||||
// return ormutil.MapCount(g.db(ctx).Where("group_id in (?)", groupIDs), "group_id")
|
|
||||||
//}
|
|
||||||
//
|
|
||||||
//func (g *GroupMemberGorm) FindJoinUserID(
|
|
||||||
// ctx context.Context,
|
|
||||||
// groupIDs []string,
|
|
||||||
//) (groupUsers map[string][]string, err error) {
|
|
||||||
// var groupMembers []*relation.GroupMemberModel
|
|
||||||
// if err := g.db(ctx).Select("group_id, user_id").Where("group_id in (?)", groupIDs).Find(&groupMembers).Error; err != nil {
|
|
||||||
// return nil, utils.Wrap(err, "")
|
|
||||||
// }
|
|
||||||
// groupUsers = make(map[string][]string)
|
|
||||||
// for _, item := range groupMembers {
|
|
||||||
// v, ok := groupUsers[item.GroupID]
|
|
||||||
// if !ok {
|
|
||||||
// groupUsers[item.GroupID] = []string{item.UserID}
|
|
||||||
// } else {
|
|
||||||
// groupUsers[item.GroupID] = append(v, item.UserID)
|
|
||||||
// }
|
|
||||||
// }
|
|
||||||
// return groupUsers, nil
|
|
||||||
//}
|
|
||||||
//
|
|
||||||
//func (g *GroupMemberGorm) FindMemberUserID(ctx context.Context, groupID string) (userIDs []string, err error) {
|
|
||||||
// return userIDs, utils.Wrap(g.db(ctx).Where("group_id = ?", groupID).Pluck("user_id", &userIDs).Error, "")
|
|
||||||
//}
|
|
||||||
//
|
|
||||||
//func (g *GroupMemberGorm) FindUserJoinedGroupID(ctx context.Context, userID string) (groupIDs []string, err error) {
|
|
||||||
// return groupIDs, utils.Wrap(g.db(ctx).Where("user_id = ?", userID).Pluck("group_id", &groupIDs).Error, "")
|
|
||||||
//}
|
|
||||||
//
|
|
||||||
//func (g *GroupMemberGorm) TakeGroupMemberNum(ctx context.Context, groupID string) (count int64, err error) {
|
|
||||||
// return count, utils.Wrap(g.db(ctx).Where("group_id = ?", groupID).Count(&count).Error, "")
|
|
||||||
//}
|
|
||||||
//
|
|
||||||
//func (g *GroupMemberGorm) FindUsersJoinedGroupID(ctx context.Context, userIDs []string) (map[string][]string, error) {
|
|
||||||
// var groupMembers []*relation.GroupMemberModel
|
|
||||||
// err := g.db(ctx).Select("group_id, user_id").Where("user_id IN (?)", userIDs).Find(&groupMembers).Error
|
|
||||||
// if err != nil {
|
|
||||||
// return nil, err
|
|
||||||
// }
|
|
||||||
// result := make(map[string][]string)
|
|
||||||
// for _, groupMember := range groupMembers {
|
|
||||||
// v, ok := result[groupMember.UserID]
|
|
||||||
// if !ok {
|
|
||||||
// result[groupMember.UserID] = []string{groupMember.GroupID}
|
|
||||||
// } else {
|
|
||||||
// result[groupMember.UserID] = append(v, groupMember.GroupID)
|
|
||||||
// }
|
|
||||||
// }
|
|
||||||
// return result, nil
|
|
||||||
//}
|
|
||||||
//
|
|
||||||
//func (g *GroupMemberGorm) FindUserManagedGroupID(ctx context.Context, userID string) (groupIDs []string, err error) {
|
|
||||||
// return groupIDs, utils.Wrap(
|
|
||||||
// g.db(ctx).
|
|
||||||
// Model(&relation.GroupMemberModel{}).
|
|
||||||
// Where("user_id = ? and (role_level = ? or role_level = ?)", userID, constant.GroupOwner, constant.GroupAdmin).
|
|
||||||
// Pluck("group_id", &groupIDs).
|
|
||||||
// Error,
|
|
||||||
// "",
|
|
||||||
// )
|
|
||||||
//}
|
|
||||||
@ -1,106 +0,0 @@
|
|||||||
// 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 (
|
|
||||||
// "context"
|
|
||||||
// "time"
|
|
||||||
//
|
|
||||||
// "github.com/OpenIMSDK/protocol/constant"
|
|
||||||
//
|
|
||||||
// "gorm.io/gorm"
|
|
||||||
//
|
|
||||||
// "github.com/OpenIMSDK/tools/errs"
|
|
||||||
// "github.com/OpenIMSDK/tools/ormutil"
|
|
||||||
// "github.com/OpenIMSDK/tools/utils"
|
|
||||||
//
|
|
||||||
// "github.com/openimsdk/open-im-server/v3/pkg/common/db/table/relation"
|
|
||||||
//)
|
|
||||||
//
|
|
||||||
//var _ relation.GroupModelInterface = (*GroupGorm)(nil)
|
|
||||||
//
|
|
||||||
//type GroupGorm struct {
|
|
||||||
// *MetaDB
|
|
||||||
//}
|
|
||||||
//
|
|
||||||
//func NewGroupDB(db *gorm.DB) relation.GroupModelInterface {
|
|
||||||
// return &GroupGorm{NewMetaDB(db, &relation.GroupModel{})}
|
|
||||||
//}
|
|
||||||
//
|
|
||||||
//func (g *GroupGorm) NewTx(tx any) relation.GroupModelInterface {
|
|
||||||
// return &GroupGorm{NewMetaDB(tx.(*gorm.DB), &relation.GroupModel{})}
|
|
||||||
//}
|
|
||||||
//
|
|
||||||
//func (g *GroupGorm) Create(ctx context.Context, groups []*relation.GroupModel) (err error) {
|
|
||||||
// return utils.Wrap(g.DB.Create(&groups).Error, "")
|
|
||||||
//}
|
|
||||||
//
|
|
||||||
//func (g *GroupGorm) UpdateMap(ctx context.Context, groupID string, args map[string]any) (err error) {
|
|
||||||
// return utils.Wrap(g.DB.Where("group_id = ?", groupID).Model(&relation.GroupModel{}).Updates(args).Error, "")
|
|
||||||
//}
|
|
||||||
//
|
|
||||||
//func (g *GroupGorm) UpdateStatus(ctx context.Context, groupID string, status int32) (err error) {
|
|
||||||
// return utils.Wrap(g.DB.Where("group_id = ?", groupID).Model(&relation.GroupModel{}).Updates(map[string]any{"status": status}).Error, "")
|
|
||||||
//}
|
|
||||||
//
|
|
||||||
//func (g *GroupGorm) Find(ctx context.Context, groupIDs []string) (groups []*relation.GroupModel, err error) {
|
|
||||||
// return groups, utils.Wrap(g.DB.Where("group_id in (?)", groupIDs).Find(&groups).Error, "")
|
|
||||||
//}
|
|
||||||
//
|
|
||||||
//func (g *GroupGorm) Take(ctx context.Context, groupID string) (group *relation.GroupModel, err error) {
|
|
||||||
// group = &relation.GroupModel{}
|
|
||||||
// return group, utils.Wrap(g.DB.Where("group_id = ?", groupID).Take(group).Error, "")
|
|
||||||
//}
|
|
||||||
//
|
|
||||||
//func (g *GroupGorm) Search(ctx context.Context, keyword string, pageNumber, showNumber int32) (total uint32, groups []*relation.GroupModel, err error) {
|
|
||||||
// db := g.DB
|
|
||||||
// db = db.WithContext(ctx).Where("status!=?", constant.GroupStatusDismissed)
|
|
||||||
// return ormutil.GormSearch[relation.GroupModel](db, []string{"name"}, keyword, pageNumber, showNumber)
|
|
||||||
//}
|
|
||||||
//
|
|
||||||
//func (g *GroupGorm) GetGroupIDsByGroupType(ctx context.Context, groupType int) (groupIDs []string, err error) {
|
|
||||||
// return groupIDs, utils.Wrap(g.DB.Model(&relation.GroupModel{}).Where("group_type = ? ", groupType).Pluck("group_id", &groupIDs).Error, "")
|
|
||||||
//}
|
|
||||||
//
|
|
||||||
//func (g *GroupGorm) CountTotal(ctx context.Context, before *time.Time) (count int64, err error) {
|
|
||||||
// db := g.db(ctx).Model(&relation.GroupModel{})
|
|
||||||
// if before != nil {
|
|
||||||
// db = db.Where("create_time < ?", before)
|
|
||||||
// }
|
|
||||||
// if err := db.Count(&count).Error; err != nil {
|
|
||||||
// return 0, err
|
|
||||||
// }
|
|
||||||
// return count, nil
|
|
||||||
//}
|
|
||||||
//
|
|
||||||
//func (g *GroupGorm) CountRangeEverydayTotal(ctx context.Context, start time.Time, end time.Time) (map[string]int64, error) {
|
|
||||||
// var res []struct {
|
|
||||||
// Date time.Time `gorm:"column:date"`
|
|
||||||
// Count int64 `gorm:"column:count"`
|
|
||||||
// }
|
|
||||||
// err := g.db(ctx).Model(&relation.GroupModel{}).Select("DATE(create_time) AS date, count(1) AS count").Where("create_time >= ? and create_time < ?", start, end).Group("date").Find(&res).Error
|
|
||||||
// if err != nil {
|
|
||||||
// return nil, errs.Wrap(err)
|
|
||||||
// }
|
|
||||||
// v := make(map[string]int64)
|
|
||||||
// for _, r := range res {
|
|
||||||
// v[r.Date.Format("2006-01-02")] = r.Count
|
|
||||||
// }
|
|
||||||
// return v, nil
|
|
||||||
//}
|
|
||||||
//
|
|
||||||
//func (g *GroupGorm) FindNotDismissedGroup(ctx context.Context, groupIDs []string) (groups []*relation.GroupModel, err error) {
|
|
||||||
// return groups, utils.Wrap(g.DB.Where("group_id in (?) and status != ?", groupIDs, constant.GroupStatusDismissed).Find(&groups).Error, "")
|
|
||||||
//}
|
|
||||||
@ -1,119 +0,0 @@
|
|||||||
// 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 (
|
|
||||||
// "context"
|
|
||||||
//
|
|
||||||
// "github.com/OpenIMSDK/tools/ormutil"
|
|
||||||
//
|
|
||||||
// "gorm.io/gorm"
|
|
||||||
//
|
|
||||||
// "github.com/OpenIMSDK/tools/utils"
|
|
||||||
//
|
|
||||||
// "github.com/openimsdk/open-im-server/v3/pkg/common/db/table/relation"
|
|
||||||
//)
|
|
||||||
//
|
|
||||||
//type GroupRequestGorm struct {
|
|
||||||
// *MetaDB
|
|
||||||
//}
|
|
||||||
//
|
|
||||||
//func NewGroupRequest(db *gorm.DB) relation.GroupRequestModelInterface {
|
|
||||||
// return &GroupRequestGorm{
|
|
||||||
// NewMetaDB(db, &relation.GroupRequestModel{}),
|
|
||||||
// }
|
|
||||||
//}
|
|
||||||
//
|
|
||||||
//func (g *GroupRequestGorm) NewTx(tx any) relation.GroupRequestModelInterface {
|
|
||||||
// return &GroupRequestGorm{NewMetaDB(tx.(*gorm.DB), &relation.GroupRequestModel{})}
|
|
||||||
//}
|
|
||||||
//
|
|
||||||
//func (g *GroupRequestGorm) Create(ctx context.Context, groupRequests []*relation.GroupRequestModel) (err error) {
|
|
||||||
// return utils.Wrap(g.DB.WithContext(ctx).Create(&groupRequests).Error, utils.GetSelfFuncName())
|
|
||||||
//}
|
|
||||||
//
|
|
||||||
//func (g *GroupRequestGorm) Delete(ctx context.Context, groupID string, userID string) (err error) {
|
|
||||||
// return utils.Wrap(
|
|
||||||
// g.DB.WithContext(ctx).
|
|
||||||
// Where("group_id = ? and user_id = ? ", groupID, userID).
|
|
||||||
// Delete(&relation.GroupRequestModel{}).
|
|
||||||
// Error,
|
|
||||||
// utils.GetSelfFuncName(),
|
|
||||||
// )
|
|
||||||
//}
|
|
||||||
//
|
|
||||||
//func (g *GroupRequestGorm) UpdateHandler(
|
|
||||||
// ctx context.Context,
|
|
||||||
// groupID string,
|
|
||||||
// userID string,
|
|
||||||
// handledMsg string,
|
|
||||||
// handleResult int32,
|
|
||||||
//) (err error) {
|
|
||||||
// return utils.Wrap(
|
|
||||||
// g.DB.WithContext(ctx).
|
|
||||||
// Model(&relation.GroupRequestModel{}).
|
|
||||||
// Where("group_id = ? and user_id = ? ", groupID, userID).
|
|
||||||
// Updates(map[string]any{
|
|
||||||
// "handle_msg": handledMsg,
|
|
||||||
// "handle_result": handleResult,
|
|
||||||
// }).
|
|
||||||
// Error,
|
|
||||||
// utils.GetSelfFuncName(),
|
|
||||||
// )
|
|
||||||
//}
|
|
||||||
//
|
|
||||||
//func (g *GroupRequestGorm) Take(
|
|
||||||
// ctx context.Context,
|
|
||||||
// groupID string,
|
|
||||||
// userID string,
|
|
||||||
//) (groupRequest *relation.GroupRequestModel, err error) {
|
|
||||||
// groupRequest = &relation.GroupRequestModel{}
|
|
||||||
// return groupRequest, utils.Wrap(
|
|
||||||
// g.DB.WithContext(ctx).Where("group_id = ? and user_id = ? ", groupID, userID).Take(groupRequest).Error,
|
|
||||||
// utils.GetSelfFuncName(),
|
|
||||||
// )
|
|
||||||
//}
|
|
||||||
//
|
|
||||||
//func (g *GroupRequestGorm) Page(
|
|
||||||
// ctx context.Context,
|
|
||||||
// userID string,
|
|
||||||
// pageNumber, showNumber int32,
|
|
||||||
//) (total uint32, groups []*relation.GroupRequestModel, err error) {
|
|
||||||
// return ormutil.GormSearch[relation.GroupRequestModel](
|
|
||||||
// g.DB.WithContext(ctx).Where("user_id = ?", userID),
|
|
||||||
// nil,
|
|
||||||
// "",
|
|
||||||
// pageNumber,
|
|
||||||
// showNumber,
|
|
||||||
// )
|
|
||||||
//}
|
|
||||||
//
|
|
||||||
//func (g *GroupRequestGorm) PageGroup(
|
|
||||||
// ctx context.Context,
|
|
||||||
// groupIDs []string,
|
|
||||||
// pageNumber, showNumber int32,
|
|
||||||
//) (total uint32, groups []*relation.GroupRequestModel, err error) {
|
|
||||||
// return ormutil.GormPage[relation.GroupRequestModel](
|
|
||||||
// g.DB.WithContext(ctx).Where("group_id in ?", groupIDs),
|
|
||||||
// pageNumber,
|
|
||||||
// showNumber,
|
|
||||||
// )
|
|
||||||
//}
|
|
||||||
//
|
|
||||||
//func (g *GroupRequestGorm) FindGroupRequests(ctx context.Context, groupID string, userIDs []string) (total int64, groupRequests []*relation.GroupRequestModel, err error) {
|
|
||||||
// err = g.DB.WithContext(ctx).Where("group_id = ? and user_id in ?", groupID, userIDs).Find(&groupRequests).Error
|
|
||||||
// return int64(len(groupRequests)), groupRequests, utils.Wrap(err, utils.GetSelfFuncName())
|
|
||||||
//}
|
|
||||||
@ -1,49 +0,0 @@
|
|||||||
package relation
|
|
||||||
|
|
||||||
//import (
|
|
||||||
// "context"
|
|
||||||
// "time"
|
|
||||||
//
|
|
||||||
// "github.com/OpenIMSDK/tools/errs"
|
|
||||||
// "github.com/OpenIMSDK/tools/ormutil"
|
|
||||||
// "gorm.io/gorm"
|
|
||||||
//
|
|
||||||
// relationtb "github.com/openimsdk/open-im-server/v3/pkg/common/db/table/relation"
|
|
||||||
//)
|
|
||||||
//
|
|
||||||
//type LogGorm struct {
|
|
||||||
// db *gorm.DB
|
|
||||||
//}
|
|
||||||
//
|
|
||||||
//func (l *LogGorm) Create(ctx context.Context, log []*relationtb.Log) error {
|
|
||||||
// return errs.Wrap(l.db.WithContext(ctx).Create(log).Error)
|
|
||||||
//}
|
|
||||||
//
|
|
||||||
//func (l *LogGorm) Search(ctx context.Context, keyword string, start time.Time, end time.Time, pageNumber int32, showNumber int32) (uint32, []*relationtb.Log, error) {
|
|
||||||
// db := l.db.WithContext(ctx).Where("create_time >= ?", start)
|
|
||||||
// if end.UnixMilli() != 0 {
|
|
||||||
// db = l.db.WithContext(ctx).Where("create_time <= ?", end)
|
|
||||||
// }
|
|
||||||
// db = db.Order("create_time desc")
|
|
||||||
// return ormutil.GormSearch[relationtb.Log](db, []string{"user_id"}, keyword, pageNumber, showNumber)
|
|
||||||
//}
|
|
||||||
//
|
|
||||||
//func (l *LogGorm) Delete(ctx context.Context, logIDs []string, userID string) error {
|
|
||||||
// if userID == "" {
|
|
||||||
// return errs.Wrap(l.db.WithContext(ctx).Where("log_id in ?", logIDs).Delete(&relationtb.Log{}).Error)
|
|
||||||
// }
|
|
||||||
// return errs.Wrap(l.db.WithContext(ctx).Where("log_id in ? and user_id=?", logIDs, userID).Delete(&relationtb.Log{}).Error)
|
|
||||||
//}
|
|
||||||
//
|
|
||||||
//func (l *LogGorm) Get(ctx context.Context, logIDs []string, userID string) ([]*relationtb.Log, error) {
|
|
||||||
// var logs []*relationtb.Log
|
|
||||||
// if userID == "" {
|
|
||||||
// return logs, errs.Wrap(l.db.WithContext(ctx).Where("log_id in ?", logIDs).Find(&logs).Error)
|
|
||||||
// }
|
|
||||||
// return logs, errs.Wrap(l.db.WithContext(ctx).Where("log_id in ? and user_id=?", logIDs, userID).Find(&logs).Error)
|
|
||||||
//}
|
|
||||||
//
|
|
||||||
//func NewLogGorm(db *gorm.DB) relationtb.LogInterface {
|
|
||||||
// db.AutoMigrate(&relationtb.Log{})
|
|
||||||
// return &LogGorm{db: db}
|
|
||||||
//}
|
|
||||||
@ -1,38 +0,0 @@
|
|||||||
// 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 (
|
|
||||||
"context"
|
|
||||||
|
|
||||||
"gorm.io/gorm"
|
|
||||||
)
|
|
||||||
|
|
||||||
type MetaDB struct {
|
|
||||||
DB *gorm.DB
|
|
||||||
table any
|
|
||||||
}
|
|
||||||
|
|
||||||
func NewMetaDB(db *gorm.DB, table any) *MetaDB {
|
|
||||||
return &MetaDB{
|
|
||||||
DB: db,
|
|
||||||
table: table,
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
func (g *MetaDB) db(ctx context.Context) *gorm.DB {
|
|
||||||
db := g.DB.WithContext(ctx).Model(g.table)
|
|
||||||
return db
|
|
||||||
}
|
|
||||||
@ -1,157 +0,0 @@
|
|||||||
// 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 (
|
|
||||||
"fmt"
|
|
||||||
"time"
|
|
||||||
|
|
||||||
"github.com/OpenIMSDK/tools/errs"
|
|
||||||
"github.com/OpenIMSDK/tools/log"
|
|
||||||
"github.com/OpenIMSDK/tools/mw/specialerror"
|
|
||||||
mysqldriver "github.com/go-sql-driver/mysql"
|
|
||||||
"gorm.io/driver/mysql"
|
|
||||||
"gorm.io/gorm"
|
|
||||||
"gorm.io/gorm/logger"
|
|
||||||
|
|
||||||
"github.com/openimsdk/open-im-server/v3/pkg/common/config"
|
|
||||||
)
|
|
||||||
|
|
||||||
const (
|
|
||||||
maxRetry = 100 // number of retries
|
|
||||||
)
|
|
||||||
|
|
||||||
type option struct {
|
|
||||||
Username string
|
|
||||||
Password string
|
|
||||||
Address []string
|
|
||||||
Database string
|
|
||||||
LogLevel int
|
|
||||||
SlowThreshold int
|
|
||||||
MaxLifeTime int
|
|
||||||
MaxOpenConn int
|
|
||||||
MaxIdleConn int
|
|
||||||
Connect func(dsn string, maxRetry int) (*gorm.DB, error)
|
|
||||||
}
|
|
||||||
|
|
||||||
// newMysqlGormDB Initialize the database connection.
|
|
||||||
func newMysqlGormDB(o *option) (*gorm.DB, error) {
|
|
||||||
err := maybeCreateTable(o)
|
|
||||||
if err != nil {
|
|
||||||
return nil, err
|
|
||||||
}
|
|
||||||
dsn := fmt.Sprintf("%s:%s@tcp(%s)/%s?charset=utf8mb4&parseTime=true&loc=Local",
|
|
||||||
o.Username, o.Password, o.Address[0], o.Database)
|
|
||||||
sqlLogger := log.NewSqlLogger(
|
|
||||||
logger.LogLevel(o.LogLevel),
|
|
||||||
true,
|
|
||||||
time.Duration(o.SlowThreshold)*time.Millisecond,
|
|
||||||
)
|
|
||||||
db, err := gorm.Open(mysql.Open(dsn), &gorm.Config{
|
|
||||||
Logger: sqlLogger,
|
|
||||||
})
|
|
||||||
if err != nil {
|
|
||||||
return nil, err
|
|
||||||
}
|
|
||||||
sqlDB, err := db.DB()
|
|
||||||
if err != nil {
|
|
||||||
return nil, err
|
|
||||||
}
|
|
||||||
sqlDB.SetConnMaxLifetime(time.Second * time.Duration(o.MaxLifeTime))
|
|
||||||
sqlDB.SetMaxOpenConns(o.MaxOpenConn)
|
|
||||||
sqlDB.SetMaxIdleConns(o.MaxIdleConn)
|
|
||||||
return db, nil
|
|
||||||
}
|
|
||||||
|
|
||||||
// maybeCreateTable creates a database if it does not exists.
|
|
||||||
func maybeCreateTable(o *option) error {
|
|
||||||
dsn := fmt.Sprintf("%s:%s@tcp(%s)/%s?charset=utf8mb4&parseTime=true&loc=Local",
|
|
||||||
o.Username, o.Password, o.Address[0], "mysql")
|
|
||||||
|
|
||||||
var db *gorm.DB
|
|
||||||
var err error
|
|
||||||
if f := o.Connect; f != nil {
|
|
||||||
db, err = f(dsn, maxRetry)
|
|
||||||
} else {
|
|
||||||
db, err = connectToDatabase(dsn, maxRetry)
|
|
||||||
}
|
|
||||||
if err != nil {
|
|
||||||
panic(err.Error() + " Open failed " + dsn)
|
|
||||||
}
|
|
||||||
|
|
||||||
sqlDB, err := db.DB()
|
|
||||||
if err != nil {
|
|
||||||
return err
|
|
||||||
}
|
|
||||||
defer sqlDB.Close()
|
|
||||||
sql := fmt.Sprintf(
|
|
||||||
"CREATE DATABASE IF NOT EXISTS `%s` default charset utf8mb4 COLLATE utf8mb4_unicode_ci",
|
|
||||||
o.Database,
|
|
||||||
)
|
|
||||||
err = db.Exec(sql).Error
|
|
||||||
if err != nil {
|
|
||||||
return fmt.Errorf("init db %w", err)
|
|
||||||
}
|
|
||||||
return nil
|
|
||||||
}
|
|
||||||
|
|
||||||
// connectToDatabase Connection retry for mysql.
|
|
||||||
func connectToDatabase(dsn string, maxRetry int) (*gorm.DB, error) {
|
|
||||||
var db *gorm.DB
|
|
||||||
var err error
|
|
||||||
for i := 0; i <= maxRetry; i++ {
|
|
||||||
db, err = gorm.Open(mysql.Open(dsn), nil)
|
|
||||||
if err == nil {
|
|
||||||
return db, nil
|
|
||||||
}
|
|
||||||
if mysqlErr, ok := err.(*mysqldriver.MySQLError); ok && mysqlErr.Number == 1045 {
|
|
||||||
return nil, err
|
|
||||||
}
|
|
||||||
time.Sleep(time.Duration(1) * time.Second)
|
|
||||||
}
|
|
||||||
return nil, err
|
|
||||||
}
|
|
||||||
|
|
||||||
// NewGormDB gorm mysql.
|
|
||||||
func NewGormDB() (*gorm.DB, error) {
|
|
||||||
specialerror.AddReplace(gorm.ErrRecordNotFound, errs.ErrRecordNotFound)
|
|
||||||
specialerror.AddErrHandler(replaceDuplicateKey)
|
|
||||||
|
|
||||||
return newMysqlGormDB(&option{
|
|
||||||
Username: config.Config.Mysql.Username,
|
|
||||||
Password: config.Config.Mysql.Password,
|
|
||||||
Address: config.Config.Mysql.Address,
|
|
||||||
Database: config.Config.Mysql.Database,
|
|
||||||
LogLevel: config.Config.Mysql.LogLevel,
|
|
||||||
SlowThreshold: config.Config.Mysql.SlowThreshold,
|
|
||||||
MaxLifeTime: config.Config.Mysql.MaxLifeTime,
|
|
||||||
MaxOpenConn: config.Config.Mysql.MaxOpenConn,
|
|
||||||
MaxIdleConn: config.Config.Mysql.MaxIdleConn,
|
|
||||||
})
|
|
||||||
}
|
|
||||||
|
|
||||||
func replaceDuplicateKey(err error) errs.CodeError {
|
|
||||||
if IsMysqlDuplicateKey(err) {
|
|
||||||
return errs.ErrDuplicateKey
|
|
||||||
}
|
|
||||||
return nil
|
|
||||||
}
|
|
||||||
|
|
||||||
func IsMysqlDuplicateKey(err error) bool {
|
|
||||||
if mysqlErr, ok := err.(*mysqldriver.MySQLError); ok {
|
|
||||||
return mysqlErr.Number == 1062
|
|
||||||
}
|
|
||||||
return false
|
|
||||||
}
|
|
||||||
@ -1,121 +0,0 @@
|
|||||||
package relation
|
|
||||||
|
|
||||||
import (
|
|
||||||
"context"
|
|
||||||
"database/sql"
|
|
||||||
"database/sql/driver"
|
|
||||||
"errors"
|
|
||||||
"fmt"
|
|
||||||
"reflect"
|
|
||||||
"testing"
|
|
||||||
|
|
||||||
"gorm.io/driver/mysql"
|
|
||||||
"gorm.io/gorm"
|
|
||||||
"gorm.io/gorm/logger"
|
|
||||||
)
|
|
||||||
|
|
||||||
func TestMaybeCreateTable(t *testing.T) {
|
|
||||||
t.Run("normal", func(t *testing.T) {
|
|
||||||
err := maybeCreateTable(&option{
|
|
||||||
Username: "root",
|
|
||||||
Password: "openIM123",
|
|
||||||
Address: []string{"172.28.0.1:13306"},
|
|
||||||
Database: "openIM_v3",
|
|
||||||
LogLevel: 4,
|
|
||||||
SlowThreshold: 500,
|
|
||||||
MaxOpenConn: 1000,
|
|
||||||
MaxIdleConn: 100,
|
|
||||||
MaxLifeTime: 60,
|
|
||||||
Connect: connect(expectExec{
|
|
||||||
query: "CREATE DATABASE IF NOT EXISTS `openIM_v3` default charset utf8mb4 COLLATE utf8mb4_unicode_ci",
|
|
||||||
args: nil,
|
|
||||||
}),
|
|
||||||
})
|
|
||||||
if err != nil {
|
|
||||||
t.Fatal(err)
|
|
||||||
}
|
|
||||||
})
|
|
||||||
|
|
||||||
t.Run("im-db", func(t *testing.T) {
|
|
||||||
err := maybeCreateTable(&option{
|
|
||||||
Username: "root",
|
|
||||||
Password: "openIM123",
|
|
||||||
Address: []string{"172.28.0.1:13306"},
|
|
||||||
Database: "im-db",
|
|
||||||
LogLevel: 4,
|
|
||||||
SlowThreshold: 500,
|
|
||||||
MaxOpenConn: 1000,
|
|
||||||
MaxIdleConn: 100,
|
|
||||||
MaxLifeTime: 60,
|
|
||||||
Connect: connect(expectExec{
|
|
||||||
query: "CREATE DATABASE IF NOT EXISTS `im-db` default charset utf8mb4 COLLATE utf8mb4_unicode_ci",
|
|
||||||
args: nil,
|
|
||||||
}),
|
|
||||||
})
|
|
||||||
if err != nil {
|
|
||||||
t.Fatal(err)
|
|
||||||
}
|
|
||||||
})
|
|
||||||
|
|
||||||
t.Run("err", func(t *testing.T) {
|
|
||||||
e := errors.New("e")
|
|
||||||
err := maybeCreateTable(&option{
|
|
||||||
Username: "root",
|
|
||||||
Password: "openIM123",
|
|
||||||
Address: []string{"172.28.0.1:13306"},
|
|
||||||
Database: "openIM_v3",
|
|
||||||
LogLevel: 4,
|
|
||||||
SlowThreshold: 500,
|
|
||||||
MaxOpenConn: 1000,
|
|
||||||
MaxIdleConn: 100,
|
|
||||||
MaxLifeTime: 60,
|
|
||||||
Connect: connect(expectExec{
|
|
||||||
err: e,
|
|
||||||
}),
|
|
||||||
})
|
|
||||||
if !errors.Is(err, e) {
|
|
||||||
t.Fatalf("err not is e: %v", err)
|
|
||||||
}
|
|
||||||
})
|
|
||||||
}
|
|
||||||
|
|
||||||
func connect(e expectExec) func(string, int) (*gorm.DB, error) {
|
|
||||||
return func(string, int) (*gorm.DB, error) {
|
|
||||||
return gorm.Open(mysql.New(mysql.Config{
|
|
||||||
SkipInitializeWithVersion: true,
|
|
||||||
Conn: sql.OpenDB(e),
|
|
||||||
}), &gorm.Config{
|
|
||||||
Logger: logger.Discard,
|
|
||||||
})
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
type expectExec struct {
|
|
||||||
err error
|
|
||||||
query string
|
|
||||||
args []driver.NamedValue
|
|
||||||
}
|
|
||||||
|
|
||||||
func (c expectExec) ExecContext(ctx context.Context, query string, args []driver.NamedValue) (driver.Result, error) {
|
|
||||||
if c.err != nil {
|
|
||||||
return nil, c.err
|
|
||||||
}
|
|
||||||
if query != c.query {
|
|
||||||
return nil, fmt.Errorf("query mismatch. expect: %s, got: %s", c.query, query)
|
|
||||||
}
|
|
||||||
if reflect.DeepEqual(args, c.args) {
|
|
||||||
return nil, fmt.Errorf("args mismatch. expect: %v, got: %v", c.args, args)
|
|
||||||
}
|
|
||||||
return noEffectResult{}, nil
|
|
||||||
}
|
|
||||||
|
|
||||||
func (e expectExec) Connect(context.Context) (driver.Conn, error) { return e, nil }
|
|
||||||
func (expectExec) Driver() driver.Driver { panic("not implemented") }
|
|
||||||
func (expectExec) Prepare(query string) (driver.Stmt, error) { panic("not implemented") }
|
|
||||||
func (expectExec) Close() (e error) { return }
|
|
||||||
func (expectExec) Begin() (driver.Tx, error) { panic("not implemented") }
|
|
||||||
|
|
||||||
type noEffectResult struct{}
|
|
||||||
|
|
||||||
func (noEffectResult) LastInsertId() (i int64, e error) { return }
|
|
||||||
func (noEffectResult) RowsAffected() (i int64, e error) { return }
|
|
||||||
@ -1,54 +0,0 @@
|
|||||||
// 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 (
|
|
||||||
// "context"
|
|
||||||
//
|
|
||||||
// "gorm.io/gorm"
|
|
||||||
//
|
|
||||||
// "github.com/OpenIMSDK/tools/errs"
|
|
||||||
//
|
|
||||||
// "github.com/openimsdk/open-im-server/v3/pkg/common/db/table/relation"
|
|
||||||
//)
|
|
||||||
//
|
|
||||||
//type ObjectInfoGorm struct {
|
|
||||||
// *MetaDB
|
|
||||||
//}
|
|
||||||
//
|
|
||||||
//func NewObjectInfo(db *gorm.DB) relation.ObjectInfoModelInterface {
|
|
||||||
// return &ObjectInfoGorm{
|
|
||||||
// NewMetaDB(db, &relation.ObjectModel{}),
|
|
||||||
// }
|
|
||||||
//}
|
|
||||||
//
|
|
||||||
//func (o *ObjectInfoGorm) NewTx(tx any) relation.ObjectInfoModelInterface {
|
|
||||||
// return &ObjectInfoGorm{
|
|
||||||
// NewMetaDB(tx.(*gorm.DB), &relation.ObjectModel{}),
|
|
||||||
// }
|
|
||||||
//}
|
|
||||||
//
|
|
||||||
//func (o *ObjectInfoGorm) SetObject(ctx context.Context, obj *relation.ObjectModel) (err error) {
|
|
||||||
// if err := o.DB.WithContext(ctx).Where("name = ?", obj.Name).FirstOrCreate(obj).Error; err != nil {
|
|
||||||
// return errs.Wrap(err)
|
|
||||||
// }
|
|
||||||
// return nil
|
|
||||||
//}
|
|
||||||
//
|
|
||||||
//func (o *ObjectInfoGorm) Take(ctx context.Context, name string) (info *relation.ObjectModel, err error) {
|
|
||||||
// info = &relation.ObjectModel{}
|
|
||||||
// return info, errs.Wrap(o.DB.WithContext(ctx).Where("name = ?", name).Take(info).Error)
|
|
||||||
//}
|
|
||||||
@ -1,138 +0,0 @@
|
|||||||
// 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 (
|
|
||||||
// "context"
|
|
||||||
// "time"
|
|
||||||
//
|
|
||||||
// "github.com/OpenIMSDK/tools/errs"
|
|
||||||
//
|
|
||||||
// "gorm.io/gorm"
|
|
||||||
//
|
|
||||||
// "github.com/OpenIMSDK/tools/utils"
|
|
||||||
//
|
|
||||||
// "github.com/openimsdk/open-im-server/v3/pkg/common/db/table/relation"
|
|
||||||
//)
|
|
||||||
//
|
|
||||||
//type UserGorm struct {
|
|
||||||
// *MetaDB
|
|
||||||
//}
|
|
||||||
//
|
|
||||||
//func NewUserGorm(db *gorm.DB) relation.UserModelInterface {
|
|
||||||
// //return &UserGorm{NewMetaDB(db, &relation.UserModel{})}
|
|
||||||
// return nil
|
|
||||||
//}
|
|
||||||
//
|
|
||||||
//// 插入多条.
|
|
||||||
//func (u *UserGorm) Create(ctx context.Context, users []*relation.UserModel) (err error) {
|
|
||||||
// return utils.Wrap(u.db(ctx).Create(&users).Error, "")
|
|
||||||
//}
|
|
||||||
//
|
|
||||||
//// 更新用户信息 零值.
|
|
||||||
//func (u *UserGorm) UpdateByMap(ctx context.Context, userID string, args map[string]any) (err error) {
|
|
||||||
// return utils.Wrap(u.db(ctx).Model(&relation.UserModel{}).Where("user_id = ?", userID).Updates(args).Error, "")
|
|
||||||
//}
|
|
||||||
//
|
|
||||||
//// 更新多个用户信息 非零值.
|
|
||||||
//func (u *UserGorm) Update(ctx context.Context, user *relation.UserModel) (err error) {
|
|
||||||
// return utils.Wrap(u.db(ctx).Model(user).Updates(user).Error, "")
|
|
||||||
//}
|
|
||||||
//
|
|
||||||
//// 获取指定用户信息 不存在，也不返回错误.
|
|
||||||
//func (u *UserGorm) Find(ctx context.Context, userIDs []string) (users []*relation.UserModel, err error) {
|
|
||||||
// err = utils.Wrap(u.db(ctx).Where("user_id in (?)", userIDs).Find(&users).Error, "")
|
|
||||||
// return users, err
|
|
||||||
//}
|
|
||||||
//
|
|
||||||
//// 获取某个用户信息 不存在，则返回错误.
|
|
||||||
//func (u *UserGorm) Take(ctx context.Context, userID string) (user *relation.UserModel, err error) {
|
|
||||||
// user = &relation.UserModel{}
|
|
||||||
// err = utils.Wrap(u.db(ctx).Where("user_id = ?", userID).Take(&user).Error, "")
|
|
||||||
// return user, err
|
|
||||||
//}
|
|
||||||
//
|
|
||||||
//// 获取用户信息 不存在，不返回错误.
|
|
||||||
//func (u *UserGorm) Page(
|
|
||||||
// ctx context.Context,
|
|
||||||
// pageNumber, showNumber int32,
|
|
||||||
//) (users []*relation.UserModel, count int64, err error) {
|
|
||||||
// err = utils.Wrap(u.db(ctx).Count(&count).Error, "")
|
|
||||||
// if err != nil {
|
|
||||||
// return
|
|
||||||
// }
|
|
||||||
// err = utils.Wrap(
|
|
||||||
// u.db(ctx).
|
|
||||||
// Limit(int(showNumber)).
|
|
||||||
// Offset(int((pageNumber-1)*showNumber)).
|
|
||||||
// Find(&users).
|
|
||||||
// Order("create_time DESC").
|
|
||||||
// Error,
|
|
||||||
// "",
|
|
||||||
// )
|
|
||||||
// return
|
|
||||||
//}
|
|
||||||
//
|
|
||||||
//// 获取所有用户ID.
|
|
||||||
//func (u *UserGorm) GetAllUserID(ctx context.Context, pageNumber, showNumber int32) (userIDs []string, err error) {
|
|
||||||
// if pageNumber == 0 || showNumber == 0 {
|
|
||||||
// return userIDs, errs.Wrap(u.db(ctx).Pluck("user_id", &userIDs).Error)
|
|
||||||
// } else {
|
|
||||||
// return userIDs, errs.Wrap(u.db(ctx).Limit(int(showNumber)).Offset(int((pageNumber-1)*showNumber)).Pluck("user_id", &userIDs).Error)
|
|
||||||
// }
|
|
||||||
//}
|
|
||||||
//
|
|
||||||
//func (u *UserGorm) GetUserGlobalRecvMsgOpt(ctx context.Context, userID string) (opt int, err error) {
|
|
||||||
// err = u.db(ctx).Model(&relation.UserModel{}).Where("user_id = ?", userID).Pluck("global_recv_msg_opt", &opt).Error
|
|
||||||
// return opt, err
|
|
||||||
//}
|
|
||||||
//
|
|
||||||
//func (u *UserGorm) CountTotal(ctx context.Context, before *time.Time) (count int64, err error) {
|
|
||||||
// db := u.db(ctx).Model(&relation.UserModel{})
|
|
||||||
// if before != nil {
|
|
||||||
// db = db.Where("create_time < ?", before)
|
|
||||||
// }
|
|
||||||
// if err := db.Count(&count).Error; err != nil {
|
|
||||||
// return 0, err
|
|
||||||
// }
|
|
||||||
// return count, nil
|
|
||||||
//}
|
|
||||||
//
|
|
||||||
//func (u *UserGorm) CountRangeEverydayTotal(
|
|
||||||
// ctx context.Context,
|
|
||||||
// start time.Time,
|
|
||||||
// end time.Time,
|
|
||||||
//) (map[string]int64, error) {
|
|
||||||
// var res []struct {
|
|
||||||
// Date time.Time `gorm:"column:date"`
|
|
||||||
// Count int64 `gorm:"column:count"`
|
|
||||||
// }
|
|
||||||
// err := u.db(ctx).
|
|
||||||
// Model(&relation.UserModel{}).
|
|
||||||
// Select("DATE(create_time) AS date, count(1) AS count").
|
|
||||||
// Where("create_time >= ? and create_time < ?", start, end).
|
|
||||||
// Group("date").
|
|
||||||
// Find(&res).
|
|
||||||
// Error
|
|
||||||
// if err != nil {
|
|
||||||
// return nil, errs.Wrap(err)
|
|
||||||
// }
|
|
||||||
// v := make(map[string]int64)
|
|
||||||
// for _, r := range res {
|
|
||||||
// v[r.Date.Format("2006-01-02")] = r.Count
|
|
||||||
// }
|
|
||||||
// return v, nil
|
|
||||||
//}
|
|
||||||
@ -1 +0,0 @@
|
|||||||
package kodo
|
|
||||||
@ -1,323 +0,0 @@
|
|||||||
package kodo
|
|
||||||
|
|
||||||
import (
|
|
||||||
"context"
|
|
||||||
"errors"
|
|
||||||
"fmt"
|
|
||||||
"net/http"
|
|
||||||
"net/url"
|
|
||||||
"strconv"
|
|
||||||
"strings"
|
|
||||||
"time"
|
|
||||||
|
|
||||||
"github.com/aws/aws-sdk-go-v2/aws"
|
|
||||||
awss3config "github.com/aws/aws-sdk-go-v2/config"
|
|
||||||
"github.com/aws/aws-sdk-go-v2/credentials"
|
|
||||||
awss3 "github.com/aws/aws-sdk-go-v2/service/s3"
|
|
||||||
awss3types "github.com/aws/aws-sdk-go-v2/service/s3/types"
|
|
||||||
"github.com/openimsdk/open-im-server/v3/pkg/common/config"
|
|
||||||
"github.com/openimsdk/open-im-server/v3/pkg/common/db/s3"
|
|
||||||
"github.com/qiniu/go-sdk/v7/auth"
|
|
||||||
"github.com/qiniu/go-sdk/v7/storage"
|
|
||||||
)
|
|
||||||
|
|
||||||
const (
|
|
||||||
minPartSize = 1024 * 1024 * 1 // 1MB
|
|
||||||
maxPartSize = 1024 * 1024 * 1024 * 5 // 5GB
|
|
||||||
maxNumSize = 10000
|
|
||||||
)
|
|
||||||
|
|
||||||
type Kodo struct {
|
|
||||||
AccessKey string
|
|
||||||
SecretKey string
|
|
||||||
Region string
|
|
||||||
Token string
|
|
||||||
Endpoint string
|
|
||||||
BucketURL string
|
|
||||||
Auth *auth.Credentials
|
|
||||||
Client *awss3.Client
|
|
||||||
PresignClient *awss3.PresignClient
|
|
||||||
}
|
|
||||||
|
|
||||||
func NewKodo() (s3.Interface, error) {
|
|
||||||
conf := config.Config.Object.Kodo
|
|
||||||
//init client
|
|
||||||
cfg, err := awss3config.LoadDefaultConfig(context.TODO(),
|
|
||||||
awss3config.WithRegion(conf.Bucket),
|
|
||||||
awss3config.WithEndpointResolverWithOptions(
|
|
||||||
aws.EndpointResolverWithOptionsFunc(func(service, region string, options ...interface{}) (aws.Endpoint, error) {
|
|
||||||
return aws.Endpoint{URL: conf.Endpoint}, nil
|
|
||||||
})),
|
|
||||||
awss3config.WithCredentialsProvider(credentials.NewStaticCredentialsProvider(
|
|
||||||
conf.AccessKeyID,
|
|
||||||
conf.AccessKeySecret,
|
|
||||||
conf.SessionToken),
|
|
||||||
),
|
|
||||||
)
|
|
||||||
if err != nil {
|
|
||||||
panic(err)
|
|
||||||
}
|
|
||||||
client := awss3.NewFromConfig(cfg)
|
|
||||||
presignClient := awss3.NewPresignClient(client)
|
|
||||||
|
|
||||||
return &Kodo{
|
|
||||||
AccessKey: conf.AccessKeyID,
|
|
||||||
SecretKey: conf.AccessKeySecret,
|
|
||||||
Region: conf.Bucket,
|
|
||||||
BucketURL: conf.BucketURL,
|
|
||||||
Auth: auth.New(conf.AccessKeyID, conf.AccessKeySecret),
|
|
||||||
Client: client,
|
|
||||||
PresignClient: presignClient,
|
|
||||||
}, nil
|
|
||||||
}
|
|
||||||
|
|
||||||
func (k Kodo) Engine() string {
|
|
||||||
return "kodo"
|
|
||||||
}
|
|
||||||
|
|
||||||
func (k Kodo) PartLimit() *s3.PartLimit {
|
|
||||||
return &s3.PartLimit{
|
|
||||||
MinPartSize: minPartSize,
|
|
||||||
MaxPartSize: maxPartSize,
|
|
||||||
MaxNumSize: maxNumSize,
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
func (k Kodo) InitiateMultipartUpload(ctx context.Context, name string) (*s3.InitiateMultipartUploadResult, error) {
|
|
||||||
result, err := k.Client.CreateMultipartUpload(ctx, &awss3.CreateMultipartUploadInput{
|
|
||||||
Bucket: aws.String(k.Region),
|
|
||||||
Key: aws.String(name),
|
|
||||||
})
|
|
||||||
if err != nil {
|
|
||||||
return nil, err
|
|
||||||
}
|
|
||||||
return &s3.InitiateMultipartUploadResult{
|
|
||||||
UploadID: aws.ToString(result.UploadId),
|
|
||||||
Bucket: aws.ToString(result.Bucket),
|
|
||||||
Key: aws.ToString(result.Key),
|
|
||||||
}, nil
|
|
||||||
}
|
|
||||||
|
|
||||||
func (k Kodo) CompleteMultipartUpload(ctx context.Context, uploadID string, name string, parts []s3.Part) (*s3.CompleteMultipartUploadResult, error) {
|
|
||||||
kodoParts := make([]awss3types.CompletedPart, len(parts))
|
|
||||||
for i, part := range parts {
|
|
||||||
kodoParts[i] = awss3types.CompletedPart{
|
|
||||||
PartNumber: aws.Int32(int32(part.PartNumber)),
|
|
||||||
ETag: aws.String(part.ETag),
|
|
||||||
}
|
|
||||||
}
|
|
||||||
result, err := k.Client.CompleteMultipartUpload(ctx, &awss3.CompleteMultipartUploadInput{
|
|
||||||
Bucket: aws.String(k.Region),
|
|
||||||
Key: aws.String(name),
|
|
||||||
UploadId: aws.String(uploadID),
|
|
||||||
MultipartUpload: &awss3types.CompletedMultipartUpload{Parts: kodoParts},
|
|
||||||
})
|
|
||||||
if err != nil {
|
|
||||||
return nil, err
|
|
||||||
}
|
|
||||||
return &s3.CompleteMultipartUploadResult{
|
|
||||||
Location: aws.ToString(result.Location),
|
|
||||||
Bucket: aws.ToString(result.Bucket),
|
|
||||||
Key: aws.ToString(result.Key),
|
|
||||||
ETag: strings.ToLower(strings.ReplaceAll(aws.ToString(result.ETag), `"`, ``)),
|
|
||||||
}, nil
|
|
||||||
}
|
|
||||||
|
|
||||||
func (k Kodo) PartSize(ctx context.Context, size int64) (int64, error) {
|
|
||||||
if size <= 0 {
|
|
||||||
return 0, errors.New("size must be greater than 0")
|
|
||||||
}
|
|
||||||
if size > maxPartSize*maxNumSize {
|
|
||||||
return 0, fmt.Errorf("size must be less than %db", maxPartSize*maxNumSize)
|
|
||||||
}
|
|
||||||
if size <= minPartSize*maxNumSize {
|
|
||||||
return minPartSize, nil
|
|
||||||
}
|
|
||||||
partSize := size / maxNumSize
|
|
||||||
if size%maxNumSize != 0 {
|
|
||||||
partSize++
|
|
||||||
}
|
|
||||||
return partSize, nil
|
|
||||||
}
|
|
||||||
|
|
||||||
func (k Kodo) AuthSign(ctx context.Context, uploadID string, name string, expire time.Duration, partNumbers []int) (*s3.AuthSignResult, error) {
|
|
||||||
result := s3.AuthSignResult{
|
|
||||||
URL: k.BucketURL + "/" + name,
|
|
||||||
Query: url.Values{"uploadId": {uploadID}},
|
|
||||||
Header: make(http.Header),
|
|
||||||
Parts: make([]s3.SignPart, len(partNumbers)),
|
|
||||||
}
|
|
||||||
for i, partNumber := range partNumbers {
|
|
||||||
part, _ := k.PresignClient.PresignUploadPart(ctx, &awss3.UploadPartInput{
|
|
||||||
Bucket: aws.String(k.Region),
|
|
||||||
UploadId: aws.String(uploadID),
|
|
||||||
Key: aws.String(name),
|
|
||||||
PartNumber: aws.Int32(int32(partNumber)),
|
|
||||||
})
|
|
||||||
result.Parts[i] = s3.SignPart{
|
|
||||||
PartNumber: partNumber,
|
|
||||||
URL: part.URL,
|
|
||||||
Header: part.SignedHeader,
|
|
||||||
}
|
|
||||||
}
|
|
||||||
return &result, nil
|
|
||||||
|
|
||||||
}
|
|
||||||
|
|
||||||
func (k Kodo) PresignedPutObject(ctx context.Context, name string, expire time.Duration) (string, error) {
|
|
||||||
object, err := k.PresignClient.PresignPutObject(ctx, &awss3.PutObjectInput{
|
|
||||||
Bucket: aws.String(k.Region),
|
|
||||||
Key: aws.String(name),
|
|
||||||
}, func(po *awss3.PresignOptions) {
|
|
||||||
po.Expires = expire
|
|
||||||
})
|
|
||||||
return object.URL, err
|
|
||||||
|
|
||||||
}
|
|
||||||
|
|
||||||
func (k Kodo) DeleteObject(ctx context.Context, name string) error {
|
|
||||||
_, err := k.Client.DeleteObject(ctx, &awss3.DeleteObjectInput{
|
|
||||||
Bucket: aws.String(k.Region),
|
|
||||||
Key: aws.String(name),
|
|
||||||
})
|
|
||||||
return err
|
|
||||||
}
|
|
||||||
|
|
||||||
func (k Kodo) CopyObject(ctx context.Context, src string, dst string) (*s3.CopyObjectInfo, error) {
|
|
||||||
result, err := k.Client.CopyObject(ctx, &awss3.CopyObjectInput{
|
|
||||||
Bucket: aws.String(k.Region),
|
|
||||||
CopySource: aws.String(k.Region + "/" + src),
|
|
||||||
Key: aws.String(dst),
|
|
||||||
})
|
|
||||||
if err != nil {
|
|
||||||
return nil, err
|
|
||||||
}
|
|
||||||
return &s3.CopyObjectInfo{
|
|
||||||
Key: dst,
|
|
||||||
ETag: strings.ToLower(strings.ReplaceAll(aws.ToString(result.CopyObjectResult.ETag), `"`, ``)),
|
|
||||||
}, nil
|
|
||||||
}
|
|
||||||
|
|
||||||
func (k Kodo) StatObject(ctx context.Context, name string) (*s3.ObjectInfo, error) {
|
|
||||||
info, err := k.Client.HeadObject(ctx, &awss3.HeadObjectInput{
|
|
||||||
Bucket: aws.String(k.Region),
|
|
||||||
Key: aws.String(name),
|
|
||||||
})
|
|
||||||
if err != nil {
|
|
||||||
return nil, err
|
|
||||||
}
|
|
||||||
res := &s3.ObjectInfo{Key: name}
|
|
||||||
res.Size = aws.ToInt64(info.ContentLength)
|
|
||||||
res.ETag = strings.ToLower(strings.ReplaceAll(aws.ToString(info.ETag), `"`, ``))
|
|
||||||
return res, nil
|
|
||||||
}
|
|
||||||
|
|
||||||
func (k Kodo) IsNotFound(err error) bool {
|
|
||||||
return true
|
|
||||||
}
|
|
||||||
|
|
||||||
func (k Kodo) AbortMultipartUpload(ctx context.Context, uploadID string, name string) error {
|
|
||||||
_, err := k.Client.AbortMultipartUpload(ctx, &awss3.AbortMultipartUploadInput{
|
|
||||||
UploadId: aws.String(uploadID),
|
|
||||||
Bucket: aws.String(k.Region),
|
|
||||||
Key: aws.String(name),
|
|
||||||
})
|
|
||||||
return err
|
|
||||||
}
|
|
||||||
|
|
||||||
func (k Kodo) ListUploadedParts(ctx context.Context, uploadID string, name string, partNumberMarker int, maxParts int) (*s3.ListUploadedPartsResult, error) {
|
|
||||||
result, err := k.Client.ListParts(ctx, &awss3.ListPartsInput{
|
|
||||||
Key: aws.String(name),
|
|
||||||
UploadId: aws.String(uploadID),
|
|
||||||
Bucket: aws.String(k.Region),
|
|
||||||
MaxParts: aws.Int32(int32(maxParts)),
|
|
||||||
PartNumberMarker: aws.String(strconv.Itoa(partNumberMarker)),
|
|
||||||
})
|
|
||||||
if err != nil {
|
|
||||||
return nil, err
|
|
||||||
}
|
|
||||||
res := &s3.ListUploadedPartsResult{
|
|
||||||
Key: aws.ToString(result.Key),
|
|
||||||
UploadID: aws.ToString(result.UploadId),
|
|
||||||
MaxParts: int(aws.ToInt32(result.MaxParts)),
|
|
||||||
UploadedParts: make([]s3.UploadedPart, len(result.Parts)),
|
|
||||||
}
|
|
||||||
// int to string
|
|
||||||
NextPartNumberMarker, err := strconv.Atoi(aws.ToString(result.NextPartNumberMarker))
|
|
||||||
if err != nil {
|
|
||||||
return nil, err
|
|
||||||
}
|
|
||||||
res.NextPartNumberMarker = NextPartNumberMarker
|
|
||||||
for i, part := range result.Parts {
|
|
||||||
res.UploadedParts[i] = s3.UploadedPart{
|
|
||||||
PartNumber: int(aws.ToInt32(part.PartNumber)),
|
|
||||||
LastModified: aws.ToTime(part.LastModified),
|
|
||||||
ETag: aws.ToString(part.ETag),
|
|
||||||
Size: aws.ToInt64(part.Size),
|
|
||||||
}
|
|
||||||
}
|
|
||||||
return res, nil
|
|
||||||
}
|
|
||||||
|
|
||||||
func (k Kodo) AccessURL(ctx context.Context, name string, expire time.Duration, opt *s3.AccessURLOption) (string, error) {
|
|
||||||
//get object head
|
|
||||||
info, err := k.Client.HeadObject(ctx, &awss3.HeadObjectInput{
|
|
||||||
Bucket: aws.String(k.Region),
|
|
||||||
Key: aws.String(name),
|
|
||||||
})
|
|
||||||
if err != nil {
|
|
||||||
return "", errors.New("AccessURL object not found")
|
|
||||||
}
|
|
||||||
if opt != nil {
|
|
||||||
if opt.ContentType != aws.ToString(info.ContentType) {
|
|
||||||
//修改文件类型
|
|
||||||
err := k.SetObjectContentType(ctx, name, opt.ContentType)
|
|
||||||
if err != nil {
|
|
||||||
return "", errors.New("AccessURL setContentType error")
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
imageMogr := ""
|
|
||||||
//image dispose
|
|
||||||
if opt != nil {
|
|
||||||
if opt.Image != nil {
|
|
||||||
//https://developer.qiniu.com/dora/8255/the-zoom
|
|
||||||
process := ""
|
|
||||||
if opt.Image.Width > 0 {
|
|
||||||
process += strconv.Itoa(opt.Image.Width) + "x"
|
|
||||||
}
|
|
||||||
if opt.Image.Height > 0 {
|
|
||||||
if opt.Image.Width > 0 {
|
|
||||||
process += strconv.Itoa(opt.Image.Height)
|
|
||||||
} else {
|
|
||||||
process += "x" + strconv.Itoa(opt.Image.Height)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
imageMogr = "imageMogr2/thumbnail/" + process
|
|
||||||
}
|
|
||||||
}
|
|
||||||
//expire
|
|
||||||
deadline := time.Now().Add(time.Second * expire).Unix()
|
|
||||||
domain := k.BucketURL
|
|
||||||
query := url.Values{}
|
|
||||||
if opt != nil && opt.Filename != "" {
|
|
||||||
query.Add("attname", opt.Filename)
|
|
||||||
}
|
|
||||||
privateURL := storage.MakePrivateURLv2WithQuery(k.Auth, domain, name, query, deadline)
|
|
||||||
if imageMogr != "" {
|
|
||||||
privateURL += "&" + imageMogr
|
|
||||||
}
|
|
||||||
return privateURL, nil
|
|
||||||
}
|
|
||||||
|
|
||||||
func (k *Kodo) SetObjectContentType(ctx context.Context, name string, contentType string) error {
|
|
||||||
//set object content-type
|
|
||||||
_, err := k.Client.CopyObject(ctx, &awss3.CopyObjectInput{
|
|
||||||
Bucket: aws.String(k.Region),
|
|
||||||
CopySource: aws.String(k.Region + "/" + name),
|
|
||||||
Key: aws.String(name),
|
|
||||||
ContentType: aws.String(contentType),
|
|
||||||
MetadataDirective: awss3types.MetadataDirectiveReplace,
|
|
||||||
})
|
|
||||||
return err
|
|
||||||
}
|
|
||||||
@ -1,51 +0,0 @@
|
|||||||
// 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 (
|
|
||||||
"time"
|
|
||||||
|
|
||||||
pbmsg "github.com/OpenIMSDK/protocol/msg"
|
|
||||||
)
|
|
||||||
|
|
||||||
const (
|
|
||||||
ChatLogModelTableName = "chat_logs"
|
|
||||||
)
|
|
||||||
|
|
||||||
type ChatLogModel struct {
|
|
||||||
ServerMsgID string `gorm:"column:server_msg_id;primary_key;type:char(64)" json:"serverMsgID"`
|
|
||||||
ClientMsgID string `gorm:"column:client_msg_id;type:char(64)" json:"clientMsgID"`
|
|
||||||
SendID string `gorm:"column:send_id;type:char(64);index:send_id,priority:2" json:"sendID"`
|
|
||||||
RecvID string `gorm:"column:recv_id;type:char(64);index:recv_id,priority:2" json:"recvID"`
|
|
||||||
SenderPlatformID int32 `gorm:"column:sender_platform_id" json:"senderPlatformID"`
|
|
||||||
SenderNickname string `gorm:"column:sender_nick_name;type:varchar(255)" json:"senderNickname"`
|
|
||||||
SenderFaceURL string `gorm:"column:sender_face_url;type:varchar(255);" json:"senderFaceURL"`
|
|
||||||
SessionType int32 `gorm:"column:session_type;index:session_type,priority:2;index:session_type_alone" json:"sessionType"`
|
|
||||||
MsgFrom int32 `gorm:"column:msg_from" json:"msgFrom"`
|
|
||||||
ContentType int32 `gorm:"column:content_type;index:content_type,priority:2;index:content_type_alone" json:"contentType"`
|
|
||||||
Content string `gorm:"column:content;type:varchar(3000)" json:"content"`
|
|
||||||
Status int32 `gorm:"column:status" json:"status"`
|
|
||||||
SendTime time.Time `gorm:"column:send_time;index:sendTime;index:content_type,priority:1;index:session_type,priority:1;index:recv_id,priority:1;index:send_id,priority:1" json:"sendTime"`
|
|
||||||
CreateTime time.Time `gorm:"column:create_time" json:"createTime"`
|
|
||||||
Ex string `gorm:"column:ex;type:varchar(1024)" json:"ex"`
|
|
||||||
}
|
|
||||||
|
|
||||||
func (ChatLogModel) TableName() string {
|
|
||||||
return ChatLogModelTableName
|
|
||||||
}
|
|
||||||
|
|
||||||
type ChatLogModelInterface interface {
|
|
||||||
Create(msg *pbmsg.MsgDataToMQ) error
|
|
||||||
}
|
|
||||||
@ -1,15 +0,0 @@
|
|||||||
// 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/openimsdk/open-im-server/v3/pkg/common/db/table/relation"
|
|
||||||
@ -1,19 +0,0 @@
|
|||||||
package tx
|
|
||||||
|
|
||||||
import (
|
|
||||||
"context"
|
|
||||||
"github.com/OpenIMSDK/tools/tx"
|
|
||||||
"go.mongodb.org/mongo-driver/bson"
|
|
||||||
"go.mongodb.org/mongo-driver/mongo"
|
|
||||||
)
|
|
||||||
|
|
||||||
func NewAuto(ctx context.Context, cli *mongo.Client) (tx.CtxTx, error) {
|
|
||||||
var res map[string]any
|
|
||||||
if err := cli.Database("admin").RunCommand(ctx, bson.M{"isMaster": 1}).Decode(&res); err != nil {
|
|
||||||
return nil, err
|
|
||||||
}
|
|
||||||
if _, ok := res["setName"]; ok {
|
|
||||||
return NewMongoTx(cli), nil
|
|
||||||
}
|
|
||||||
return NewInvalidTx(), nil
|
|
||||||
}
|
|
||||||
@ -1,16 +0,0 @@
|
|||||||
package tx
|
|
||||||
|
|
||||||
import (
|
|
||||||
"context"
|
|
||||||
"github.com/OpenIMSDK/tools/tx"
|
|
||||||
)
|
|
||||||
|
|
||||||
func NewInvalidTx() tx.CtxTx {
|
|
||||||
return invalid{}
|
|
||||||
}
|
|
||||||
|
|
||||||
type invalid struct{}
|
|
||||||
|
|
||||||
func (m invalid) Transaction(ctx context.Context, fn func(ctx context.Context) error) error {
|
|
||||||
return fn(ctx)
|
|
||||||
}
|
|
||||||
@ -1,28 +0,0 @@
|
|||||||
package tx
|
|
||||||
|
|
||||||
import (
|
|
||||||
"context"
|
|
||||||
"github.com/OpenIMSDK/tools/tx"
|
|
||||||
"go.mongodb.org/mongo-driver/mongo"
|
|
||||||
)
|
|
||||||
|
|
||||||
func NewMongoTx(client *mongo.Client) tx.CtxTx {
|
|
||||||
return &mongoTx{
|
|
||||||
client: client,
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
type mongoTx struct {
|
|
||||||
client *mongo.Client
|
|
||||||
}
|
|
||||||
|
|
||||||
func (m *mongoTx) Transaction(ctx context.Context, fn func(ctx context.Context) error) error {
|
|
||||||
sess, err := m.client.StartSession()
|
|
||||||
if err != nil {
|
|
||||||
return err
|
|
||||||
}
|
|
||||||
_, err = sess.WithTransaction(ctx, func(ctx mongo.SessionContext) (interface{}, error) {
|
|
||||||
return nil, fn(ctx)
|
|
||||||
})
|
|
||||||
return err
|
|
||||||
}
|
|
||||||
Loading…
Reference in new issue