forked from apache/cassandra-gocql-driver
-
Notifications
You must be signed in to change notification settings - Fork 59
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
tablet_integration_test.go: Integration tests checking if TokenAwareHostPolicy and shard awareness works correctly when using tablets, and if adding new table changes tablets table. tablet_test.go: Unit tests checking if searching in tablets list works correctly.
- Loading branch information
1 parent
39500a7
commit a98a393
Showing
6 changed files
with
659 additions
and
14 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,134 @@ | ||
| //go:build tablet | ||
| // +build tablet | ||
|
|
||
| package gocql | ||
|
|
||
| import ( | ||
| "bytes" | ||
| "context" | ||
| "fmt" | ||
| "regexp" | ||
| "strings" | ||
| "testing" | ||
| ) | ||
|
|
||
| // Check if TokenAwareHostPolicy works correctly when using tablets | ||
| func TestTablets(t *testing.T) { | ||
| cluster := createMultiNodeCluster() | ||
|
|
||
| fallback := RoundRobinHostPolicy() | ||
| cluster.PoolConfig.HostSelectionPolicy = TokenAwareHostPolicy(fallback) | ||
|
|
||
| session := createSessionFromMultiNodeCluster(cluster, t) | ||
| defer session.Close() | ||
|
|
||
| if err := createTable(session, fmt.Sprintf(`CREATE TABLE %s.%s (pk int, ck int, v int, PRIMARY KEY (pk, ck)); | ||
| `, "test1", "table1")); err != nil { | ||
| panic(fmt.Sprintf("unable to create table: %v", err)) | ||
| } | ||
|
|
||
| hosts, _, err := session.hostSource.GetHosts() | ||
| assertTrue(t, "err == nil", err == nil) | ||
|
|
||
| hostAddresses := []string{} | ||
| for _, host := range hosts { | ||
| hostAddresses = append(hostAddresses, host.connectAddress.String()) | ||
| } | ||
|
|
||
| ctx := context.Background() | ||
|
|
||
| i := 0 | ||
| for i < 50 { | ||
| i = i + 1 | ||
| err = session.Query(`INSERT INTO test1.table1 (pk, ck, v) VALUES (?, ?, ?);`, i, i%5, i%2).WithContext(ctx).Exec() | ||
| if err != nil { | ||
| t.Fatal(err) | ||
| } | ||
| } | ||
|
|
||
| i = 0 | ||
| for i < 50 { | ||
| i = i + 1 | ||
|
|
||
| var pk int | ||
| var ck int | ||
| var v int | ||
|
|
||
| buf := &bytes.Buffer{} | ||
| trace := NewTraceWriter(session, buf) | ||
|
|
||
| err = session.Query(`SELECT pk, ck, v FROM test1.table1 WHERE pk = ?;`, i).WithContext(ctx).Consistency(One).Trace(trace).Scan(&pk, &ck, &v) | ||
| if err != nil { | ||
| t.Fatal(err) | ||
| } | ||
|
|
||
| queriedHosts := 0 | ||
| for _, hostAddress := range hostAddresses { | ||
| if strings.Contains(buf.String(), hostAddress) { | ||
| queriedHosts = queriedHosts + 1 | ||
| } | ||
| } | ||
|
|
||
| assertEqual(t, "queriedHosts", 1, queriedHosts) | ||
| } | ||
| } | ||
|
|
||
| // Check if shard awareness works correctly when using tablets | ||
| func TestTabletsShardAwareness(t *testing.T) { | ||
| cluster := createMultiNodeCluster() | ||
|
|
||
| fallback := RoundRobinHostPolicy() | ||
| cluster.PoolConfig.HostSelectionPolicy = TokenAwareHostPolicy(fallback) | ||
|
|
||
| session := createSessionFromMultiNodeCluster(cluster, t) | ||
| defer session.Close() | ||
|
|
||
| if err := createTable(session, fmt.Sprintf(`CREATE TABLE %s.%s (pk int, ck int, v int, PRIMARY KEY (pk, ck)); | ||
| `, "test1", "table_shard")); err != nil { | ||
| panic(fmt.Sprintf("unable to create table: %v", err)) | ||
| } | ||
|
|
||
| ctx := context.Background() | ||
|
|
||
| i := 0 | ||
| for i < 50 { | ||
| i = i + 1 | ||
| err := session.Query(`INSERT INTO test1.table_shard (pk, ck, v) VALUES (?, ?, ?);`, i, i%5, i%2).WithContext(ctx).Exec() | ||
| if err != nil { | ||
| t.Fatal(err) | ||
| } | ||
| } | ||
|
|
||
| i = 0 | ||
| for i < 50 { | ||
| i = i + 1 | ||
|
|
||
| var pk int | ||
| var ck int | ||
| var v int | ||
|
|
||
| buf := &bytes.Buffer{} | ||
| trace := NewTraceWriter(session, buf) | ||
|
|
||
| err := session.Query(`SELECT pk, ck, v FROM test1.table_shard WHERE pk = ?;`, i).WithContext(ctx).Consistency(One).Trace(trace).Scan(&pk, &ck, &v) | ||
| if err != nil { | ||
| t.Fatal(err) | ||
| } | ||
|
|
||
| re := regexp.MustCompile(`\[shard .*\]`) | ||
|
|
||
| shards := re.FindAllString(buf.String(), -1) | ||
|
|
||
| // find duplicates to check how many shards are used | ||
| allShards := make(map[string]bool) | ||
| shardList := []string{} | ||
| for _, item := range shards { | ||
| if _, value := allShards[item]; !value { | ||
| allShards[item] = true | ||
| shardList = append(shardList, item) | ||
| } | ||
| } | ||
|
|
||
| assertEqual(t, "shardList", 1, len(shardList)) | ||
| } | ||
| } |
Oops, something went wrong.