mirror of https://github.com/v2ray/v2ray-core
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
76 lines
1.4 KiB
76 lines
1.4 KiB
package io
|
|
|
|
import (
|
|
"errors"
|
|
)
|
|
|
|
const (
|
|
SizeSmall = 16
|
|
SizeMedium = 128
|
|
SizeLarge = 512
|
|
)
|
|
|
|
var (
|
|
ErrorNoChannel = errors.New("No suitable channels found.")
|
|
)
|
|
|
|
type BufferSet struct {
|
|
small chan []byte
|
|
medium chan []byte
|
|
large chan []byte
|
|
}
|
|
|
|
func NewBufferSet() *BufferSet {
|
|
bSet := new(BufferSet)
|
|
bSet.small = make(chan []byte, 128)
|
|
bSet.medium = make(chan []byte, 128)
|
|
bSet.large = make(chan []byte, 128)
|
|
return bSet
|
|
}
|
|
|
|
func (bSet *BufferSet) detectBucket(size int, strict bool) (chan []byte, error) {
|
|
if strict {
|
|
if size == SizeSmall {
|
|
return bSet.small, nil
|
|
} else if size == SizeMedium {
|
|
return bSet.medium, nil
|
|
} else if size == SizeLarge {
|
|
return bSet.large, nil
|
|
}
|
|
} else {
|
|
if size <= SizeSmall {
|
|
return bSet.small, nil
|
|
} else if size <= SizeMedium {
|
|
return bSet.medium, nil
|
|
} else if size <= SizeLarge {
|
|
return bSet.large, nil
|
|
}
|
|
}
|
|
return nil, ErrorNoChannel
|
|
}
|
|
|
|
func (bSet *BufferSet) FetchBuffer(minSize int) []byte {
|
|
var buffer []byte
|
|
byteChan, err := bSet.detectBucket(minSize, false)
|
|
if err != nil {
|
|
return make([]byte, minSize)
|
|
}
|
|
select {
|
|
case buffer = <-byteChan:
|
|
default:
|
|
buffer = make([]byte, minSize)
|
|
}
|
|
return buffer
|
|
}
|
|
|
|
func (bSet *BufferSet) ReturnBuffer(buffer []byte) {
|
|
byteChan, err := bSet.detectBucket(len(buffer), true)
|
|
if err != nil {
|
|
return
|
|
}
|
|
select {
|
|
case byteChan <- buffer:
|
|
default:
|
|
}
|
|
}
|