snapshot of https://github.com/centrifugal/centrifugo.git (master, 2026-09-16; blobs > 131072B omitted)
This commit is contained in:
45
internal/gen/api/gen_handlers_consumer.go
Normal file
45
internal/gen/api/gen_handlers_consumer.go
Normal file
@@ -0,0 +1,45 @@
|
||||
package main
|
||||
|
||||
func generateHandlersConsumer() {
|
||||
generateToFile(
|
||||
headerHandlersConsuming, templateFuncHandlersConsuming, "internal/api/consuming_gen.go",
|
||||
nil,
|
||||
[]string{
|
||||
"Publish", "Broadcast", "Subscribe", "Unsubscribe", "Disconnect", "HistoryRemove", "Refresh",
|
||||
},
|
||||
)
|
||||
}
|
||||
|
||||
var headerHandlersConsuming = `// Code generated by internal/gen/api/main.go. DO NOT EDIT.
|
||||
|
||||
package api
|
||||
|
||||
import (
|
||||
"context"
|
||||
|
||||
. "github.com/centrifugal/centrifugo/v6/internal/apiproto"
|
||||
"github.com/centrifugal/centrifugo/v6/internal/metrics"
|
||||
|
||||
"go.opentelemetry.io/otel/codes"
|
||||
"go.opentelemetry.io/otel/trace"
|
||||
)
|
||||
`
|
||||
|
||||
var templateFuncHandlersConsuming = `
|
||||
func (h *ConsumingHandler) handle{{ .RequestCapitalized }}(ctx context.Context, data []byte) (*{{ .RequestCapitalized }}Result, error) {
|
||||
req, err := requestDecoder.Decode{{ .RequestCapitalized }}(data)
|
||||
if err != nil {
|
||||
return nil, ErrInvalidData
|
||||
}
|
||||
resp := h.api.{{ .RequestCapitalized }}(ctx, req)
|
||||
if h.config.UseOpenTelemetry && resp.Error != nil {
|
||||
span := trace.SpanFromContext(ctx)
|
||||
span.SetStatus(codes.Error, resp.Error.Error())
|
||||
}
|
||||
if resp.Error != nil {
|
||||
metrics.IncAPIError(h.api.config.Protocol, "{{ .RequestSnake }}", resp.Error.Code)
|
||||
return nil, resp.Error
|
||||
}
|
||||
return resp.Result, nil
|
||||
}
|
||||
`
|
||||
Reference in New Issue
Block a user