TRKJ/EdgeNode
2
0
Fork 0
mirror of https://github.com/TeaOSLab/EdgeNode.git synced 2026-02-26 00:25:41 +08:00
Code Issues Packages Projects Releases Wiki Activity

优化代码

Browse Source
This commit is contained in:
GoEdgeLab
2024-05-03 15:31:40 +08:00
parent 6db792a7c9
commit d839d81ea7
14 changed files with 117 additions and 111 deletions
Download Patch File Download Diff File Expand all files Collapse all files

52
internal/utils/byte_pool.go
View File

@@ -1,52 +0,0 @@
package utils
import (
"sync"
)
var BytePool1k = NewBytePool(1 << 10)
var BytePool4k = NewBytePool(4 << 10)
var BytePool16k = NewBytePool(16 << 10)
var BytePool32k = NewBytePool(32 << 10)
type BytesBuf struct {
Bytes []byte
}
// BytePool pool for get byte slice
type BytePool struct {
length int
rawPool *sync.Pool
}
// NewBytePool 创建新对象
func NewBytePool(length int) *BytePool {
if length < 0 {
length = 1024
}
return &BytePool{
length: length,
rawPool: &sync.Pool{
New: func() any {
return &BytesBuf{
Bytes: make([]byte, length),
}
},
},
}
}
// Get 获取一个新的byte slice
func (this *BytePool) Get() *BytesBuf {
return this.rawPool.Get().(*BytesBuf)
}
// Put 放回一个使用过的byte slice
func (this *BytePool) Put(ptr *BytesBuf) {
this.rawPool.Put(ptr)
}
// Length 单个字节slice长度
func (this *BytePool) Length() int {
return this.length
}

52
internal/utils/bytepool/byte_pool.go Normal file
View File

@@ -0,0 +1,52 @@
package bytepool
import (
"sync"
)
var Pool1k = NewPool(1 << 10)
var Pool4k = NewPool(4 << 10)
var Pool16k = NewPool(16 << 10)
var Pool32k = NewPool(32 << 10)
type Buf struct {
Bytes []byte
}
// Pool for get byte slice
type Pool struct {
length int
rawPool *sync.Pool
}
// NewPool 创建新对象
func NewPool(length int) *Pool {
if length < 0 {
length = 1024
}
return &Pool{
length: length,
rawPool: &sync.Pool{
New: func() any {
return &Buf{
Bytes: make([]byte, length),
}
},
},
}
}
// Get 获取一个新的byte slice
func (this *Pool) Get() *Buf {
return this.rawPool.Get().(*Buf)
}
// Put 放回一个使用过的byte slice
func (this *Pool) Put(ptr *Buf) {
this.rawPool.Put(ptr)
}
// Length 单个字节slice长度
func (this *Pool) Length() int {
return this.length
}

34
internal/utils/byte_pool_test.go → internal/utils/bytepool/byte_pool_test.go
View File

@@ -1,8 +1,8 @@
package utils_test
package bytepool_test
import (
"bytes"
"github.com/TeaOSLab/EdgeNode/internal/utils"
"github.com/TeaOSLab/EdgeNode/internal/utils/bytepool"
"runtime"
"sync"
"testing"
@@ -12,9 +12,9 @@ func TestBytePool_Memory(t *testing.T) {
var stat1 = &runtime.MemStats{}
runtime.ReadMemStats(stat1)
var pool = utils.NewBytePool(32 * 1024)
var pool = bytepool.NewPool(32 * 1024)
for i := 0; i < 20480; i++ {
pool.Put(&utils.BytesBuf{
pool.Put(&bytepool.Buf{
Bytes: make([]byte, 32*1024),
})
}
@@ -33,7 +33,7 @@ func TestBytePool_Memory(t *testing.T) {
func BenchmarkBytePool_Get(b *testing.B) {
runtime.GOMAXPROCS(1)
var pool = utils.NewBytePool(1)
var pool = bytepool.NewPool(1)
b.ResetTimer()
for i := 0; i < b.N; i++ {
@@ -46,7 +46,7 @@ func BenchmarkBytePool_Get(b *testing.B) {
func BenchmarkBytePool_Get_Parallel(b *testing.B) {
runtime.GOMAXPROCS(1)
var pool = utils.NewBytePool(1024)
var pool = bytepool.NewPool(1024)
b.ResetTimer()
b.RunParallel(func(pb *testing.PB) {
@@ -82,7 +82,7 @@ func BenchmarkBytePool_Get_Sync2(b *testing.B) {
var pool = &sync.Pool{
New: func() any {
return &utils.BytesBuf{
return &bytepool.Buf{
Bytes: make([]byte, 1024),
}
},
@@ -127,7 +127,7 @@ func BenchmarkBytePool_Copy_Wrapper_4(b *testing.B) {
var pool = &sync.Pool{
New: func() any {
return &utils.BytesBuf{
return &bytepool.Buf{
Bytes: make([]byte, size),
}
},
@@ -136,7 +136,7 @@ func BenchmarkBytePool_Copy_Wrapper_4(b *testing.B) {
b.RunParallel(func(pb *testing.PB) {
for pb.Next() {
var buf = pool.Get().(*utils.BytesBuf)
var buf = pool.Get().(*bytepool.Buf)
copy(buf.Bytes, data)
pool.Put(buf)
}
@@ -171,7 +171,7 @@ func BenchmarkBytePool_Copy_Wrapper_16(b *testing.B) {
var pool = &sync.Pool{
New: func() any {
return &utils.BytesBuf{
return &bytepool.Buf{
Bytes: make([]byte, size),
}
},
@@ -180,7 +180,7 @@ func BenchmarkBytePool_Copy_Wrapper_16(b *testing.B) {
b.RunParallel(func(pb *testing.PB) {
for pb.Next() {
var buf = pool.Get().(*utils.BytesBuf)
var buf = pool.Get().(*bytepool.Buf)
copy(buf.Bytes, data)
pool.Put(buf)
}
@@ -194,7 +194,7 @@ func BenchmarkBytePool_Copy_Wrapper_Buf_16(b *testing.B) {
var pool = &sync.Pool{
New: func() any {
return &utils.BytesBuf{
return &bytepool.Buf{
Bytes: make([]byte, size),
}
},
@@ -203,7 +203,7 @@ func BenchmarkBytePool_Copy_Wrapper_Buf_16(b *testing.B) {
b.RunParallel(func(pb *testing.PB) {
for pb.Next() {
var bytesPtr = pool.Get().(*utils.BytesBuf)
var bytesPtr = pool.Get().(*bytepool.Buf)
var buf = bytesPtr.Bytes
copy(buf, data)
pool.Put(bytesPtr)
@@ -220,9 +220,9 @@ func BenchmarkBytePool_Copy_Wrapper_BytePool_16(b *testing.B) {
b.RunParallel(func(pb *testing.PB) {
for pb.Next() {
var bytesPtr = utils.BytePool16k.Get()
var bytesPtr = bytepool.Pool16k.Get()
copy(bytesPtr.Bytes, data)
utils.BytePool16k.Put(bytesPtr)
bytepool.Pool16k.Put(bytesPtr)
}
})
}
@@ -255,7 +255,7 @@ func BenchmarkBytePool_Copy_Wrapper_32(b *testing.B) {
var pool = &sync.Pool{
New: func() any {
return &utils.BytesBuf{
return &bytepool.Buf{
Bytes: make([]byte, size),
}
},
@@ -264,7 +264,7 @@ func BenchmarkBytePool_Copy_Wrapper_32(b *testing.B) {
b.RunParallel(func(pb *testing.PB) {
for pb.Next() {
var buf = pool.Get().(*utils.BytesBuf)
var buf = pool.Get().(*bytepool.Buf)
copy(buf.Bytes, data)
pool.Put(buf)
}