Function clientWSSections

http/codegen/websocket.go:405–453

clientWSSections returns section templates that contain client WebSocket specific code for the given service.

(data *ServiceData)

// clientWSSections returns section templates that contain client WebSocket
// specific code for the given service.
func clientWSSections(data *ServiceData) []*codegen.SectionTemplate {
	var sections []*codegen.SectionTemplate
	sections = append(sections, &codegen.SectionTemplate{
		Name:    "client-websocket-conn-configurer-struct-init",
		Source:  httpTemplates.Read(websocketConnConfigurerStructInitT),
		Data:    data,
		FuncMap: map[string]any{"isWebSocketEndpoint": IsWebSocketEndpoint},
	})
	for _, e := range data.Endpoints {
		if e.ClientWebSocket != nil {
			if e.ClientWebSocket.RecvTypeRef != "" {
				sections = append(sections, &codegen.SectionTemplate{
					Name:    "client-websocket-recv",
					Source:  httpTemplates.Read(websocketRecvT, websocketUpgradeP),
					Data:    e.ClientWebSocket,
					FuncMap: map[string]any{"upgradeParams": upgradeParams},
				})
			}
			switch e.ClientWebSocket.Kind {
			case expr.ClientStreamKind, expr.BidirectionalStreamKind:
				sections = append(sections, &codegen.SectionTemplate{
					Name:   "client-websocket-send",
					Source: httpTemplates.Read(websocketSendT, websocketUpgradeP),
					Data:   e.ClientWebSocket,
					FuncMap: map[string]any{
						"upgradeParams":    upgradeParams,
						"viewedServerBody": viewedServerBody,
					},
				})
			}
			if e.ClientWebSocket.MustClose {
				sections = append(sections, &codegen.SectionTemplate{
					Name:    "client-websocket-close",
					Source:  httpTemplates.Read(websocketCloseT),
					Data:    e.ClientWebSocket,
					FuncMap: map[string]any{"upgradeParams": upgradeParams},
				})
			}
			if e.Method.ViewedResult != nil && e.Method.ViewedResult.ViewName == "" {
				sections = append(sections, &codegen.SectionTemplate{
					Name:   "client-websocket-set-view",
					Source: httpTemplates.Read(websocketSetViewT),
					Data:   e.ClientWebSocket,
				})
			}
		}
	}
	return sections
}

// HasWebSocket returns true if at least one of the endpoints in the service
// defines a streaming payload or result.