leveldb.go 2.0 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111
  1. package ezdb
  2. import (
  3. "os"
  4. "github.com/syndtr/goleveldb/leveldb"
  5. "github.com/syndtr/goleveldb/leveldb/opt"
  6. )
  7. type LevelDBCollection[T any] struct {
  8. path string
  9. db *leveldb.DB
  10. m DocumentMarshaler[T, []byte]
  11. optOpen *opt.Options
  12. optRead *opt.ReadOptions
  13. optWrite *opt.WriteOptions
  14. }
  15. func (c *LevelDBCollection[T]) Close() error {
  16. if c.db != nil {
  17. if err := c.db.Close(); err != nil {
  18. return err
  19. }
  20. c.db = nil
  21. }
  22. return nil
  23. }
  24. func (c *LevelDBCollection[T]) Delete(key string) error {
  25. return c.db.Delete([]byte(key), c.optWrite)
  26. }
  27. // Destroy the database completely, removing it from disk.
  28. func (c *LevelDBCollection[T]) Destroy() error {
  29. if err := c.Close(); err != nil {
  30. return err
  31. }
  32. return os.RemoveAll(c.path)
  33. }
  34. func (c *LevelDBCollection[T]) Get(key string) (T, error) {
  35. dest := c.m.Factory()
  36. src, err := c.db.Get([]byte(key), c.optRead)
  37. if err != nil {
  38. return dest, err
  39. }
  40. err = c.m.Unmarshal(src, dest)
  41. return dest, err
  42. }
  43. func (c *LevelDBCollection[T]) Has(key string) (bool, error) {
  44. return c.db.Has([]byte(key), c.optRead)
  45. }
  46. func (c *LevelDBCollection[T]) Iter() Iterator[T] {
  47. i := &LevelDBIterator[T]{
  48. i: c.db.NewIterator(nil, c.optRead),
  49. m: c.m,
  50. }
  51. return i
  52. }
  53. func (c *LevelDBCollection[T]) Open() error {
  54. if c.db == nil {
  55. db, err := leveldb.OpenFile(c.path, c.optOpen)
  56. if err != nil {
  57. return err
  58. }
  59. c.db = db
  60. }
  61. return nil
  62. }
  63. func (c *LevelDBCollection[T]) Put(key string, src T) error {
  64. if err := ValidateKey(key); err != nil {
  65. return err
  66. }
  67. dest, err := c.m.Marshal(src)
  68. if err != nil {
  69. return err
  70. }
  71. return c.db.Put([]byte(key), dest, c.optWrite)
  72. }
  73. // LevelDB creates a new collection using LevelDB storage.
  74. func LevelDB[T any](path string, m DocumentMarshaler[T, []byte], o *LevelDBOptions) *LevelDBCollection[T] {
  75. c := &LevelDBCollection[T]{
  76. path: path,
  77. m: m,
  78. // Unpack options now to reduce nil checks
  79. optOpen: o.GetOpen(),
  80. optRead: o.GetRead(),
  81. optWrite: o.GetWrite(),
  82. }
  83. return c
  84. }