interceptor.go

package interceptor

import (
	"fmt"
	"go.temporal.io/sdk/interceptor"
	"go.temporal.io/sdk/workflow"
)

type blobSizeLimitInterceptor struct {
	interceptor.WorkerInterceptorBase
}

func NewBlobSizeLimitInterceptor() interceptor.WorkerInterceptor {
	return &blobSizeLimitInterceptor{}
}

func (*blobSizeLimitInterceptor) InterceptWorkflow(
	ctx workflow.Context,
	next interceptor.WorkflowInboundInterceptor,
) interceptor.WorkflowInboundInterceptor {
	i := &blobSizeLimitWorkflowInboundInterceptor{}
	i.Next = next
	return i
}

type blobSizeLimitWorkflowInboundInterceptor struct {
	interceptor.WorkflowInboundInterceptorBase
}

func (i *blobSizeLimitWorkflowInboundInterceptor) Init(outbound interceptor.WorkflowOutboundInterceptor) error {
	injected := &blobSizeLimitWorkflowOutboundInterceptor{}
	injected.Next = outbound
	return i.Next.Init(injected)
}

type blobSizeLimitWorkflowOutboundInterceptor struct {
	interceptor.WorkflowOutboundInterceptorBase
}

func (i *blobSizeLimitWorkflowOutboundInterceptor) ExecuteActivity(
	ctx workflow.Context,
	activityType string,
	args ...interface{},
) workflow.Future {
	fmt.Printf("executeActivity header %s(%+v): %+v\n", activityType, args, interceptor.WorkflowHeader(ctx))
	return i.Next.ExecuteActivity(ctx, activityType, args...)
}