mirror of
https://github.com/openimsdk/open-im-server.git
synced 2025-12-03 10:52:33 +08:00
155 lines
4.8 KiB
Go
155 lines
4.8 KiB
Go
// 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
|
|
// }
|