friend_request_model.go

package im_mysql_model

import (
	"time"

	"github.com/erbaner/be/pkg/common/db"
	"github.com/erbaner/be/pkg/utils"
)

//type FriendRequest struct {
//	FromUserID    string    `gorm:"column:from_user_id;primaryKey;"`
//	ToUserID      string    `gorm:"column:to_user_id;primaryKey;"`
//	HandleResult  int32     `gorm:"column:handle_result"`
//	ReqMessage    string    `gorm:"column:req_message"`
//	CreateTime    time.Time `gorm:"column:create_time"`
//	HandlerUserID string    `gorm:"column:handler_user_id"`
//	HandleMsg     string    `gorm:"column:handle_msg"`
//	HandleTime    time.Time `gorm:"column:handle_time"`
//	Ex            string    `gorm:"column:ex"`
//}

// who apply to add me
func GetReceivedFriendsApplicationListByUserID(ToUserID string) ([]db.FriendRequest, error) {
	var usersInfo []db.FriendRequest
	err := db.DB.MysqlDB.DefaultGormDB().Table("friend_requests").Where("to_user_id=?", ToUserID).Find(&usersInfo).Error
	if err != nil {
		return nil, err
	}
	return usersInfo, nil
}

// I apply to add somebody
func GetSendFriendApplicationListByUserID(FromUserID string) ([]db.FriendRequest, error) {
	var usersInfo []db.FriendRequest
	err := db.DB.MysqlDB.DefaultGormDB().Table("friend_requests").Where("from_user_id=?", FromUserID).Find(&usersInfo).Error
	if err != nil {
		return nil, err
	}
	return usersInfo, nil
}

// FromUserId apply to add ToUserID
func GetFriendApplicationByBothUserID(FromUserID, ToUserID string) (*db.FriendRequest, error) {
	var friendRequest db.FriendRequest
	err := db.DB.MysqlDB.DefaultGormDB().Table("friend_requests").Where("from_user_id=? and to_user_id=?", FromUserID, ToUserID).Take(&friendRequest).Error
	if err != nil {
		return nil, err
	}
	return &friendRequest, nil
}

func UpdateFriendApplication(friendRequest *db.FriendRequest) error {
	friendRequest.CreateTime = time.Now()
	return db.DB.MysqlDB.DefaultGormDB().Table("friend_requests").Where("from_user_id=? and to_user_id=?",
		friendRequest.FromUserID, friendRequest.ToUserID).Updates(&friendRequest).Error
}

func InsertFriendApplication(friendRequest *db.FriendRequest, args map[string]interface{}) error {
	if err := db.DB.MysqlDB.DefaultGormDB().Table("friend_requests").Create(friendRequest).Error; err == nil {
		return nil
	}

	//t := dbConn.Debug().Table("friend_requests").Where("from_user_id = ? and to_user_id = ?", friendRequest.FromUserID, friendRequest.ToUserID).Select("*").Updates(*friendRequest)
	//if t.RowsAffected == 0 {
	//	return utils.Wrap(errors.New("RowsAffected == 0"), "no update")
	//}
	//return utils.Wrap(t.Error, "")

	friendRequest.CreateTime = time.Now()
	args["create_time"] = friendRequest.CreateTime
	u := db.DB.MysqlDB.DefaultGormDB().Model(friendRequest).Updates(args)
	//u := dbConn.Table("friend_requests").Where("from_user_id=? and to_user_id=?",
	// friendRequest.FromUserID, friendRequest.ToUserID).Update(&friendRequest)
	//u := dbConn.Table("friend_requests").Where("from_user_id=? and to_user_id=?",
	//	friendRequest.FromUserID, friendRequest.ToUserID).Update(&friendRequest)
	if u.RowsAffected != 0 {
		return nil
	}

	if friendRequest.CreateTime.Unix() < 0 {
		friendRequest.CreateTime = time.Now()
	}
	if friendRequest.HandleTime.Unix() < 0 {
		friendRequest.HandleTime = utils.UnixSecondToTime(0)
	}
	err := db.DB.MysqlDB.DefaultGormDB().Table("friend_requests").Create(friendRequest).Error
	if err != nil {
		return err
	}
	return nil
}