@Override public void batch_mutate(String kvsMethodName, Map<ByteBuffer, Map<String, List<Mutation>>> mutation_map, ConsistencyLevel consistency_level) throws InvalidRequestException, UnavailableException, TimedOutException, TException { int numberOfRowsMutated = mutation_map.size(); try (CloseableTrace trace = startLocalTrace("client.batch_mutate(number of mutations {}, consistency {})" + " on kvs.{}", numberOfRowsMutated, consistency_level, kvsMethodName)) { client.batch_mutate(kvsMethodName, mutation_map, consistency_level); } }
private CassandraClient instrumentClient(Client rawClient) { CassandraClient client = new CassandraClientImpl(rawClient); client = new ProfilingCassandraClient(client); client = new TracingCassandraClient(client); // TODO(ssouza): use the kvsMethodName to tag the timers. client = AtlasDbMetrics.instrument(metricsManager.getRegistry(), CassandraClient.class, client); client = new InstrumentedCassandraClient(client, metricsManager.getTaggedRegistry()); client = QosCassandraClient.instrumentWithMetrics(client, metricsManager); return client; }
private CassandraClient instrumentClient(Client rawClient) { CassandraClient client = new CassandraClientImpl(rawClient); client = new ProfilingCassandraClient(client); client = new TracingCassandraClient(client); // TODO(ssouza): use the kvsMethodName to tag the timers. client = AtlasDbMetrics.instrument(metricsManager.getRegistry(), CassandraClient.class, client); client = new InstrumentedCassandraClient(client, metricsManager.getTaggedRegistry()); client = QosCassandraClient.instrumentWithMetrics(client, metricsManager); return client; }
@Override public void remove(String kvsMethodName, TableReference tableRef, byte[] row, long timestamp, ConsistencyLevel consistency_level) throws InvalidRequestException, UnavailableException, TimedOutException, TException { try (CloseableTrace trace = startLocalTrace( "client.remove(consistency {}) on kvs.{}", consistency_level, kvsMethodName)) { client.remove(kvsMethodName, tableRef, row, timestamp, consistency_level); } }
@Override public ColumnOrSuperColumn get(TableReference tableReference, ByteBuffer key, byte[] column, ConsistencyLevel consistency_level) throws InvalidRequestException, NotFoundException, UnavailableException, TimedOutException, TException { try (CloseableTrace trace = startLocalTrace("client.get(table {}, consistency {})", LoggingArgs.safeTableOrPlaceholder(tableReference), consistency_level)) { return client.get(tableReference, key, column, consistency_level); } }
@Override public CqlResult execute_cql3_query(CqlQuery cqlQuery, Compression compression, ConsistencyLevel consistency) throws InvalidRequestException, UnavailableException, TimedOutException, SchemaDisagreementException, TException { try (CloseableTrace trace = startLocalTrace("cqlExecutor.execute_cql3_query(query {})", cqlQuery.getLazySafeLoggableObject())) { return client.execute_cql3_query(cqlQuery, compression, consistency); } }
@Override public Map<ByteBuffer, List<ColumnOrSuperColumn>> multiget_slice(String kvsMethodName, TableReference tableRef, List<ByteBuffer> keys, SlicePredicate predicate, ConsistencyLevel consistency_level) throws InvalidRequestException, UnavailableException, TimedOutException, TException { int numberOfKeys = keys.size(); int numberOfColumns = predicate.slice_range.count; try (CloseableTrace trace = startLocalTrace( "client.multiget_slice(table {}, number of keys {}, number of columns {}, consistency {}) on kvs.{}", LoggingArgs.safeTableOrPlaceholder(tableRef), numberOfKeys, numberOfColumns, consistency_level, kvsMethodName)) { return client.multiget_slice(kvsMethodName, tableRef, keys, predicate, consistency_level); } }
@Override public CASResult cas(TableReference tableReference, ByteBuffer key, List<Column> expected, List<Column> updates, ConsistencyLevel serial_consistency_level, ConsistencyLevel commit_consistency_level) throws InvalidRequestException, UnavailableException, TimedOutException, TException { try (CloseableTrace trace = startLocalTrace("client.cas(table {})", LoggingArgs.safeTableOrPlaceholder(tableReference))) { return client.cas(tableReference, key, expected, updates, serial_consistency_level, commit_consistency_level); } }
@Override public List<KeySlice> get_range_slices(String kvsMethodName, TableReference tableRef, SlicePredicate predicate, KeyRange range, ConsistencyLevel consistency_level) throws InvalidRequestException, UnavailableException, TimedOutException, TException { int numberOfKeys = predicate.slice_range.count; int numberOfColumns = range.count; try (CloseableTrace trace = startLocalTrace( "client.get_range_slices(table {}, number of keys {}, number of columns {}, consistency {}) on kvs.{}", LoggingArgs.safeTableOrPlaceholder(tableRef), numberOfKeys, numberOfColumns, consistency_level, kvsMethodName)) { return client.get_range_slices(kvsMethodName, tableRef, predicate, range, consistency_level); } }
@Override public void remove(String kvsMethodName, TableReference tableRef, byte[] row, long timestamp, ConsistencyLevel consistency_level) throws InvalidRequestException, UnavailableException, TimedOutException, TException { try (CloseableTrace trace = startLocalTrace( "client.remove(consistency {}) on kvs.{}", consistency_level, kvsMethodName)) { client.remove(kvsMethodName, tableRef, row, timestamp, consistency_level); } }
@Override public void batch_mutate(String kvsMethodName, Map<ByteBuffer, Map<String, List<Mutation>>> mutation_map, ConsistencyLevel consistency_level) throws InvalidRequestException, UnavailableException, TimedOutException, TException { int numberOfRowsMutated = mutation_map.size(); try (CloseableTrace trace = startLocalTrace("client.batch_mutate(number of mutations {}, consistency {})" + " on kvs.{}", numberOfRowsMutated, consistency_level, kvsMethodName)) { client.batch_mutate(kvsMethodName, mutation_map, consistency_level); } }
@Override public ColumnOrSuperColumn get(TableReference tableReference, ByteBuffer key, byte[] column, ConsistencyLevel consistency_level) throws InvalidRequestException, NotFoundException, UnavailableException, TimedOutException, TException { try (CloseableTrace trace = startLocalTrace("client.get(table {}, consistency {})", LoggingArgs.safeTableOrPlaceholder(tableReference), consistency_level)) { return client.get(tableReference, key, column, consistency_level); } }
@Override public Map<ByteBuffer, List<ColumnOrSuperColumn>> multiget_slice(String kvsMethodName, TableReference tableRef, List<ByteBuffer> keys, SlicePredicate predicate, ConsistencyLevel consistency_level) throws InvalidRequestException, UnavailableException, TimedOutException, TException { int numberOfKeys = keys.size(); int numberOfColumns = predicate.slice_range.count; try (CloseableTrace trace = startLocalTrace( "client.multiget_slice(table {}, number of keys {}, number of columns {}, consistency {}) on kvs.{}", LoggingArgs.safeTableOrPlaceholder(tableRef), numberOfKeys, numberOfColumns, consistency_level, kvsMethodName)) { return client.multiget_slice(kvsMethodName, tableRef, keys, predicate, consistency_level); } }
@Override public CqlResult execute_cql3_query(CqlQuery cqlQuery, Compression compression, ConsistencyLevel consistency) throws InvalidRequestException, UnavailableException, TimedOutException, SchemaDisagreementException, TException { try (CloseableTrace trace = startLocalTrace("cqlExecutor.execute_cql3_query(query {})", cqlQuery.getLazySafeLoggableObject())) { return client.execute_cql3_query(cqlQuery, compression, consistency); } }
@Override public List<KeySlice> get_range_slices(String kvsMethodName, TableReference tableRef, SlicePredicate predicate, KeyRange range, ConsistencyLevel consistency_level) throws InvalidRequestException, UnavailableException, TimedOutException, TException { int numberOfKeys = predicate.slice_range.count; int numberOfColumns = range.count; try (CloseableTrace trace = startLocalTrace( "client.get_range_slices(table {}, number of keys {}, number of columns {}, consistency {}) on kvs.{}", LoggingArgs.safeTableOrPlaceholder(tableRef), numberOfKeys, numberOfColumns, consistency_level, kvsMethodName)) { return client.get_range_slices(kvsMethodName, tableRef, predicate, range, consistency_level); } }
@Override public CASResult cas(TableReference tableReference, ByteBuffer key, List<Column> expected, List<Column> updates, ConsistencyLevel serial_consistency_level, ConsistencyLevel commit_consistency_level) throws InvalidRequestException, UnavailableException, TimedOutException, TException { try (CloseableTrace trace = startLocalTrace("client.cas(table {})", LoggingArgs.safeTableOrPlaceholder(tableReference))) { return client.cas(tableReference, key, expected, updates, serial_consistency_level, commit_consistency_level); } }