Split starwars models out from resolvers

example/starwars/model.go

@@ -0,0 +1,128 @@
+package starwars
+
+import (
+	"context"
+	"encoding/base64"
+	"fmt"
+	"strconv"
+	"strings"
+	"time"
+)
+
+type Human struct {
+	ID           string
+	Name         string
+	FriendIds    []string
+	StarshipIds  []string
+	AppearsIn    []string
+	heightMeters float64
+	Mass         float64
+}
+
+func (h *Human) Height(unit string) float64 {
+	switch unit {
+	case "METER", "":
+		return h.heightMeters
+	case "FOOT":
+		return h.heightMeters * 3.28084
+	default:
+		panic("invalid unit")
+	}
+}
+
+type Starship struct {
+	ID           string
+	Name         string
+	History      [][2]int
+	lengthMeters float64
+}
+
+func (s *Starship) Length(unit string) float64 {
+	switch unit {
+	case "METER", "":
+		return s.lengthMeters
+	case "FOOT":
+		return s.lengthMeters * 3.28084
+	default:
+		panic("invalid unit")
+	}
+}
+
+type Review struct {
+	Stars      int
+	Commentary *string
+	Time       time.Time
+}
+
+type Droid struct {
+	ID              string
+	Name            string
+	FriendIds       []string
+	AppearsIn       []string
+	PrimaryFunction string
+}
+
+type Character interface{}
+type SearchResult interface{}
+
+func (r *Resolver) resolveFriendConnection(ctx context.Context, ids []string, first *int, after *string) (FriendsConnection, error) {
+	from := 0
+	if after != nil {
+		b, err := base64.StdEncoding.DecodeString(*after)
+		if err != nil {
+			return FriendsConnection{}, err
+		}
+		i, err := strconv.Atoi(strings.TrimPrefix(string(b), "cursor"))
+		if err != nil {
+			return FriendsConnection{}, err
+		}
+		from = i
+	}
+
+	to := len(ids)
+	if first != nil {
+		to = from + *first
+		if to > len(ids) {
+			to = len(ids)
+		}
+	}
+
+	return FriendsConnection{
+		ids:  ids,
+		from: from,
+		to:   to,
+	}, nil
+}
+
+type FriendsConnection struct {
+	ids  []string
+	from int
+	to   int
+}
+
+func (f *FriendsConnection) TotalCount() int {
+	return len(f.ids)
+}
+
+func (f *FriendsConnection) PageInfo() PageInfo {
+	return PageInfo{
+		StartCursor: encodeCursor(f.from),
+		EndCursor:   encodeCursor(f.to - 1),
+		HasNextPage: f.to < len(f.ids),
+	}
+}
+
+func encodeCursor(i int) string {
+	return base64.StdEncoding.EncodeToString([]byte(fmt.Sprintf("cursor%d", i+1)))
+}
+
+type FriendsEdge struct {
+	Cursor string
+	Node   Character
+}
+
+type PageInfo struct {
+	StartCursor string
+	EndCursor   string
+	HasNextPage bool
+}

example/starwars/starwars.go → example/starwars/resolvers.go

@@ -4,12 +4,8 @@ package starwars
 
 import (
 	"context"
-	"encoding/base64"
-	"strconv"
 	"strings"
 	"time"
-
-	"fmt"
 )
 
 type Resolver struct {
@@ -274,121 +270,3 @@ func NewResolver() *Resolver {
 
 	return &r
 }
-
-type Human struct {
-	ID           string
-	Name         string
-	FriendIds    []string
-	StarshipIds  []string
-	AppearsIn    []string
-	heightMeters float64
-	Mass         float64
-}
-
-func (h *Human) Height(unit string) float64 {
-	switch unit {
-	case "METER", "":
-		return h.heightMeters
-	case "FOOT":
-		return h.heightMeters * 3.28084
-	default:
-		panic("invalid unit")
-	}
-}
-
-type Starship struct {
-	ID           string
-	Name         string
-	History      [][2]int
-	lengthMeters float64
-}
-
-func (s *Starship) Length(unit string) float64 {
-	switch unit {
-	case "METER", "":
-		return s.lengthMeters
-	case "FOOT":
-		return s.lengthMeters * 3.28084
-	default:
-		panic("invalid unit")
-	}
-}
-
-type Review struct {
-	Stars      int
-	Commentary *string
-	Time       time.Time
-}
-
-type Droid struct {
-	ID              string
-	Name            string
-	FriendIds       []string
-	AppearsIn       []string
-	PrimaryFunction string
-}
-
-type Character interface{}
-type SearchResult interface{}
-
-func (r *Resolver) resolveFriendConnection(ctx context.Context, ids []string, first *int, after *string) (FriendsConnection, error) {
-	from := 0
-	if after != nil {
-		b, err := base64.StdEncoding.DecodeString(*after)
-		if err != nil {
-			return FriendsConnection{}, err
-		}
-		i, err := strconv.Atoi(strings.TrimPrefix(string(b), "cursor"))
-		if err != nil {
-			return FriendsConnection{}, err
-		}
-		from = i
-	}
-
-	to := len(ids)
-	if first != nil {
-		to = from + *first
-		if to > len(ids) {
-			to = len(ids)
-		}
-	}
-
-	return FriendsConnection{
-		ids:  ids,
-		from: from,
-		to:   to,
-	}, nil
-}
-
-type FriendsConnection struct {
-	ids  []string
-	from int
-	to   int
-}
-
-func (f *FriendsConnection) TotalCount() int {
-	return len(f.ids)
-}
-
-func (f *FriendsConnection) PageInfo() PageInfo {
-	return PageInfo{
-		StartCursor: encodeCursor(f.from),
-		EndCursor:   encodeCursor(f.to - 1),
-		HasNextPage: f.to < len(f.ids),
-	}
-}
-
-func encodeCursor(i int) string {
-	return base64.StdEncoding.EncodeToString([]byte(fmt.Sprintf("cursor%d", i+1)))
-}
-
-type FriendsEdge struct {
-	Cursor string
-	Node   Character
-}
-
-type PageInfo struct {
-	StartCursor string
-	EndCursor   string
-	HasNextPage bool
-}