213
func (b *EventStore) Finish(
214
flushBufferEvent bool,
216
>
defer func() {
217
>
b.state = HistoryBuilderStateSealed
218
>
}()
219
221
_, _ = b.FlushBufferToCurrentBatch()
222
}
224
>
225
>
dbEventsBatches := b.memEventsBatches
226
>
dbClearBuffer := b.dbClearBuffer
227
>
dbBufferBatch := b.memBufferBatch
228
>
memBufferBatch := b.dbBufferBatch
229
>
memBufferBatch = append(memBufferBatch, dbBufferBatch...)
230
>
scheduledIDToStartedID := b.scheduledIDToStartedID
231
>
requestIDToEventID := b.requestIDToEventID
232
>
233
>
b.memEventsBatches = nil
234
>
b.memBufferBatch = nil
235
>
b.memLatestBatch = nil
236
>
b.memLatestBatchSize = 0
237
>
b.dbClearBuffer = false
238
>
b.dbBufferBatch = nil
239
>
b.scheduledIDToStartedID = nil
240
>
241
>
if err := b.assignTaskIDs(dbEventsBatches); err != nil {
242
return nil, err
243
}
244
246
>
DBEventsBatches: dbEventsBatches,
247
>
DBClearBuffer: dbClearBuffer,
248
>
DBBufferBatch: dbBufferBatch,
249
>
MemBufferBatch: memBufferBatch,
250
>
ScheduledIDToStartedID: scheduledIDToStartedID,
251
>
RequestIDToEventID: requestIDToEventID,
252
>
}, nil
253
}
254
255
func (b *EventStore) assignTaskIDs(
256
dbEventsBatches [][]*historypb.HistoryEvent,
258
>
b.assertNotSealed()
259
>
260
>
if b.state == HistoryBuilderStateImmutable {
261
return nil
262
}