@@ -24,12 +24,13 @@ package execution
2424import (
2525 "context"
2626 "fmt"
27+ "maps"
2728 "math/rand"
2829 "runtime/debug"
30+ "slices"
2931 "time"
3032
3133 "github.com/pborman/uuid"
32- "golang.org/x/exp/maps"
3334
3435 "github.com/uber/cadence/common"
3536 "github.com/uber/cadence/common/backoff"
@@ -1486,18 +1487,18 @@ func (e *mutableStateBuilder) CloseTransactionAsMutation(
14861487 ExecutionInfo : e .executionInfo ,
14871488 VersionHistories : e .versionHistories ,
14881489
1489- UpsertActivityInfos : maps .Values (e .updateActivityInfos ),
1490- DeleteActivityInfos : maps .Keys (e .deleteActivityInfos ),
1491- UpsertTimerInfos : maps .Values (e .updateTimerInfos ),
1492- DeleteTimerInfos : maps .Keys (e .deleteTimerInfos ),
1493- UpsertChildExecutionInfos : maps .Values (e .updateChildExecutionInfos ),
1494- DeleteChildExecutionInfos : maps .Keys (e .deleteChildExecutionInfos ),
1495- UpsertRequestCancelInfos : maps .Values (e .updateRequestCancelInfos ),
1496- DeleteRequestCancelInfos : maps .Keys (e .deleteRequestCancelInfos ),
1497- UpsertSignalInfos : maps .Values (e .updateSignalInfos ),
1498- DeleteSignalInfos : maps .Keys (e .deleteSignalInfos ),
1499- UpsertSignalRequestedIDs : maps .Keys (e .updateSignalRequestedIDs ),
1500- DeleteSignalRequestedIDs : maps .Keys (e .deleteSignalRequestedIDs ),
1490+ UpsertActivityInfos : slices . Collect ( maps .Values (e .updateActivityInfos ) ),
1491+ DeleteActivityInfos : slices . Collect ( maps .Keys (e .deleteActivityInfos ) ),
1492+ UpsertTimerInfos : slices . Collect ( maps .Values (e .updateTimerInfos ) ),
1493+ DeleteTimerInfos : slices . Collect ( maps .Keys (e .deleteTimerInfos ) ),
1494+ UpsertChildExecutionInfos : slices . Collect ( maps .Values (e .updateChildExecutionInfos ) ),
1495+ DeleteChildExecutionInfos : slices . Collect ( maps .Keys (e .deleteChildExecutionInfos ) ),
1496+ UpsertRequestCancelInfos : slices . Collect ( maps .Values (e .updateRequestCancelInfos ) ),
1497+ DeleteRequestCancelInfos : slices . Collect ( maps .Keys (e .deleteRequestCancelInfos ) ),
1498+ UpsertSignalInfos : slices . Collect ( maps .Values (e .updateSignalInfos ) ),
1499+ DeleteSignalInfos : slices . Collect ( maps .Keys (e .deleteSignalInfos ) ),
1500+ UpsertSignalRequestedIDs : slices . Collect ( maps .Keys (e .updateSignalRequestedIDs ) ),
1501+ DeleteSignalRequestedIDs : slices . Collect ( maps .Keys (e .deleteSignalRequestedIDs ) ),
15011502 NewBufferedEvents : e .updateBufferedEvents ,
15021503 ClearBufferedEvents : e .clearBufferedEvents ,
15031504
@@ -1575,12 +1576,12 @@ func (e *mutableStateBuilder) CloseTransactionAsSnapshot(
15751576 ExecutionInfo : e .executionInfo ,
15761577 VersionHistories : e .versionHistories ,
15771578
1578- ActivityInfos : maps .Values (e .pendingActivityInfoIDs ),
1579- TimerInfos : maps .Values (e .pendingTimerInfoIDs ),
1580- ChildExecutionInfos : maps .Values (e .pendingChildExecutionInfoIDs ),
1581- RequestCancelInfos : maps .Values (e .pendingRequestCancelInfoIDs ),
1582- SignalInfos : maps .Values (e .pendingSignalInfoIDs ),
1583- SignalRequestedIDs : maps .Keys (e .pendingSignalRequestedIDs ),
1579+ ActivityInfos : slices . Collect ( maps .Values (e .pendingActivityInfoIDs ) ),
1580+ TimerInfos : slices . Collect ( maps .Values (e .pendingTimerInfoIDs ) ),
1581+ ChildExecutionInfos : slices . Collect ( maps .Values (e .pendingChildExecutionInfoIDs ) ),
1582+ RequestCancelInfos : slices . Collect ( maps .Values (e .pendingRequestCancelInfoIDs ) ),
1583+ SignalInfos : slices . Collect ( maps .Values (e .pendingSignalInfoIDs ) ),
1584+ SignalRequestedIDs : slices . Collect ( maps .Keys (e .pendingSignalRequestedIDs ) ),
15841585
15851586 TasksByCategory : map [persistence.HistoryTaskCategory ][]persistence.Task {
15861587 persistence .HistoryTaskCategoryTransfer : e .insertTransferTasks ,
0 commit comments