88 lines
2.1 KiB
Go
88 lines
2.1 KiB
Go
package arango
|
|
|
|
import (
|
|
"strings"
|
|
)
|
|
|
|
// Query provides a simple way to build ArangoDB queries.
|
|
type Query struct {
|
|
Lines []string
|
|
Params map[string]any
|
|
}
|
|
|
|
// Append adds a line to the query, attaching (optional) values to bind parameters.
|
|
//
|
|
// Any variables given are assigned to bind parameters in the same order as they appear in the line.
|
|
// Keep in mind the following behaviours:
|
|
//
|
|
// - If more variables are provided than there are parameters in the line, leftover variables are discarded
|
|
// - If more parameters are used than variables are given, remaining parameters are left unmapped
|
|
// - Reused parameters are overwritten
|
|
func (query *Query) Append(line string, values ...any) *Query {
|
|
var params map[string]any = nil
|
|
|
|
names := ReadParams(line)
|
|
if len(names) > 0 {
|
|
params = map[string]any{}
|
|
for i, name := range names {
|
|
if i > len(values) {
|
|
break
|
|
}
|
|
params[name] = values[i]
|
|
}
|
|
}
|
|
|
|
query.Lines = append(query.Lines, line)
|
|
return query.AssignMap(params)
|
|
}
|
|
|
|
// Assign assigns a value to a single bind parameter.
|
|
func (query *Query) Assign(name string, value any) *Query {
|
|
query.Params[name] = value
|
|
return query
|
|
}
|
|
|
|
// AssignMap assigns values to bind parameters.
|
|
func (query *Query) AssignMap(params map[string]any) *Query {
|
|
if params != nil {
|
|
for name, value := range params {
|
|
query.Params[name] = value
|
|
}
|
|
}
|
|
return query
|
|
}
|
|
|
|
// Copy creates a copy of the query.
|
|
func (query *Query) Copy() *Query {
|
|
newQuery := NewQuery()
|
|
for _, line := range query.Lines {
|
|
newQuery.Lines = append(newQuery.Lines, line)
|
|
}
|
|
for name, value := range query.Params {
|
|
newQuery.Params[name] = value
|
|
}
|
|
return newQuery
|
|
}
|
|
|
|
// L (for "Line") is a shorthand for Append.
|
|
func (query *Query) L(line string, values ...any) *Query {
|
|
return query.Append(line, values...)
|
|
}
|
|
|
|
// P (for "Parameter") is a shorthand for Assign.
|
|
func (query *Query) P(name string, value any) *Query {
|
|
return query.Assign(name, value)
|
|
}
|
|
|
|
func (query *Query) String() string {
|
|
return strings.Join(query.Lines, "\n")
|
|
}
|
|
|
|
// NewQuery creates a new Query.
|
|
func NewQuery() *Query {
|
|
return &Query{
|
|
Lines: []string{},
|
|
Params: map[string]any{},
|
|
}
|
|
}
|