Function pushGoStructUnmarshalJSON

scripts/codegen/go.ts:1121–1186

(lines: string[], typeName: string, fields: GoStructField[], ctx: GoCodegenCtx)

}

function pushGoStructUnmarshalJSON(lines: string[], typeName: string, fields: GoStructField[], ctx: GoCodegenCtx): void {
    const unionFields = fields
        .map((field) => ({ field, unionField: goDiscriminatedUnionField(field.goType, ctx) }))
        .filter((entry): entry is { field: GoStructField; unionField: GoDiscriminatedUnionField } => entry.unionField !== undefined);
    if (unionFields.length === 0) return;

    const blockLines: string[] = [];
    blockLines.push(`func (r *${typeName}) UnmarshalJSON(data []byte) error {`);
    blockLines.push(`\ttype raw${typeName} struct {`);
    for (const field of fields) {
        const unionField = goDiscriminatedUnionField(field.goType, ctx);
        let rawType = field.goType;
        if (unionField?.kind === "single") rawType = "json.RawMessage";
        if (unionField?.kind === "slice") rawType = "[]json.RawMessage";
        if (unionField?.kind === "map") rawType = "map[string]json.RawMessage";
        blockLines.push(`\t\t${field.goName} ${rawType} \`${field.jsonTag}\``);
    }
    blockLines.push(`\t}`);
    blockLines.push(`\tvar raw raw${typeName}`);
    blockLines.push(`\tif err := json.Unmarshal(data, &raw); err != nil {`);
    blockLines.push(`\t\treturn err`);
    blockLines.push(`\t}`);

    for (const field of fields) {
        const unionField = goDiscriminatedUnionField(field.goType, ctx);
        if (!unionField) {
            blockLines.push(`\tr.${field.goName} = raw.${field.goName}`);
            continue;
        }

        if (unionField.kind === "single") {
            blockLines.push(`\tif raw.${field.goName} != nil {`);
            blockLines.push(`\t\tvalue, err := ${unionField.unionInfo.unmarshalFuncName}(raw.${field.goName})`);
            blockLines.push(`\t\tif err != nil {`);
            blockLines.push(`\t\t\treturn err`);
            blockLines.push(`\t\t}`);
            blockLines.push(`\t\tr.${field.goName} = value`);
            blockLines.push(`\t}`);
        } else if (unionField.kind === "slice") {
            blockLines.push(`\tif raw.${field.goName} != nil {`);
            blockLines.push(`\t\tr.${field.goName} = make([]${unionField.unionInfo.typeName}, 0, len(raw.${field.goName}))`);
            blockLines.push(`\t\tfor _, rawItem := range raw.${field.goName} {`);
            blockLines.push(`\t\t\tvalue, err := ${unionField.unionInfo.unmarshalFuncName}(rawItem)`);
            blockLines.push(`\t\t\tif err != nil {`);
            blockLines.push(`\t\t\t\treturn err`);
            blockLines.push(`\t\t\t}`);
            blockLines.push(`\t\t\tr.${field.goName} = append(r.${field.goName}, value)`);
            blockLines.push(`\t\t}`);
            blockLines.push(`\t}`);
        } else {
            blockLines.push(`\tif raw.${field.goName} != nil {`);
            blockLines.push(`\t\tr.${field.goName} = make(map[string]${unionField.unionInfo.typeName}, len(raw.${field.goName}))`);
            blockLines.push(`\t\tfor key, rawValue := range raw.${field.goName} {`);
            blockLines.push(`\t\t\tvalue, err := ${unionField.unionInfo.unmarshalFuncName}(rawValue)`);
            blockLines.push(`\t\t\tif err != nil {`);
            blockLines.push(`\t\t\t\treturn err`);
            blockLines.push(`\t\t\t}`);
            blockLines.push(`\t\t\tr.${field.goName}[key] = value`);