-
Notifications
You must be signed in to change notification settings - Fork 914
/
driver.go
58 lines (47 loc) · 1.63 KB
/
driver.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
// Copyright 2019-present Facebook Inc. All rights reserved.
// This source code is licensed under the Apache 2.0 license found
// in the LICENSE file in the root directory of this source tree.
package gremlin
import (
"context"
"fmt"
"entgo.io/ent/dialect"
"entgo.io/ent/dialect/gremlin/graph/dsl"
)
// Driver is a dialect.Driver implementation for TinkerPop gremlin.
type Driver struct {
*Client
}
// NewDriver returns a new dialect.Driver implementation for gremlin.
func NewDriver(c *Client) *Driver {
c.Transport = ExpandBindings(c.Transport)
return &Driver{c}
}
// Dialect implements the dialect.Dialect method.
func (Driver) Dialect() string { return dialect.Gremlin }
// Exec implements the dialect.Exec method.
func (c *Driver) Exec(ctx context.Context, query string, args, v any) error {
vr, ok := v.(*Response)
if !ok {
return fmt.Errorf("dialect/gremlin: invalid type %T. expect *gremlin.Response", v)
}
bindings, ok := args.(dsl.Bindings)
if !ok {
return fmt.Errorf("dialect/gremlin: invalid type %T. expect map[string]any for bindings", args)
}
res, err := c.Do(ctx, NewEvalRequest(query, WithBindings(bindings)))
if err != nil {
return err
}
*vr = *res
return nil
}
// Query implements the dialect.Query method.
func (c *Driver) Query(ctx context.Context, query string, args, v any) error {
return c.Exec(ctx, query, args, v)
}
// Close is a nop close call. It should close the connection in case of WS client.
func (Driver) Close() error { return nil }
// Tx returns a nop transaction.
func (c *Driver) Tx(context.Context) (dialect.Tx, error) { return dialect.NopTx(c), nil }
var _ dialect.Driver = (*Driver)(nil)