bounds check elimation in core loop

xtaci · Oct 8, 2018 · f48db19 · f48db19
1 parent 90edee5
commit f48db19
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 3 deletions.
diff --git a/kcp.go b/kcp.go
@@ -738,8 +738,10 @@ func (kcp *KCP) flush(ackOnly bool) uint32 {
 	current := currentMs()
 	var change, lost, lostSegs, fastRetransSegs, earlyRetransSegs uint64
 	minrto := int32(kcp.interval)
-	for k := range kcp.snd_buf {
-		segment := &kcp.snd_buf[k]
+
+	ref := kcp.snd_buf[:len(kcp.snd_buf)] // for bounds check elimination
+	for k := range ref {
+		segment := &ref[k]
 		needsend := false
 		if segment.xmit == 0 { // initial transmit
 			needsend = true

diff --git a/kcp_test.go b/kcp_test.go
@@ -287,7 +287,7 @@ func TestNetwork(t *testing.T) {
 
 func BenchmarkFlush(b *testing.B) {
 	kcp := NewKCP(1, func(buf []byte, size int) {})
-	kcp.snd_buf = make([]segment, 32)
+	kcp.snd_buf = make([]segment, 1024)
 	for k := range kcp.snd_buf {
 		kcp.snd_buf[k].xmit = 1
 		kcp.snd_buf[k].resendts = currentMs() + 10000