56 lines
1.2 KiB
Go
56 lines
1.2 KiB
Go
// Copyright (C) 2017 ScyllaDB
|
|
// Use of this source code is governed by a ALv2-style
|
|
// license that can be found in the LICENSE file.
|
|
|
|
package qb
|
|
|
|
import (
|
|
"bytes"
|
|
"strconv"
|
|
)
|
|
|
|
// value is a CQL value expression for use in an initializer, assignment,
|
|
// or comparison.
|
|
type value interface {
|
|
// writeCql writes the bytes for this value to the buffer and returns the
|
|
// list of names of parameters which need substitution.
|
|
writeCql(cql *bytes.Buffer) (names []string)
|
|
}
|
|
|
|
// param is a named CQL '?' parameter.
|
|
type param string
|
|
|
|
func (p param) writeCql(cql *bytes.Buffer) (names []string) {
|
|
cql.WriteByte('?')
|
|
return []string{string(p)}
|
|
}
|
|
|
|
// param is a named CQL tuple '?' parameter.
|
|
type tupleParam struct {
|
|
param param
|
|
count int
|
|
}
|
|
|
|
func (t tupleParam) writeCql(cql *bytes.Buffer) (names []string) {
|
|
baseName := string(t.param)
|
|
cql.WriteByte('(')
|
|
for i := 0; i < t.count-1; i++ {
|
|
cql.WriteByte('?')
|
|
cql.WriteByte(',')
|
|
names = append(names, baseName+"["+strconv.Itoa(i)+"]")
|
|
}
|
|
cql.WriteByte('?')
|
|
cql.WriteByte(')')
|
|
names = append(names, baseName+"["+strconv.Itoa(t.count-1)+"]")
|
|
|
|
return
|
|
}
|
|
|
|
// lit is a literal CQL value.
|
|
type lit string
|
|
|
|
func (l lit) writeCql(cql *bytes.Buffer) (names []string) {
|
|
cql.WriteString(string(l))
|
|
return nil
|
|
}
|