87 lines
1.8 KiB
Go
87 lines
1.8 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
|
|
|
|
// INSERT reference:
|
|
// https://cassandra.apache.org/doc/latest/cql/dml.html#insert
|
|
|
|
import (
|
|
"bytes"
|
|
)
|
|
|
|
// InsertBuilder builds CQL INSERT statements.
|
|
type InsertBuilder struct {
|
|
table string
|
|
columns columns
|
|
unique bool
|
|
using using
|
|
}
|
|
|
|
// Insert returns a new InsertBuilder with the given table name.
|
|
func Insert(table string) *InsertBuilder {
|
|
return &InsertBuilder{
|
|
table: table,
|
|
}
|
|
}
|
|
|
|
// ToCql builds the query into a CQL string and named args.
|
|
func (b *InsertBuilder) ToCql() (stmt string, names []string) {
|
|
cql := bytes.Buffer{}
|
|
|
|
cql.WriteString("INSERT ")
|
|
|
|
cql.WriteString("INTO ")
|
|
cql.WriteString(b.table)
|
|
cql.WriteByte(' ')
|
|
|
|
cql.WriteByte('(')
|
|
b.columns.writeCql(&cql)
|
|
names = append(names, b.columns...)
|
|
cql.WriteString(") ")
|
|
|
|
cql.WriteString("VALUES (")
|
|
placeholders(&cql, len(b.columns))
|
|
cql.WriteString(") ")
|
|
|
|
names = append(names, b.using.writeCql(&cql)...)
|
|
|
|
if b.unique {
|
|
cql.WriteString("IF NOT EXISTS ")
|
|
}
|
|
|
|
stmt = cql.String()
|
|
return
|
|
}
|
|
|
|
// Into sets the INTO clause of the query.
|
|
func (b *InsertBuilder) Into(table string) *InsertBuilder {
|
|
b.table = table
|
|
return b
|
|
}
|
|
|
|
// Columns adds insert columns to the query.
|
|
func (b *InsertBuilder) Columns(columns ...string) *InsertBuilder {
|
|
b.columns = append(b.columns, columns...)
|
|
return b
|
|
}
|
|
|
|
// Unique sets a IF NOT EXISTS clause on the query.
|
|
func (b *InsertBuilder) Unique() *InsertBuilder {
|
|
b.unique = true
|
|
return b
|
|
}
|
|
|
|
// Timestamp sets a USING TIMESTAMP clause on the query.
|
|
func (b *InsertBuilder) Timestamp() *InsertBuilder {
|
|
b.using.timestamp = true
|
|
return b
|
|
}
|
|
|
|
// TTL sets a USING TTL clause on the query.
|
|
func (b *InsertBuilder) TTL() *InsertBuilder {
|
|
b.using.ttl = true
|
|
return b
|
|
}
|