gostore/internal/store_disk/store_disk.go

// package store_disk -- хранилище на диске
package store_disk

import (
	"fmt"
	"log"
	"os"
	"strings"

	"github.com/syndtr/goleveldb/leveldb"
	"github.com/syndtr/goleveldb/leveldb/util"

	"git.p78su.freemyip.com/svi/gostore/pkg/types"
)

// StoreDisk -- хранилище на диске
type StoreDisk struct {
	serv types.IService
	db   *leveldb.DB
}

// NewStoreDisk -- возвращает новое хранилище на диске
func NewStoreDisk(serv types.IService) (types.IStoreDisk, error) {
	if serv == nil {
		return nil, fmt.Errorf("NewStoreDisk(): IService==nil")
	}
	_ = os.MkdirAll("./store/db", 0750)
	db, err := leveldb.OpenFile("./store/db", nil)
	if err != nil {
		if !strings.Contains(err.Error(), "leveldb: manifest corrupted") {
			return nil, fmt.Errorf("NewStoreDisk(): in create IStoreDisk, err=\n\t%w", err)
		}
		db, err = leveldb.RecoverFile("./stor/db", nil)
		if err != nil {
			return nil, fmt.Errorf("NewStoreDisk(): in recovery DB, err=\n\t%w", err)
		}
	}
	sf := &StoreDisk{
		serv: serv,
		db:   db,
	}
	go sf.close()
	return sf, nil
}

// Put -- размещает в хранилище ключ и значение
func (sf *StoreDisk) Put(key string, val []byte) error {
	err := sf.db.Put([]byte(key), val, nil)
	if err != nil {
		return fmt.Errorf("StoreDisk.Put(): key=%q\terr=\n\t%w", key, err)
	}
	return nil
}

// Get -- возвращает значение ключа
func (sf *StoreDisk) Get(key string) ([]byte, error) {
	val, err := sf.db.Get([]byte(key), nil)
	if err != nil {
		return nil, fmt.Errorf("StoreDisk.Get(): key=%q\terr=\n\t%w", key, err)
	}
	return val, nil
}

// Find -- ищет ключи по префиксу
func (sf *StoreDisk) Find(prefixKey string) ([]string, error) {
	lstKey := []string{}
	iter := sf.db.NewIterator(util.BytesPrefix([]byte(prefixKey)), nil)
	for iter.Next() {
		key := iter.Key()
		lstKey = append(lstKey, string(key))
	}
	iter.Release()
	err := iter.Error()
	if err != nil {
		return nil, fmt.Errorf("StoreDisk.Find(): preefixKey=%q\terr=\n\t%w", prefixKey, err)
	}
	return lstKey, nil
}

// Del -- удаление ключа из базы
func (sf *StoreDisk) Del(key string) {
	err := sf.db.Delete([]byte(key), nil)
	if err != nil {
		log.Printf("StoreDisk.Del(): key=%q\terr=\n\t%v\n", key, err)
	}
}

// Ожидание закрытия приложения
func (sf *StoreDisk) close() {
	<-sf.serv.Ctx().Done()
	sf.db.Close()
}