(keyspace_name)
| 51 | |
| 52 | |
| 53 | def setup_connection(keyspace_name): |
| 54 | connection.setup([CASSANDRA_IP], |
| 55 | consistency=ConsistencyLevel.ONE, |
| 56 | protocol_version=PROTOCOL_VERSION, |
| 57 | allow_beta_protocol_version=ALLOW_BETA_PROTOCOL, |
| 58 | default_keyspace=keyspace_name) |
| 59 | |
| 60 | |
| 61 | class StatementCounter(object): |