1
0
mirror of https://github.com/Mrs4s/MiraiGo.git synced 2025-05-04 11:07:40 +08:00

drop(pool): unnecessay nil judge

This commit is contained in:
fumiama 2021-11-23 15:35:52 +08:00
parent ec053573f1
commit 26579583cc
2 changed files with 8 additions and 11 deletions

View File

@ -16,11 +16,10 @@ var bufferPool = sync.Pool{
// SelectWriter 从池中取出一个 Writer
func SelectWriter() *Writer {
w := bufferPool.Get().(*Writer)
if w == nil {
return new(Writer)
}
return w
// 因为 bufferPool 定义有 New 函数
// 所以 bufferPool.Get() 永不为 nil
// 不用判空
return bufferPool.Get().(*Writer)
}
// PutWriter 将 Writer 放回池中
@ -103,9 +102,9 @@ var b256kPool = sync.Pool{
// Get256KBytes 获取一个128k大小 []byte
func Get256KBytes() *[]byte {
buf := b256kPool.Get().(*[]byte)
if buf == nil {
return make128kSlicePointer()
}
// 因为 b256kPool 定义有 New 函数
// 所以 b256kPool.Get() 永不为 nil
// 不用判空
if cap(*buf) < size256k {
return make128kSlicePointer()
}

View File

@ -69,9 +69,7 @@ func (w *Writer) WriteBool(b bool) {
}
func (w *Writer) EncryptAndWrite(key []byte, data []byte) {
tea := NewTeaCipher(key)
ed := tea.Encrypt(data)
w.Write(ed)
w.Write(NewTeaCipher(key).Encrypt(data))
}
func (w *Writer) WriteIntLvPacket(offset int, f func(writer *Writer)) {