53 lines
1.3 KiB
Go
53 lines
1.3 KiB
Go
|
package document
|
|||
|
|
|
|||
|
|
import (
|
|||
|
|
"fmt"
|
|||
|
|
|
|||
|
|
"go.mongodb.org/mongo-driver/bson"
|
|||
|
|
"git.pyer.club/kingecg/godocdb/storage"
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
// DocumentStore 管理文档的存储和检索
|
|||
|
|
type DocumentStore struct {
|
|||
|
|
storage *storage.LevelDBStorage
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// NewDocumentStore 创建新的文档存储实例
|
|||
|
|
func NewDocumentStore(path string) (*DocumentStore, error) {
|
|||
|
|
storage, err := storage.NewLevelDBStorage(path)
|
|||
|
|
if err != nil {
|
|||
|
|
return nil, err
|
|||
|
|
}
|
|||
|
|
return &DocumentStore{storage: storage}, nil
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// StoreDocument 存储文档
|
|||
|
|
func (ds *DocumentStore) StoreDocument(id string, doc interface{}) error {
|
|||
|
|
// 使用BSON序列化文档
|
|||
|
|
data, err := bson.Marshal(doc)
|
|||
|
|
if err != nil {
|
|||
|
|
return fmt.Errorf("failed to marshal document: %v", err)
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 存储文档（格式： collection:id → BSON数据）
|
|||
|
|
key := []byte(fmt.Sprintf("documents:%s", id))
|
|||
|
|
return ds.storage.Put(key, data)
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// GetDocument 获取文档
|
|||
|
|
func (ds *DocumentStore) GetDocument(id string, result interface{}) error {
|
|||
|
|
key := []byte(fmt.Sprintf("documents:%s", id))
|
|||
|
|
data, err := ds.storage.Get(key)
|
|||
|
|
if err != nil {
|
|||
|
|
return fmt.Errorf("document not found: %v", err)
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 使用BSON反序列化文档
|
|||
|
|
return bson.Unmarshal(data, result)
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// DeleteDocument 删除文档
|
|||
|
|
func (ds *DocumentStore) DeleteDocument(id string) error {
|
|||
|
|
key := []byte(fmt.Sprintf("documents:%s", id))
|
|||
|
|
return ds.storage.Delete(key)
|
|||
|
|
}
|