friend_request_model.go 1.9 KB
Newer Older
落凡尘.'s avatar
落凡尘. 已提交
1 2 3
package im_mysql_model

import (
4
	"Open_IM/pkg/common/db"
落凡尘.'s avatar
落凡尘. 已提交
5 6 7
	"time"
)

W
wenxu12345 已提交
8
// reqId add userId
9
func ReplaceIntoFriendReq(reqId, userId string, flag int32, reqMessage string) error {
落凡尘.'s avatar
落凡尘. 已提交
10 11 12 13
	dbConn, err := db.DB.MysqlDB.DefaultGormDB()
	if err != nil {
		return err
	}
14
	err = dbConn.Exec("replace into friend_request(req_id,user_id,flag,req_message,create_time) values(?,?,?,?,?)", reqId, userId, flag, reqMessage, time.Now()).Error
落凡尘.'s avatar
落凡尘. 已提交
15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34
	if err != nil {
		return err
	}
	return nil
}

func FindFriendsApplyFromFriendReq(userId string) ([]FriendRequest, error) {
	dbConn, err := db.DB.MysqlDB.DefaultGormDB()
	if err != nil {
		return nil, err
	}
	var usersInfo []FriendRequest
	//dbConn.LogMode(true)
	err = dbConn.Table("friend_request").Where("user_id=?", userId).Find(&usersInfo).Error
	if err != nil {
		return nil, err
	}
	return usersInfo, nil
}

落凡尘.'s avatar
落凡尘. 已提交
35 36 37 38 39 40 41 42 43 44 45 46 47
func FindSelfApplyFromFriendReq(userId string) ([]FriendRequest, error) {
	dbConn, err := db.DB.MysqlDB.DefaultGormDB()
	if err != nil {
		return nil, err
	}
	var usersInfo []FriendRequest
	err = dbConn.Table("friend_request").Where("req_id=?", userId).Find(&usersInfo).Error
	if err != nil {
		return nil, err
	}
	return usersInfo, nil
}

W
wenxu12345 已提交
48
//reqId apply to add userId already
落凡尘.'s avatar
落凡尘. 已提交
49
func FindFriendApplyFromFriendReqByUid(reqId, userId string) (*FriendRequest, error) {
落凡尘.'s avatar
落凡尘. 已提交
50 51 52 53 54 55 56 57 58 59 60 61
	dbConn, err := db.DB.MysqlDB.DefaultGormDB()
	if err != nil {
		return nil, err
	}
	var friendRequest FriendRequest
	err = dbConn.Table("friend_request").Where("req_id=? and user_id=?", reqId, userId).Find(&friendRequest).Error
	if err != nil {
		return nil, err
	}
	return &friendRequest, nil
}

W
wenxu12345 已提交
62
//userId process reqId
落凡尘.'s avatar
落凡尘. 已提交
63 64 65 66 67 68 69 70 71 72 73
func UpdateFriendRelationshipToFriendReq(reqId, userId string, flag int32) error {
	dbConn, err := db.DB.MysqlDB.DefaultGormDB()
	if err != nil {
		return err
	}
	err = dbConn.Exec("update friend_request set flag=? where req_id=? and user_id=?", flag, reqId, userId).Error
	if err != nil {
		return err
	}
	return nil
}