72 lines
1.6 KiB
Go
72 lines
1.6 KiB
Go
package docstore
|
|
|
|
import (
|
|
"context"
|
|
"os"
|
|
"path/filepath"
|
|
"strings"
|
|
|
|
"github.com/tech/sendico/pkg/merrors"
|
|
"github.com/tech/sendico/pkg/mlogger"
|
|
"go.uber.org/zap"
|
|
)
|
|
|
|
type LocalStore struct {
|
|
logger mlogger.Logger
|
|
rootPath string
|
|
}
|
|
|
|
func NewLocalStore(logger mlogger.Logger, cfg LocalConfig) (*LocalStore, error) {
|
|
root := strings.TrimSpace(cfg.RootPath)
|
|
if root == "" {
|
|
return nil, merrors.InvalidArgument("docstore: local root_path is empty")
|
|
}
|
|
|
|
store := &LocalStore{
|
|
logger: logger.Named("docstore").Named("local"),
|
|
rootPath: root,
|
|
}
|
|
store.logger.Info("Document storage initialised", zap.String("root_path", root))
|
|
|
|
return store, nil
|
|
}
|
|
|
|
func (s *LocalStore) Save(ctx context.Context, key string, data []byte) error {
|
|
if err := ctx.Err(); err != nil {
|
|
return err
|
|
}
|
|
|
|
path := filepath.Join(s.rootPath, filepath.Clean(key))
|
|
if err := os.MkdirAll(filepath.Dir(path), 0o755); err != nil {
|
|
s.logger.Warn("Failed to create document directory", zap.Error(err), zap.String("path", path))
|
|
|
|
return err
|
|
}
|
|
if err := os.WriteFile(path, data, 0o600); err != nil {
|
|
s.logger.Warn("Failed to write document file", zap.Error(err), zap.String("path", path))
|
|
|
|
return err
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
func (s *LocalStore) Load(ctx context.Context, key string) ([]byte, error) {
|
|
if err := ctx.Err(); err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
path := filepath.Join(s.rootPath, filepath.Clean(key))
|
|
|
|
data, err := os.ReadFile(path)
|
|
if err != nil {
|
|
s.logger.Warn("Failed to read document file", zap.Error(err), zap.String("path", path))
|
|
|
|
return nil, err
|
|
}
|
|
|
|
return data, nil
|
|
}
|
|
|
|
var _ Store = (*LocalStore)(nil)
|