61 lines
1.3 KiB
Go
61 lines
1.3 KiB
Go
package report
|
|
|
|
import (
|
|
"context"
|
|
)
|
|
|
|
type (
|
|
// DatasourceProvider provides access to system datasources, such as ComposeRecords
|
|
DatasourceProvider interface {
|
|
// Datasource initializes and returns the Datasource the reporter can use
|
|
Datasource(context.Context, *LoadStepDefinition) (Datasource, error)
|
|
}
|
|
|
|
// Loader returns the next Frame from the Datasource
|
|
// @todo better memory reuse
|
|
Loader func(cap int) ([]*Frame, error)
|
|
// Closer closes the Datasource
|
|
Closer func()
|
|
|
|
DatasourceSet []Datasource
|
|
Datasource interface {
|
|
Name() string
|
|
Load(context.Context, ...*FrameDefinition) (Loader, Closer, error)
|
|
}
|
|
|
|
// GroupableDatasource is able to provide groupped data
|
|
GroupableDatasource interface {
|
|
Datasource
|
|
Group(GroupDefinition, string) (bool, error)
|
|
}
|
|
|
|
// @todo TransformableDatasource
|
|
)
|
|
|
|
// Merge merges the two DatasourceSets and overwrites any duplicates
|
|
func (dd DatasourceSet) Merge(mm DatasourceSet) DatasourceSet {
|
|
outer:
|
|
for _, m := range mm {
|
|
for i, d := range dd {
|
|
if d.Name() == m.Name() {
|
|
dd[i] = m
|
|
continue outer
|
|
}
|
|
}
|
|
dd = append(dd, m)
|
|
}
|
|
|
|
return dd
|
|
}
|
|
|
|
// Find searches for the Datasource by name
|
|
func (dd DatasourceSet) Find(name string) Datasource {
|
|
for _, d := range dd {
|
|
if d.Name() == name {
|
|
return d
|
|
}
|
|
}
|
|
|
|
return nil
|
|
}
|