From b4e9a0aad67c241916798cc766488142d4719ed9 Mon Sep 17 00:00:00 2001
From: kingecg <kingecg@hotmail.com>
Date: Sat, 7 Jun 2025 17:52:14 +0800
Subject: [PATCH] Initial storage layer implementation with tests

---
 go.mod                  |  5 +++++
 go.sum                  | 18 +++++++++++++++++
 storage/leveldb.go      | 39 +++++++++++++++++++++++++++++++++++
 storage/leveldb_test.go | 45 +++++++++++++++++++++++++++++++++++++++++
 4 files changed, 107 insertions(+)
 create mode 100644 go.sum
 create mode 100644 storage/leveldb.go
 create mode 100644 storage/leveldb_test.go

diff --git a/go.mod b/go.mod
index 504716b..c4f6464 100644
--- a/go.mod
+++ b/go.mod
@@ -1,3 +1,8 @@
 module git.pyer.club/kingecg/godocdb
 
 go 1.23.1
+
+require (
+	github.com/golang/snappy v0.0.0-20180518054509-2e65f85255db // indirect
+	github.com/syndtr/goleveldb v1.0.0 // indirect
+)
diff --git a/go.sum b/go.sum
new file mode 100644
index 0000000..e188042
--- /dev/null
+++ b/go.sum
@@ -0,0 +1,18 @@
+github.com/fsnotify/fsnotify v1.4.7/go.mod h1:jwhsz4b93w/PPRr/qN1Yymfu8t87LnFCMoQvtojpjFo=
+github.com/golang/protobuf v1.2.0/go.mod h1:6lQm79b+lXiMfvg/cZm0SGofjICqVBUtrP5yJMmIC1U=
+github.com/golang/snappy v0.0.0-20180518054509-2e65f85255db h1:woRePGFeVFfLKN/pOkfl+p/TAqKOfFu+7KPlMVpok/w=
+github.com/golang/snappy v0.0.0-20180518054509-2e65f85255db/go.mod h1:/XxbfmMg8lxefKM7IXC3fBNl/7bRcc72aCRzEWrmP2Q=
+github.com/hpcloud/tail v1.0.0/go.mod h1:ab1qPbhIpdTxEkNHXyeSf5vhxWSCs/tWer42PpOxQnU=
+github.com/onsi/ginkgo v1.6.0/go.mod h1:lLunBs/Ym6LB5Z9jYTR76FiuTmxDTDusOGeTQH+WWjE=
+github.com/onsi/ginkgo v1.7.0/go.mod h1:lLunBs/Ym6LB5Z9jYTR76FiuTmxDTDusOGeTQH+WWjE=
+github.com/onsi/gomega v1.4.3/go.mod h1:ex+gbHU/CVuBBDIJjb2X0qEXbFg53c61hWP/1CpauHY=
+github.com/syndtr/goleveldb v1.0.0 h1:fBdIW9lB4Iz0n9khmH8w27SJ3QEJ7+IgjPEwGSZiFdE=
+github.com/syndtr/goleveldb v1.0.0/go.mod h1:ZVVdQEZoIme9iO1Ch2Jdy24qqXrMMOU6lpPAyBWyWuQ=
+golang.org/x/net v0.0.0-20180906233101-161cd47e91fd/go.mod h1:mL1N/T3taQHkDXs73rZJwtUhF3w3ftmwwsq0BUmARs4=
+golang.org/x/sync v0.0.0-20180314180146-1d60e4601c6f/go.mod h1:RxMgew5VJxzue5/jJTE5uejpjVlOe/izrB70Jof72aM=
+golang.org/x/sys v0.0.0-20180909124046-d0be0721c37e/go.mod h1:STP8DvDyc/dI5b8T5hshtkjS+E42TnysNCUPdjciGhY=
+golang.org/x/text v0.3.0/go.mod h1:NqM8EUOU14njkJ3fqMW+pc6Ldnwhi/IjpwHt7yyuwOQ=
+gopkg.in/check.v1 v0.0.0-20161208181325-20d25e280405/go.mod h1:Co6ibVJAznAaIkqp8huTwlJQCZ016jof/cbN4VW5Yz0=
+gopkg.in/fsnotify.v1 v1.4.7/go.mod h1:Tz8NjZHkW78fSQdbUxIjBTcgA1z1m8ZHf0WmKUhAMys=
+gopkg.in/tomb.v1 v1.0.0-20141024135613-dd632973f1e7/go.mod h1:dt/ZhP58zS4L8KSrWDmTeBkI65Dw0HsyUHuEVlX15mw=
+gopkg.in/yaml.v2 v2.2.1/go.mod h1:hI93XBmqTisBFMUTm0b8Fm+jr3Dg1NNxqwp+5A1VGuI=
diff --git a/storage/leveldb.go b/storage/leveldb.go
new file mode 100644
index 0000000..7a95d96
--- /dev/null
+++ b/storage/leveldb.go
@@ -0,0 +1,39 @@
+package storage
+
+import (
+	"github.com/syndtr/goleveldb/leveldb"
+)
+
+// LevelDBStorage 封装LevelDB基础操作
+type LevelDBStorage struct {
+	db *leveldb.DB
+}
+
+// NewLevelDBStorage 初始化LevelDB实例
+func NewLevelDBStorage(path string) (*LevelDBStorage, error) {
+	db, err := leveldb.OpenFile(path, nil)
+	if err != nil {
+		return nil, err
+	}
+	return &LevelDBStorage{db: db}, nil
+}
+
+// Put 存储键值对
+func (s *LevelDBStorage) Put(key []byte, value []byte) error {
+	return s.db.Put(key, value, nil)
+}
+
+// Get 获取键对应的值
+func (s *LevelDBStorage) Get(key []byte) ([]byte, error) {
+	return s.db.Get(key, nil)
+}
+
+// Delete 删除指定键
+func (s *LevelDBStorage) Delete(key []byte) error {
+	return s.db.Delete(key, nil)
+}
+
+// Close 关闭数据库连接
+func (s *LevelDBStorage) Close() {
+	s.db.Close()
+}
\ No newline at end of file
diff --git a/storage/leveldb_test.go b/storage/leveldb_test.go
new file mode 100644
index 0000000..0b3b598
--- /dev/null
+++ b/storage/leveldb_test.go
@@ -0,0 +1,45 @@
+package storage
+
+import (
+	"os"
+	"testing"
+)
+
+func TestLevelDBStorage(t *testing.T) {
+	// 测试目录
+	dir := "./testdb"
+	defer os.RemoveAll(dir)
+
+	// 初始化存储
+	storage, err := NewLevelDBStorage(dir)
+	if err != nil {
+		t.Fatalf("Failed to create LevelDB: %v", err)
+	}
+	defer storage.Close()
+
+	// 测试Put和Get
+	key := []byte("test_key")
+	value := []byte("test_value")
+	
+	if err := storage.Put(key, value); err != nil {
+		t.Errorf("Put failed: %v", err)
+	}
+
+	retrieved, err := storage.Get(key)
+	if err != nil {
+		t.Errorf("Get failed: %v", err)
+	}
+	if string(retrieved) != string(value) {
+		t.Errorf("Get returned wrong value: got %s want %s", retrieved, value)
+	}
+
+	// 测试Delete
+	if err := storage.Delete(key); err != nil {
+		t.Errorf("Delete failed: %v", err)
+	}
+
+	_, err = storage.Get(key)
+	if err == nil {
+		t.Errorf("Expected error after Delete")
+	}
+}
\ No newline at end of file