add Batch wrapper with BindStruct method

batchx_test.go

@@ -0,0 +1,102 @@
+// Copyright (C) 2017 ScyllaDB
+// Use of this source code is governed by a ALv2-style
+// license that can be found in the LICENSE file.
+
+//go:build all || integration
+// +build all integration
+
+package gocqlx_test
+
+import (
+	"testing"
+
+	"github.com/gocql/gocql"
+	"github.com/google/go-cmp/cmp"
+	"github.com/scylladb/gocqlx/v2"
+	"github.com/scylladb/gocqlx/v2/gocqlxtest"
+	"github.com/scylladb/gocqlx/v2/qb"
+)
+
+func TestBatch(t *testing.T) {
+	t.Parallel()
+
+	cluster := gocqlxtest.CreateCluster()
+	if err := gocqlxtest.CreateKeyspace(cluster, "batch_test"); err != nil {
+		t.Fatal("create keyspace:", err)
+	}
+
+	session, err := gocqlx.WrapSession(cluster.CreateSession())
+	if err != nil {
+		t.Fatal("create session:", err)
+	}
+	t.Cleanup(func() {
+		session.Close()
+	})
+
+	basicCreateAndPopulateKeyspace(t, session, "batch_test")
+
+	song := Song{
+		ID:     mustParseUUID("60fc234a-8481-4343-93bb-72ecab404863"),
+		Title:  "La Petite Tonkinoise",
+		Album:  "Bye Bye Blackbird",
+		Artist: "Joséphine Baker",
+		Tags:   []string{"jazz"},
+		Data:   []byte("music"),
+	}
+	playlist := PlaylistItem{
+		ID:     mustParseUUID("6a6255d9-680f-4cb5-b9a2-27cf4a810344"),
+		Title:  "La Petite Tonkinoise",
+		Album:  "Bye Bye Blackbird",
+		Artist: "Joséphine Baker",
+		SongID: mustParseUUID("60fc234a-8481-4343-93bb-72ecab404863"),
+	}
+
+	insertSong := qb.Insert("batch_test.songs").
+		Columns("id", "title", "album", "artist", "tags", "data").Query(session)
+	insertPlaylist := qb.Insert("batch_test.playlists").
+		Columns("id", "title", "album", "artist", "song_id").Query(session)
+	selectSong := qb.Select("batch_test.songs").Where(qb.Eq("id")).Query(session)
+	selectPlaylist := qb.Select("batch_test.playlists").Where(qb.Eq("id")).Query(session)
+
+	t.Run("batch inserts", func(t *testing.T) {
+		t.Parallel()
+
+		type batchQry struct {
+			qry *gocqlx.Queryx
+			arg interface{}
+		}
+
+		qrys := []batchQry{
+			{qry: insertSong, arg: song},
+			{qry: insertPlaylist, arg: playlist},
+		}
+
+		b := session.NewBatch(gocql.LoggedBatch)
+		for _, qry := range qrys {
+			if err := b.BindStruct(qry.qry, qry.arg); err != nil {
+				t.Fatal("BindStruct failed:", err)
+			}
+		}
+		if err := session.ExecuteBatch(b); err != nil {
+			t.Fatal("batch execution:", err)
+		}
+
+		// verify song was inserted
+		var gotSong Song
+		if err := selectSong.BindStruct(song).Get(&gotSong); err != nil {
+			t.Fatal("select song:", err)
+		}
+		if diff := cmp.Diff(gotSong, song); diff != "" {
+			t.Errorf("expected %v song, got %v, diff: %q", song, gotSong, diff)
+		}
+
+		// verify playlist item was inserted
+		var gotPlayList PlaylistItem
+		if err := selectPlaylist.BindStruct(playlist).Get(&gotPlayList); err != nil {
+			t.Fatal("select song:", err)
+		}
+		if diff := cmp.Diff(gotPlayList, playlist); diff != "" {
+			t.Errorf("expected %v playList, got %v, diff: %q", playlist, gotPlayList, diff)
+		}
+	})
+}