Function wrapperFile

codegen/service/interceptors.go:133–216

wrapperFile returns the file containing the interceptor wrappers.

(svc *Data)

// wrapperFile returns the file containing the interceptor wrappers.
func wrapperFile(svc *Data) *codegen.File {
	path := filepath.Join(codegen.Gendir, svc.PathName, "interceptor_wrappers.go")

	var sections []*codegen.SectionTemplate
	sections = append(sections, codegen.Header("Interceptor wrappers", svc.PkgName, []*codegen.ImportSpec{
		{Path: "context"},
		{Path: "fmt"},
		codegen.GoaImport(""),
	}))

	// Generate any interceptor stream wrapper struct types first
	var wrappedServerStreams, wrappedClientStreams []*StreamInterceptorData
	if len(svc.ServerInterceptors) > 0 {
		wrappedServerStreams = collectWrappedStreams(svc.ServerInterceptors, true)
		if len(wrappedServerStreams) > 0 {
			sections = append(sections, &codegen.SectionTemplate{
				Name:   "server-interceptor-stream-wrapper-types",
				Source: serviceTemplates.Read(serverInterceptorStreamWrapperTypesT),
				Data: map[string]any{
					"WrappedServerStreams": wrappedServerStreams,
				},
			})
		}
	}
	if len(svc.ClientInterceptors) > 0 {
		wrappedClientStreams = collectWrappedStreams(svc.ClientInterceptors, false)
		if len(wrappedClientStreams) > 0 {
			sections = append(sections, &codegen.SectionTemplate{
				Name:   "client-interceptor-stream-wrapper-types",
				Source: serviceTemplates.Read(clientInterceptorStreamWrapperTypesT),
				Data: map[string]any{
					"WrappedClientStreams": wrappedClientStreams,
				},
			})
		}
	}

	// Generate the interceptor wrapper functions next (only once)
	if len(svc.ServerInterceptors) > 0 {
		sections = append(sections, &codegen.SectionTemplate{
			Name:   "server-interceptor-wrappers",
			Source: serviceTemplates.Read(serverInterceptorWrappersT),
			Data: map[string]any{
				"Service":            svc.Name,
				"ServerInterceptors": svc.ServerInterceptors,
			},
		})
	}
	if len(svc.ClientInterceptors) > 0 {
		sections = append(sections, &codegen.SectionTemplate{
			Name:   "client-interceptor-wrappers",
			Source: serviceTemplates.Read(clientInterceptorWrappersT),
			Data: map[string]any{
				"Service":            svc.Name,
				"ClientInterceptors": svc.ClientInterceptors,
			},
		})
	}