mirror of
https://github.com/TeaOSLab/EdgeNode.git
synced 2025-11-03 15:00:26 +08:00
37 lines
633 B
Go
37 lines
633 B
Go
// Copyright 2024 GoEdge CDN goedge.cdn@gmail.com. All rights reserved. Official site: https://goedge.cloud .
|
|
|
|
package percpu
|
|
|
|
import (
|
|
"runtime"
|
|
)
|
|
|
|
type Chan[T any] struct {
|
|
c chan T
|
|
|
|
count int
|
|
cList []chan T
|
|
}
|
|
|
|
func NewChan[T any](size int) *Chan[T] {
|
|
var count = max(runtime.NumCPU(), runtime.GOMAXPROCS(0))
|
|
var cList []chan T
|
|
for i := 0; i < count; i++ {
|
|
cList = append(cList, make(chan T, size))
|
|
}
|
|
|
|
return &Chan[T]{
|
|
c: make(chan T, size),
|
|
count: count,
|
|
cList: cList,
|
|
}
|
|
}
|
|
|
|
func (this *Chan[T]) C() chan T {
|
|
var procId = GetProcId()
|
|
if procId < this.count {
|
|
return this.cList[procId]
|
|
}
|
|
return this.c
|
|
}
|