forked from datastax/cpp-driver
-
Notifications
You must be signed in to change notification settings - Fork 0
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
- Loading branch information
Showing
8 changed files
with
188 additions
and
20 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,128 @@ | ||
| #define BOOST_TEST_DYN_LINK | ||
| #ifdef STAND_ALONE | ||
| # define BOOST_TEST_MODULE cassandra | ||
| #endif | ||
|
|
||
| #include "cassandra.h" | ||
| #include "test_utils.hpp" | ||
|
|
||
| #include <algorithm> | ||
| #include <future> | ||
|
|
||
| #include <boost/test/unit_test.hpp> | ||
| #include <boost/test/debug.hpp> | ||
| #include <boost/lexical_cast.hpp> | ||
| #include <boost/cstdint.hpp> | ||
| #include <boost/format.hpp> | ||
| #include <boost/thread.hpp> | ||
|
|
||
| struct STRESS_CCM_SETUP : test_utils::CCM_SETUP { | ||
| STRESS_CCM_SETUP() : CCM_SETUP(3, 0) {} | ||
| }; | ||
|
|
||
| BOOST_FIXTURE_TEST_SUITE(stress, STRESS_CCM_SETUP) | ||
|
|
||
| void bind_and_execute_insert(CassSession* session, CassStatement* statement) { | ||
| std::chrono::system_clock::time_point now(std::chrono::system_clock::now()); | ||
| std::chrono::milliseconds event_time(std::chrono::duration_cast<std::chrono::milliseconds>(now.time_since_epoch())); | ||
| std::string text_sample(test_utils::string_from_time_point(now)); | ||
|
|
||
| BOOST_REQUIRE(cass_statement_bind_uuid(statement, 0, | ||
| test_utils::generate_time_uuid().uuid) == CASS_OK); | ||
| BOOST_REQUIRE(cass_statement_bind_int64(statement, 1, | ||
| event_time.count()) == CASS_OK); | ||
| BOOST_REQUIRE(cass_statement_bind_string(statement, 2, | ||
| cass_string_init2(text_sample.data(), text_sample.size())) == CASS_OK); | ||
|
|
||
| test_utils::StackPtr<CassFuture> future(cass_session_execute(session, statement)); | ||
| cass_future_wait(future.get()); | ||
| CassError code = cass_future_error_code(future.get()); | ||
| if(code != CASS_OK && code != CASS_ERROR_LIB_REQUEST_TIMED_OUT) { // Timeout is okay | ||
| CassString message = cass_future_error_message(future.get()); | ||
| BOOST_FAIL("Error occured during insert '" << std::string(message.data, message.length) << "' (" << code << ")"); | ||
| } | ||
| } | ||
|
|
||
| void insert_task(CassSession* session, const std::string& query, CassConsistency consistency, int rows_per_id) { | ||
| for(int i = 0; i < rows_per_id; ++i) { | ||
| test_utils::StackPtr<CassStatement> statement(cass_statement_new(cass_string_init(query.c_str()), 3, consistency)); | ||
| bind_and_execute_insert(session, statement.get()); | ||
| } | ||
| } | ||
|
|
||
| void insert_prepared_task(CassSession* session, const CassPrepared* prepared, CassConsistency consistency, int rows_per_id) { | ||
| for(int i = 0; i < rows_per_id; ++i) { | ||
| test_utils::StackPtr<CassStatement> statement(cass_prepared_bind(prepared, 3, consistency)); | ||
| bind_and_execute_insert(session, statement.get()); | ||
| } | ||
| } | ||
|
|
||
| void select_task(CassSession* session, const std::string& query, CassConsistency consistency, int num_iterations) { | ||
| test_utils::StackPtr<CassStatement> statement(cass_statement_new(cass_string_init(query.c_str()), 0, consistency)); | ||
| for(int i = 0; i < num_iterations; ++i) { | ||
| test_utils::StackPtr<CassFuture> future(cass_session_execute(session, statement.get())); | ||
| cass_future_wait(future.get()); | ||
|
|
||
| CassError code = cass_future_error_code(future.get()); | ||
| if(code != CASS_OK | ||
| && code != CASS_ERROR_LIB_REQUEST_TIMED_OUT | ||
| && code != CASS_ERROR_SERVER_READ_TIMEOUT) { // Timeout is okay | ||
| CassString message = cass_future_error_message(future.get()); | ||
| BOOST_FAIL("Error occured during select '" << std::string(message.data, message.length) << "' (" << code << ")"); | ||
| } | ||
|
|
||
| if(code == CASS_OK) { | ||
| test_utils::StackPtr<const CassResult> result(cass_future_get_result(future.get())); | ||
| BOOST_REQUIRE(cass_result_row_count(result.get()) > 0); | ||
| } | ||
| } | ||
| } | ||
|
|
||
| BOOST_AUTO_TEST_CASE(insert_and_select) | ||
| { | ||
| test_utils::StackPtr<CassFuture> session_future; | ||
| test_utils::StackPtr<CassSession> session(cass_cluster_connect(cluster, session_future.address_of())); | ||
| test_utils::wait_and_check_error(session_future.get()); | ||
|
|
||
| test_utils::execute_query(session.get(), "CREATE KEYSPACE tester WITH replication = {'class': 'SimpleStrategy', 'replication_factor' : 3};"); | ||
| test_utils::execute_query(session.get(), "USE tester;"); | ||
|
|
||
| std::string table_name = str(boost::format("table_%s") % test_utils::generate_unique_str()); | ||
|
|
||
| test_utils::execute_query(session.get(), str(boost::format(test_utils::CREATE_TABLE_TIME_SERIES) % table_name)); | ||
|
|
||
| std::string insert_query = str(boost::format("INSERT INTO %s (id, event_time, text_sample) VALUES (?, ?, ?)") % table_name); | ||
| std::string select_query = str(boost::format("SELECT * FROM %s LIMIT 10000") % table_name); | ||
|
|
||
| test_utils::StackPtr<CassFuture> prepared_future(cass_session_prepare(session.get(), | ||
| cass_string_init2(insert_query.data(), insert_query.size()))); | ||
|
|
||
| test_utils::wait_and_check_error(prepared_future.get()); | ||
| test_utils::StackPtr<const CassPrepared> prepared(cass_future_get_prepared(prepared_future.get())); | ||
|
|
||
| int rows_per_id = 100; | ||
| int num_iterations = 10; | ||
|
|
||
| std::vector<std::future<void>> tasks; | ||
|
|
||
| for(int i = 0; i < 10; ++i) { | ||
| tasks.push_back(std::async(std::launch::async, insert_task, session.get(), insert_query, CASS_CONSISTENCY_QUORUM, rows_per_id)); | ||
| tasks.push_back(std::async(std::launch::async, select_task, session.get(), select_query, CASS_CONSISTENCY_QUORUM, num_iterations)); | ||
| tasks.push_back(std::async(std::launch::async, insert_prepared_task, session.get(), prepared.get(), CASS_CONSISTENCY_QUORUM, rows_per_id)); | ||
|
|
||
| tasks.push_back(std::async(std::launch::async, select_task, session.get(), select_query, CASS_CONSISTENCY_QUORUM, num_iterations)); | ||
| tasks.push_back(std::async(std::launch::async, insert_task, session.get(), insert_query, CASS_CONSISTENCY_QUORUM, rows_per_id)); | ||
| tasks.push_back(std::async(std::launch::async, insert_prepared_task, session.get(), prepared.get(), CASS_CONSISTENCY_QUORUM, rows_per_id)); | ||
| tasks.push_back(std::async(std::launch::async, insert_task, session.get(), insert_query, CASS_CONSISTENCY_QUORUM, rows_per_id)); | ||
|
|
||
| tasks.push_back(std::async(std::launch::async, insert_prepared_task, session.get(), prepared.get(), CASS_CONSISTENCY_QUORUM, rows_per_id)); | ||
| tasks.push_back(std::async(std::launch::async, insert_prepared_task, session.get(), prepared.get(), CASS_CONSISTENCY_QUORUM, rows_per_id)); | ||
| tasks.push_back(std::async(std::launch::async, select_task, session.get(), select_query, CASS_CONSISTENCY_QUORUM, num_iterations)); | ||
| } | ||
|
|
||
| for(auto& task : tasks) { | ||
| task.wait(); | ||
| } | ||
| } | ||
|
|
||
| BOOST_AUTO_TEST_SUITE_END() |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters