dave/smpp-tester
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Make bufpool use uint

Browse Source 
I mean we can't have a pool of negative size?
This commit is contained in:
PhatPhuckDave
2024-07-22 23:13:34 +02:00
parent 3d0d8ce676
commit 4ea320c784
1 changed files with 4 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
pdu/bufpool.go
View File

@@ -5,17 +5,17 @@ import (
)
type BufferPoolManager struct {
pools map[int]*sync.Pool
pools map[uint]*sync.Pool
mu sync.Mutex
}
func NewBufferPoolManager() *BufferPoolManager {
return &BufferPoolManager{
pools: make(map[int]*sync.Pool),
pools: make(map[uint]*sync.Pool),
}
}
func (bpm *BufferPoolManager) GetBuffer(size int) *([]uint8) {
func (bpm *BufferPoolManager) GetBuffer(size uint) *([]uint8) {
bpm.mu.Lock()
pool, exists := bpm.pools[size]
if !exists {
@@ -32,7 +32,7 @@ func (bpm *BufferPoolManager) GetBuffer(size int) *([]uint8) {
}
func (bpm *BufferPoolManager) PutBuffer(buf *([]uint8)) {
size := len(*buf)
size := uint(len(*buf))
bpm.mu.Lock()
pool, exists := bpm.pools[size]
if !exists {
@@ -41,7 +41,6 @@ func (bpm *BufferPoolManager) PutBuffer(buf *([]uint8)) {
}
bpm.mu.Unlock()
// Reset buffer (optional)
for i := range *buf {
(*buf)[i] = 0
}