gameui/webp
2
0
Fork 0
Code Issues Pull Requests Packages Projects Releases 1 Wiki Activity

token 简化

Browse Source
This commit is contained in:
satori
2024-12-02 11:11:40 +08:00
parent 51c8a7fe2e
commit 06070edded
5 changed files with 15 additions and 31 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
api/graphql.go
View File

@@ -36,6 +36,13 @@ func InitDefault(config *viper.Viper) {
}
}
func ParseToken(token string) (user_id int) {
if err := db.Table("web_auth").Select("user_id").Where("token = ?", token).Scan(&user_id).Error; err != nil {
fmt.Println("token解析失败", err)
}
return user_id
}
// 定时检查补全颜色字段
func CheckColorNullRows(offset int) {
for {

8
api/image.go
View File

@@ -87,9 +87,11 @@ var imageType = graphql.NewObject(graphql.ObjectConfig{
var user_id = p.Context.Value("user_id").(int)
if user_id != 0 {
var praise int64
if err := db.Table("web_praise").Where("user_id = ?", user_id).Where("praise_id = ?", p.Source.(Image).ID).Where("type = ?", 4).Count(&praise); err != nil {
if err := db.Table("web_praise").Where("user_id = ?", user_id).Where("praise_id = ?", p.Source.(Image).ID).Where("type = ?", 4).Count(&praise).Error; err != nil {
fmt.Println(user_id, p.Source.(Image).ID, praise, "E", err)
return false, nil
}
fmt.Println(user_id, p.Source.(Image).ID, praise)
if praise > 0 {
return true, nil
}
@@ -100,7 +102,7 @@ var imageType = graphql.NewObject(graphql.ObjectConfig{
var user_id = p.Context.Value("user_id").(int)
if user_id != 0 {
var collect int64
if err := db.Table("web_collect").Where("user_id = ?", user_id).Where("collect_id = ?", p.Source.(Image).ID).Where("type = ?", 1).Count(&collect); err != nil {
if err := db.Table("web_collect").Where("user_id = ?", user_id).Where("image_id = ?", p.Source.(Image).ID).Where("type = ?", 1).Count(&collect).Error; err != nil {
return false, nil
}
if collect > 0 {
@@ -113,7 +115,7 @@ var imageType = graphql.NewObject(graphql.ObjectConfig{
var user_id = p.Context.Value("user_id").(int)
if user_id != 0 {
var collect_id int
if err := db.Table("web_collect").Where("user_id = ?", user_id).Where("image_id = ?", p.Source.(Image).ID).First(&collect_id); err != nil {
if err := db.Table("web_collect").Where("user_id = ?", user_id).Where("image_id = ?", p.Source.(Image).ID).First(&collect_id).Error; err != nil {
return nil, nil
}
return collect_id, nil