mirror of
https://github.com/hanzoai/tasks.git
synced 2026-11-03 03:56:04 +00:00
zeekay
5a601d081b
Full module path rebrand from Temporal to Hanzo Tasks: - go.mod module: github.com/hanzoai/tasks - All internal Go imports updated (1906 files) - cmd/server renamed to cmd/tasksd - Binary name: tasksd (was temporal-server) - Docker images: hanzoai/tasks (was temporalio/server) - README rewritten with Hanzo Tasks branding - LLM.md created with integration notes - Makefile, goreleaser, CI workflows updated - go.temporal.io/sdk and go.temporal.io/api imports preserved (separate repos) Fork of Temporal.io (MIT license).
105 lines
3.3 KiB
Go
105 lines
3.3 KiB
Go
// Code generated by protoc-gen-go-chasm. DO NOT EDIT.
|
|
package testspb
|
|
|
|
import (
|
|
"context"
|
|
"math/rand"
|
|
"time"
|
|
|
|
"github.com/hanzoai/tasks/client/history"
|
|
"github.com/hanzoai/tasks/common"
|
|
"github.com/hanzoai/tasks/common/backoff"
|
|
"github.com/hanzoai/tasks/common/config"
|
|
"github.com/hanzoai/tasks/common/dynamicconfig"
|
|
"github.com/hanzoai/tasks/common/headers"
|
|
"github.com/hanzoai/tasks/common/log"
|
|
"github.com/hanzoai/tasks/common/membership"
|
|
"github.com/hanzoai/tasks/common/metrics"
|
|
"github.com/hanzoai/tasks/common/primitives"
|
|
"google.golang.org/grpc"
|
|
)
|
|
|
|
// TestServiceLayeredClient is a client for TestService.
|
|
type TestServiceLayeredClient struct {
|
|
metricsHandler metrics.Handler
|
|
numShards int32
|
|
redirector history.Redirector[TestServiceClient]
|
|
retryPolicy backoff.RetryPolicy
|
|
}
|
|
|
|
// NewTestServiceLayeredClient initializes a new TestServiceLayeredClient.
|
|
func NewTestServiceLayeredClient(
|
|
dc *dynamicconfig.Collection,
|
|
rpcFactory common.RPCFactory,
|
|
monitor membership.Monitor,
|
|
config *config.Persistence,
|
|
logger log.Logger,
|
|
metricsHandler metrics.Handler,
|
|
) (TestServiceClient, error) {
|
|
resolver, err := monitor.GetResolver(primitives.HistoryService)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
connections := history.NewConnectionPool(resolver, rpcFactory, NewTestServiceClient)
|
|
var redirector history.Redirector[TestServiceClient]
|
|
if dynamicconfig.HistoryClientOwnershipCachingEnabled.Get(dc)() {
|
|
redirector = history.NewCachingRedirector(
|
|
connections,
|
|
resolver,
|
|
logger,
|
|
dynamicconfig.HistoryClientOwnershipCachingStaleTTL.Get(dc),
|
|
)
|
|
} else {
|
|
redirector = history.NewBasicRedirector(connections, resolver)
|
|
}
|
|
return &TestServiceLayeredClient{
|
|
metricsHandler: metricsHandler,
|
|
redirector: redirector,
|
|
numShards: config.NumHistoryShards,
|
|
retryPolicy: common.CreateHistoryClientRetryPolicy(),
|
|
}, nil
|
|
}
|
|
func (c *TestServiceLayeredClient) callTestNoRetry(
|
|
ctx context.Context,
|
|
request *TestRequest,
|
|
opts ...grpc.CallOption,
|
|
) (*TestResponse, error) {
|
|
var response *TestResponse
|
|
var err error
|
|
startTime := time.Now().UTC()
|
|
// the caller is a namespace, hence the tag below.
|
|
caller := headers.GetCallerInfo(ctx).CallerName
|
|
metricsHandler := c.metricsHandler.WithTags(
|
|
metrics.OperationTag("TestService.Test"),
|
|
metrics.NamespaceTag(caller),
|
|
metrics.ServiceRoleTag(metrics.HistoryRoleTagValue),
|
|
)
|
|
metrics.ClientRequests.With(metricsHandler).Record(1)
|
|
defer func() {
|
|
if err != nil {
|
|
metrics.ClientFailures.With(metricsHandler).Record(1, metrics.ServiceErrorTypeTag(err))
|
|
}
|
|
metrics.ClientLatency.With(metricsHandler).Record(time.Since(startTime))
|
|
}()
|
|
shardID := int32(rand.Intn(int(c.numShards)) + 1)
|
|
op := func(ctx context.Context, client TestServiceClient) error {
|
|
var err error
|
|
ctx, cancel := context.WithTimeout(ctx, history.DefaultTimeout)
|
|
defer cancel()
|
|
response, err = client.Test(ctx, request, opts...)
|
|
return err
|
|
}
|
|
err = c.redirector.Execute(ctx, shardID, op)
|
|
return response, err
|
|
}
|
|
func (c *TestServiceLayeredClient) Test(
|
|
ctx context.Context,
|
|
request *TestRequest,
|
|
opts ...grpc.CallOption,
|
|
) (*TestResponse, error) {
|
|
call := func(ctx context.Context) (*TestResponse, error) {
|
|
return c.callTestNoRetry(ctx, request, opts...)
|
|
}
|
|
return backoff.ThrottleRetryContextWithReturn(ctx, call, c.retryPolicy, common.IsServiceClientTransientError)
|
|
}
|