Add ability to set GraphQL struct name for the returned object

Author: debugger84

internal/result.go

@@ -186,6 +186,7 @@ func buildQueries(req *plugin.GenerateRequest, options *opts.Options, structs []
 		comments := query.Comments
 		var extendedType string
 		resolverName := query.Name
+		returnType := ""
 		for i, comment := range comments {
 			parts := strings.Split(comment, ":")
 			if len(parts) > 1 && strings.Trim(parts[0], " ") == "gql" {
@@ -195,6 +196,9 @@ func buildQueries(req *plugin.GenerateRequest, options *opts.Options, structs []
 				if len(resolverInfo) > 1 {
 					resolverName = resolverInfo[1]
 				}
+				if len(parts) > 2 {
+					returnType = strings.Trim(parts[2], " ")
+				}
 				comments = append(comments[:i], comments[i+1:]...)
 				break
 			}
@@ -228,6 +232,10 @@ func buildQueries(req *plugin.GenerateRequest, options *opts.Options, structs []
 			CursorPagination: cursorPagination,
 		}
 
+		if returnType == "" {
+			returnType = gq.MethodName + "Row"
+		}
+
 		qpl := int(*options.QueryParameterLimit)
 
 		if paginated {
@@ -366,13 +374,15 @@ func buildQueries(req *plugin.GenerateRequest, options *opts.Options, structs []
 					)
 				}
 				var err error
-				gs, err = columnsToStruct(req, options, gq.MethodName+"Row", columns, true)
+				gs, err = columnsToStruct(req, options, returnType, columns, true)
 				if err != nil {
 					return nil, err
 				}
 				emit = true
 				modelPath = options.Package + "." + gq.MethodName + "Row"
 			}
+			gs.ModelPath = modelPath
+
 			gq.Ret = QueryValue{
 				Emit:      emit,
 				Name:      "i",

internal/struct.go

@@ -10,10 +10,11 @@ import (
 )
 
 type Struct struct {
-	Table   *plugin.Identifier
-	Name    string
-	Fields  []Field
-	Comment string
+	Table     *plugin.Identifier
+	Name      string
+	Fields    []Field
+	Comment   string
+	ModelPath string
 }
 
 func StructName(name string, options *opts.Options) string {

internal/templates/schema.tmpl

@@ -28,7 +28,7 @@ enum {{.Name}}  @goModel(model: "{{$.ModelPackage}}.{{.Name}}") {
 {{ .Comment}}
 """
     {{- end }}
-type {{.Name}} @goModel(model: "{{$.ModelPackage}}.{{.Name}}") {
+type {{.Name}} @goModel(model: "{{.ModelPath}}") {
 {{- range .Fields -}}
     {{ if .Comment }}
     """