diff options
Diffstat (limited to 'pkg/db/storage.go')
| -rw-r--r-- | pkg/db/storage.go | 44 |
1 files changed, 44 insertions, 0 deletions
diff --git a/pkg/db/storage.go b/pkg/db/storage.go new file mode 100644 index 0000000..1c1c428 --- /dev/null +++ b/pkg/db/storage.go @@ -0,0 +1,44 @@ +package db + +import ( + "bytes" + "fmt" + "io/ioutil" + "os" + "path/filepath" + + "github.com/xlgmokha/x/pkg/env" + "github.com/xlgmokha/x/pkg/serde" + "github.com/xlgmokha/x/pkg/x" +) + +type Entity interface { + Identifier() string +} + +type Storage[T Entity] struct { + dir string +} + +func New[T Entity](dir string) *Storage[T] { + fullPath := x.Must(filepath.Abs(dir)) + x.Check(os.MkdirAll(fullPath, 0700)) + + return &Storage[T]{ + dir: fullPath, + } +} + +func (db *Storage[T]) Save(item T) error { + w := new(bytes.Buffer) + x.Check(serde.To(w, item, serde.YAML)) + if env.Fetch("DUMP", "") != "" { + fmt.Println(w.String()) + } + + return ioutil.WriteFile(db.filePathFor(item), w.Bytes(), 0700) +} + +func (db *Storage[T]) filePathFor(item T) string { + return fmt.Sprintf("%v/%v.yaml", db.dir, item.Identifier()) +} |
