Created
June 10, 2019 07:12
-
-
Save leafsummer/76b891afea2fcad2aa14656948a721f7 to your computer and use it in GitHub Desktop.
[leakybuf]
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
// Package leakybuf provides leaky buffer. | |
// It's based on the example in Effective Go. | |
package leakybuf | |
type LeakyBuf struct { | |
bufSize int // size of each buffer | |
freeList chan []byte | |
} | |
// NewLeakyBuf creates a leaky buffer which can hold at most n buffer, each | |
// with bufSize bytes. | |
func NewLeakyBuf(n, bufSize int) *LeakyBuf { | |
return &LeakyBuf{ | |
bufSize: bufSize, | |
freeList: make(chan []byte, n), | |
} | |
} | |
// Get returns a buffer from the leaky buffer or create a new buffer. | |
func (lb *LeakyBuf) Get() (b []byte) { | |
select { | |
case b = <-lb.freeList: | |
default: | |
b = make([]byte, lb.bufSize) | |
} | |
return | |
} | |
// Put add the buffer into the free buffer pool for reuse. Panic if the buffer | |
// size is not the same with the leaky buffer's. This is intended to expose | |
// error usage of leaky buffer. | |
func (lb *LeakyBuf) Put(b []byte) { | |
if len(b) != lb.bufSize { | |
panic("invalid buffer size that's put into leaky buffer") | |
} | |
select { | |
case lb.freeList <- b: | |
default: | |
} | |
return | |
} |
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment