big bug on subschunk

pkg/obichunk/options.go

@@ -0,0 +1,151 @@
+package obichunk
+
+type __options__ struct {
+	statsOn         []string
+	categories      []string
+	navalue         string
+	cacheOnDisk     bool
+	batchCount      int
+	bufferSize      int
+	batchSize       int
+	parallelWorkers int
+}
+
+type Options struct {
+	pointer *__options__
+}
+
+type WithOption func(Options)
+
+func MakeOptions(setters []WithOption) Options {
+	o := __options__{
+		statsOn:         make([]string, 0, 100),
+		categories:      make([]string, 0, 100),
+		navalue:         "NA",
+		cacheOnDisk:     false,
+		batchCount:      100,
+		bufferSize:      2,
+		batchSize:       5000,
+		parallelWorkers: 4,
+	}
+
+	opt := Options{&o}
+
+	for _, set := range setters {
+		set(opt)
+	}
+
+	return opt
+}
+
+func (opt Options) Categories() []string {
+	return opt.pointer.categories
+}
+
+func (opt Options) PopCategories() string {
+	if len(opt.pointer.categories) > 0 {
+		c := opt.pointer.categories[0]
+		opt.pointer.categories = opt.pointer.categories[1:]
+		return c
+	}
+	return ""
+}
+
+func (opt Options) StatsOn() []string {
+	return opt.pointer.statsOn
+}
+
+func (opt Options) NAValue() string {
+	return opt.pointer.navalue
+}
+
+func (opt Options) BatchCount() int {
+	return opt.pointer.batchCount
+}
+
+func (opt Options) BufferSize() int {
+	return opt.pointer.bufferSize
+}
+
+func (opt Options) BatchSize() int {
+	return opt.pointer.batchSize
+}
+
+func (opt Options) ParallelWorkers() int {
+	return opt.pointer.parallelWorkers
+}
+
+func (opt Options) SortOnDisk() bool {
+	return opt.pointer.cacheOnDisk
+}
+
+func OptionSortOnDisk() WithOption {
+	f := WithOption(func(opt Options) {
+		opt.pointer.cacheOnDisk = true
+	})
+
+	return f
+}
+
+func OptionSortOnMemory() WithOption {
+	f := WithOption(func(opt Options) {
+		opt.pointer.cacheOnDisk = false
+	})
+
+	return f
+}
+func OptionSubCategory(keys ...string) WithOption {
+	f := WithOption(func(opt Options) {
+		opt.pointer.categories = append(opt.pointer.categories, keys...)
+	})
+
+	return f
+}
+
+func OptionNAValue(na string) WithOption {
+	f := WithOption(func(opt Options) {
+		opt.pointer.navalue = na
+	})
+
+	return f
+}
+
+func OptionStatOn(keys ...string) WithOption {
+	f := WithOption(func(opt Options) {
+		opt.pointer.statsOn = append(opt.pointer.categories, keys...)
+	})
+
+	return f
+}
+
+func OptionBatchCount(number int) WithOption {
+	f := WithOption(func(opt Options) {
+		opt.pointer.batchCount = number
+	})
+
+	return f
+}
+
+func OptionsParallelWorkers(nworkers int) WithOption {
+	f := WithOption(func(opt Options) {
+		opt.pointer.parallelWorkers = nworkers
+	})
+
+	return f
+}
+
+func OptionsBatchSize(size int) WithOption {
+	f := WithOption(func(opt Options) {
+		opt.pointer.batchSize = size
+	})
+
+	return f
+}
+
+func OptionsBufferSize(size int) WithOption {
+	f := WithOption(func(opt Options) {
+		opt.pointer.bufferSize = size
+	})
+
+	return f
+}