cdrs 2.2.0

Cassandra DB driver written in Rust
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63

#[cfg(feature = "e2e-tests")]
use cdrs::authenticators::NoneAuthenticator;
#[cfg(feature = "e2e-tests")]
use cdrs::cluster::session::{new as new_session, Session};
#[cfg(feature = "e2e-tests")]
use cdrs::cluster::{ClusterTcpConfig, NodeTcpConfigBuilder, TcpConnectionPool};
#[cfg(feature = "e2e-tests")]
use cdrs::error::Result;
#[cfg(feature = "e2e-tests")]
use cdrs::load_balancing::RoundRobin;
#[cfg(feature = "e2e-tests")]
use cdrs::query::QueryExecutor;
#[cfg(feature = "e2e-tests")]
use regex::Regex;

#[cfg(feature = "e2e-tests")]
const ADDR: &'static str = "127.0.0.1:9042";

#[cfg(feature = "e2e-tests")]
type CurrentSession = Session<RoundRobin<TcpConnectionPool<NoneAuthenticator>>>;

#[cfg(feature = "e2e-tests")]
pub fn setup(create_table_cql: &'static str) -> Result<CurrentSession> {
  setup_multiple(&[create_table_cql])
}

#[cfg(feature = "e2e-tests")]
pub fn setup_multiple(create_cqls: &[&'static str]) -> Result<CurrentSession> {
  let node = NodeTcpConfigBuilder::new(ADDR, NoneAuthenticator {}).build();
  let cluster_config = ClusterTcpConfig(vec![node]);
  let lb = RoundRobin::new();
  let session = new_session(&cluster_config, lb).expect("session should be created");
  let re_table_name = Regex::new(r"CREATE TABLE IF NOT EXISTS (\w+\.\w+)").unwrap();

  let create_keyspace_query = "CREATE KEYSPACE IF NOT EXISTS cdrs_test WITH \
                               replication = {'class': 'SimpleStrategy', 'replication_factor': 1} \
                               AND durable_writes = false";
  session.query(create_keyspace_query)?;

  for create_cql in create_cqls.iter() {
    let table_name = re_table_name
      .captures(create_cql)
      .map(|cap| cap.get(1).unwrap().as_str());

    // Re-using tables is a lot faster than creating/dropping them for every test.
    // But if table definitions change while editing tests
    // the old tables need to be dropped. For example by uncommenting the following lines.
    // if let Some(table_name) = table_name {
    //     let cql = format!("DROP TABLE IF EXISTS {}", table_name);
    //     let query = QueryBuilder::new(cql).finalize();
    //     session.query(query, true, true)?;
    // }

    session.query(create_cql.to_owned())?;

    if let Some(table_name) = table_name {
      let cql = format!("TRUNCATE TABLE {}", table_name);
      session.query(cql)?;
    }
  }

  Ok(session)
}