/**
   * The purpose of this test is to make sure that: When a TimedOutException is thrown out from
   * invoke method, the CassandraProxyClient will try at least maxAttempts times before throwing the
   * exception to the client.
   *
   * <p>A counter from the ThriftServer in BriskErrorDaemon tracks how many times the method has
   * been called.
   *
   * @throws Exception
   */
  @Test
  public void testReconnect() throws Exception {

    EmbeddedBriskErrorServer.startBrisk();

    Brisk.Iface client =
        CassandraProxyClient.newProxyConnection(
            "localhost", DatabaseDescriptor.getRpcPort(), true, ConnectionStrategy.STICKY);
    List<KsDef> ks = client.describe_keyspaces();

    assertTrue(ks.size() > 0);

    try {
      client.get(ByteBufferUtil.EMPTY_BYTE_BUFFER, new ColumnPath("test"), ConsistencyLevel.ALL);
      fail("Expect a TimedoutException");
    } catch (TimedOutException e) {
      // This is expected.
    }

    assertEquals(
        11,
        client.get_count(
            ByteBufferUtil.EMPTY_BYTE_BUFFER,
            new ColumnParent("test"),
            new SlicePredicate(),
            ConsistencyLevel.ALL));
  }
Пример #2
0
 protected Configuration buildConfiguration() {
   Configuration conf = new Configuration();
   conf.set(CassandraClientHolder.CONF_PARAM_HOST, "localhost");
   conf.setInt(CassandraClientHolder.CONF_PARAM_PORT, DatabaseDescriptor.getRpcPort());
   conf.setBoolean(CassandraClientHolder.CONF_PARAM_FRAMED, true);
   conf.set(CassandraClientHolder.CONF_PARAM_CONNECTION_STRATEGY, "STICKY");
   conf.set("hive.metastore.warehouse.dir", "cfs:///user/hive/warehouse");
   return conf;
 }
  /**
   * When a cassandra server is not up, we expect that an IOException is thrown out from the method.
   */
  @Test
  public void testNodeDown() {
    Brisk.Iface client = null;

    try {
      client =
          CassandraProxyClient.newProxyConnection(
              "localhost", DatabaseDescriptor.getRpcPort(), true, ConnectionStrategy.STICKY);
      fail("This should error");
    } catch (IOException e) {

    }
  }
    public void init(String keyspace) {
      outputHandler.output(
          String.format(
              "Starting client (and waiting %d seconds for gossip) ...",
              StorageService.RING_DELAY / 1000));
      try {
        // Init gossip
        StorageService.instance.initClient();

        Set<InetAddress> hosts = Gossiper.instance.getLiveMembers();
        hosts.remove(FBUtilities.getLocalAddress());
        if (hosts.isEmpty())
          throw new IllegalStateException(
              "Cannot load any sstable, no live member found in the cluster");

        // Query endpoint to ranges map and schemas from thrift
        String host = hosts.iterator().next().toString().substring(1);
        int port = DatabaseDescriptor.getRpcPort();

        Cassandra.Client client = createThriftClient(host, port);
        List<TokenRange> tokenRanges = client.describe_ring(keyspace);
        List<KsDef> ksDefs = client.describe_keyspaces();

        Token.TokenFactory tkFactory = StorageService.getPartitioner().getTokenFactory();

        try {
          for (TokenRange tr : tokenRanges) {
            Range range =
                new Range(tkFactory.fromString(tr.start_token), tkFactory.fromString(tr.end_token));
            for (String ep : tr.endpoints) {
              addRangeForEndpoint(range, InetAddress.getByName(ep));
            }
          }
        } catch (UnknownHostException e) {
          throw new RuntimeException("Got an unknow host from describe_ring()", e);
        }

        for (KsDef ksDef : ksDefs) {
          Set<String> cfs = new HashSet<String>();
          for (CfDef cfDef : ksDef.cf_defs) cfs.add(cfDef.name);
          knownCfs.put(ksDef.name, cfs);
        }
      } catch (Exception e) {
        throw new RuntimeException(e);
      }
    }