forked from apache/tinkerpop
/
client.go
88 lines (77 loc) · 2.83 KB
/
client.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
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
/*
Licensed to the Apache Software Foundation (ASF) under one
or more contributor license agreements. See the NOTICE file
distributed with this work for additional information
regarding copyright ownership. The ASF licenses this file
to you under the Apache License, Version 2.0 (the
"License"); you may not use this file except in compliance
with the License. You may obtain a copy of the License at
http://www.apache.org/licenses/LICENSE-2.0
Unless required by applicable law or agreed to in writing,
software distributed under the License is distributed on an
"AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
KIND, either express or implied. See the License for the
specific language governing permissions and limitations
under the License.
*/
package gremlingo
import (
"golang.org/x/text/language"
)
// ClientSettings is used to modify a Client's settings on initialization.
type ClientSettings struct {
TransporterType TransporterType
LogVerbosity LogVerbosity
Logger Logger
Language language.Tag
}
// Client is used to connect and interact with a Gremlin-supported server.
type Client struct {
host string
port int
logHandler *logHandler
transporterType TransporterType
connection *connection
}
// NewClient creates a Client and configures it with the given parameters.
func NewClient(host string, port int, configurations ...func(settings *ClientSettings)) (*Client, error) {
settings := &ClientSettings{
TransporterType: Gorilla,
LogVerbosity: Info,
Logger: &defaultLogger{},
Language: language.English,
}
for _, configuration := range configurations {
configuration(settings)
}
logHandler := newLogHandler(settings.Logger, settings.LogVerbosity, settings.Language)
conn, err := createConnection(host, port, logHandler)
if err != nil {
return nil, err
}
client := &Client{
host: host,
port: port,
logHandler: logHandler,
transporterType: settings.TransporterType,
connection: conn,
}
return client, nil
}
// Close closes the client via connection
func (client *Client) Close() error {
return client.connection.close()
}
// Submit submits a Gremlin script to the server and returns a ResultSet
func (client *Client) Submit(traversalString string) (ResultSet, error) {
// TODO AN-982: Obtain connection from pool of connections held by the client.
client.logHandler.logf(Debug, submitStartedString, traversalString)
request := makeStringRequest(traversalString)
return client.connection.write(&request)
}
// SubmitBytecode submits bytecode to the server to execute and returns a ResultSet
func (client *Client) SubmitBytecode(bytecode *bytecode) (ResultSet, error) {
client.logHandler.logf(Debug, submitStartedBytecode, *bytecode)
request := makeBytecodeRequest(bytecode)
return client.connection.write(&request)
}