mirror of
https://github.com/MontFerret/ferret.git
synced 2025-11-23 21:54:45 +02:00
@@ -1,102 +0,0 @@
|
||||
package compiler
|
||||
|
||||
import (
|
||||
"github.com/MontFerret/ferret/pkg/parser"
|
||||
"github.com/MontFerret/ferret/pkg/runtime"
|
||||
"github.com/MontFerret/ferret/pkg/runtime/core"
|
||||
"github.com/MontFerret/ferret/pkg/stdlib"
|
||||
"github.com/pkg/errors"
|
||||
"strings"
|
||||
)
|
||||
|
||||
type FqlCompiler struct {
|
||||
funcs map[string]core.Function
|
||||
}
|
||||
|
||||
func New(setters ...Option) *FqlCompiler {
|
||||
c := &FqlCompiler{}
|
||||
opts := &Options{}
|
||||
|
||||
for _, setter := range setters {
|
||||
setter(opts)
|
||||
}
|
||||
|
||||
if !opts.noStdlib {
|
||||
c.funcs = stdlib.NewLib()
|
||||
} else {
|
||||
c.funcs = make(map[string]core.Function)
|
||||
}
|
||||
|
||||
return &FqlCompiler{
|
||||
stdlib.NewLib(),
|
||||
}
|
||||
}
|
||||
|
||||
func (c *FqlCompiler) RegisterFunction(name string, fun core.Function) error {
|
||||
_, exists := c.funcs[name]
|
||||
|
||||
if exists {
|
||||
return errors.Errorf("function already exists: %s", name)
|
||||
}
|
||||
|
||||
c.funcs[strings.ToUpper(name)] = fun
|
||||
|
||||
return nil
|
||||
}
|
||||
|
||||
func (c *FqlCompiler) RegisterFunctions(funcs map[string]core.Function) error {
|
||||
for name, fun := range funcs {
|
||||
if err := c.RegisterFunction(name, fun); err != nil {
|
||||
return err
|
||||
}
|
||||
}
|
||||
|
||||
return nil
|
||||
}
|
||||
|
||||
func (c *FqlCompiler) Compile(query string) (program *runtime.Program, err error) {
|
||||
if query == "" {
|
||||
return nil, ErrEmptyQuery
|
||||
}
|
||||
|
||||
p := parser.New(query)
|
||||
p.AddErrorListener(&errorListener{})
|
||||
|
||||
defer func() {
|
||||
if r := recover(); r != nil {
|
||||
// find out exactly what the error was and set err
|
||||
switch x := r.(type) {
|
||||
case string:
|
||||
err = errors.New(x)
|
||||
case error:
|
||||
err = x
|
||||
default:
|
||||
err = errors.New("unknown panic")
|
||||
}
|
||||
|
||||
program = nil
|
||||
}
|
||||
}()
|
||||
|
||||
l := newVisitor(query, c.funcs)
|
||||
|
||||
res := p.Visit(l).(*result)
|
||||
|
||||
if res.Ok() {
|
||||
program = res.Data().(*runtime.Program)
|
||||
} else {
|
||||
err = res.Error()
|
||||
}
|
||||
|
||||
return program, err
|
||||
}
|
||||
|
||||
func (c *FqlCompiler) MustCompile(query string) *runtime.Program {
|
||||
program, err := c.Compile(query)
|
||||
|
||||
if err != nil {
|
||||
panic(err)
|
||||
}
|
||||
|
||||
return program
|
||||
}
|
||||
Reference in New Issue
Block a user