text字段筛选

api/graphql.go

@@ -3,6 +3,7 @@ package api
 import (
 	"fmt"
 	"log"
+	"reflect"
 	"strings"
 
 	"git.satori.love/gameui/webp/models"
@@ -12,6 +13,39 @@ import (
 	"github.com/mitchellh/mapstructure"
 )
 
+// 自动生成 GraphQL 类型的函数
+func generateGraphQLType(model interface{}) (*graphql.Object, error) {
+	modelType := reflect.TypeOf(model)
+	if modelType.Kind() != reflect.Struct {
+		return nil, fmt.Errorf("model must be a struct")
+	}
+
+	fields := graphql.Fields{}
+	for i := 0; i < modelType.NumField(); i++ {
+		field := modelType.Field(i)
+		fieldType := graphql.String // 默认使用字符串类型
+
+		// 这里可以根据需要添加更多类型映射
+		switch field.Type.Kind() {
+		case reflect.String:
+			fieldType = graphql.String
+		case reflect.Int:
+			fieldType = graphql.Int
+		case reflect.Bool:
+			fieldType = graphql.Boolean
+		}
+
+		fields[field.Name] = &graphql.Field{
+			Type: fieldType,
+		}
+	}
+
+	return graphql.NewObject(graphql.ObjectConfig{
+		Name:   modelType.Name(),
+		Fields: fields,
+	}), nil
+}
+
 func NewSchema(config Config) (graphql.Schema, error) {
 
 	// 打开数据库连接
@@ -27,8 +61,10 @@ func NewSchema(config Config) (graphql.Schema, error) {
 		log.Fatalln("连接数据库失败", err)
 	}
 
+	var user, article, text *graphql.Object
+
 	// 用户的可选字段
-	user := graphql.NewObject(graphql.ObjectConfig{
+	user = graphql.NewObject(graphql.ObjectConfig{
 		Name:        "User",
 		Description: "用户信息",
 		Fields: graphql.Fields{
@@ -43,7 +79,7 @@ func NewSchema(config Config) (graphql.Schema, error) {
 	})
 
 	// 文章的可选字段
-	article := graphql.NewObject(graphql.ObjectConfig{
+	article = graphql.NewObject(graphql.ObjectConfig{
 		Name:        "Article",
 		Description: "文章信息",
 		Fields: graphql.Fields{
@@ -57,7 +93,7 @@ func NewSchema(config Config) (graphql.Schema, error) {
 	})
 
 	// 图像中的文字提取
-	text := graphql.NewObject(graphql.ObjectConfig{
+	text = graphql.NewObject(graphql.ObjectConfig{
 		Name:        "Text",
 		Description: "图像中的文字提取",
 		Fields: graphql.Fields{
@@ -80,18 +116,41 @@ func NewSchema(config Config) (graphql.Schema, error) {
 			"description":             &graphql.Field{Type: graphql.String, Description: "图像描述"},
 			"tags":                    &graphql.Field{Type: graphql.String, Description: "图像标签"},
 			"rank":                    &graphql.Field{Type: graphql.String, Description: "图像等级"},
-			"text":                    &graphql.Field{Type: graphql.NewList(text), Description: "图像中的文字"},
 			"comment_num":             &graphql.Field{Type: graphql.Int, Description: "评论数"},
 			"article_category_top_id": &graphql.Field{Type: graphql.Int, Description: "文章分类顶级ID"},
 			"praise_count":            &graphql.Field{Type: graphql.Int, Description: "点赞数"},
 			"collect_count":           &graphql.Field{Type: graphql.Int, Description: "收藏数"},
 			"create_time":             &graphql.Field{Type: graphql.DateTime, Description: "图像创建时间"},
 			"update_time":             &graphql.Field{Type: graphql.DateTime, Description: "图像更新时间"},
-			"user":                    &graphql.Field{Type: user, Description: "图像所属用户"},
 			"article":                 &graphql.Field{Type: article, Description: "图像所属文章"},
+			"text": &graphql.Field{
+				Type:        graphql.NewList(text),
+				Description: "图像中的文字",
+				Args: graphql.FieldConfigArgument{
+					"text": &graphql.ArgumentConfig{Type: graphql.String, Description: "筛选含有指定文字的列"},
+				},
+				Resolve: func(p graphql.ResolveParams) (interface{}, error) {
+					if p.Args["text"] != nil {
+						var texts TextList
+						for _, text := range p.Source.(Image).Text {
+							fmt.Println("san", text.Text)
+							if strings.Contains(text.Text, p.Args["text"].(string)) {
+								texts = append(texts, text)
+							}
+						}
+						return texts, nil
+					}
+					return p.Source.(Image).Text, nil
+				},
+			},
 		},
 	})
 
+	image.AddFieldConfig("user", &graphql.Field{Type: user, Description: "图像所属用户"})
+	image.AddFieldConfig("similars", &graphql.Field{Type: graphql.NewList(image), Description: "相似的图像", Resolve: func(p graphql.ResolveParams) (interface{}, error) {
+		return []Image{}, nil
+	}})
+
 	// 将 list 中的字段提取出来用于查询
 	get_fields := func(requestedFields []ast.Selection) (fields []string) {
 		for _, field := range requestedFields {
@@ -108,6 +167,10 @@ func NewSchema(config Config) (graphql.Schema, error) {
 							fields = append(fields, "article_id")
 							continue
 						}
+						if fieldAST.Name.Value == "similars" {
+							// 跳过自定义字段
+							continue
+						}
 						fields = append(fields, fieldAST.Name.Value)
 					}
 				}
@@ -154,18 +217,15 @@ func NewSchema(config Config) (graphql.Schema, error) {
 				if where_str != "" {
 					where_str = "WHERE " + where_str
 				}
-
 				var query strings.Builder
 				var users []User
 				var total int
-
 				fields := strings.Join(get_fields(p.Info.FieldASTs[0].SelectionSet.Selections), ",")
 				query.WriteString(fmt.Sprintf("SELECT %s FROM web_member %s LIMIT %d OFFSET %d", fields, where_str, 10, 0))
 				if err := connection.Select(&users, query.String()); err != nil {
 					fmt.Println("获取用户列表失败", err)
 					return nil, err
 				}
-
 				if len(users) > 0 {
 					query.Reset()
 					query.WriteString(fmt.Sprintf("SELECT COUNT(*) FROM web_member %s", where_str))
@@ -174,7 +234,6 @@ func NewSchema(config Config) (graphql.Schema, error) {
 						return nil, err
 					}
 				}
-
 				return map[string]interface{}{
 					"list":  users,
 					"total": total,
@@ -308,7 +367,7 @@ func NewSchema(config Config) (graphql.Schema, error) {
 							"total": 0,
 						}, nil
 					}
-					where = append(where, fmt.Sprintf("id IN (%s) LIMIT %d", id_list_str, len(id_list)))
+					where = append(where, fmt.Sprintf("id IN (%s)", id_list_str))
 				}
 
 				where_str := strings.Join(where, " AND ")
@@ -342,17 +401,11 @@ func NewSchema(config Config) (graphql.Schema, error) {
 
 				var images ImageList
 				var q = query.String()
-				fmt.Println(q)
 				if err := connection.Select(&images, q); err != nil {
 					fmt.Println("获取图像列表失败", err)
 					return nil, err
 				}
 
-				// 按照 id_list 的顺序重新排序
-				//if len(id_list) > 0 {
-				//	images.SortByIDList(id_list)
-				//}
-
 				// 获取用户信息(如果图像列表不为空且请求字段中包含user)
 				if len(images) > 0 && strings.Contains(fields, "user") {
 					user_ids_str := images.ToAllUserID().ToString()