Function runWALTestWithAppendMode

tempodb/wal/wal_test.go:360–422

(t testing.TB, encoding string, appendTrace bool, runner func([][]byte, []*tempopb.Trace, common.WALBlock))

}

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

	blockID := uuid.New()

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

	enc := model.MustNewSegmentDecoder(model.CurrentEncoding)

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

		trace.SortTrace(obj)

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

		if appendTrace {
			err = block.AppendTrace(id, obj, 0, 0, true)
			require.NoError(t, err)
		} else {
			b1, err := enc.PrepareForWrite(obj, 0, 0)
			require.NoError(t, err)

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

			err = block.Append(id, b2, 0, 0, true)
			require.NoError(t, err)
		}

		if i%100 == 0 {
			err = block.Flush()
			require.NoError(t, err)
		}
	}
	err = block.Flush()
	require.NoError(t, err)

	runner(ids, objs, block)

	// rescan blocks
	blocks, err := wal.RescanBlocks(0, log.NewNopLogger())
	require.NoError(t, err, "unexpected error getting blocks")
	require.Len(t, blocks, 1)