83 lines
1.9 KiB
Go
83 lines
1.9 KiB
Go
|
package models
|
||
|
|
|
||
|
|
import (
|
||
|
|
"kefu/types"
|
||
|
|
"time"
|
||
|
|
)
|
||
|
|
|
||
|
|
type Oauth struct {
|
||
|
|
ID uint `gorm:"primary_key" json:"id"`
|
||
|
|
UserId string `json:"user_id"`
|
||
|
|
OauthId string `json:"oauth_id"`
|
||
|
|
status uint `json:"status"`
|
||
|
|
CreatedAt types.Time `json:"created_at"`
|
||
|
|
}
|
||
|
|
|
||
|
|
func CreateOauth(userId, oauthId string) Oauth {
|
||
|
|
|
||
|
|
model := Oauth{
|
||
|
|
OauthId: oauthId,
|
||
|
|
UserId: userId,
|
||
|
|
CreatedAt: types.Time{
|
||
|
|
time.Now(),
|
||
|
|
},
|
||
|
|
}
|
||
|
|
oauth := FindOauthsByOpenId(oauthId)
|
||
|
|
if oauth.ID != 0 {
|
||
|
|
UpdateOauthById(model, oauthId)
|
||
|
|
return model
|
||
|
|
}
|
||
|
|
DB.Create(&model)
|
||
|
|
return model
|
||
|
|
}
|
||
|
|
func FindOauthById(userId string) Oauth {
|
||
|
|
var oauth Oauth
|
||
|
|
DB.Where("user_id = ?", userId).First(&oauth)
|
||
|
|
return oauth
|
||
|
|
}
|
||
|
|
func FindOauth(userId, oauthId string) Oauth {
|
||
|
|
var oauth Oauth
|
||
|
|
DB.Where("user_id = ? and oauth_id = ?", userId, oauthId).First(&oauth)
|
||
|
|
return oauth
|
||
|
|
}
|
||
|
|
func DelOauth(oauthId string) {
|
||
|
|
DB.Where("oauth_id = ?", oauthId).Delete(&Oauth{})
|
||
|
|
}
|
||
|
|
func DelOauthKefuName(kefuName string) {
|
||
|
|
DB.Where("user_id = ?", kefuName).Delete(&Oauth{})
|
||
|
|
}
|
||
|
|
func FindOauthsById(userId string) []Oauth {
|
||
|
|
var oauths []Oauth
|
||
|
|
DB.Where("user_id = ?", userId).Find(&oauths)
|
||
|
|
return oauths
|
||
|
|
}
|
||
|
|
func FindOauthsByOpenId(openId string) Oauth {
|
||
|
|
var oauths Oauth
|
||
|
|
DB.Where("oauth_id = ?", openId).First(&oauths)
|
||
|
|
return oauths
|
||
|
|
}
|
||
|
|
func FindOauthsQuery(query interface{}, args []interface{}) Oauth {
|
||
|
|
var oauths Oauth
|
||
|
|
DB.Where(query, args...).Find(&oauths)
|
||
|
|
return oauths
|
||
|
|
}
|
||
|
|
func UpdateOauthById(oauth Oauth, oauthId string) Oauth {
|
||
|
|
DB.Table("oauth").Where("oauth_id = ?", oauthId).Update(&oauth)
|
||
|
|
return oauth
|
||
|
|
}
|
||
|
|
func FindOauthsByVisitors(visitorIds []string) map[string]uint {
|
||
|
|
var oauths []Oauth
|
||
|
|
DB.Where("user_id in (? ) ", visitorIds).Find(&oauths)
|
||
|
|
res := make(map[string]uint)
|
||
|
|
for _, oauth := range oauths {
|
||
|
|
res[oauth.UserId] = 2
|
||
|
|
}
|
||
|
|
|
||
|
|
for _, visitorId := range visitorIds {
|
||
|
|
if res[visitorId] != 2 {
|
||
|
|
res[visitorId] = 1
|
||
|
|
}
|
||
|
|
}
|
||
|
|
return res
|
||
|
|
}
|