func rebalanceLeases(t testing.TB, tc *testcluster.TestCluster) { kvDB := tc.Server(0).KVClient().(*client.DB) txn := client.NewTxn(context.Background(), *kvDB) rangeDescs, err := sql.AllRangeDescriptors(txn) if err != nil { t.Fatal(err) } for _, r := range rangeDescs { target := tc.Target(int(r.RangeID) % tc.NumServers()) if err := tc.TransferRangeLease(r, target); err != nil { t.Fatal(err) } } }
// verifyTables ensures that the correct number of tables were created and that // they all correspond to individual table descriptor IDs in the correct range // of values. func verifyTables( t *testing.T, tc *testcluster.TestCluster, completed chan int, expectedNumOfTables int, descIDStart int64, ) { descIDEnd := descIDStart + int64(expectedNumOfTables) usedTableIDs := make(map[sqlbase.ID]string) var count int for id := range completed { count++ tableName := fmt.Sprintf("table_%d", id) kvDB := tc.Servers[count%tc.NumServers()].KVClient().(*client.DB) tableDesc := sqlbase.GetTableDescriptor(kvDB, "test", tableName) if int64(tableDesc.ID) < descIDStart || int64(tableDesc.ID) >= descIDEnd { t.Fatalf( "table %s's ID %d is not within the expected range of %d to %d", tableName, tableDesc.ID, descIDStart, descIDEnd, ) } usedTableIDs[tableDesc.ID] = tableName } if e, a := expectedNumOfTables, len(usedTableIDs); e != a { t.Fatalf("expected %d tables created, only got %d", e, a) } kvDB := tc.Servers[count%tc.NumServers()].KVClient().(*client.DB) if descID, err := kvDB.Get(context.Background(), keys.DescIDGenerator); err != nil { t.Fatal(err) } else { if e, a := descIDEnd, descID.ValueInt(); e != a { t.Fatalf("expected next descriptor ID to be %d, got %d", e, a) } } }