Method Remaining

references.go:31–61

Remaining will call the next function for each type that has not been generated but should be.

(next func(object types.Object) error)

// Remaining will call the next function for each type that has not been generated
// but should be.
func (r *referencedTypes) Remaining(next func(object types.Object) error) error {
	// Keep looping over the referenced types until we don't generate anything new
	// TODO: This could be optimized with a queue vs a full loop every time.
	tried := make(map[string]map[string]struct{})
	for {
		generatedSomething := false
		for pkg, types := range r.ReferencedTypes {
			if _, ok := tried[pkg]; !ok {
				tried[pkg] = make(map[string]struct{})
			}

			for ty, generated := range types {
				if !generated.Generated {
					if _, ok := tried[pkg][ty]; ok {
						return fmt.Errorf("circular generation detected for %s.%s, infinite loop will not end", pkg, ty)
					}
					tried[pkg][ty] = struct{}{}
					err := next(generated.Object)
					if err != nil {
						return err
					}
					generatedSomething = true
				}
			}
		}
		if !generatedSomething {
			break
		}
	}
	return nil
}

func (r *referencedTypes) MarkReferenced(ty types.Object) {
	if r.state(ty) == nil {