Function collectTypes

codegen/service/service_data.go:1021–1063

collectTypes recurses through the attribute to gather all user types and records them in userTypes.

(at *expr.AttributeExpr, scope *codegen.NameScope, seen map[string]struct{}, loc *codegen.Location)

// collectTypes recurses through the attribute to gather all user types and
// records them in userTypes.
func collectTypes(at *expr.AttributeExpr, scope *codegen.NameScope, seen map[string]struct{}, loc *codegen.Location) (data []*UserTypeData) {
	if at == nil || at.Type == expr.Empty {
		return data
	}
	collect := func(at *expr.AttributeExpr, loc *codegen.Location) []*UserTypeData {
		return collectTypes(at, scope, seen, loc)
	}
	switch dt := at.Type.(type) {
	case expr.UserType:
		if _, ok := seen[dt.ID()]; ok {
			return nil
		}
		typeLoc := codegen.UserTypeLocation(dt)
		if typeLoc == nil {
			typeLoc = loc
		}
		data = append(data, &UserTypeData{
			Name:        dt.Name(),
			VarName:     scope.GoTypeName(at),
			Description: dt.Attribute().Description,
			Def:         scope.GoTypeDef(dt.Attribute(), false, true),
			Ref:         scope.GoTypeRef(at),
			Loc:         typeLoc,
			Type:        dt,
		})
		seen[dt.ID()] = struct{}{}
		data = append(data, collect(dt.Attribute(), typeLoc)...)
	case *expr.Object:
		for _, nat := range *dt {
			data = append(data, collect(nat.Attribute, loc)...)
		}
	case *expr.Array:
		data = append(data, collect(dt.ElemType, loc)...)
	case *expr.Map:
		data = append(data, collect(dt.KeyType, loc)...)
		data = append(data, collect(dt.ElemType, loc)...)
	case *expr.Union:
		for _, nat := range dt.Values {
			data = append(data, collect(nat.Attribute, loc)...)
		}
	}
	return data
}

// collectUnionTypes traverses the attribute to gather all union sum-type
// definitions referenced by the service. It records each union by its hash to