12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970 |
- package sync
- import (
- "github.com/unknwon/com"
- )
- type UniqueQueue struct {
- table *StatusTable
- queue chan string
- }
- func NewUniqueQueue(queueLength int) *UniqueQueue {
- if queueLength <= 0 {
- queueLength = 100
- }
- return &UniqueQueue{
- table: NewStatusTable(),
- queue: make(chan string, queueLength),
- }
- }
- func (q *UniqueQueue) Queue() <-chan string {
- return q.queue
- }
- func (q *UniqueQueue) Exist(id interface{}) bool {
- return q.table.IsRunning(com.ToStr(id))
- }
- func (q *UniqueQueue) AddFunc(id interface{}, fn func()) {
- if q.Exist(id) {
- return
- }
- idStr := com.ToStr(id)
- q.table.Lock()
- q.table.pool[idStr] = true
- if fn != nil {
- fn()
- }
- q.table.Unlock()
- q.queue <- idStr
- }
- func (q *UniqueQueue) Add(id interface{}) {
- q.AddFunc(id, nil)
- }
- func (q *UniqueQueue) Remove(id interface{}) {
- q.table.Stop(com.ToStr(id))
- }
|