-
What happened?I want to bind What did you expect?type User struct {
gorm.Model //<-- here
ID Int
}Minimal graphql.schema and models to reproducetype User {
id:ID!
}versions
|
Beta Was this translation helpful? Give feedback.
Replies: 2 comments 5 replies
-
|
Hi there! I've been using go for around the last 8 hours, so I don't know how legal this is, but I had the same question as you and I solved it with this code: Code// generate/main.go
package main
import (
"fmt"
"go/importer"
"go/token"
"os"
"github.com/99designs/gqlgen/api"
"github.com/99designs/gqlgen/codegen/config"
"github.com/99designs/gqlgen/plugin/modelgen"
)
func mutateHook(b *modelgen.ModelBuild) *modelgen.ModelBuild {
fset := token.NewFileSet()
tpatata, err := importer.ForCompiler(fset, "source", nil).Import("gorm.io/gorm")
if err != nil {
panic("Error importing gorm" + err.Error())
}
for _, model := range b.Models {
model.Fields = append(model.Fields, &modelgen.Field{Type: tpatata.Scope().Lookup("Model").Type()})
for _, field := range model.Fields {
field.Tag += ` test"`
}
}
return b
}
func main() {
cfg, err := config.LoadConfigFromDefaultLocations()
if err != nil {
fmt.Fprintln(os.Stderr, "failed to load config", err.Error())
os.Exit(2)
}
// Attaching the mutation function onto modelgen plugin
p := modelgen.Plugin{
MutateHook: mutateHook,
}
err = api.Generate(cfg, api.ReplacePlugin(&p))
if err != nil {
fmt.Fprintln(os.Stderr, err.Error())
os.Exit(3)
}
}It took me a while to figure out how the Code generated// Code generated by github.com/99designs/gqlgen, DO NOT EDIT.
package model
import (
"gorm.io/gorm"
)
type NewTodo struct {
Text string `json:"text" test"`
UserID string `json:"userId" test"`
gorm.Model ` test"`
}
type NewUser struct {
Name string `json:"name" test"`
gorm.Model ` test"`
}
type Todo struct {
ID string `json:"id" test"`
Text string `json:"text" test"`
Done bool `json:"done" test"`
User *User `json:"user" test"`
gorm.Model ` test"`
}
type User struct {
ID string `json:"id" test"`
Name string `json:"name" test"`
gorm.Model ` test"`
}From the following schema: Schema# GraphQL schema example
#
# https://gqlgen.com/getting-started/
type Todo {
id: ID!
text: String!
done: Boolean!
user: User!
}
type User {
id: ID!
name: String!
}
type Query {
todos: [Todo!]!
users: [User!]!
}
input NewTodo {
text: String!
userId: String!
}
input NewUser {
name: String!
}
type Mutation {
createTodo(input: NewTodo!): Todo!
createUser(input: NewUser!): User!
}
type Subscription {
testSub: String
}It's not perfect, but as you can see it's adding |
Beta Was this translation helpful? Give feedback.
-
|
Hey, so, small update: I've updated the plugin, and I made it react to directives. You can now mark types to make them include gorm.Model and mark fields to add gorm related metadata to their tags. I've also added "belogs to" relationships as a proof of concept and it looks like it working nicely. Basically, you can now turn this: schema.graphqls# GraphQL schema example
#
# https://gqlgen.com/getting-started/
directive @db on OBJECT
directive @gorm(
value: String
belongsTo: String
) on FIELD_DEFINITION
type Todo @db {
id: ID! @gorm(value: "primaryKey")
text: String!
done: Boolean!
user: User! @gorm(belongsTo: "User")
}
type User @db {
id: ID! @gorm(value: "primaryKey")
name: String! @gorm(value: "unique")
}
type Query {
todos: [Todo!]!
users: [User!]!
}
input NewTodo {
text: String!
userId: String!
}
input NewUser {
name: String!
}
type Mutation {
createTodo(input: NewTodo!): Todo!
createUser(input: NewUser!): User!
}
type Subscription {
testSub: String
}Into this: models_gen.go// Code generated by github.com/99designs/gqlgen, DO NOT EDIT.
package model
import (
"gorm.io/gorm"
)
type NewTodo struct {
Text string `json:"text"`
UserID string `json:"userId"`
}
type NewUser struct {
Name string `json:"name"`
}
type Todo struct {
gorm.Model ``
ID string `json:"id" gorm:"primaryKey"`
Text string `json:"text"`
Done bool `json:"done"`
User *User `json:"user"`
UserID string ``
}
type User struct {
gorm.Model ``
ID string `json:"id" gorm:"primaryKey"`
Name string `json:"name" gorm:"unique"`
}The code is a bit more complex now, so if you still have a use case for this concept, let me know and I'll upload the code to a repo Cheers! |
Beta Was this translation helpful? Give feedback.
-
|
Hey @rmortes have you published this already somewhere? |
Beta Was this translation helpful? Give feedback.
-
|
I don't think so... I'll try to look for that code, but it's probably lost, sorry! |
Beta Was this translation helpful? Give feedback.
-
|
No worries, thanks |
Beta Was this translation helpful? Give feedback.
-
|
@nephix Long time ago now sure, but this is how I did it via directives 😄 You can see to get the directives you have to get it via the fieldHook in the modelgen.Plugin, via the ast FieldDefinition, using a const I set for directive names/arguments I can check against the names of each field and apply anything I want to it, check types etc etc. I don't plan on expanding this to inject the gorm.Model but you would do that on the MutateHook side on the modelgen.Plugin. as mentioned above in previous comments mutate_hook.gofunc main() {
trace.SetLogger("debug")
cfg, err := config.LoadConfigFromDefaultLocations()
if err != nil {
logrus.Fatal("failed to load config", err.Error())
}
// Attaching the mutation function onto modelgen plugin
p := modelgen.Plugin{
MutateHook: func(b *modelgen.ModelBuild) *modelgen.ModelBuild {
return b
},
FieldHook: func(td *ast.Definition, fd *ast.FieldDefinition, f *modelgen.Field) (*modelgen.Field, error) {
f = gormDirectiveHook(td, fd, f)
return f, nil
},
}
err = api.Generate(cfg, api.ReplacePlugin(&p))
if err != nil {
logrus.Fatal("failed to generate code", err.Error())
}
}
const (
GormDirectiveName string = "gorm"
GormFieldTagsName string = "fieldTags"
)
func logrusEntry(fieldName string) *logrus.Entry {
return logrus.WithFields(logrus.Fields{
"field_name": fieldName,
})
}
// gormDirectiveHook
func gormDirectiveHook(td *ast.Definition, fd *ast.FieldDefinition, f *modelgen.Field) *modelgen.Field {
if len(fd.Directives) == 0 {
logrusEntry(f.Name).Debug("no directives found")
return f
}
for _, directive := range fd.Directives {
name := strcase.ToLowerCamel(directive.Name)
if name == GormDirectiveName {
logrusEntry(f.Name).Debug("found gorm directive")
args := directive.Arguments
for _, arg := range args {
logrusEntry(f.Name).Debugf("found argument %s", arg.Name)
// handle fieldTags directive
name := strcase.ToLowerCamel(arg.Name)
if name == GormFieldTagsName {
// decode the values into an array of string
if arg.Value.Kind != ast.ListValue {
logrusEntry(f.Name).Warn("fieldTags argument must be a list")
continue
}
var tag string = " gorm:\""
for _, v := range arg.Value.Children {
if v.Value.Kind != ast.StringValue {
logrusEntry(f.Name).Warn("fieldTags argument must be a list of strings")
continue
}
if v.Value.Raw == "" {
logrusEntry(f.Name).Warn("fieldTags argument must not be empty")
continue
}
logrusEntry(f.Name).Debugf("found field tag %s", v.Value.Raw)
// apply the tag to the field
tag += v.Value.Raw + ";\""
}
// apply the tag to the field
f.Tag += tag
logrus.Infof("tag is now: %s", f.Tag)
}
}
}
}
return f
}schema.graphql directive @gorm(fieldTags: [String]) on FIELD_DEFINITION
type MediaDirectorySettings {
id: UUID! @gorm(fieldTags: ["primaryKey"])
createdAt: Time
updatedAt: Time
deleteAt: Time
friendlyName: String
directoryPath: String
} |
Beta Was this translation helpful? Give feedback.
-
|
Thanks @shanehughes1990 🙏 |
Beta Was this translation helpful? Give feedback.
Hi there! I've been using go for around the last 8 hours, so I don't know how legal this is, but I had the same question as you and I solved it with this code:
Code