72 lines
2.6 KiB
Go
72 lines
2.6 KiB
Go
package mock
|
|
|
|
import (
|
|
"context"
|
|
|
|
"github.com/influxdata/flux/memory"
|
|
"github.com/influxdata/influxdb/v2/query"
|
|
)
|
|
|
|
type StorageReader struct {
|
|
ReadFilterFn func(ctx context.Context, spec query.ReadFilterSpec, alloc *memory.Allocator) (query.TableIterator, error)
|
|
ReadGroupFn func(ctx context.Context, spec query.ReadGroupSpec, alloc *memory.Allocator) (query.TableIterator, error)
|
|
ReadTagKeysFn func(ctx context.Context, spec query.ReadTagKeysSpec, alloc *memory.Allocator) (query.TableIterator, error)
|
|
ReadTagValuesFn func(ctx context.Context, spec query.ReadTagValuesSpec, alloc *memory.Allocator) (query.TableIterator, error)
|
|
CloseFn func()
|
|
}
|
|
|
|
func (s *StorageReader) ReadFilter(ctx context.Context, spec query.ReadFilterSpec, alloc *memory.Allocator) (query.TableIterator, error) {
|
|
return s.ReadFilterFn(ctx, spec, alloc)
|
|
}
|
|
|
|
func (s *StorageReader) ReadGroup(ctx context.Context, spec query.ReadGroupSpec, alloc *memory.Allocator) (query.TableIterator, error) {
|
|
return s.ReadGroupFn(ctx, spec, alloc)
|
|
}
|
|
|
|
func (s *StorageReader) ReadTagKeys(ctx context.Context, spec query.ReadTagKeysSpec, alloc *memory.Allocator) (query.TableIterator, error) {
|
|
return s.ReadTagKeysFn(ctx, spec, alloc)
|
|
}
|
|
|
|
func (s *StorageReader) ReadTagValues(ctx context.Context, spec query.ReadTagValuesSpec, alloc *memory.Allocator) (query.TableIterator, error) {
|
|
return s.ReadTagValuesFn(ctx, spec, alloc)
|
|
}
|
|
|
|
func (s *StorageReader) Close() {
|
|
// Only invoke the close function if it is set.
|
|
// We want this to be a no-op and work without
|
|
// explicitly setting up a close function.
|
|
if s.CloseFn != nil {
|
|
s.CloseFn()
|
|
}
|
|
}
|
|
|
|
type GroupStoreReader struct {
|
|
*StorageReader
|
|
GroupCapabilityFn func(ctx context.Context) query.GroupCapability
|
|
}
|
|
|
|
func (s *GroupStoreReader) GetGroupCapability(ctx context.Context) query.GroupCapability {
|
|
if s.GroupCapabilityFn != nil {
|
|
return s.GroupCapabilityFn(ctx)
|
|
}
|
|
return nil
|
|
}
|
|
|
|
type WindowAggregateStoreReader struct {
|
|
*StorageReader
|
|
GetWindowAggregateCapabilityFn func(ctx context.Context) query.WindowAggregateCapability
|
|
ReadWindowAggregateFn func(ctx context.Context, spec query.ReadWindowAggregateSpec, alloc *memory.Allocator) (query.TableIterator, error)
|
|
}
|
|
|
|
func (s *WindowAggregateStoreReader) GetWindowAggregateCapability(ctx context.Context) query.WindowAggregateCapability {
|
|
// Use the function if it exists.
|
|
if s.GetWindowAggregateCapabilityFn != nil {
|
|
return s.GetWindowAggregateCapabilityFn(ctx)
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func (s *WindowAggregateStoreReader) ReadWindowAggregate(ctx context.Context, spec query.ReadWindowAggregateSpec, alloc *memory.Allocator) (query.TableIterator, error) {
|
|
return s.ReadWindowAggregateFn(ctx, spec, alloc)
|
|
}
|