support queue

2yz · May 21, 2017 · 672af32 · 672af32
1 parent da5a00f
commit 672af32
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 16 deletions.
diff --git a/.gitignore b/.gitignore
@@ -0,0 +1 @@
+.idea/
diff --git a/pool.go b/pool.go
@@ -75,23 +75,24 @@ func (wp *WorkerPool) clearWorkerQueue() {
 }
 
 func (wp *WorkerPool) Queue(unit WorkUnit) error {
-	return wp.queue(unit, NO_WAIT)
+	go wp.queue(unit)
+	return nil
 }
 
 func (wp *WorkerPool) QueueAndWait(unit WorkUnit) error {
-	return wp.queue(unit, WAIT)
+	return wp.queue(unit)
 }
 
-func (wp *WorkerPool) queue(unit WorkUnit, isWait bool) error {
-	w, err := wp.getWorker(isWait)
+func (wp *WorkerPool) queue(unit WorkUnit) error {
+	w, err := wp.getWorker()
 	if err != nil {
 		return err
 	}
 	w.ch <- unit
 	return nil
 }
 
-func (wp *WorkerPool) getWorker(isWait bool) (*Worker, error) {
+func (wp *WorkerPool) getWorker() (*Worker, error) {
 	createWorker := false
 
 	wp.cond.L.Lock()
@@ -111,17 +112,6 @@ func (wp *WorkerPool) getWorker(isWait bool) (*Worker, error) {
 		return w, nil
 	}
 
-	// NO_WAIT
-	if !isWait {
-		select {
-		case w := <-wp.workerQueue:
-			return w, nil
-		default:
-			return nil, errors.New("No Worker Available")
-		}
-	}
-
-	// WAIT
 	tick := time.Tick(10 * time.Millisecond)
 	for {
 		select {