shanshanzhong
c582087c0f
Some checks failed
Build docker and publish / build (20.15.1) (push) Failing after 6m27s
feat: 添加版本和构建时间变量 fix: 修正短信队列类型注释错误 style: 清理未使用的代码和测试文件 docs: 更新安装文档中的下载链接 chore: 迁移数据库脚本添加日志和订阅配置
49 lines
1.1 KiB
Go
49 lines
1.1 KiB
Go
package syncx
|
|
|
|
import (
|
|
"errors"
|
|
|
|
"github.com/perfect-panel/server/pkg/lang"
|
|
)
|
|
|
|
// ErrLimitReturn indicates that the more than borrowed elements were returned.
|
|
var ErrLimitReturn = errors.New("discarding limited token, resource pool is full, someone returned multiple times")
|
|
|
|
// Limit controls the concurrent requests.
|
|
type Limit struct {
|
|
pool chan lang.PlaceholderType
|
|
}
|
|
|
|
// NewLimit creates a Limit that can borrow n elements from it concurrently.
|
|
func NewLimit(n int) Limit {
|
|
return Limit{
|
|
pool: make(chan lang.PlaceholderType, n),
|
|
}
|
|
}
|
|
|
|
// Borrow borrows an element from Limit in blocking mode.
|
|
func (l Limit) Borrow() {
|
|
l.pool <- lang.Placeholder
|
|
}
|
|
|
|
// Return returns the borrowed resource, returns error only if returned more than borrowed.
|
|
func (l Limit) Return() error {
|
|
select {
|
|
case <-l.pool:
|
|
return nil
|
|
default:
|
|
return ErrLimitReturn
|
|
}
|
|
}
|
|
|
|
// TryBorrow tries to borrow an element from Limit, in non-blocking mode.
|
|
// If success, true returned, false for otherwise.
|
|
func (l Limit) TryBorrow() bool {
|
|
select {
|
|
case l.pool <- lang.Placeholder:
|
|
return true
|
|
default:
|
|
return false
|
|
}
|
|
}
|