f2fc168abe
- Replace prazdnikisegodnyaru with celebratoday as first is banned in Russia - Update .golangci.yml rule set - Megatons of fixes for linter - Update README.md
131 lines
4.3 KiB
Go
131 lines
4.3 KiB
Go
// Package pcachedb implements persistent cache using PebbleDB key-value library.
|
|
package pcachedb
|
|
|
|
import (
|
|
"aleesa-webapp-go/internal/config"
|
|
"aleesa-webapp-go/internal/log"
|
|
"crypto/sha256"
|
|
"errors"
|
|
"fmt"
|
|
"io/fs"
|
|
|
|
"github.com/cockroachdb/errors/oserror"
|
|
"github.com/cockroachdb/pebble"
|
|
)
|
|
|
|
// StoreKV сохраняет в указанной бд ключ и значение.
|
|
func StoreKV(db *pebble.DB, key string, value string) error {
|
|
var (
|
|
kArray = []byte(key)
|
|
vArray = []byte(value)
|
|
)
|
|
|
|
err := db.Set(kArray, vArray, pebble.Sync)
|
|
|
|
return err
|
|
}
|
|
|
|
// FetchV достаёт значение по ключу.
|
|
func FetchV(db *pebble.DB, key string) (string, error) {
|
|
var (
|
|
kArray = []byte(key)
|
|
vArray []byte
|
|
valueString = ""
|
|
)
|
|
|
|
vArray, closer, err := db.Get(kArray)
|
|
|
|
if err != nil {
|
|
return valueString, err
|
|
}
|
|
|
|
valueString = string(vArray)
|
|
err = closer.Close()
|
|
|
|
return valueString, err
|
|
}
|
|
|
|
// GetValue достанем данные из БД.
|
|
func GetValue(cfg *config.MyConfig, cacheID string, setting string) string {
|
|
var err error
|
|
|
|
cacheHash := sha256.Sum256([]byte(cacheID))
|
|
database := fmt.Sprintf("cache_db/%x", cacheHash)
|
|
|
|
// Если БД не открыта, откроем её.
|
|
if _, ok := cfg.PcacheDB[database]; !ok {
|
|
var options pebble.Options
|
|
// По дефолту ограничение ставится на мегабайты данных, а не на количество файлов, поэтому с дефолтными
|
|
// настройками порождается огромное количество файлов. Умолчальное ограничение на количество файлов - 500 штук,
|
|
// что нас не устраивает, поэтому немного снизим эту цифру до более приемлемых значений.
|
|
options.L0CompactionFileThreshold = 8
|
|
cfg.PcacheDB[database], err = pebble.Open(cfg.DataDir+"/"+database, &options)
|
|
|
|
if err != nil {
|
|
log.Errorf("Unable to open db, %s: %s\n", database, err)
|
|
|
|
return ""
|
|
}
|
|
}
|
|
|
|
value, err := FetchV(cfg.PcacheDB[database], setting)
|
|
|
|
// Если из базы ничего не вынулось, по каким-то причинам, то просто вернём пустую строку.
|
|
if err != nil {
|
|
switch {
|
|
case errors.Is(err, pebble.ErrNotFound):
|
|
log.Debugf("Unable to get value for %s: no record found in db %s", setting, database)
|
|
case errors.Is(err, fs.ErrNotExist):
|
|
log.Debugf("Unable to get value for %s: db dir %s does not exist", setting, database)
|
|
case errors.Is(err, oserror.ErrNotExist):
|
|
log.Debugf("Unable to get value for %s: db dir %s does not exist", setting, database)
|
|
default:
|
|
log.Errorf("Unable to get value for %s in db dir %s: %s", setting, database, err)
|
|
}
|
|
|
|
return ""
|
|
}
|
|
|
|
return value
|
|
}
|
|
|
|
// SaveKeyWithValue сохраним данные в БД.
|
|
func SaveKeyWithValue(cfg *config.MyConfig, cacheID string, setting string, value string) error {
|
|
var (
|
|
cacheHash = sha256.Sum256([]byte(cacheID))
|
|
database = fmt.Sprintf("cache_db/%x", cacheHash)
|
|
err error
|
|
)
|
|
|
|
// Если БД не открыта, откроем её.
|
|
if _, ok := cfg.PcacheDB[database]; !ok {
|
|
var options pebble.Options
|
|
// По дефолту ограничение ставится на мегабайты данных, а не на количество файлов, поэтому с дефолтными
|
|
// настройками порождается огромное количество файлов. Умолчальное ограничение на количество файлов - 500 штук,
|
|
// что нас не устраивает, поэтому немного снизим эту цифру до более приемлемых значений.
|
|
options.L0CompactionFileThreshold = 8
|
|
|
|
cfg.PcacheDB[database], err = pebble.Open(cfg.DataDir+"/"+database, &options)
|
|
|
|
if err != nil {
|
|
log.Errorf("Unable to open settings db, %s: %s\n", database, err)
|
|
|
|
return err
|
|
}
|
|
}
|
|
|
|
if err := StoreKV(cfg.PcacheDB[database], setting, value); err != nil {
|
|
log.Errorf("Unable to save setting %s in database %s: %s", setting, database, err)
|
|
|
|
return err
|
|
}
|
|
|
|
if _, err = cfg.PcacheDB[database].AsyncFlush(); err != nil {
|
|
log.Errorf("Unable to flush data to disk for database %s: %s", database, err)
|
|
}
|
|
|
|
return err
|
|
}
|
|
|
|
/* vim: set ft=go noet ai ts=4 sw=4 sts=4: */
|