(t *testing.T, f func(w *walBlock, ids []common.ID, trs []*tempopb.Trace))
| 459 | } |
| 460 | |
| 461 | func testWalBlock(t *testing.T, f func(w *walBlock, ids []common.ID, trs []*tempopb.Trace)) { |
| 462 | meta := backend.NewBlockMeta("fake", uuid.New(), VersionString) |
| 463 | w, err := createWALBlock(meta, t.TempDir(), model.CurrentEncoding, 0) |
| 464 | require.NoError(t, err) |
| 465 | |
| 466 | decoder := model.MustNewSegmentDecoder(model.CurrentEncoding) |
| 467 | |
| 468 | count := 30 |
| 469 | ids := make([]common.ID, count) |
| 470 | trs := make([]*tempopb.Trace, count) |
| 471 | for i := 0; i < count; i++ { |
| 472 | ids[i] = test.ValidTraceID(nil) |
| 473 | trs[i] = test.MakeTrace(10, ids[i]) |
| 474 | trace.SortTrace(trs[i]) |
| 475 | |
| 476 | b1, err := decoder.PrepareForWrite(trs[i], 0, 0) |
| 477 | require.NoError(t, err) |
| 478 | |
| 479 | b2, err := decoder.ToObject([][]byte{b1}) |
| 480 | require.NoError(t, err) |
| 481 | |
| 482 | err = w.Append(ids[i], b2, 0, 0, true) |
| 483 | require.NoError(t, err) |
| 484 | |
| 485 | if i%10 == 0 { |
| 486 | require.NoError(t, w.Flush()) |
| 487 | } |
| 488 | } |
| 489 | |
| 490 | require.NoError(t, w.Flush()) |
| 491 | |
| 492 | f(w, ids, trs) |
| 493 | } |
| 494 | |
| 495 | func BenchmarkWalTraceQL(b *testing.B) { |
| 496 | reqs := []string{ |
no test coverage detected