Function sseClientFile

jsonrpc/codegen/sse.go:70–113

sseClientFile returns the file implementing the SSE client streaming implementation if any.

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

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

	// Check if any endpoint has SSE
	hasSSE := false
	for _, ed := range data.Endpoints {
		if ed.SSE != nil {
			hasSSE = true
			break
		}
	}
	if !hasSSE {
		return nil
	}

	path := filepath.Join(codegen.Gendir, "jsonrpc", codegen.SnakeCase(svc.Name()), "client", "stream.go")
	tmplSections := sseClientStreamSections(data)
	sections := make([]*codegen.SectionTemplate, 0, 1+len(tmplSections))
	sections = append(sections,
		codegen.Header(
			"stream",
			"client",
			[]*codegen.ImportSpec{
				{Path: "bufio"},
				{Path: "bytes"},
				{Path: "context"},
				{Path: "encoding/json"},
				{Path: "fmt"},
				{Path: "io"},
				{Path: "net/http"},
				{Path: "strings"},
				{Path: "sync"},
				codegen.GoaImport("jsonrpc"),
				codegen.GoaNamedImport("http", "goahttp"),
				{Path: genpkg + "/" + codegen.SnakeCase(svc.Name()), Name: data.Service.PkgName},
			},
		),
	)
	sections = append(sections, tmplSections...)
	return &codegen.File{Path: path, SectionTemplates: sections}
}

// sseServerStreamSections returns section templates for SSE server endpoints.
func sseServerStreamSections(data *httpcodegen.ServiceData) []*codegen.SectionTemplate {