graphql 从数据库查询

models/graphql.go

@@ -2,50 +2,70 @@ package models
 
 import (
 	"fmt"
+	"strings"
 
 	"github.com/graphql-go/graphql"
+	"github.com/graphql-go/graphql/language/ast"
 )
 
 func NewSchema() (graphql.Schema, error) {
 	user := graphql.NewObject(graphql.ObjectConfig{
 		Name: "User",
 		Fields: graphql.Fields{
-			"id":     &graphql.Field{Type: graphql.Int},
-			"name":   &graphql.Field{Type: graphql.String},
-			"age":    &graphql.Field{Type: graphql.Int},
-			"info":   &graphql.Field{Type: graphql.String},
-			"price":  &graphql.Field{Type: graphql.Float},
-			"avatar": &graphql.Field{Type: graphql.String},
+			"id":          &graphql.Field{Type: graphql.Int},
+			"user_name":   &graphql.Field{Type: graphql.String},
+			"avatar":      &graphql.Field{Type: graphql.String},
+			"rank":        &graphql.Field{Type: graphql.String},
+			"price":       &graphql.Field{Type: graphql.Float},
+			"create_time": &graphql.Field{Type: graphql.DateTime},
+			"update_time": &graphql.Field{Type: graphql.DateTime},
 		},
 	})
 
 	users := &graphql.Field{
 		Type: graphql.NewList(user),
 		Args: graphql.FieldConfigArgument{
-			"id": &graphql.ArgumentConfig{
-				Type: graphql.Int,
-			},
+			"id":          &graphql.ArgumentConfig{Type: graphql.Int},
+			"user_name":   &graphql.ArgumentConfig{Type: graphql.String},
+			"avatar":      &graphql.ArgumentConfig{Type: graphql.String},
+			"rank":        &graphql.ArgumentConfig{Type: graphql.String},
+			"create_time": &graphql.ArgumentConfig{Type: graphql.DateTime},
+			"update_time": &graphql.ArgumentConfig{Type: graphql.DateTime},
 		},
 		Resolve: func(p graphql.ResolveParams) (interface{}, error) {
-			fmt.Println("p.Args:", p.Args)
-			return []interface{}{
-				map[string]interface{}{
-					"id":     1,
-					"name":   "user1",
-					"age":    10,
-					"info":   "info",
-					"price":  1.1,
-					"avatar": "",
-				},
-				map[string]interface{}{
-					"id":     2,
-					"name":   "user2",
-					"age":    20,
-					"info":   "info",
-					"price":  2.2,
-					"avatar": "",
-				},
-			}, nil
+			var fields []string
+			requestedFields := p.Info.FieldASTs[0].SelectionSet.Selections
+			for _, field := range requestedFields {
+				fieldAST, ok := field.(*ast.Field)
+				if ok {
+					fields = append(fields, fieldAST.Name.Value)
+				}
+			}
+			fields_str := strings.Join(fields, ",")
+
+			var where []string
+			if p.Args["id"] != nil {
+				where = append(where, fmt.Sprintf("id=%d", p.Args["id"]))
+			}
+			if p.Args["user_name"] != nil {
+				where = append(where, fmt.Sprintf("user_name='%s'", p.Args["user_name"]))
+			}
+
+			// 筛选条件
+			where_str := strings.Join(where, " AND ")
+			fmt.Println(where_str)
+
+			var query strings.Builder
+			query.WriteString(fmt.Sprintf("SELECT %s FROM web_member WHERE %s LIMIT %s", fields_str, where_str, "10"))
+			fmt.Println(query.String())
+
+			var users []User
+			if err := connection.Select(&users, query.String()); err != nil {
+				fmt.Println("获取用户列表失败", err)
+				return nil, err
+			}
+			fmt.Println(users)
+			return users, nil
 		},
 	}
 
@@ -76,6 +96,9 @@ func NewSchema() (graphql.Schema, error) {
 		},
 		Resolve: func(p graphql.ResolveParams) (interface{}, error) {
 			fmt.Println("p.Args:", p.Args)
+
+			//connection.Query("SELECT ")
+
 			return []interface{}{
 				map[string]interface{}{
 					"id":                      1,