pikiwidb/codis/pkg/proxy/request_test.go at c0ca29129c9281685d1beaa577b062235af0d9fb · OpenAtomFoundation/pikiwidb · GitHub

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
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
// Copyright 2016 CodisLabs. All Rights Reserved.
// Licensed under the MIT (MIT-LICENSE.txt) license.

package proxy

import (
	"runtime"
	"sync"
	"testing"
	"time"

	"pika/codis/v2/pkg/proxy/redis"
	"pika/codis/v2/pkg/utils/assert"
)

func TestRequestChan1(t *testing.T) {
	var ch = NewRequestChanBuffer(0)
	for i := 0; i < 8192; i++ {
		n := ch.PushBack(&Request{ReceiveTime: int64(i)})
		assert.Must(n == i+1)
	}
	for i := 0; i < 8192; i++ {
		r, ok := ch.PopFront()
		assert.Must(ok && r.ReceiveTime == int64(i))
	}
	assert.Must(ch.Buffered() == 0)

	ch.Close()

	_, ok := ch.PopFront()
	assert.Must(!ok)
}

func TestRequestChan2(t *testing.T) {
	var ch = NewRequestChanBuffer(512)
	for i := 0; i < 8192; i++ {
		n := ch.PushBack(&Request{ReceiveTime: int64(i)})
		assert.Must(n == i+1)
	}
	ch.Close()

	assert.Must(ch.Buffered() == 8192)

	for i := 0; i < 8192; i++ {
		r, ok := ch.PopFront()
		assert.Must(ok && r.ReceiveTime == int64(i))
	}
	assert.Must(ch.Buffered() == 0)

	_, ok := ch.PopFront()
	assert.Must(!ok)
}

func TestRequestChan3(t *testing.T) {
	var wg sync.WaitGroup
	var ch = NewRequestChanBuffer(512)

	const n = 1000 * 1000 * 4

	wg.Add(1)
	go func() {
		defer wg.Done()
		for i := 0; i < n; i++ {
			ch.PushBack(&Request{ReceiveTime: int64(i)})
			if i%1024 == 0 {
				runtime.Gosched()
			}
		}
	}()

	wg.Add(1)
	go func() {
		defer wg.Done()
		for i := 0; i < n; i++ {
			r, ok := ch.PopFront()
			assert.Must(ok && r.ReceiveTime == int64(i))
			if i%4096 == 0 {
				runtime.Gosched()
			}
		}
	}()

	wg.Wait()

	go func() {
		defer ch.Close()
		time.Sleep(time.Millisecond * 10)
	}()

	_, ok := ch.PopFront()
	assert.Must(!ok)
}

func BenchmarkRequestGoChannel(b *testing.B) {
	var request = &Request{
		Multi: make([]*redis.Resp, 1024*1024),
	}
	var ch = make(chan *Request, 1024)
	go func() {
		for i := 0; i < b.N; i++ {
			ch <- request
		}
	}()

	for i := 0; i < b.N; i++ {
		<-ch
	}
}

func benchmarkRequestChanN(b *testing.B, n int) {
	var request = &Request{
		Multi: make([]*redis.Resp, 1024*1024),
	}
	var ch = NewRequestChanBuffer(n)
	go func() {
		defer ch.Close()
		for i := 0; i < b.N; i++ {
			ch.PushBack(request)
			if i%1024 == 0 {
				runtime.Gosched()
			}
		}
	}()

	ch.PopFrontAllVoid(func(r *Request) {})
}

func BenchmarkRequestChan128(b *testing.B)  { benchmarkRequestChanN(b, 128) }
func BenchmarkRequestChan256(b *testing.B)  { benchmarkRequestChanN(b, 256) }
func BenchmarkRequestChan512(b *testing.B)  { benchmarkRequestChanN(b, 512) }
func BenchmarkRequestChan1024(b *testing.B) { benchmarkRequestChanN(b, 1024) }
func BenchmarkRequestChan2048(b *testing.B) { benchmarkRequestChanN(b, 2048) }