diff options
Diffstat (limited to 'scheduler/worker_pool.go')
-rw-r--r-- | scheduler/worker_pool.go | 34 |
1 files changed, 34 insertions, 0 deletions
diff --git a/scheduler/worker_pool.go b/scheduler/worker_pool.go new file mode 100644 index 00000000..b753f893 --- /dev/null +++ b/scheduler/worker_pool.go @@ -0,0 +1,34 @@ +// Copyright 2017 Frédéric Guillot. All rights reserved. +// Use of this source code is governed by the Apache 2.0 +// license that can be found in the LICENSE file. + +package scheduler + +import ( + "github.com/miniflux/miniflux2/model" + "github.com/miniflux/miniflux2/reader/feed" +) + +// WorkerPool handle a pool of workers. +type WorkerPool struct { + queue chan model.Job +} + +// Push send a job on the queue. +func (w *WorkerPool) Push(job model.Job) { + w.queue <- job +} + +// NewWorkerPool creates a pool of background workers. +func NewWorkerPool(feedHandler *feed.Handler, nbWorkers int) *WorkerPool { + workerPool := &WorkerPool{ + queue: make(chan model.Job), + } + + for i := 0; i < nbWorkers; i++ { + worker := &Worker{id: i, feedHandler: feedHandler} + go worker.Run(workerPool.queue) + } + + return workerPool +} |