从 demo 引用

This commit is contained in:
2023-11-20 23:25:18 +08:00
parent e43fde4116
commit 9186827b51
4 changed files with 154 additions and 122 deletions

View File

@@ -1,155 +0,0 @@
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},
"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},
"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) {
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
},
}
image := graphql.NewObject(graphql.ObjectConfig{
Name: "Image",
Fields: graphql.Fields{
"id": &graphql.Field{Type: graphql.Int},
"width": &graphql.Field{Type: graphql.Int},
"height": &graphql.Field{Type: graphql.Int},
"content": &graphql.Field{Type: graphql.String},
"article_category_top_id": &graphql.Field{Type: graphql.Int},
"praise_count": &graphql.Field{Type: graphql.Int},
"collect_count": &graphql.Field{Type: graphql.Int},
"create_time": &graphql.Field{Type: graphql.DateTime},
"update_time": &graphql.Field{Type: graphql.DateTime},
"user": &graphql.Field{
Type: user,
},
},
})
images := &graphql.Field{
Type: graphql.NewList(image),
Args: graphql.FieldConfigArgument{
"id": &graphql.ArgumentConfig{
Type: graphql.Int,
},
},
Resolve: func(p graphql.ResolveParams) (interface{}, error) {
fmt.Println("p.Args:", p.Args)
//connection.Query("SELECT ")
return []interface{}{
map[string]interface{}{
"id": 1,
"width": 100,
"height": 100,
"content": "content",
"article_category_top_id": 1,
"praise_count": 1,
"collect_count": 1,
"create_time": "2018-01-01 00:00:00",
"update_time": "2018-01-01 00:00:00",
"user": map[string]interface{}{
"id": 1,
"name": "user1",
"age": 10,
"info": "info",
"price": 1.1,
"avatar": "",
},
},
map[string]interface{}{
"id": 2,
"width": 100,
"height": 100,
"content": "content",
"article_category_top_id": 1,
"praise_count": 1,
"collect_count": 1,
"create_time": "2018-01-01 00:00:00",
"update_time": "2018-01-01 00:00:00",
"user": map[string]interface{}{
"id": 2,
"name": "user2",
"age": 20,
"info": "info",
"price": 2.2,
"avatar": "",
},
},
}, nil
},
}
rootQuery := graphql.ObjectConfig{Name: "RootQuery", Fields: graphql.Fields{
"users": users,
"images": images,
}}
schemaConfig := graphql.SchemaConfig{Query: graphql.NewObject(rootQuery)}
schema, err := graphql.NewSchema(schemaConfig)
if err != nil {
return schema, err
}
return schema, nil
}