Add an implementation of an INSERT statement

statement.go

@@ -67,6 +67,42 @@ type InsertStatement struct {
 	ttl      time.Duration          // ttl of the row
 }
 
+// Query provides the CQL query string for an UPDATE query
+func (s InsertStatement) Query() string {
+	query, _ := s.queryAndValues()
+	return query
+}
+
+// Values provide the binding values for an UPDATE query
+func (s InsertStatement) Values() []interface{} {
+	_, values := s.queryAndValues()
+	return values
+}
+
+func (s InsertStatement) queryAndValues() (string, []interface{}) {
+	query := []string{"INSERT INTO", fmt.Sprintf("%s.%s", s.keyspace, s.table)}
+
+	fieldNames := make([]string, 0, len(s.fieldMap))
+	placeholders := make([]string, 0, len(s.fieldMap))
+	values := make([]interface{}, 0, len(s.fieldMap))
+	for _, field := range sortedKeys(s.fieldMap) {
+		fieldNames = append(fieldNames, strings.ToLower(field))
+		placeholders = append(placeholders, "?")
+		values = append(values, s.fieldMap[field])
+	}
+
+	query = append(query, "("+strings.Join(fieldNames, ", ")+")")
+	query = append(query, "VALUES ("+strings.Join(placeholders, ", ")+")")
+
+	// Determine if we need to set a TTL
+	if s.ttl > 0 {
+		query = append(query, "USING TTL ?")
+		values = append(values, int(s.ttl.Seconds()))
+	}
+
+	return strings.Join(query, " "), values
+}
+
 // UpdateStatement represents an UPDATE query to update some data in C*
 // It satisfies the Statement interface
 type UpdateStatement struct {

statement_test.go

@@ -45,6 +45,21 @@ func TestStatement(t *testing.T) {
 	}
 }
 
+func TestInsertStatement(t *testing.T) {
+	stmt := InsertStatement{keyspace: "ks1", table: "tbl1"}
+	stmt.fieldMap = map[string]interface{}{"a": "b"}
+	assert.Equal(t, "INSERT INTO ks1.tbl1 (a) VALUES (?)", stmt.Query())
+	assert.Equal(t, []interface{}{"b"}, stmt.Values())
+
+	stmt.fieldMap = map[string]interface{}{"a": "b", "c": "d"}
+	assert.Equal(t, "INSERT INTO ks1.tbl1 (a, c) VALUES (?, ?)", stmt.Query())
+	assert.Equal(t, []interface{}{"b", "d"}, stmt.Values())
+
+	stmt.ttl = 1 * time.Hour
+	assert.Equal(t, "INSERT INTO ks1.tbl1 (a, c) VALUES (?, ?) USING TTL ?", stmt.Query())
+	assert.Equal(t, []interface{}{"b", "d", 3600}, stmt.Values())
+}
+
 func TestUpdateStatement(t *testing.T) {
 	stmt := UpdateStatement{keyspace: "ks1", table: "tbl1"}
 	stmt.fieldMap = map[string]interface{}{"a": "b"}