diff --git a/go.mod b/go.mod index d39ce55ad7..f8c3600a46 100644 --- a/go.mod +++ b/go.mod @@ -60,7 +60,7 @@ require ( go.opentelemetry.io/otel/sdk v1.34.0 go.opentelemetry.io/otel/sdk/metric v1.34.0 go.opentelemetry.io/otel/trace v1.34.0 - go.temporal.io/api v1.62.2-0.20260212001044-b64b7ab17efc + go.temporal.io/api v1.62.2-0.20260213194545-c89ebac64f01 go.temporal.io/sdk v1.38.0 go.uber.org/fx v1.24.0 go.uber.org/mock v0.6.0 diff --git a/go.sum b/go.sum index a2fd2af1e5..54e56f011b 100644 --- a/go.sum +++ b/go.sum @@ -375,8 +375,8 @@ go.opentelemetry.io/otel/trace v1.34.0 h1:+ouXS2V8Rd4hp4580a8q23bg0azF2nI8cqLYnC go.opentelemetry.io/otel/trace v1.34.0/go.mod h1:Svm7lSjQD7kG7KJ/MUHPVXSDGz2OX4h0M2jHBhmSfRE= go.opentelemetry.io/proto/otlp v1.5.0 h1:xJvq7gMzB31/d406fB8U5CBdyQGw4P399D1aQWU/3i4= go.opentelemetry.io/proto/otlp v1.5.0/go.mod h1:keN8WnHxOy8PG0rQZjJJ5A2ebUoafqWp0eVQ4yIXvJ4= -go.temporal.io/api v1.62.2-0.20260212001044-b64b7ab17efc h1:hFmYOJKWlLJVG5wfziY8SLv+iXEswyGVnm9c7ebMi7k= -go.temporal.io/api v1.62.2-0.20260212001044-b64b7ab17efc/go.mod h1:iaxoP/9OXMJcQkETTECfwYq4cw/bj4nwov8b3ZLVnXM= +go.temporal.io/api v1.62.2-0.20260213194545-c89ebac64f01 h1:PTeSnyAfgKbp3E8CID1WY+Fi21HmMtEvCJ2/Qrw8pQU= +go.temporal.io/api v1.62.2-0.20260213194545-c89ebac64f01/go.mod h1:iaxoP/9OXMJcQkETTECfwYq4cw/bj4nwov8b3ZLVnXM= go.temporal.io/sdk v1.38.0 h1:4Bok5LEdED7YKpsSjIa3dDqram5VOq+ydBf4pyx0Wo4= go.temporal.io/sdk v1.38.0/go.mod h1:a+R2Ej28ObvHoILbHaxMyind7M6D+W0L7edt5UJF4SE= go.uber.org/atomic v1.5.0/go.mod h1:sABNBOSYdrvTF6hTgEIbc7YasKWGhgEQZyfxyTvoXHQ= diff --git a/service/frontend/namespace_handler.go b/service/frontend/namespace_handler.go index 764a705d60..604c69eaa7 100644 --- a/service/frontend/namespace_handler.go +++ b/service/frontend/namespace_handler.go @@ -864,6 +864,7 @@ func (d *namespaceHandler) createResponse( WorkerHeartbeats: d.config.WorkerHeartbeatsEnabled(info.Name), WorkflowPause: d.config.WorkflowPauseEnabled(info.Name), StandaloneActivities: d.config.Activity.Enabled(info.Name), + WorkerPollCompleteOnShutdown: d.config.EnableCancelWorkerPollsOnShutdown(info.Name), }, Limits: &namespacepb.NamespaceInfo_Limits{ BlobSizeLimitError: int64(d.config.BlobSizeLimitError(info.Name)), diff --git a/service/frontend/namespace_handler_test.go b/service/frontend/namespace_handler_test.go index 5011e768c1..d586a1a95a 100644 --- a/service/frontend/namespace_handler_test.go +++ b/service/frontend/namespace_handler_test.go @@ -386,6 +386,7 @@ func (s *namespaceHandlerCommonSuite) TestCapabilitiesAndLimits() { s.True(resp.NamespaceInfo.Capabilities.WorkerHeartbeats) s.False(resp.NamespaceInfo.Capabilities.WorkflowPause) s.False(resp.NamespaceInfo.Capabilities.StandaloneActivities) + s.False(resp.NamespaceInfo.Capabilities.WorkerPollCompleteOnShutdown) s.Equal(int64(2*1024*1024), resp.NamespaceInfo.Limits.BlobSizeLimitError) s.Equal(int64(2*1024*1024), resp.NamespaceInfo.Limits.MemoSizeLimitError) @@ -399,6 +400,7 @@ func (s *namespaceHandlerCommonSuite) TestCapabilitiesAndLimits() { s.config.Activity.Enabled = dc.GetBoolPropertyFnFilteredByNamespace(true) s.config.BlobSizeLimitError = dc.GetIntPropertyFnFilteredByNamespace(1024) s.config.MemoSizeLimitError = dc.GetIntPropertyFnFilteredByNamespace(512) + s.config.EnableCancelWorkerPollsOnShutdown = dc.GetBoolPropertyFnFilteredByNamespace(true) resp, err = s.handler.DescribeNamespace(context.Background(), &workflowservice.DescribeNamespaceRequest{ Namespace: "ns", @@ -411,6 +413,7 @@ func (s *namespaceHandlerCommonSuite) TestCapabilitiesAndLimits() { s.False(resp.NamespaceInfo.Capabilities.WorkerHeartbeats) s.True(resp.NamespaceInfo.Capabilities.WorkflowPause) s.True(resp.NamespaceInfo.Capabilities.StandaloneActivities) + s.True(resp.NamespaceInfo.Capabilities.WorkerPollCompleteOnShutdown) s.Equal(int64(1024), resp.NamespaceInfo.Limits.BlobSizeLimitError) s.Equal(int64(512), resp.NamespaceInfo.Limits.MemoSizeLimitError) }