-
-
package index
-
-import (
- "fmt"
-
- "github.com/iancoleman/orderedmap"
- "go.mongodb.org/mongo-driver/bson"
- "git.pyer.club/kingecg/godocdb/storage"
-)
-
-// IndexType 表示索引类型
-type IndexType string
-
-const (
- SingleField IndexType = "single"
- Composite IndexType = "composite"
-)
-
-// IndexMetadata 索引元数据
-type IndexMetadata struct {
- Name string
- Type IndexType
- KeyFields []string
-}
-
-// IndexStore 管理索引的存储和查询
-type IndexStore struct {
- storage *storage.LevelDBStorage
-}
-
-// NewIndexStore 创建新的索引存储实例
-func NewIndexStore(path string) (*IndexStore, error) {
- storage, err := storage.NewLevelDBStorage(path)
- if err != nil {
- return nil, err
- }
- return &IndexStore{storage: storage}, nil
-}
-
-// CreateIndex 创建索引
-func (is *IndexStore) CreateIndex(indexName string, indexType IndexType, keyFields []string) error {
- // 存储索引元数据
- metadata := IndexMetadata{
- Name: indexName,
- Type: indexType,
- KeyFields: keyFields,
- }
-
- data, err := bson.Marshal(metadata)
- if err != nil {
- return fmt.Errorf("failed to marshal index metadata: %v", err)
- }
-
- key := []byte(fmt.Sprintf("indexes:metadata:%s", indexName))
- if err := is.storage.Put(key, data); err != nil {
- return err
- }
-
- // 初始化索引存储结构
- indexKey := fmt.Sprintf("indexes:data:%s", indexName)
- index := orderedmap.New()
- indexData, _ := bson.Marshal(index)
- return is.storage.Put([]byte(indexKey), indexData)
-}
-
-// DropIndex 删除索引
-func (is *IndexStore) DropIndex(indexName string) error {
- // 删除元数据
- metadataKey := []byte(fmt.Sprintf("indexes:metadata:%s", indexName))
- if err := is.storage.Delete(metadataKey); err != nil {
- return err
- }
-
- // 删除索引数据
- indexKey := []byte(fmt.Sprintf("indexes:data:%s", indexName))
- return is.storage.Delete(indexKey)
-}
-
-// GetIndexMetadata 获取索引元数据
-func (is *IndexStore) GetIndexMetadata(indexName string) (*IndexMetadata, error) {
- key := []byte(fmt.Sprintf("indexes:metadata:%s", indexName))
- rawData, err := is.storage.Get(key)
- if err != nil {
- return nil, fmt.Errorf("index not found: %v", err)
- }
-
- var metadata IndexMetadata
- if err := bson.Unmarshal(rawData, &metadata); err != nil {
- return nil, err
- }
-
- return &metadata, nil
-}
-
-