bitcask/options.go

package bitcask

import (
	"encoding/json"
	"errors"
	"io/ioutil"
	"path/filepath"
)

const (
	// DefaultMaxDatafileSize is the default maximum datafile size in bytes
	DefaultMaxDatafileSize = 1 << 20 // 1MB

	// DefaultMaxKeySize is the default maximum key size in bytes
	DefaultMaxKeySize = 64 // 64 bytes

	// DefaultMaxValueSize is the default value size in bytes
	DefaultMaxValueSize = 1 << 16 // 65KB
)

var (
	// ErrMaxConcurrencyLowerEqZero is the error returned for
	// maxConcurrency option not greater than zero
	ErrMaxConcurrencyLowerEqZero = errors.New("error: maxConcurrency must be greater than zero")
)

// Option is a function that takes a config struct and modifies it
type Option func(*config) error

type config struct {
	maxDatafileSize int
	maxKeySize      int
	maxValueSize    int
	maxConcurrency  *int
}

func (c *config) MarshalJSON() ([]byte, error) {
	return json.Marshal(struct {
		MaxDatafileSize int `json:"max_datafile_size"`
		MaxKeySize      int `json:"max_key_size"`
		MaxValueSize    int `json:"max_value_size"`
	}{
		MaxDatafileSize: c.maxDatafileSize,
		MaxKeySize:      c.maxKeySize,
		MaxValueSize:    c.maxValueSize,
	})
}

func getConfig(path string) (*config, error) {
	type Config struct {
		MaxDatafileSize int `json:"max_datafile_size"`
		MaxKeySize      int `json:"max_key_size"`
		MaxValueSize    int `json:"max_value_size"`
	}

	var cfg Config

	data, err := ioutil.ReadFile(filepath.Join(path, "config.json"))
	if err != nil {
		return nil, err
	}

	if err := json.Unmarshal(data, &cfg); err != nil {
		return nil, err
	}

	return &config{
		maxDatafileSize: cfg.MaxDatafileSize,
		maxKeySize:      cfg.MaxKeySize,
		maxValueSize:    cfg.MaxValueSize,
	}, nil
}

func newDefaultConfig() *config {
	return &config{
		maxDatafileSize: DefaultMaxDatafileSize,
		maxKeySize:      DefaultMaxKeySize,
		maxValueSize:    DefaultMaxValueSize,
	}
}

// WithMaxDatafileSize sets the maximum datafile size option
func WithMaxDatafileSize(size int) Option {
	return func(cfg *config) error {
		cfg.maxDatafileSize = size
		return nil
	}
}

// WithMaxKeySize sets the maximum key size option
func WithMaxKeySize(size int) Option {
	return func(cfg *config) error {
		cfg.maxKeySize = size
		return nil
	}
}

// WithMaxValueSize sets the maximum value size option
func WithMaxValueSize(size int) Option {
	return func(cfg *config) error {
		cfg.maxValueSize = size
		return nil
	}
}

// WithMemPool configures usage of a memory pool to avoid allocations
func WithMemPool(maxConcurrency int) Option {
	return func(cfg *config) error {
		if maxConcurrency <= 0 {
			return ErrMaxConcurrencyLowerEqZero
		}
		cfg.maxConcurrency = &maxConcurrency
		return nil
	}
}