Function generateGoRpcTypeCode

scripts/codegen/go.ts:3059–3096

(definitions: Record<string, JSONSchema7>, definitionCollections: DefinitionCollections)

}

function generateGoRpcTypeCode(definitions: Record<string, JSONSchema7>, definitionCollections: DefinitionCollections): GoGeneratedTypeCode {
    const ctx: GoCodegenCtx = {
        structs: [],
        encoding: [],
        enums: [],
        enumsByName: new Map(),
        discriminatedUnions: new Map(),
        generatedNames: new Set(),
        definitions: definitionCollections,
        packageName: "rpc",
    };
    ctx.skipDefinitionTypeNames = collectGoDiscriminatedUnionVariantDefinitionTypeNames(definitions, ctx);
    const schemaKeysByTypeName = new Map<string, string>();
    const entries = Object.entries(definitions)
        .sort(([left], [right]) => goDefinitionName(left).localeCompare(goDefinitionName(right)));

    for (const [definitionName, definition] of entries) {
        const typeName = goDefinitionName(definitionName);
        if (ctx.skipDefinitionTypeNames.has(typeName)) continue;
        const schemaKey = stableStringify(resolveSchema(definition, definitionCollections) ?? definition);
        const existingSchemaKey = schemaKeysByTypeName.get(typeName);
        if (existingSchemaKey && existingSchemaKey !== schemaKey) {
            throw new Error(`Conflicting Go RPC type name "${typeName}" for different schemas. Add a schema title/withTypeName to disambiguate.`);
        }
        schemaKeysByTypeName.set(typeName, schemaKey);
        emitGoRpcDefinition(definitionName, definition, ctx);
    }

    const lines: string[] = [];
    pushGoCodeBlocks(lines, sortedGoDeclaredTypeBlocks(ctx.structs));
    pushGoCodeBlocks(lines, sortedGoDeclaredTypeBlocks(ctx.enums));

    return {
        typeCode: joinGoCode(lines),
        encodingCode: goEncodingBlocksCode(ctx.encoding),
        discriminatedUnions: new Map(ctx.discriminatedUnions),
    };
}

function goDeclaredTypeName(code: string): string {
    return /^type\s+(\w+)\b/m.exec(code)?.[1] ?? code;