浏览代码

optimize for 0 buffer size

Darien Raymond 9 年之前
父节点
当前提交
9ecf89657c
共有 1 个文件被更改,包括 7 次插入3 次删除
  1. 7 3
      common/alloc/buffer_pool.go

+ 7 - 3
common/alloc/buffer_pool.go

@@ -87,7 +87,7 @@ const (
 )
 
 var (
-	mediumPool *BufferPool
+	mediumPool Pool
 	smallPool  = NewSyncPool(2048)
 )
 
@@ -100,6 +100,10 @@ func init() {
 			size = uint32(customSize)
 		}
 	}
-	totalByteSize := size * 1024 * 1024
-	mediumPool = NewBufferPool(mediumBufferByteSize, totalByteSize/mediumBufferByteSize)
+	if size > 0 {
+		totalByteSize := size * 1024 * 1024
+		mediumPool = NewBufferPool(mediumBufferByteSize, totalByteSize/mediumBufferByteSize)
+	} else {
+		mediumPool = NewSyncPool(mediumBufferByteSize)
+	}
 }