mirror of
https://github.com/hanzoai/tasks.git
synced 2026-10-04 02:04:03 +00:00
Will DuanandGitHub
87b1a5687d
## What changed? Add businessID at namespace.ClusterNames() api ## Why? To allow different replication configuration for different workflows. ## How did you test it? - [x] built - [x] run locally and tested manually - [x] covered by existing tests - [ ] added new unit test(s) - [ ] added new functional test(s) ## Potential risks No risk.
419 lines
14 KiB
Go
419 lines
14 KiB
Go
package replication
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"math/rand"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/google/uuid"
|
|
"github.com/stretchr/testify/require"
|
|
"github.com/stretchr/testify/suite"
|
|
commonpb "go.temporal.io/api/common/v1"
|
|
"go.temporal.io/api/serviceerror"
|
|
enumsspb "go.temporal.io/server/api/enums/v1"
|
|
historyspb "go.temporal.io/server/api/history/v1"
|
|
"go.temporal.io/server/api/historyservice/v1"
|
|
"go.temporal.io/server/api/historyservicemock/v1"
|
|
persistencespb "go.temporal.io/server/api/persistence/v1"
|
|
replicationspb "go.temporal.io/server/api/replication/v1"
|
|
"go.temporal.io/server/common/cluster"
|
|
"go.temporal.io/server/common/collection"
|
|
"go.temporal.io/server/common/namespace"
|
|
"go.temporal.io/server/common/persistence"
|
|
"go.temporal.io/server/common/resourcetest"
|
|
serviceerrors "go.temporal.io/server/common/serviceerror"
|
|
"go.temporal.io/server/service/history/configs"
|
|
deletemanager "go.temporal.io/server/service/history/deletemanager"
|
|
"go.temporal.io/server/service/history/replication/eventhandler"
|
|
"go.temporal.io/server/service/history/shard"
|
|
"go.temporal.io/server/service/history/tests"
|
|
wcache "go.temporal.io/server/service/history/workflow/cache"
|
|
"go.uber.org/mock/gomock"
|
|
"google.golang.org/protobuf/types/known/timestamppb"
|
|
)
|
|
|
|
type (
|
|
taskExecutorSuite struct {
|
|
suite.Suite
|
|
*require.Assertions
|
|
controller *gomock.Controller
|
|
|
|
remoteCluster string
|
|
mockResource *resourcetest.Test
|
|
mockShard *shard.ContextTest
|
|
config *configs.Config
|
|
historyClient *historyservicemock.MockHistoryServiceClient
|
|
mockNamespaceCache *namespace.MockRegistry
|
|
clusterMetadata *cluster.MockMetadata
|
|
workflowCache *wcache.MockCache
|
|
remoteHistoryFetcher *eventhandler.MockHistoryPaginatedFetcher
|
|
replicationTaskExecutor *taskExecutorImpl
|
|
}
|
|
)
|
|
|
|
func TestTaskExecutorSuite(t *testing.T) {
|
|
s := new(taskExecutorSuite)
|
|
suite.Run(t, s)
|
|
}
|
|
|
|
func (s *taskExecutorSuite) SetupSuite() {
|
|
|
|
}
|
|
|
|
func (s *taskExecutorSuite) TearDownSuite() {
|
|
|
|
}
|
|
|
|
func (s *taskExecutorSuite) SetupTest() {
|
|
s.Assertions = require.New(s.T())
|
|
s.controller = gomock.NewController(s.T())
|
|
s.remoteCluster = cluster.TestAlternativeClusterName
|
|
|
|
s.config = tests.NewDynamicConfig()
|
|
s.mockShard = shard.NewTestContext(
|
|
s.controller,
|
|
&persistencespb.ShardInfo{
|
|
ShardId: 0,
|
|
RangeId: 1,
|
|
ReplicationDlqAckLevel: map[string]int64{
|
|
cluster.TestAlternativeClusterName: persistence.EmptyQueueMessageID,
|
|
},
|
|
},
|
|
s.config,
|
|
)
|
|
s.mockResource = s.mockShard.Resource
|
|
s.mockNamespaceCache = s.mockResource.NamespaceCache
|
|
s.clusterMetadata = s.mockResource.ClusterMetadata
|
|
|
|
s.historyClient = historyservicemock.NewMockHistoryServiceClient(s.controller)
|
|
s.workflowCache = wcache.NewMockCache(s.controller)
|
|
s.remoteHistoryFetcher = eventhandler.NewMockHistoryPaginatedFetcher(s.controller)
|
|
|
|
s.clusterMetadata.EXPECT().GetCurrentClusterName().Return(cluster.TestCurrentClusterName).AnyTimes()
|
|
s.mockNamespaceCache.EXPECT().GetNamespaceName(gomock.Any()).Return(tests.Namespace, nil).AnyTimes()
|
|
s.mockShard.SetHistoryClientForTesting(s.historyClient)
|
|
s.replicationTaskExecutor = NewTaskExecutor(
|
|
s.remoteCluster,
|
|
s.mockShard,
|
|
s.remoteHistoryFetcher,
|
|
deletemanager.NewMockDeleteManager(s.controller),
|
|
s.workflowCache,
|
|
).(*taskExecutorImpl)
|
|
}
|
|
|
|
func (s *taskExecutorSuite) TearDownTest() {
|
|
s.controller.Finish()
|
|
s.mockShard.StopForTest()
|
|
}
|
|
|
|
func (s *taskExecutorSuite) TestFilterTask_Apply() {
|
|
namespaceID := namespace.ID(uuid.NewString())
|
|
s.mockNamespaceCache.EXPECT().
|
|
GetNamespaceByID(namespaceID).
|
|
Return(namespace.NewGlobalNamespaceForTest(
|
|
nil,
|
|
nil,
|
|
&persistencespb.NamespaceReplicationConfig{Clusters: []string{
|
|
cluster.TestCurrentClusterName,
|
|
cluster.TestAlternativeClusterName,
|
|
}},
|
|
0,
|
|
), nil)
|
|
ok, err := s.replicationTaskExecutor.filterTask(namespaceID, "test-workflow-id", false)
|
|
s.NoError(err)
|
|
s.True(ok)
|
|
}
|
|
|
|
func (s *taskExecutorSuite) TestFilterTask_NotApply() {
|
|
namespaceID := namespace.ID(uuid.NewString())
|
|
s.mockNamespaceCache.EXPECT().
|
|
GetNamespaceByID(namespaceID).
|
|
Return(namespace.NewGlobalNamespaceForTest(
|
|
nil,
|
|
nil,
|
|
&persistencespb.NamespaceReplicationConfig{Clusters: []string{cluster.TestAlternativeClusterName}},
|
|
0,
|
|
), nil)
|
|
ok, err := s.replicationTaskExecutor.filterTask(namespaceID, "test-workflow-id", false)
|
|
s.NoError(err)
|
|
s.False(ok)
|
|
}
|
|
|
|
func (s *taskExecutorSuite) TestFilterTask_Error() {
|
|
namespaceID := namespace.ID(uuid.NewString())
|
|
s.mockNamespaceCache.EXPECT().
|
|
GetNamespaceByID(namespaceID).
|
|
Return(nil, fmt.Errorf("random error"))
|
|
ok, err := s.replicationTaskExecutor.filterTask(namespaceID, "test-workflow-id", false)
|
|
s.Error(err)
|
|
s.False(ok)
|
|
}
|
|
|
|
func (s *taskExecutorSuite) TestFilterTask_EnforceApply() {
|
|
namespaceID := namespace.ID(uuid.NewString())
|
|
ok, err := s.replicationTaskExecutor.filterTask(namespaceID, "test-workflow-id", true)
|
|
s.NoError(err)
|
|
s.True(ok)
|
|
}
|
|
|
|
func (s *taskExecutorSuite) TestFilterTask_NamespaceNotFound() {
|
|
namespaceID := namespace.ID(uuid.NewString())
|
|
s.mockNamespaceCache.EXPECT().
|
|
GetNamespaceByID(namespaceID).
|
|
Return(nil, &serviceerror.NamespaceNotFound{})
|
|
ok, err := s.replicationTaskExecutor.filterTask(namespaceID, "test-workflow-id", false)
|
|
s.NoError(err)
|
|
s.False(ok)
|
|
}
|
|
|
|
func (s *taskExecutorSuite) TestProcessTaskOnce_SyncActivityReplicationTask() {
|
|
namespaceID := namespace.ID(uuid.NewString())
|
|
workflowID := uuid.NewString()
|
|
runID := uuid.NewString()
|
|
task := &replicationspb.ReplicationTask{
|
|
TaskType: enumsspb.REPLICATION_TASK_TYPE_SYNC_ACTIVITY_TASK,
|
|
Attributes: &replicationspb.ReplicationTask_SyncActivityTaskAttributes{
|
|
SyncActivityTaskAttributes: &replicationspb.SyncActivityTaskAttributes{
|
|
NamespaceId: namespaceID.String(),
|
|
WorkflowId: workflowID,
|
|
RunId: runID,
|
|
Version: 1234,
|
|
ScheduledEventId: 2345,
|
|
ScheduledTime: nil,
|
|
StartedEventId: 2346,
|
|
StartedTime: nil,
|
|
LastHeartbeatTime: nil,
|
|
Attempt: 10,
|
|
LastFailure: nil,
|
|
LastWorkerIdentity: "",
|
|
LastStartedBuildId: "ABC",
|
|
LastStartedRedirectCounter: 8,
|
|
},
|
|
},
|
|
}
|
|
request := &historyservice.SyncActivityRequest{
|
|
NamespaceId: namespaceID.String(),
|
|
WorkflowId: workflowID,
|
|
RunId: runID,
|
|
Version: 1234,
|
|
ScheduledEventId: 2345,
|
|
ScheduledTime: nil,
|
|
StartedEventId: 2346,
|
|
StartedTime: nil,
|
|
LastHeartbeatTime: nil,
|
|
Attempt: 10,
|
|
LastFailure: nil,
|
|
LastWorkerIdentity: "",
|
|
LastStartedBuildId: "ABC",
|
|
LastStartedRedirectCounter: 8,
|
|
}
|
|
|
|
s.historyClient.EXPECT().SyncActivity(gomock.Any(), request).Return(&historyservice.SyncActivityResponse{}, nil)
|
|
err := s.replicationTaskExecutor.Execute(context.Background(), task, true)
|
|
s.NoError(err)
|
|
}
|
|
|
|
func (s *taskExecutorSuite) TestProcessTaskOnce_SyncActivityReplicationTask_Resend() {
|
|
namespaceID := namespace.ID(uuid.NewString())
|
|
workflowID := uuid.NewString()
|
|
runID := uuid.NewString()
|
|
task := &replicationspb.ReplicationTask{
|
|
TaskType: enumsspb.REPLICATION_TASK_TYPE_SYNC_ACTIVITY_TASK,
|
|
Attributes: &replicationspb.ReplicationTask_SyncActivityTaskAttributes{
|
|
SyncActivityTaskAttributes: &replicationspb.SyncActivityTaskAttributes{
|
|
NamespaceId: namespaceID.String(),
|
|
WorkflowId: workflowID,
|
|
RunId: runID,
|
|
Version: 1234,
|
|
ScheduledEventId: 2345,
|
|
ScheduledTime: nil,
|
|
StartedEventId: 2346,
|
|
StartedTime: nil,
|
|
LastHeartbeatTime: nil,
|
|
Attempt: 10,
|
|
LastFailure: nil,
|
|
LastWorkerIdentity: "",
|
|
},
|
|
},
|
|
}
|
|
request := &historyservice.SyncActivityRequest{
|
|
NamespaceId: namespaceID.String(),
|
|
WorkflowId: workflowID,
|
|
RunId: runID,
|
|
Version: 1234,
|
|
ScheduledEventId: 2345,
|
|
ScheduledTime: nil,
|
|
StartedEventId: 2346,
|
|
StartedTime: nil,
|
|
LastHeartbeatTime: nil,
|
|
Attempt: 10,
|
|
LastFailure: nil,
|
|
LastWorkerIdentity: "",
|
|
}
|
|
|
|
resendErr := serviceerrors.NewRetryReplication(
|
|
"some random error message",
|
|
namespaceID.String(),
|
|
workflowID,
|
|
runID,
|
|
123,
|
|
234,
|
|
345,
|
|
456,
|
|
)
|
|
s.historyClient.EXPECT().SyncActivity(gomock.Any(), request).Return(nil, resendErr)
|
|
emptyIterator := collection.NewPagingIterator(func(paginationToken []byte) ([]*eventhandler.HistoryBatch, []byte, error) {
|
|
return nil, nil, nil
|
|
})
|
|
s.remoteHistoryFetcher.EXPECT().GetSingleWorkflowHistoryPaginatedIteratorExclusive(
|
|
gomock.Any(),
|
|
s.remoteCluster,
|
|
namespaceID,
|
|
workflowID,
|
|
runID,
|
|
int64(123),
|
|
int64(234),
|
|
int64(345),
|
|
int64(456),
|
|
).Return(emptyIterator)
|
|
|
|
s.historyClient.EXPECT().SyncActivity(gomock.Any(), request).Return(&historyservice.SyncActivityResponse{}, nil)
|
|
err := s.replicationTaskExecutor.Execute(context.Background(), task, true)
|
|
s.NoError(err)
|
|
}
|
|
|
|
func (s *taskExecutorSuite) TestProcess_HistoryReplicationTask() {
|
|
namespaceID := namespace.ID(uuid.NewString())
|
|
workflowID := uuid.NewString()
|
|
runID := uuid.NewString()
|
|
task := &replicationspb.ReplicationTask{
|
|
TaskType: enumsspb.REPLICATION_TASK_TYPE_HISTORY_V2_TASK,
|
|
Attributes: &replicationspb.ReplicationTask_HistoryTaskAttributes{
|
|
HistoryTaskAttributes: &replicationspb.HistoryTaskAttributes{
|
|
NamespaceId: namespaceID.String(),
|
|
WorkflowId: workflowID,
|
|
RunId: runID,
|
|
VersionHistoryItems: []*historyspb.VersionHistoryItem{{EventId: 233, Version: 2333}},
|
|
Events: nil,
|
|
NewRunEvents: nil,
|
|
},
|
|
},
|
|
}
|
|
request := &historyservice.ReplicateEventsV2Request{
|
|
NamespaceId: namespaceID.String(),
|
|
WorkflowExecution: &commonpb.WorkflowExecution{
|
|
WorkflowId: workflowID,
|
|
RunId: runID,
|
|
},
|
|
VersionHistoryItems: []*historyspb.VersionHistoryItem{{EventId: 233, Version: 2333}},
|
|
Events: nil,
|
|
NewRunEvents: nil,
|
|
}
|
|
s.historyClient.EXPECT().ReplicateEventsV2(gomock.Any(), request).Return(&historyservice.ReplicateEventsV2Response{}, nil)
|
|
err := s.replicationTaskExecutor.Execute(context.Background(), task, true)
|
|
s.NoError(err)
|
|
}
|
|
|
|
func (s *taskExecutorSuite) TestProcess_HistoryReplicationTask_Resend() {
|
|
namespaceID := namespace.ID(uuid.NewString())
|
|
workflowID := uuid.NewString()
|
|
runID := uuid.NewString()
|
|
task := &replicationspb.ReplicationTask{
|
|
TaskType: enumsspb.REPLICATION_TASK_TYPE_HISTORY_V2_TASK,
|
|
Attributes: &replicationspb.ReplicationTask_HistoryTaskAttributes{
|
|
HistoryTaskAttributes: &replicationspb.HistoryTaskAttributes{
|
|
NamespaceId: namespaceID.String(),
|
|
WorkflowId: workflowID,
|
|
RunId: runID,
|
|
VersionHistoryItems: []*historyspb.VersionHistoryItem{{EventId: 233, Version: 2333}},
|
|
Events: nil,
|
|
NewRunEvents: nil,
|
|
},
|
|
},
|
|
}
|
|
request := &historyservice.ReplicateEventsV2Request{
|
|
NamespaceId: namespaceID.String(),
|
|
WorkflowExecution: &commonpb.WorkflowExecution{
|
|
WorkflowId: workflowID,
|
|
RunId: runID,
|
|
},
|
|
VersionHistoryItems: []*historyspb.VersionHistoryItem{{EventId: 233, Version: 2333}},
|
|
Events: nil,
|
|
NewRunEvents: nil,
|
|
}
|
|
|
|
resendErr := serviceerrors.NewRetryReplication(
|
|
"some random error message",
|
|
namespaceID.String(),
|
|
workflowID,
|
|
runID,
|
|
123,
|
|
234,
|
|
345,
|
|
456,
|
|
)
|
|
s.historyClient.EXPECT().ReplicateEventsV2(gomock.Any(), request).Return(nil, resendErr)
|
|
emptyIterator := collection.NewPagingIterator(func(paginationToken []byte) ([]*eventhandler.HistoryBatch, []byte, error) {
|
|
return nil, nil, nil
|
|
})
|
|
s.remoteHistoryFetcher.EXPECT().GetSingleWorkflowHistoryPaginatedIteratorExclusive(
|
|
gomock.Any(),
|
|
s.remoteCluster,
|
|
namespaceID,
|
|
workflowID,
|
|
runID,
|
|
int64(123),
|
|
int64(234),
|
|
int64(345),
|
|
int64(456),
|
|
).Return(emptyIterator)
|
|
|
|
s.historyClient.EXPECT().ReplicateEventsV2(gomock.Any(), request).Return(&historyservice.ReplicateEventsV2Response{}, nil)
|
|
err := s.replicationTaskExecutor.Execute(context.Background(), task, true)
|
|
s.NoError(err)
|
|
}
|
|
|
|
func (s *taskExecutorSuite) TestProcessTaskOnce_SyncWorkflowStateTask() {
|
|
namespaceID := namespace.ID(uuid.NewString())
|
|
task := &replicationspb.ReplicationTask{
|
|
TaskType: enumsspb.REPLICATION_TASK_TYPE_SYNC_WORKFLOW_STATE_TASK,
|
|
Attributes: &replicationspb.ReplicationTask_SyncWorkflowStateTaskAttributes{
|
|
SyncWorkflowStateTaskAttributes: &replicationspb.SyncWorkflowStateTaskAttributes{
|
|
WorkflowState: &persistencespb.WorkflowMutableState{
|
|
ExecutionInfo: &persistencespb.WorkflowExecutionInfo{
|
|
NamespaceId: namespaceID.String(),
|
|
},
|
|
},
|
|
},
|
|
},
|
|
}
|
|
s.historyClient.EXPECT().ReplicateWorkflowState(gomock.Any(), gomock.Any()).Return(&historyservice.ReplicateWorkflowStateResponse{}, nil)
|
|
|
|
err := s.replicationTaskExecutor.Execute(context.Background(), task, true)
|
|
s.NoError(err)
|
|
}
|
|
|
|
func (s *taskExecutorSuite) TestProcessTaskOnce_SyncHSMTask() {
|
|
namespaceID := namespace.ID(uuid.NewString())
|
|
workflowID := uuid.NewString()
|
|
runID := uuid.NewString()
|
|
task := &replicationspb.ReplicationTask{
|
|
SourceTaskId: rand.Int63(),
|
|
TaskType: enumsspb.REPLICATION_TASK_TYPE_SYNC_HSM_TASK,
|
|
Attributes: &replicationspb.ReplicationTask_SyncHsmAttributes{SyncHsmAttributes: &replicationspb.SyncHSMAttributes{
|
|
NamespaceId: namespaceID.String(),
|
|
WorkflowId: workflowID,
|
|
RunId: runID,
|
|
VersionHistory: &historyspb.VersionHistory{},
|
|
StateMachineNode: &persistencespb.StateMachineNode{},
|
|
}},
|
|
VisibilityTime: timestamppb.New(time.Now()),
|
|
}
|
|
|
|
// not handling SyncHSMTask in deprecated replication task processing code path
|
|
err := s.replicationTaskExecutor.Execute(context.Background(), task, true)
|
|
s.ErrorIs(err, ErrUnknownReplicationTask)
|
|
}
|