forked from mirror/ledisdb
49 lines
825 B
Go
49 lines
825 B
Go
package store
|
|
|
|
import (
|
|
"github.com/ledisdb/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) GetSlice(key []byte) (Slice, error) {
|
|
if d, ok := s.ISnapshot.(driver.ISliceGeter); ok {
|
|
v, err := d.GetSlice(key)
|
|
s.st.statGet(v, err)
|
|
return v, err
|
|
} else {
|
|
v, err := s.Get(key)
|
|
if err != nil {
|
|
return nil, err
|
|
} else if v == nil {
|
|
return nil, nil
|
|
} else {
|
|
return driver.GoSlice(v), nil
|
|
}
|
|
}
|
|
}
|
|
|
|
func (s *Snapshot) Close() {
|
|
s.st.SnapshotCloseNum.Add(1)
|
|
s.ISnapshot.Close()
|
|
}
|