9d4c7842c1
Conflicts: cassandra/cluster.py cassandra/connection.py cassandra/io/asyncorereactor.py cassandra/io/libevreactor.py setup.py tests/integration/long/test_large_data.py
110 lines
3.2 KiB
Python
110 lines
3.2 KiB
Python
from __future__ import print_function
|
|
import logging
|
|
import time
|
|
|
|
from collections import defaultdict
|
|
|
|
from cassandra.query import named_tuple_factory
|
|
|
|
from tests.integration import get_node
|
|
|
|
|
|
log = logging.getLogger(__name__)
|
|
|
|
|
|
class CoordinatorStats():
|
|
|
|
def __init__(self):
|
|
self.coordinator_counts = defaultdict(int)
|
|
|
|
def add_coordinator(self, future):
|
|
coordinator = future._current_host.address
|
|
self.coordinator_counts[coordinator] += 1
|
|
|
|
if future._errors:
|
|
log.error('future._errors: %s', future._errors)
|
|
future.result()
|
|
|
|
def reset_counts(self):
|
|
self.coordinator_counts = defaultdict(int)
|
|
|
|
def assert_query_count_equals(self, testcase, node, expected):
|
|
ip = '127.0.0.%d' % node
|
|
if self.coordinator_counts[ip] != expected:
|
|
testcase.fail('Expected %d queries to %s, but got %d. Query counts: %s' % (
|
|
expected, ip, self.coordinator_counts[ip], dict(self.coordinator_counts)))
|
|
|
|
|
|
def create_schema(session, keyspace, simple_strategy=True,
|
|
replication_factor=1, replication_strategy=None):
|
|
row_factory = session.row_factory
|
|
session.row_factory = named_tuple_factory
|
|
|
|
results = session.execute(
|
|
'SELECT keyspace_name FROM system.schema_keyspaces')
|
|
existing_keyspaces = [row[0] for row in results]
|
|
if keyspace in existing_keyspaces:
|
|
session.execute('DROP KEYSPACE %s' % keyspace, timeout=20)
|
|
|
|
if simple_strategy:
|
|
ddl = "CREATE KEYSPACE %s WITH replication" \
|
|
" = {'class': 'SimpleStrategy', 'replication_factor': '%s'}"
|
|
session.execute(ddl % (keyspace, replication_factor), timeout=10)
|
|
else:
|
|
if not replication_strategy:
|
|
raise Exception('replication_strategy is not set')
|
|
|
|
ddl = "CREATE KEYSPACE %s" \
|
|
" WITH replication = { 'class' : 'NetworkTopologyStrategy', %s }"
|
|
session.execute(ddl % (keyspace, str(replication_strategy)[1:-1]), timeout=10)
|
|
|
|
ddl = 'CREATE TABLE %s.cf (k int PRIMARY KEY, i int)'
|
|
session.execute(ddl % keyspace, timeout=10)
|
|
session.execute('USE %s' % keyspace)
|
|
|
|
session.row_factory = row_factory
|
|
|
|
|
|
def start(node):
|
|
get_node(node).start()
|
|
|
|
|
|
def stop(node):
|
|
get_node(node).stop()
|
|
|
|
|
|
def force_stop(node):
|
|
log.debug("Forcing stop of node %s", node)
|
|
get_node(node).stop(wait=False, gently=False)
|
|
log.debug("Node %s was stopped", node)
|
|
|
|
|
|
def ring(node):
|
|
print('From node%s:' % node)
|
|
get_node(node).nodetool('ring')
|
|
|
|
|
|
def wait_for_up(cluster, node, wait=True):
|
|
while True:
|
|
host = cluster.metadata.get_host('127.0.0.%s' % node)
|
|
time.sleep(0.1)
|
|
if host and host.is_up:
|
|
# BUG: shouldn't have to, but we do
|
|
if wait:
|
|
time.sleep(5)
|
|
return
|
|
|
|
|
|
def wait_for_down(cluster, node, wait=True):
|
|
log.debug("Waiting for node %s to be down", node)
|
|
while True:
|
|
host = cluster.metadata.get_host('127.0.0.%s' % node)
|
|
time.sleep(0.1)
|
|
if not host or not host.is_up:
|
|
# BUG: shouldn't have to, but we do
|
|
if wait:
|
|
log.debug("Sleeping 5s until host is up")
|
|
time.sleep(5)
|
|
log.debug("Done waiting for node %s to be down", node)
|
|
return
|