23 lines
441 B
Go
23 lines
441 B
Go
|
package querytest
|
||
|
|
|
||
|
|
import (
|
||
|
|
"context"
|
||
|
|
|
||
|
|
"github.com/influxdata/platform/query"
|
||
|
|
)
|
||
|
|
|
||
|
|
// FromCSVCompiler wraps a compiler and replaces all From operations with FromCSV
|
||
|
|
type FromCSVCompiler struct {
|
||
|
|
query.Compiler
|
||
|
|
InputFile string
|
||
|
|
}
|
||
|
|
|
||
|
|
func (c FromCSVCompiler) Compile(ctx context.Context) (*query.Spec, error) {
|
||
|
|
spec, err := c.Compiler.Compile(ctx)
|
||
|
|
if err != nil {
|
||
|
|
return nil, err
|
||
|
|
}
|
||
|
|
ReplaceFromSpec(spec, c.InputFile)
|
||
|
|
return spec, nil
|
||
|
|
}
|