Add option for limit a number of parallel reads

defaults.go

@@ -0,0 +1,5 @@
+package zkv
+
+var defaultOptions = Options{
+	MaxParallelReads: 64,
+}

options.go

@@ -0,0 +1,13 @@
+package zkv
+
+type Options struct {
+	MaxParallelReads uint
+}
+
+func (o *Options) Validate() error {
+	if o.MaxParallelReads == 0 {
+		o.MaxParallelReads = defaultOptions.MaxParallelReads
+	}
+
+	return nil
+}

zkv.go

@@ -18,6 +18,10 @@ type Database struct {
 	filePath   string
 	offset     int64
 
+	options Options
+
+	readOrderChan chan struct{}
+
 	mu sync.RWMutex
 }
 
@@ -63,6 +67,9 @@ func (db *Database) Get(key, value interface{}) error {
 	db.mu.RLock()
 	defer db.mu.RUnlock()
 
+	db.readOrderChan <- struct{}{}
+	defer func() { <-db.readOrderChan }()
+
 	hashToFind, err := hashInterface(key)
 	if err != nil {
 		return err
@@ -102,14 +109,20 @@ func (db *Database) Get(key, value interface{}) error {
 	return decode(record.ValueBytes, value)
 }
 
-func Open(filePath string) (*Database, error) {
+func OpenWithOptions(filePath string, options Options) (*Database, error) {
 	f, err := os.OpenFile(filePath, os.O_APPEND|os.O_CREATE|os.O_WRONLY, 0644)
 	if err != nil {
+		f.Close()
 		return nil, fmt.Errorf("ошибка при открытии файла для записи: %v", err)
 	}
 
+	if options.Validate() != nil {
+		return nil, err
+	}
+
 	compressor, err := zstd.NewWriter(f)
 	if err != nil {
+		f.Close()
 		return nil, fmt.Errorf("ошибка при инициализации компрессора: %v", err)
 	}
 
@@ -118,7 +131,9 @@ func Open(filePath string) (*Database, error) {
 		offset:        0,
 		file:          f,
 		compressor:    compressor,
-		filePath:   filePath}
+		filePath:      filePath,
+		options:       options,
+		readOrderChan: make(chan struct{}, int(options.MaxParallelReads))}
 
 	// restore file data
 	readF, err := os.Open(filePath)
@@ -130,6 +145,7 @@ func Open(filePath string) (*Database, error) {
 
 	decompressor, err := zstd.NewReader(readF)
 	if err != nil {
+		f.Close()
 		return nil, fmt.Errorf("ошибка при инициализации декомпрессора: %v", err)
 	}
 	defer decompressor.Close()
@@ -158,6 +174,10 @@ func Open(filePath string) (*Database, error) {
 	return database, nil
 }
 
+func Open(filePath string) (*Database, error) {
+	return OpenWithOptions(filePath, defaultOptions)
+}
+
 func (db *Database) Delete(key interface{}) error {
 	db.mu.Lock()
 	defer db.mu.Unlock()