package store

import (
	"github.com/siddontang/ledisdb/store/driver"
)

type Snapshot struct {
	driver.ISnapshot
	st *Stat
}

func (s *Snapshot) NewIterator() *Iterator {
	it := new(Iterator)
	it.it = s.ISnapshot.NewIterator()
	it.st = s.st

	s.st.IterNum.Add(1)

	return it
}

func (s *Snapshot) Get(key []byte) ([]byte, error) {
	v, err := s.ISnapshot.Get(key)
	s.st.statGet(v, err)
	return v, err
}

func (s *Snapshot) Close() {
	s.st.SnapshotCloseNum.Add(1)
	s.ISnapshot.Close()
}