Browse Source

Merge pull request #6350 from nekto0n/fix_message_limit

rafthttp: fix misprint in readBytesLimit value
Gyu-Ho Lee 9 years ago
parent
commit
1ebeef5cbf
2 changed files with 7 additions and 1 deletions
  1. 1 1
      rafthttp/msg_codec.go
  2. 6 0
      rafthttp/msg_codec_test.go

+ 1 - 1
rafthttp/msg_codec.go

@@ -43,7 +43,7 @@ type messageDecoder struct {
 }
 
 var (
-	readBytesLimit     uint64 = 512 * 1024 // 512 MB
+	readBytesLimit     uint64 = 512 * 1024 * 1024 // 512 MB
 	ErrExceedSizeLimit        = errors.New("rafthttp: error limit exceeded")
 )
 

+ 6 - 0
rafthttp/msg_codec_test.go

@@ -23,6 +23,12 @@ import (
 )
 
 func TestMessage(t *testing.T) {
+	// Lower readBytesLimit to make test pass in restricted resources environment
+	originalLimit := readBytesLimit
+	readBytesLimit = 1000
+	defer func() {
+		readBytesLimit = originalLimit
+	}()
 	tests := []struct {
 		msg       raftpb.Message
 		encodeErr error