feat: add input helpers

proto/block.go

@@ -2,6 +2,7 @@ package proto
 
 import (
 	"fmt"
+	"strings"
 
 	"github.com/go-faster/errors"
 )
@@ -62,6 +63,23 @@ func (i *BlockInfo) Decode(r *Reader) error {
 	}
 }
 
+// Input of query.
+type Input []InputColumn
+
+// Into returns INSERT INTO table (c0, c..., cn) VALUES query.
+func (i Input) Into(table string) string {
+	return fmt.Sprintf("INSERT INTO %s %s VALUES", table, i.Columns())
+}
+
+// Columns returns "(foo, bar, baz)" formatted list of Input column names.
+func (i Input) Columns() string {
+	var names []string
+	for _, v := range i {
+		names = append(names, v.Name)
+	}
+	return fmt.Sprintf("(%s)", strings.Join(names, ", "))
+}
+
 type InputColumn struct {
 	Name string
 	Data ColInput

query.go

@@ -121,7 +121,7 @@ type Query struct {
 	QuotaKey string
 
 	// Input columns for INSERT operations.
-	Input []proto.InputColumn
+	Input proto.Input
 	// OnInput is called to allow ingesting more data to Input.
 	//
 	// The io.EOF reports that no more input should be ingested.

query_test.go

@@ -47,6 +47,37 @@ func TestClient_Query(t *testing.T) {
 		require.Len(t, data, 4)
 		require.Equal(t, data, gotData)
 	})
+	t.Run("InsertHelper", func(t *testing.T) {
+		t.Parallel()
+		conn := Conn(t)
+
+		// Create table, no data fetch.
+		createTable := Query{
+			Body: "CREATE TABLE test_table (id UInt8) ENGINE = MergeTree ORDER BY id",
+		}
+		require.NoError(t, conn.Query(ctx, createTable), "create table")
+
+		data := proto.ColUInt8{1, 2, 3, 4}
+		input := proto.Input{
+			{Name: "id", Data: &data},
+		}
+		insertQuery := Query{
+			Body:  input.Into("test_table"),
+			Input: input,
+		}
+		require.NoError(t, conn.Query(ctx, insertQuery), "insert")
+
+		var gotData proto.ColUInt8
+		selectData := Query{
+			Body: "SELECT * FROM test_table",
+			Result: proto.Results{
+				{Name: "id", Data: &gotData},
+			},
+		}
+		require.NoError(t, conn.Query(ctx, selectData), "select")
+		require.Len(t, data, 4)
+		require.Equal(t, data, gotData)
+	})
 	t.Run("InsertEnum8", func(t *testing.T) {
 		t.Parallel()
 		conn := Conn(t)