From 51e383fd39aabd73ee7ab30697193690c3fb7405 Mon Sep 17 00:00:00 2001
From: satori <huan0016@gmail.com>
Date: Fri, 29 Nov 2024 03:25:08 +0800
Subject: [PATCH] images "total"

---
 api/graphql.go | 17 +++++++++++++++++
 1 file changed, 17 insertions(+)

diff --git a/api/graphql.go b/api/graphql.go
index a7b8311..4eb0eb9 100644
--- a/api/graphql.go
+++ b/api/graphql.go
@@ -1181,6 +1181,23 @@ func NewSchema(config Config) (graphql.Schema, error) {
 					}
 				}
 
+				existField := func(selections []ast.Selection, name string) bool {
+					for _, field := range selections {
+						if f, ok := field.(*ast.Field); ok && f.Name.Value == name {
+							return true
+						}
+					}
+					return false
+				}
+
+				if existField(p.Info.FieldASTs[0].SelectionSet.Selections, "total") {
+					sql, _, _ := query.ToSQL()
+					sql = strings.Replace(sql, "SELECT *", "SELECT COUNT(*)", 1)
+					if err := db.Raw(sql).Scan(&total).Error; err != nil {
+						return nil, err
+					}
+				}
+
 				// 如果没有外部排序则使用指定排序(正则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), "")