reshackpool.go

package hackpool

import "sync"

type ResHackPool struct {
	numGo    int
	messages chan arg
	function func(...interface{}) interface{}
}

type Result struct {
	ID    int
	Tag   string
	Value interface{}
}

type arg struct {
	ID   int
	Tag  string
	Data []interface{}
}

func NewRes(numGoroutine int, function func(...interface{}) interface{}) *ResHackPool {
	return &ResHackPool{
		numGo:    numGoroutine,
		messages: make(chan arg),
		function: function,
	}
}

func (c *ResHackPool) Push(id int, tag string, data ...interface{}) {
	c.messages <- arg{ID: id, Tag: tag, Data: data}
}

func (c *ResHackPool) CloseQueue() {
	close(c.messages)
}

func (c *ResHackPool) Run() <-chan Result {
	var wg sync.WaitGroup
	rc := make(chan Result)

	wg.Add(c.numGo)
	for i := 0; i < c.numGo; i++ {
		go func() {
			for arg := range c.messages {
				r := c.function(arg.Data...)
				rc <- Result{ID: arg.ID, Tag: arg.Tag, Value: r}
			}
			wg.Done()
		}()
	}

	go func() {
		wg.Wait()
		close(rc)
	}()
	return rc
}