267
}
268
270
>
numTasks := 20
271
>
272
>
shardID := rand.Int31()
273
>
minTaskID := int64(1)
274
>
taskID := minTaskID
275
>
maxTaskID := taskID + int64(numTasks)
276
>
277
>
var tasks []sqlplugin.TransferTasksRow
278
>
for range numTasks {
279
>
task := s.newRandomTransferTaskRow(shardID, taskID)
280
>
taskID++
281
>
tasks = append(tasks, task)
282
>
}
283
>
result, err := s.store.InsertIntoTransferTasks(newExecutionContext(), tasks)
284
>
s.NoError(err)
285
>
rowsAffected, err := result.RowsAffected()
286
>
s.NoError(err)
287
>
s.Equal(numTasks, int(rowsAffected))
288
>
289
>
filter := sqlplugin.TransferTasksRangeFilter{
290
>
ShardID: shardID,
291
>
InclusiveMinTaskID: minTaskID,
292
>
ExclusiveMaxTaskID: maxTaskID,
293
>
PageSize: int(maxTaskID - minTaskID),
294
>
}
295
>
result, err = s.store.RangeDeleteFromTransferTasks(newExecutionContext(), filter)
296
>
s.NoError(err)
297
>
rowsAffected, err = result.RowsAffected()
298
>
s.NoError(err)
299
>
s.Equal(numTasks, int(rowsAffected))
300
>
301
>
rows, err := s.store.RangeSelectFromTransferTasks(newExecutionContext(), filter)
302
>
s.NoError(err)
303
>
for index := range rows {
304
rows[index].ShardID = shardID
305
}
307
}
308