/
is_remote.go
60 lines (48 loc) · 1.32 KB
/
is_remote.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
package function
import (
"fmt"
"reflect"
"gopkg.in/src-d/go-git.v4/plumbing"
"github.com/src-d/go-mysql-server/sql"
"github.com/src-d/go-mysql-server/sql/expression"
)
// IsRemote checks the given string is a remote reference.
type IsRemote struct {
expression.UnaryExpression
}
// NewIsRemote creates a new IsRemote function.
func NewIsRemote(e sql.Expression) sql.Expression {
return &IsRemote{expression.UnaryExpression{Child: e}}
}
// Eval implements the expression interface.
func (f *IsRemote) Eval(ctx *sql.Context, row sql.Row) (interface{}, error) {
span, ctx := ctx.Span("gitbase.IsRemote")
defer span.Finish()
val, err := f.Child.Eval(ctx, row)
if err != nil {
return nil, err
}
if val == nil {
return false, nil
}
name, ok := val.(string)
if !ok {
return nil, sql.ErrInvalidType.New(reflect.TypeOf(val).String())
}
return plumbing.ReferenceName(name).IsRemote(), nil
}
func (f IsRemote) String() string {
return fmt.Sprintf("is_remote(%s)", f.Child)
}
// TransformUp implements the Expression interface.
func (f IsRemote) TransformUp(fn sql.TransformExprFunc) (sql.Expression, error) {
child, err := f.Child.TransformUp(fn)
if err != nil {
return nil, err
}
return fn(NewIsRemote(child))
}
// Type implements the Expression interface.
func (IsRemote) Type() sql.Type {
return sql.Boolean
}