diff options
| author | mo khan <mo@mokhan.ca> | 2024-06-19 11:04:03 -0600 |
|---|---|---|
| committer | mo khan <mo@mokhan.ca> | 2024-06-19 11:04:03 -0600 |
| commit | 77bddf64c1991acf97c7c8896203bc9872012918 (patch) | |
| tree | 44c80a5deccf9dbe412a6b0813b0aa86bdf1ebc4 /pkg/db/storage.go | |
| parent | ee8260d57d8194288c52056ee27b0d0a7c111bdd (diff) | |
Rename Paramable to Entity
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()) +} |
