webp/api/work.go

package api

import (
	"fmt"
	"regexp"
	"time"

	"github.com/doug-martin/goqu/v9"
	"github.com/graphql-go/graphql"
)

type Work struct {
	ID           int       `json:"id"            gorm:"primaryKey"`
	Title        string    `json:"title"`
	Orientation  string    `json:"orientation"`
	Device       string    `json:"device"`
	Era          string    `json:"era"`
	Tags         string    `json:"tags"`
	UserId       int       `json:"user_id"`
	Content      string    `json:"content"`
	Image        string    `json:"image"`
	Images       string    `json:"images"`
	User         User      `json:"user"          gorm:"foreignKey:UserId"`
	CollectCount int       `json:"collect_count" gorm:"column:collect_num"`
	PraiseCount  int       `json:"praise_count"  gorm:"column:praise_num"`
	CreateTime   time.Time `json:"create_time"`
	UpdateTime   time.Time `json:"update_time"`
}

func (Work) TableName() string {
	return "web_article"
}

var workType = graphql.NewObject(graphql.ObjectConfig{
	Name:        "Work",
	Description: "作品",
	Fields: graphql.Fields{
		"id":            &graphql.Field{Type: graphql.Int, Description: "作品ID"},
		"title":         &graphql.Field{Type: graphql.String, Description: "作品标题"},
		"create_time":   &graphql.Field{Type: graphql.DateTime, Description: "作品创建时间"},
		"update_time":   &graphql.Field{Type: graphql.DateTime, Description: "作品更新时间"},
		"praise_count":  &graphql.Field{Type: graphql.Int, Description: "点赞数"},
		"collect_count": &graphql.Field{Type: graphql.Int, Description: "收藏数"},
		"praise": &graphql.Field{Type: graphql.Boolean, Description: "当前用户是否点赞", Resolve: func(p graphql.ResolveParams) (interface{}, error) {
			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.(Work).ID).Where("type = ?", 0).Count(&praise); err != nil {
					return false, nil
				}
				if praise > 0 {
					return true, nil
				}
			}
			return false, nil
		}},
		"collect": &graphql.Field{Type: graphql.Boolean, Description: "当前用户是否收藏", Resolve: func(p graphql.ResolveParams) (interface{}, error) {
			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.(Work).ID).Where("type = ?", 0).Count(&collect); err != nil {
					return false, nil
				}
				if collect > 0 {
					return true, nil
				}
			}
			return false, nil
		}},
	},
})

var WorkItems = &graphql.Field{
	Name:        "works",
	Description: "作品列表",
	Type: graphql.NewObject(graphql.ObjectConfig{
		Name:        "WorkConnection",
		Description: "条件筛选作品列表",
		Fields: graphql.Fields{
			"list": &graphql.Field{Type: graphql.NewList(workType), Description: "作品列表"},
			//"total": &graphql.Field{Type: graphql.Int, Description: "作品总数"},
		},
	}),
	Args: graphql.FieldConfigArgument{
		"id":          &graphql.ArgumentConfig{Type: graphql.Int, Description: "筛选作品中指定ID的"},
		"title":       &graphql.ArgumentConfig{Type: graphql.String, Description: "筛选作品中含有指定标题的"},
		"tags":        &graphql.ArgumentConfig{Type: graphql.String, Description: "筛选作品中含有指定标签的"},
		"create_time": &graphql.ArgumentConfig{Type: graphql.DateTime, Description: "筛选作品中创建时间等于指定值的"},
		"update_time": &graphql.ArgumentConfig{Type: graphql.DateTime, Description: "筛选作品中更新时间等于指定值的"},
		"sort":        &graphql.ArgumentConfig{Type: graphql.String, Description: "按指定字段排序游戏", DefaultValue: "id"},
		"order":       &graphql.ArgumentConfig{Type: orderType, Description: "排序类型(升序或降序)", DefaultValue: "ASC"},
		"first":       &graphql.ArgumentConfig{Type: graphql.Int, Description: "翻页参数(傳回清單中的前n個元素)"},
		"last":        &graphql.ArgumentConfig{Type: graphql.Int, Description: "翻页参数(傳回清單中的最後n個元素)"},
		"after":       &graphql.ArgumentConfig{Type: graphql.String, Description: "翻页参数(傳回清單中指定遊標之後的元素)"},
		"before":      &graphql.ArgumentConfig{Type: graphql.String, Description: "翻页参数(傳回清單中指定遊標之前的元素)"},
	},
	Resolve: func(p graphql.ResolveParams) (interface{}, error) {
		var works []Work
		var total int
		var err error

		var query = goqu.Dialect("mysql").From("web_article").Where(goqu.Ex{"category_top_id": 1})

		// 筛选条件
		for _, format := range []string{"id", "tags"} {
			if p.Args[format] != nil {
				query = query.Where(goqu.C(format).Eq(p.Args[format]))
			}
		}

		// 排序条件
		if p.Args["sort"] != nil {
			if p.Args["order"].(string) == "ASC" {
				query = query.Order(goqu.C(p.Args["sort"].(string)).Asc())
			}
			if p.Args["order"].(string) == "DESC" {
				query = query.Order(goqu.C(p.Args["sort"].(string)).Desc())
			}
		}

		// 如果没有外部排序则使用指定排序(正则sort只能是字母数字下划下)
		if p.Args["text"] == nil && p.Args["similar"] == nil && p.Args["interest"] == nil {
			sort := regexp.MustCompile(`[^a-zA-Z0-9_]`).ReplaceAllString(p.Args["sort"].(string), "")
			query = query.Select("web_article.id", goqu.L(
				fmt.Sprintf("ROW_NUMBER() OVER(ORDER BY web_article.%s %s)", sort, p.Args["order"]),
			).As("row_num"))
		}

		// 取所有数据的前N条
		sql, _, _ := query.ToSQL()
		fmt.Println(sql)

		// 遊標截取篩選結果集的前N条
		var cursor string
		if p.Args["after"] != nil {
			cursor = fmt.Sprintf(`WHERE row_num > (SELECT row_num FROM RankedArticles WHERE RankedArticles.id = %d)`, p.Args["after"].(int))
		}

		var limit int = 10
		if p.Args["first"] != nil {
			limit = p.Args["first"].(int)
		} else if p.Args["last"] != nil {
			limit = p.Args["last"].(int)
		}

		sql = fmt.Sprintf(`
		WITH RankedArticles AS (%s)
		SELECT * FROM web_article INNER JOIN(
			SELECT id, row_num FROM RankedArticles %s
		) AS LimitedRanked ON LimitedRanked.id = web_article.id
		ORDER BY LimitedRanked.row_num ASC LIMIT %d
		`, sql, cursor, limit)

		if err := db.Raw(sql).Scan(&works).Error; err != nil {
			fmt.Println("获取游戏列表失败", err)
			return nil, err
		}

		return map[string]interface{}{
			"list":  works,
			"total": total,
		}, err
	},
}