obitools4/pkg/obiformats/dispatcher.go

package obiformats

import (
	"fmt"
	log "github.com/sirupsen/logrus"
	"sync"

	"git.metabarcoding.org/lecasofts/go/obitools/pkg/obiiter"
)

type SequenceBatchWriterToFile func(iterator obiiter.IBioSequenceBatch,
	filename string,
	options ...WithOption) (obiiter.IBioSequenceBatch, error)

func WriterDispatcher(prototypename string,
	dispatcher obiiter.IDistribute,
	formater SequenceBatchWriterToFile,
	options ...WithOption) {

	jobDone := sync.WaitGroup{}
	jobDone.Add(1)

	go func() {
		for newflux := range dispatcher.News() {
			jobDone.Add(1)
			go func(newflux int) {
				data, err := dispatcher.Outputs(newflux)

				if err != nil {
					log.Fatalf("Cannot retreive the new chanel : %v", err)
				}

				out, err := formater(data,
					fmt.Sprintf(prototypename, dispatcher.Classifier().Value(newflux)),
					options...)

				if err != nil {
					log.Fatalf("cannot open the output file for key %s",
						dispatcher.Classifier().Value(newflux))
				}

				out.Recycle()
				jobDone.Done()
			}(newflux)
		}
		jobDone.Done()
	}()

	jobDone.Wait()
}