82 lines
1.8 KiB
Go
82 lines
1.8 KiB
Go
/*
|
||
* Copyright (C) THL A29 Limited, a Tencent company. All rights reserved.
|
||
*
|
||
* SPDX-License-Identifier: Apache-2.0
|
||
*
|
||
*/
|
||
|
||
package lockfree
|
||
|
||
import (
|
||
"errors"
|
||
"reflect"
|
||
"runtime"
|
||
"unsafe"
|
||
)
|
||
|
||
const (
|
||
activeSpin = 4
|
||
passiveSpin = 2
|
||
READY = 0 // 模块的状态之就绪态
|
||
RUNNING = 1 // 模块的状态之运行态
|
||
StartErrorFormat = "start model [%s] error"
|
||
CloseErrorFormat = "close model [%s] error"
|
||
)
|
||
|
||
var (
|
||
ncpu = runtime.NumCPU()
|
||
spin = 0
|
||
ClosedError = errors.New("the queue has been closed")
|
||
)
|
||
|
||
func init() {
|
||
if ncpu > 1 {
|
||
spin = activeSpin
|
||
}
|
||
}
|
||
|
||
//go:linkname procyield runtime.procyield
|
||
func procyield(cycles uint32)
|
||
|
||
// //go:linkname osyield runtime.osyield
|
||
// func osyield()
|
||
func osyield() {
|
||
runtime.Gosched() // 主动放弃CPU，让渡给其他goroutine，兼容所有Go版本
|
||
}
|
||
|
||
// byteArrayPointerWithUint8 创建uint8切片，返回其对应实际内容（Data）的指针
|
||
func byteArrayPointerWithUint8(capacity int) unsafe.Pointer {
|
||
bytes := make([]uint8, capacity)
|
||
rs := (*reflect.SliceHeader)(unsafe.Pointer(&bytes))
|
||
return unsafe.Pointer(rs.Data)
|
||
}
|
||
|
||
// byteArrayPointer 创建uint32切片，返回其对应实际内容（Data）的指针
|
||
func byteArrayPointerWithUint32(capacity int) unsafe.Pointer {
|
||
bytes := make([]uint32, capacity)
|
||
rs := (*reflect.SliceHeader)(unsafe.Pointer(&bytes))
|
||
return unsafe.Pointer(rs.Data)
|
||
}
|
||
|
||
// byteArrayPointer 创建int64切片，返回其对应实际内容（Data）的指针
|
||
func byteArrayPointerWithInt64(capacity int) unsafe.Pointer {
|
||
bytes := make([]int64, capacity)
|
||
rs := (*reflect.SliceHeader)(unsafe.Pointer(&bytes))
|
||
return unsafe.Pointer(rs.Data)
|
||
}
|
||
|
||
// minSuitableCap 最小的合适的数量
|
||
func minSuitableCap(v int) int {
|
||
if v <= 0 {
|
||
return 2
|
||
}
|
||
v--
|
||
v |= v >> 1
|
||
v |= v >> 2
|
||
v |= v >> 4
|
||
v |= v >> 8
|
||
v |= v >> 16
|
||
v++
|
||
return v
|
||
}
|