livekit-protocol 0.7.5

Livekit protocol and utilities for the Rust SDK
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91

package utils

import (
	"testing"
	"time"

	"github.com/stretchr/testify/require"

	"github.com/livekit/protocol/utils/mono"
)

func TestTimeSizeCache(t *testing.T) {
	cache := NewTimeSizeCache[string](TimeSizeCacheParams{
		TTL:     5 * time.Second,
		MaxSize: 10,
	})

	now := mono.Now()
	one := "one"
	cache.AddAt(&one, len(one), now.UnixNano())
	items := cache.Get()
	require.Equal(t, 1, len(items))
	require.Equal(t, one, *items[0])
	require.Equal(t, 3, cache.itemsSize)

	now = now.Add(time.Second)
	two := "two"
	cache.AddAt(&two, len(two), now.UnixNano())
	items = cache.Get()
	require.Equal(t, 2, len(items))
	require.Equal(t, one, *items[0])
	require.Equal(t, two, *items[1])
	require.Equal(t, 6, cache.itemsSize)

	// this should evict the "one" based on size
	now = now.Add(time.Second)
	three := "three"
	cache.AddAt(&three, len(three), now.UnixNano())
	items = cache.Get()
	require.Equal(t, 2, len(items))
	require.Equal(t, two, *items[0])
	require.Equal(t, three, *items[1])
	require.Equal(t, 8, cache.itemsSize)

	// this should evict the "two" based on size
	now = now.Add(time.Second)
	four := "four"
	cache.AddAt(&four, len(four), now.UnixNano())
	items = cache.Get()
	require.Equal(t, 2, len(items))
	require.Equal(t, three, *items[0])
	require.Equal(t, four, *items[1])
	require.Equal(t, 9, cache.itemsSize)

	// this should not evict any element as existing elements should not have aged enough
	now = now.Add(time.Second)
	cache.PruneAt(now.UnixNano())
	items = cache.Get()
	require.Equal(t, 2, len(items))
	require.Equal(t, 9, cache.itemsSize)

	// should evict "three" based on age
	now = now.Add(4 * time.Second)
	cache.PruneAt(now.UnixNano())
	items = cache.Get()
	require.Equal(t, 1, len(items))
	require.Equal(t, 4, cache.itemsSize)

	// should evict all
	now = now.Add(time.Second)
	cache.PruneAt(now.UnixNano())
	items = cache.Get()
	require.Equal(t, 0, len(items))
	require.Equal(t, 0, cache.itemsSize)

	// add something that fits
	cache.Add(&four, len(four))
	items = cache.Get()
	require.Equal(t, 1, len(items))
	require.Equal(t, four, *items[0])
	require.Equal(t, 4, cache.itemsSize)

	// add something which will exceed size threshold by itself,
	// should clean up all items in cache as items are FIFO and the "four" added above
	// gets removed first and the long item added also should get removed as it does not fit.
	reallyLong := "really-long"
	cache.Add(&reallyLong, len(reallyLong))
	items = cache.Get()
	require.Equal(t, 0, len(items))
	require.Equal(t, 0, cache.itemsSize)
}