go-study/routine/options.go

package routine

type T any

type options struct {
	workers      int
	capacity     int
	taskFn       func(T)
	panicHandler func(any)
}

// Option function
type Option func(*options)

// loadOptions 设置默认值
func loadOptions(opt ...Option) options {
	opts := options{
		workers:  1,
		capacity: 1,
	}

	for _, o := range opt {
		o(&opts)
	}

	return opts
}

// WithWorkers 设置协程数
func WithWorkers(workers int) Option {
	if workers <= 0 {
		workers = 1
	}
	return func(o *options) {
		o.workers = workers
	}
}

// WithCapacity 设置任务队列容量
func WithCapacity(capacity int) Option {
	if capacity <= 0 {
		capacity = 1
	}
	return func(o *options) {
		o.capacity = capacity
	}
}

// WithTaskFn 设置任务函数
func WithTaskFn(taskFn func(T)) Option {
	return func(o *options) {
		o.taskFn = taskFn
	}
}

// WithPanicHandler 设置panic处理函数
func WithPanicHandler(panicHandler func(any)) Option {
	return func(o *options) {
		o.panicHandler = panicHandler
	}
}