Function sseServerStreamFile

jsonrpc/codegen/sse_server.go:14–100

sseServerStreamFile returns the file implementing the JSON-RPC SSE server streaming implementation if any.

(genpkg string, svc *expr.HTTPServiceExpr, services *httpcodegen.ServicesData)

// sseServerStreamFile returns the file implementing the JSON-RPC SSE server
// streaming implementation if any.
func sseServerStreamFile(genpkg string, svc *expr.HTTPServiceExpr, services *httpcodegen.ServicesData) *codegen.File {
	data := services.Get(svc.Name())
	if data == nil {
		return nil
	}

	// Check if service has streaming methods
	hasStreaming := false
	for _, m := range data.Service.Methods {
		if m.ServerStream != nil {
			hasStreaming = true
			break
		}
	}
	if !hasStreaming {
		return nil
	}

	funcs := map[string]any{
		"lowerInitial": lowerInitial,
		"allErrors":    allErrors,
		"hasErrors": func() bool {
			for _, m := range data.Service.Methods {
				if len(m.Errors) > 0 {
					return true
				}
			}
			return false
		},
		"hasStreamingPayload": func() bool {
			for _, m := range data.Service.Methods {
				if m.StreamingPayload != "" {
					return true
				}
			}
			return false
		},
		// dedupeBySSEEvent returns endpoints with unique SSE event type
		"dedupeBySSEEvent": func(eds []*httpcodegen.EndpointData) []*httpcodegen.EndpointData {
			seen := make(map[string]struct{})
			out := make([]*httpcodegen.EndpointData, 0, len(eds))
			for _, e := range eds {
				if e == nil || e.SSE == nil || e.SSE.EventTypeRef == "" {
					continue
				}
				if _, ok := seen[e.SSE.EventTypeRef]; ok {
					continue
				}
				seen[e.SSE.EventTypeRef] = struct{}{}
				out = append(out, e)
			}
			return out
		},
	}
	svcName := data.Service.PathName
	title := fmt.Sprintf("%s SSE server streaming", svc.Name())
	imports := make([]*codegen.ImportSpec, 0, 11+len(data.Service.UserTypeImports))
	imports = append(imports,