Function runWALBenchmarkWithAppendMode

tempodb/wal/wal_test.go:507–565

(b *testing.B, encoding string, flushCount int, appendTrace bool, runner func([][]byte, []*tempopb.Trace, common.WALBlock))

}

func runWALBenchmarkWithAppendMode(b *testing.B, encoding string, flushCount int, appendTrace bool, runner func([][]byte, []*tempopb.Trace, common.WALBlock)) {
	wal, err := New(&Config{
		Filepath: b.TempDir(),
		Version:  encoding,
	})
	require.NoError(b, err, "unexpected error creating temp wal")

	blockID := uuid.New()

	meta := backend.NewBlockMeta("fake", blockID, encoding)
	block, err := wal.NewBlock(meta, model.CurrentEncoding)
	require.NoError(b, err, "unexpected error creating block")

	dec := model.MustNewSegmentDecoder(model.CurrentEncoding)

	objects := 250
	traces := make([]*tempopb.Trace, 0, objects)
	objs := make([][]byte, 0, objects)
	ids := make([][]byte, 0, objects)
	for i := 0; i < objects; i++ {
		id := make([]byte, 16)
		_, err = crand.Read(id)
		require.NoError(b, err)
		obj := test.MakeTrace(rand.Int()%10+1, id)

		trace.SortTrace(obj)

		ids = append(ids, id)
		traces = append(traces, obj)

		b1, err := dec.PrepareForWrite(obj, 0, 0)
		require.NoError(b, err)

		b2, err := dec.ToObject([][]byte{b1})
		require.NoError(b, err)

		objs = append(objs, b2)
	}

	b.ResetTimer()

	for flush := 0; flush < flushCount; flush++ {

		for i := range traces {
			if appendTrace {
				require.NoError(b, block.AppendTrace(ids[i], traces[i], 0, 0, true))
			} else {
				require.NoError(b, block.Append(ids[i], objs[i], 0, 0, true))
			}
		}

		err = block.Flush()
		require.NoError(b, err)
	}

	if runner != nil {
		runner(ids, traces, block)
	}