Beispiel #1
0
// Cursor returns an iterator for a key.
func (tx *Tx) Cursor(key string, direction tsdb.Direction) tsdb.Cursor {
	// Retrieve key bucket.
	b := tx.Bucket([]byte(key))

	tx.engine.mu.RLock()
	defer tx.engine.mu.RUnlock()

	// Ignore if there is no bucket or points in the cache.
	partitionID := WALPartition([]byte(key))
	if b == nil && len(tx.engine.cache[partitionID][key]) == 0 {
		return nil
	}

	// Retrieve a copy of the in-cache points for the key.
	cache := make([][]byte, len(tx.engine.cache[partitionID][key]))
	copy(cache, tx.engine.cache[partitionID][key])

	// Build a cursor that merges the bucket and cache together.
	cur := &Cursor{cache: cache, direction: direction}
	if b != nil {
		cur.cursor = b.Cursor()
	}

	// If it's a reverse cursor, set the current location to the end.
	if direction.Reverse() {
		cur.index = len(cache) - 1
		if cur.cursor != nil {
			cur.cursor.Last()
		}
	}
	return cur
}
Beispiel #2
0
// NewCursor returns a new instance of Cursor.
func NewCursor(direction tsdb.Direction, items []CursorItem) *Cursor {
	index := 0
	sort.Sort(CursorItems(items))

	if direction.Reverse() {
		index = len(items)
	}
	return &Cursor{direction: direction, items: items, index: index}
}
Beispiel #3
0
func newCursor(cache [][]byte, direction tsdb.Direction) *cursor {
	// position is set such that a call to Next will successfully advance
	// to the next postion and return the value.
	c := &cursor{cache: cache, direction: direction, position: -1}
	if direction.Reverse() {
		c.position = len(c.cache)
	}
	return c
}
Beispiel #4
0
// Cursor returns an iterator for a key.
func (tx *Tx) Cursor(key string, direction tsdb.Direction) tsdb.Cursor {
	walCursor := tx.wal.Cursor(key, direction)

	// Retrieve points bucket. Ignore if there is no bucket.
	b := tx.Bucket([]byte("points")).Bucket([]byte(key))
	if b == nil {
		return walCursor
	}

	c := &Cursor{
		cursor:    b.Cursor(),
		direction: direction,
	}

	if direction.Reverse() {
		c.last()
	}

	return tsdb.MultiCursor(direction, walCursor, c)
}