mirror of
https://github.com/pyihe/go-pkg.git
synced 2025-09-27 04:16:05 +08:00
95 lines
1.5 KiB
Go
95 lines
1.5 KiB
Go
package bytes
|
||
|
||
import (
|
||
"unsafe"
|
||
|
||
"github.com/valyala/bytebufferpool"
|
||
)
|
||
|
||
type ByteBuffer = bytebufferpool.ByteBuffer
|
||
|
||
var (
|
||
Get = bytebufferpool.Get()
|
||
Put = func(b *ByteBuffer) {
|
||
if b != nil {
|
||
bytebufferpool.Put(b)
|
||
}
|
||
}
|
||
)
|
||
|
||
// String []byte转换为string
|
||
func String(b []byte) string {
|
||
return *(*string)(unsafe.Pointer(&b))
|
||
}
|
||
|
||
// Equal 判断两个字节切片每个元素是否相等
|
||
func Equal(a, b []byte) bool {
|
||
aLen, bLen := len(a), len(b)
|
||
if aLen != bLen {
|
||
return false
|
||
}
|
||
|
||
if (a == nil) != (b == nil) {
|
||
return false
|
||
}
|
||
|
||
b = b[:aLen]
|
||
for i, v := range a {
|
||
if b[i] != v {
|
||
return false
|
||
}
|
||
}
|
||
return true
|
||
}
|
||
|
||
// Contain 判断字节切片b是否包含ele元素
|
||
func Contain(b []byte, ele byte) bool {
|
||
for _, v := range b {
|
||
if v == ele {
|
||
return true
|
||
}
|
||
}
|
||
return false
|
||
}
|
||
|
||
// Reverse 翻转字节切片
|
||
func Reverse(b []byte) {
|
||
l := len(b)
|
||
for i := l/2 - 1; i >= 0; i-- {
|
||
opp := l - i - 1
|
||
b[i], b[opp] = b[opp], b[i]
|
||
}
|
||
}
|
||
|
||
// Remove 对于eles的每个元素，只删除一次
|
||
func Remove(b *[]byte, eles ...byte) {
|
||
if b == nil || len(*b) == 0 {
|
||
return
|
||
}
|
||
for _, e := range eles {
|
||
for i := 0; i < len(*b); {
|
||
if e == (*b)[i] {
|
||
copy((*b)[i:], (*b)[i+1:])
|
||
(*b)[len(*b)-1] = 0
|
||
*b = (*b)[:len(*b)-1]
|
||
break
|
||
} else {
|
||
i++
|
||
}
|
||
}
|
||
}
|
||
}
|
||
|
||
// RemoveAll 删除所有的ele
|
||
func RemoveAll(b *[]byte, eles ...byte) {
|
||
for _, e := range eles {
|
||
for i := 0; i < len(*b); {
|
||
if (*b)[i] == e {
|
||
*b = append((*b)[:i], (*b)[i+1:]...)
|
||
} else {
|
||
i++
|
||
}
|
||
}
|
||
}
|
||
}
|