• Home
  • Features
  • Pricing
  • Docs
  • Announcements
  • Sign In

uber / cadence / 018f6432-870f-4576-92cf-48f7eedf1f30

10 May 2024 08:29PM UTC coverage: 68.971% (+0.03%) from 68.94%
018f6432-870f-4576-92cf-48f7eedf1f30

push

buildkite

web-flow
Add unit test for replication task fetcher (#6009)

23 of 24 new or added lines in 2 files covered. (95.83%)

38 existing lines in 12 files now uncovered.

101328 of 146914 relevant lines covered (68.97%)

2650.69 hits per line

Source File
Press 'n' to go to next uncovered line, 'b' for previous

96.07
/common/persistence/statsComputer.go
1
// Copyright (c) 2017-2020 Uber Technologies, Inc.
2
// Portions of the Software are attributed to Copyright (c) 2020 Temporal Technologies Inc.
3
//
4
// Permission is hereby granted, free of charge, to any person obtaining a copy
5
// of this software and associated documentation files (the "Software"), to deal
6
// in the Software without restriction, including without limitation the rights
7
// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
8
// copies of the Software, and to permit persons to whom the Software is
9
// furnished to do so, subject to the following conditions:
10
//
11
// The above copyright notice and this permission notice shall be included in
12
// all copies or substantial portions of the Software.
13
//
14
// THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
15
// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
16
// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
17
// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
18
// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
19
// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
20
// THE SOFTWARE.
21

22
package persistence
23

24
type (
25
        // statsComputer is to computing struct sizes after serialization
26
        statsComputer struct{}
27
)
28

29
func (sc *statsComputer) computeMutableStateStats(req *InternalGetWorkflowExecutionResponse) *MutableStateStats {
739✔
30
        executionInfoSize := computeExecutionInfoSize(req.State.ExecutionInfo)
739✔
31

739✔
32
        activityInfoCount := 0
739✔
33
        activityInfoSize := 0
739✔
34
        for _, ai := range req.State.ActivityInfos {
860✔
35
                activityInfoCount++
121✔
36
                activityInfoSize += computeActivityInfoSize(ai)
121✔
37
        }
121✔
38

39
        timerInfoCount := 0
739✔
40
        timerInfoSize := 0
739✔
41
        for _, ti := range req.State.TimerInfos {
740✔
42
                timerInfoCount++
1✔
43
                timerInfoSize += computeTimerInfoSize(ti)
1✔
44
        }
1✔
45

46
        childExecutionInfoCount := 0
739✔
47
        childExecutionInfoSize := 0
739✔
48
        for _, ci := range req.State.ChildExecutionInfos {
743✔
49
                childExecutionInfoCount++
4✔
50
                childExecutionInfoSize += computeChildInfoSize(ci)
4✔
51
        }
4✔
52

53
        signalInfoCount := 0
739✔
54
        signalInfoSize := 0
739✔
55
        for _, si := range req.State.SignalInfos {
739✔
UNCOV
56
                signalInfoCount++
×
UNCOV
57
                signalInfoSize += computeSignalInfoSize(si)
×
UNCOV
58
        }
×
59

60
        bufferedEventsCount := 0
739✔
61
        bufferedEventsSize := 0
739✔
62

739✔
63
        for _, be := range req.State.BufferedEvents {
742✔
64
                bufferedEventsCount++
3✔
65
                bufferedEventsSize += len(be.Data)
3✔
66
        }
3✔
67

68
        requestCancelInfoCount := len(req.State.RequestCancelInfos)
739✔
69

739✔
70
        totalSize := executionInfoSize
739✔
71
        totalSize += activityInfoSize
739✔
72
        totalSize += timerInfoSize
739✔
73
        totalSize += childExecutionInfoSize
739✔
74
        totalSize += signalInfoSize
739✔
75
        totalSize += bufferedEventsSize
739✔
76

739✔
77
        return &MutableStateStats{
739✔
78
                MutableStateSize:       totalSize,
739✔
79
                ExecutionInfoSize:      executionInfoSize,
739✔
80
                ActivityInfoSize:       activityInfoSize,
739✔
81
                TimerInfoSize:          timerInfoSize,
739✔
82
                ChildInfoSize:          childExecutionInfoSize,
739✔
83
                SignalInfoSize:         signalInfoSize,
739✔
84
                BufferedEventsSize:     bufferedEventsSize,
739✔
85
                ActivityInfoCount:      activityInfoCount,
739✔
86
                TimerInfoCount:         timerInfoCount,
739✔
87
                ChildInfoCount:         childExecutionInfoCount,
739✔
88
                SignalInfoCount:        signalInfoCount,
739✔
89
                BufferedEventsCount:    bufferedEventsCount,
739✔
90
                RequestCancelInfoCount: requestCancelInfoCount,
739✔
91
        }
739✔
92
}
93

94
func (sc *statsComputer) computeMutableStateUpdateStats(req *InternalUpdateWorkflowExecutionRequest) *MutableStateUpdateSessionStats {
4,452✔
95
        if req.NewWorkflowSnapshot != nil {
4,627✔
96
                return mergeMutableStateUpdateSessionStats(sc.computeWorkflowMutationStats(&req.UpdateWorkflowMutation), sc.computeWorkflowSnapshotStats(req.NewWorkflowSnapshot))
175✔
97
        }
175✔
98
        return sc.computeWorkflowMutationStats(&req.UpdateWorkflowMutation)
4,280✔
99
}
100

101
func (sc *statsComputer) computeMutableStateCreateStats(req *InternalCreateWorkflowExecutionRequest) *MutableStateUpdateSessionStats {
525✔
102
        return sc.computeWorkflowSnapshotStats(&req.NewWorkflowSnapshot)
525✔
103
}
525✔
104

105
func (sc *statsComputer) computeMutableStateConflictResolveStats(req *InternalConflictResolveWorkflowExecutionRequest) *MutableStateUpdateSessionStats {
3✔
106
        mss := sc.computeWorkflowSnapshotStats(&req.ResetWorkflowSnapshot)
3✔
107
        if req.NewWorkflowSnapshot != nil {
3✔
108
                mss = mergeMutableStateUpdateSessionStats(mss, sc.computeWorkflowSnapshotStats(req.NewWorkflowSnapshot))
×
109
        }
×
110
        if req.CurrentWorkflowMutation != nil {
3✔
111
                mss = mergeMutableStateUpdateSessionStats(mss, sc.computeWorkflowMutationStats(req.CurrentWorkflowMutation))
×
112
        }
×
113
        return mss
3✔
114
}
115

116
func (sc *statsComputer) computeWorkflowMutationStats(req *InternalWorkflowMutation) *MutableStateUpdateSessionStats {
4,453✔
117
        executionInfoSize := computeExecutionInfoSize(req.ExecutionInfo)
4,453✔
118

4,453✔
119
        activityInfoCount := 0
4,453✔
120
        activityInfoSize := 0
4,453✔
121
        for _, ai := range req.UpsertActivityInfos {
5,738✔
122
                activityInfoCount++
1,285✔
123
                activityInfoSize += computeActivityInfoSize(ai)
1,285✔
124
        }
1,285✔
125

126
        timerInfoCount := 0
4,453✔
127
        timerInfoSize := 0
4,453✔
128
        for _, ti := range req.UpsertTimerInfos {
4,489✔
129
                timerInfoCount++
36✔
130
                timerInfoSize += computeTimerInfoSize(ti)
36✔
131
        }
36✔
132

133
        childExecutionInfoCount := 0
4,453✔
134
        childExecutionInfoSize := 0
4,453✔
135
        for _, ci := range req.UpsertChildExecutionInfos {
4,496✔
136
                childExecutionInfoCount++
43✔
137
                childExecutionInfoSize += computeChildInfoSize(ci)
43✔
138
        }
43✔
139

140
        signalInfoCount := 0
4,453✔
141
        signalInfoSize := 0
4,453✔
142
        for _, si := range req.UpsertSignalInfos {
4,469✔
143
                signalInfoCount++
16✔
144
                signalInfoSize += computeSignalInfoSize(si)
16✔
145
        }
16✔
146

147
        bufferedEventsSize := 0
4,453✔
148
        if req.NewBufferedEvents != nil {
5,096✔
149
                bufferedEventsSize = len(req.NewBufferedEvents.Data)
643✔
150
        }
643✔
151

152
        requestCancelInfoCount := len(req.UpsertRequestCancelInfos)
4,453✔
153

4,453✔
154
        deleteActivityInfoCount := len(req.DeleteActivityInfos)
4,453✔
155

4,453✔
156
        deleteTimerInfoCount := len(req.DeleteTimerInfos)
4,453✔
157

4,453✔
158
        deleteChildInfoCount := len(req.DeleteChildExecutionInfos)
4,453✔
159

4,453✔
160
        deleteSignalInfoCount := len(req.DeleteSignalInfos)
4,453✔
161

4,453✔
162
        deleteRequestCancelInfoCount := len(req.DeleteRequestCancelInfos)
4,453✔
163

4,453✔
164
        transferTasksCount := len(req.TransferTasks)
4,453✔
165
        crossClusterTasksCount := len(req.CrossClusterTasks)
4,453✔
166
        timerTasksCount := len(req.TimerTasks)
4,453✔
167
        replicationTasksCount := len(req.ReplicationTasks)
4,453✔
168

4,453✔
169
        totalSize := executionInfoSize
4,453✔
170
        totalSize += activityInfoSize
4,453✔
171
        totalSize += timerInfoSize
4,453✔
172
        totalSize += childExecutionInfoSize
4,453✔
173
        totalSize += signalInfoSize
4,453✔
174
        totalSize += bufferedEventsSize
4,453✔
175

4,453✔
176
        return &MutableStateUpdateSessionStats{
4,453✔
177
                MutableStateSize:             totalSize,
4,453✔
178
                ExecutionInfoSize:            executionInfoSize,
4,453✔
179
                ActivityInfoSize:             activityInfoSize,
4,453✔
180
                TimerInfoSize:                timerInfoSize,
4,453✔
181
                ChildInfoSize:                childExecutionInfoSize,
4,453✔
182
                SignalInfoSize:               signalInfoSize,
4,453✔
183
                BufferedEventsSize:           bufferedEventsSize,
4,453✔
184
                ActivityInfoCount:            activityInfoCount,
4,453✔
185
                TimerInfoCount:               timerInfoCount,
4,453✔
186
                ChildInfoCount:               childExecutionInfoCount,
4,453✔
187
                SignalInfoCount:              signalInfoCount,
4,453✔
188
                RequestCancelInfoCount:       requestCancelInfoCount,
4,453✔
189
                DeleteActivityInfoCount:      deleteActivityInfoCount,
4,453✔
190
                DeleteTimerInfoCount:         deleteTimerInfoCount,
4,453✔
191
                DeleteChildInfoCount:         deleteChildInfoCount,
4,453✔
192
                DeleteSignalInfoCount:        deleteSignalInfoCount,
4,453✔
193
                DeleteRequestCancelInfoCount: deleteRequestCancelInfoCount,
4,453✔
194
                TransferTasksCount:           transferTasksCount,
4,453✔
195
                CrossClusterTaskCount:        crossClusterTasksCount,
4,453✔
196
                TimerTasksCount:              timerTasksCount,
4,453✔
197
                ReplicationTasksCount:        replicationTasksCount,
4,453✔
198
        }
4,453✔
199
}
200

201
func (sc *statsComputer) computeWorkflowSnapshotStats(req *InternalWorkflowSnapshot) *MutableStateUpdateSessionStats {
698✔
202
        executionInfoSize := computeExecutionInfoSize(req.ExecutionInfo)
698✔
203

698✔
204
        activityInfoCount := 0
698✔
205
        activityInfoSize := 0
698✔
206
        for _, ai := range req.ActivityInfos {
721✔
207
                activityInfoCount++
23✔
208
                activityInfoSize += computeActivityInfoSize(ai)
23✔
209
        }
23✔
210

211
        timerInfoCount := 0
698✔
212
        timerInfoSize := 0
698✔
213
        for _, ti := range req.TimerInfos {
701✔
214
                timerInfoCount++
3✔
215
                timerInfoSize += computeTimerInfoSize(ti)
3✔
216
        }
3✔
217

218
        childExecutionInfoCount := 0
698✔
219
        childExecutionInfoSize := 0
698✔
220
        for _, ci := range req.ChildExecutionInfos {
703✔
221
                childExecutionInfoCount++
5✔
222
                childExecutionInfoSize += computeChildInfoSize(ci)
5✔
223
        }
5✔
224

225
        signalInfoCount := 0
698✔
226
        signalInfoSize := 0
698✔
227
        for _, si := range req.SignalInfos {
699✔
228
                signalInfoCount++
1✔
229
                signalInfoSize += computeSignalInfoSize(si)
1✔
230
        }
1✔
231

232
        requestCancelInfoCount := len(req.RequestCancelInfos)
698✔
233

698✔
234
        transferTasksCount := len(req.TransferTasks)
698✔
235
        crossClusterTasksCount := len(req.CrossClusterTasks)
698✔
236
        timerTasksCount := len(req.TimerTasks)
698✔
237
        replicationTasksCount := len(req.ReplicationTasks)
698✔
238

698✔
239
        totalSize := executionInfoSize
698✔
240
        totalSize += activityInfoSize
698✔
241
        totalSize += timerInfoSize
698✔
242
        totalSize += childExecutionInfoSize
698✔
243
        totalSize += signalInfoSize
698✔
244

698✔
245
        return &MutableStateUpdateSessionStats{
698✔
246
                MutableStateSize:       totalSize,
698✔
247
                ExecutionInfoSize:      executionInfoSize,
698✔
248
                ActivityInfoSize:       activityInfoSize,
698✔
249
                TimerInfoSize:          timerInfoSize,
698✔
250
                ChildInfoSize:          childExecutionInfoSize,
698✔
251
                SignalInfoSize:         signalInfoSize,
698✔
252
                ActivityInfoCount:      activityInfoCount,
698✔
253
                TimerInfoCount:         timerInfoCount,
698✔
254
                ChildInfoCount:         childExecutionInfoCount,
698✔
255
                SignalInfoCount:        signalInfoCount,
698✔
256
                RequestCancelInfoCount: requestCancelInfoCount,
698✔
257
                TransferTasksCount:     transferTasksCount,
698✔
258
                CrossClusterTaskCount:  crossClusterTasksCount,
698✔
259
                TimerTasksCount:        timerTasksCount,
698✔
260
                ReplicationTasksCount:  replicationTasksCount,
698✔
261
        }
698✔
262
}
263

264
func mergeMutableStateUpdateSessionStats(stats ...*MutableStateUpdateSessionStats) *MutableStateUpdateSessionStats {
175✔
265
        result := &MutableStateUpdateSessionStats{}
175✔
266
        for _, s := range stats {
522✔
267
                result.MutableStateSize += s.MutableStateSize
347✔
268

347✔
269
                result.ExecutionInfoSize += s.ExecutionInfoSize
347✔
270
                result.ActivityInfoSize += s.ActivityInfoSize
347✔
271
                result.TimerInfoSize += s.TimerInfoSize
347✔
272
                result.ChildInfoSize += s.ChildInfoSize
347✔
273
                result.SignalInfoSize += s.SignalInfoSize
347✔
274
                result.BufferedEventsSize += s.BufferedEventsSize
347✔
275

347✔
276
                result.ActivityInfoCount += s.ActivityInfoCount
347✔
277
                result.TimerInfoCount += s.TimerInfoCount
347✔
278
                result.ChildInfoCount += s.ChildInfoCount
347✔
279
                result.SignalInfoCount += s.SignalInfoCount
347✔
280
                result.RequestCancelInfoCount += s.RequestCancelInfoCount
347✔
281

347✔
282
                result.DeleteActivityInfoCount += s.DeleteActivityInfoCount
347✔
283
                result.DeleteTimerInfoCount += s.DeleteTimerInfoCount
347✔
284
                result.DeleteChildInfoCount += s.DeleteChildInfoCount
347✔
285
                result.DeleteSignalInfoCount += s.DeleteSignalInfoCount
347✔
286
                result.DeleteRequestCancelInfoCount += s.DeleteRequestCancelInfoCount
347✔
287

347✔
288
                result.TransferTasksCount += s.TransferTasksCount
347✔
289
                result.CrossClusterTaskCount += s.CrossClusterTaskCount
347✔
290
                result.TimerInfoCount += s.TimerInfoCount
347✔
291
                result.ReplicationTasksCount += s.ReplicationTasksCount
347✔
292
        }
347✔
293
        return result
175✔
294
}
295

296
func computeExecutionInfoSize(executionInfo *InternalWorkflowExecutionInfo) int {
5,886✔
297
        size := len(executionInfo.WorkflowID)
5,886✔
298
        size += len(executionInfo.TaskList)
5,886✔
299
        size += len(executionInfo.WorkflowTypeName)
5,886✔
300
        size += len(executionInfo.ParentWorkflowID)
5,886✔
301

5,886✔
302
        return size
5,886✔
303
}
5,886✔
304

305
func computeActivityInfoSize(ai *InternalActivityInfo) int {
1,425✔
306
        size := len(ai.ActivityID)
1,425✔
307
        if ai.ScheduledEvent != nil {
1,427✔
308
                size += len(ai.ScheduledEvent.Data)
2✔
309
        }
2✔
310
        if ai.StartedEvent != nil {
1,427✔
311
                size += len(ai.StartedEvent.Data)
2✔
312
        }
2✔
313
        size += len(ai.Details)
1,425✔
314

1,425✔
315
        return size
1,425✔
316
}
317

318
func computeTimerInfoSize(ti *TimerInfo) int {
42✔
319
        size := len(ti.TimerID)
42✔
320

42✔
321
        return size
42✔
322
}
42✔
323

324
func computeChildInfoSize(ci *InternalChildExecutionInfo) int {
52✔
325
        size := 0
52✔
326
        if ci.InitiatedEvent != nil {
52✔
327
                size += len(ci.InitiatedEvent.Data)
×
328
        }
×
329
        if ci.StartedEvent != nil {
52✔
330
                size += len(ci.StartedEvent.Data)
×
331
        }
×
332
        return size
52✔
333
}
334

335
func computeSignalInfoSize(si *SignalInfo) int {
19✔
336
        size := len(si.SignalName)
19✔
337
        size += len(si.Input)
19✔
338
        size += len(si.Control)
19✔
339

19✔
340
        return size
19✔
341
}
19✔
STATUS · Troubleshooting · Open an Issue · Sales · Support · CAREERS · ENTERPRISE · START FREE · SCHEDULE DEMO
ANNOUNCEMENTS · TWITTER · TOS & SLA · Supported CI Services · What's a CI service? · Automated Testing

© 2026 Coveralls, Inc