package simple_bitcask_kv

import (
	"bytes"
	"simple_bitcask_kv/index"
)

// Iterator 迭代器
// 使用Iterator过程中需要上锁嘛？虽然应该每个用户各自用一个iterator
type Iterator struct {
	indexIter index.Iterator // 索引迭代器 - 像这样的分层非常多
	db        *DB
	options   IteratorOptions
}

func (db *DB) NewIterator(opts IteratorOptions) *Iterator {
	indexIter := db.index.Iterator(opts.Reverse)
	it := &Iterator{
		indexIter: indexIter,
		db:        db,
		options:   opts,
	}
	it.Rewind() // 之前忘加了这句，iterator初始位置就要考虑Prefix
	return it
}

// Rewind 重新回到迭代器第一个元素
func (it *Iterator) Rewind() {
	it.indexIter.Rewind()
	it.skipToNext()
}

// Seek 定位到第一个>=key的位置，从这里开始遍历
// 可以用二分查找
func (it *Iterator) Seek(key []byte) {
	it.indexIter.Seek(key)
	it.skipToNext()
}

// Next 跳转到下一个key
func (it *Iterator) Next() {
	it.indexIter.Next()
	it.skipToNext()
}

// Valid 迭代器是否有效，用于检查是否遍历结束
func (it *Iterator) Valid() bool {
	return it.indexIter.Valid()
}

// Key 当前位置key数据
func (it *Iterator) Key() []byte {
	return it.indexIter.Key()
}

// Value 当前位置Value数据 - 注意这里跟indexIterator直接返回LogRecordPos，而是要直接返回数值
// 一般是在确定Valid情况下返回Value，这样不用管err，有err直接返回nil
func (it *Iterator) Value() []byte {
	pos := it.indexIter.Value()
	val, err := it.db.getValueByPos(pos)
	if err != nil {
		return nil
	}
	return val
}

// Close 关闭迭代器，释放资源
func (it *Iterator) Close() {
	it.indexIter.Close()
}

// 筛选出满足prefix的value 在Rewind/Next/Seek中使用
func (it *Iterator) skipToNext() {
	prefixLen := len(it.options.Prefix)
	if prefixLen == 0 { // 没有指定prefix，不处理
		return
	}
	for it.Valid() {
		key := it.indexIter.Key()
		if prefixLen <= len(key) &&
			bytes.Compare(it.options.Prefix, key[:prefixLen]) == 0 {
			return
		}
		it.Next()
	}
}