Files
genqlient/generate/unmarshal.go.tmpl
T
Ben KraftandGitHub 4c38cb7759 Add support for interfaces, part 1: the simplest cases (#52)
## Summary:
In this commit I begin the journey to add the long-awaited support for
interfaces (part of #8).  Well, it's not the beginning: I already had
some half-written broken code around.  But it's the first fully
functional support, and especially, the first *tested* support; it's
probably best to review the nontrivially-changed code as if it were new.

Conceptually, the code so far is pretty simple: we generate an interface
type, and the implementations.  (That code is in fact mostly unchanged.)
The complexity comes in because encoding/json doesn't know how to
unmarshal that.  So we have to add an UnmarshalJSON method, which
actually has to be on the types with interface-type fields, that knows
how.  I factored it into two methods, such that that UnmarshalJSON
method is just glue, and then there's a separate function, corresponding
to each interface-type, that actually does all the work.  (If only one
could just write it as an actual method!)  The method uses the same
trick suggested to me by a few others in another context to deserialize
all but one field, then handle that field specially, which is discussed
in the code.

This still has some limitations, which will be lifted in future commits:
- it doesn't allow for list-of-interface fields
- it requires that you manually ask for `__typename`
- it doesn't support fragments, i.e. you can only query for interface
  fields, not concrete-type-specific ones
But it works, even in integration tests, which is progress!

As a part of this, I added a proper config option for the "allow broken
features" flag, since I need to be able to set it from the integration
tests which are in a separate package (and actually shell out via `go
generate`).  I also renamed what was to be the first case
(InterfaceNoFragments), and replaced it with a further-simplified
version (avoiding list-of-interface fields.

[1] https://github.com/benjaminjkraft/notes/blob/master/go-json-interfaces.md

Issue: https://github.com/Khan/genqlient/issues/8

## Test plan:
make tesc

Author: benjaminjkraft

Reviewers: dnerdy, benjaminjkraft, aberkan, csilvers, MiguelCastillo

Required Reviewers: 

Approved by: dnerdy

Checks:  Test (1.17),  Test (1.16),  Test (1.15),  Test (1.14),  Test (1.13),  Lint,  Test (1.17),  Test (1.16),  Test (1.15),  Test (1.14),  Test (1.13),  Lint

Pull request URL: https://github.com/Khan/genqlient/pull/52
2021-08-25 11:51:26 -07:00

44 lines
1.7 KiB
Cheetah

{{/* (the blank lines at the start are intentional, to separate
UnmarshalJSON from the function it follows) */}}
func (v *{{.GoName}}) UnmarshalJSON(b []byte) error {
{{/* We want to specially handle the abstract fields (.AbstractFields),
but unmarshal everything else normally. To handle abstract fields,
first we unmarshal them into a json.RawMessage, and then handle those
further, below. For the rest, we just want to call json.Unmarshal.
But if we do that naively on a value of type `.Type`, it will call
this function again, and recurse infinitely. So we make a wrapper
type -- with a different name, thus different methods, but the same
fields, and unmarshal into that. For more on why this is so
difficult, see
https://github.com/benjaminjkraft/notes/blob/master/go-json-interfaces.md
TODO(benkraft)): Ensure `{{.Type}}Wrapper` won't collide with any
other type we need. (For the most part it being locally-scoped saves
us; it's not clear if this can be a problem in practice.)
*/}}
type {{.GoName}}Wrapper {{.GoName}}
var firstPass struct{
*{{.GoName}}Wrapper
{{range .AbstractFields -}}
{{.GoName}} {{ref "encoding/json.RawMessage"}} `json:"{{.JSONName}}"`
{{end}}
}
firstPass.{{.GoName}}Wrapper = (*{{.GoName}}Wrapper)(v)
err := {{ref "encoding/json.Unmarshal"}}(b, &firstPass)
if err != nil {
return err
}
{{/* Now, for each field, call out to the unmarshal-helper. */}}
{{range .AbstractFields -}}
err = __unmarshal{{.GoType.Reference}}(
&v.{{.GoName}}, firstPass.{{.GoName}})
if err != nil {
return err
}
{{end}}
return nil
}