scheduler/worker_pool.go


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
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
}