int replay(ObUpsMutator& mutator) { data_len_ = 0; mutator.serialize(log_data_, BUFSIZ, data_len_); cond_.wait(lock_); if (is_error_) { return OB_ERROR; } else { return OB_SUCCESS; } }
TEST_F(TestObUpsLogMgr, test_init) { int64_t log_file_max_size = 1L << 28; ObRoleMgr role_mgr; SlaveMgr4Test sm4t; ObUpsLogMgr log_mgr1; ASSERT_EQ(OB_SUCCESS, log_mgr1.init(log_dir, log_file_max_size, sm4t.get_slave_mgr(), &role_mgr, 0)); ASSERT_EQ(1U, log_mgr1.get_replay_point()); erase_file(1); uint64_t log_file_max_id = 1100; uint64_t log_max_seq = 22000; char lf[BUFSIZ]; int64_t lf_pos = 0; ObUpsMutator mutator; ASSERT_EQ(OB_SUCCESS, mutator.serialize(lf, BUFSIZ, lf_pos)); uint64_t new_log_file_id; ObLogWriter log_writer; ASSERT_EQ(OB_SUCCESS, log_writer.init(log_dir, log_file_max_size, sm4t.get_slave_mgr(), 0)); ASSERT_EQ(OB_SUCCESS, log_writer.start_log(log_file_max_id, log_max_seq)); ASSERT_EQ(OB_SUCCESS, log_writer.switch_log_file(new_log_file_id)); ASSERT_EQ(OB_SUCCESS, log_writer.switch_log_file(new_log_file_id)); ASSERT_EQ(OB_SUCCESS, log_writer.write_log(OB_LOG_UPS_MUTATOR, lf, lf_pos)); ASSERT_EQ(OB_SUCCESS, log_writer.flush_log()); ASSERT_EQ(OB_SUCCESS, log_writer.switch_log_file(new_log_file_id)); ASSERT_EQ(OB_SUCCESS, log_writer.write_log(OB_LOG_UPS_MUTATOR, lf, lf_pos)); ASSERT_EQ(OB_SUCCESS, log_writer.flush_log()); ASSERT_EQ(OB_SUCCESS, log_writer.switch_log_file(new_log_file_id)); ASSERT_EQ(false, exists(ObUpsLogMgr::UPS_LOG_REPLAY_POINT_FILE)); ObUpsLogMgr log_mgr2; ASSERT_EQ(OB_SUCCESS, log_mgr2.init(log_dir, log_file_max_size, sm4t.get_slave_mgr(), &role_mgr, 0)); ASSERT_EQ(log_file_max_id, log_mgr2.get_replay_point()); log_mgr2.set_replay_point(100); ASSERT_EQ(100U, log_mgr2.get_replay_point()); log_mgr2.set_replay_point(log_file_max_id + 1); ASSERT_EQ(log_file_max_id + 1, log_mgr2.get_replay_point()); ObUpsTableMgr table_mgr; ASSERT_EQ(OB_SUCCESS, table_mgr.init()); ASSERT_EQ(0U, log_mgr2.get_cur_log_seq()); ASSERT_EQ(OB_SUCCESS, log_mgr2.replay_log(table_mgr)); ASSERT_NE(0U, log_mgr2.get_cur_log_seq()); ASSERT_EQ(OB_SUCCESS, log_mgr2.write_log(OB_LOG_UPS_MUTATOR, lf, lf_pos)); ASSERT_EQ(OB_SUCCESS, log_mgr2.flush_log()); ASSERT_EQ(OB_SUCCESS, log_mgr2.switch_log_file(new_log_file_id)); log_mgr2.set_replay_point(new_log_file_id); }
TEST_F(TestObUpsReplayRunnable, test_init) { int64_t log_file_max_size = 1L << 24; uint64_t log_file_max_id = 1100; uint64_t log_max_seq = 22000; char lf[BUFSIZ]; int64_t lf_pos = 0; ObUpsMutator mutator; ASSERT_EQ(OB_SUCCESS, mutator.serialize(lf, BUFSIZ, lf_pos)); SlaveMgr4Test sm4t; TestUpdateServerMain test_main; TestTableMgr &table_mgr = test_main.get_update_server().get_table_mgr(); uint64_t new_log_file_id; ObLogWriter log_writer; ASSERT_EQ(OB_SUCCESS, log_writer.init(log_dir, log_file_max_size, sm4t.get_slave_mgr(), 0)); ASSERT_EQ(OB_SUCCESS, log_writer.start_log(log_file_max_id, log_max_seq)); ASSERT_EQ(OB_SUCCESS, log_writer.switch_log_file(new_log_file_id)); ASSERT_EQ(log_file_max_id + 1, new_log_file_id); ASSERT_EQ(OB_SUCCESS, log_writer.switch_log_file(new_log_file_id)); ASSERT_EQ(OB_SUCCESS, log_writer.write_log(OB_LOG_UPS_MUTATOR, lf, lf_pos)); ASSERT_EQ(OB_SUCCESS, log_writer.flush_log()); ASSERT_EQ(OB_SUCCESS, log_writer.switch_log_file(new_log_file_id)); ASSERT_EQ(OB_SUCCESS, log_writer.write_log(OB_LOG_UPS_MUTATOR, lf, lf_pos)); ASSERT_EQ(OB_SUCCESS, log_writer.flush_log()); ASSERT_EQ(OB_SUCCESS, log_writer.switch_log_file(new_log_file_id)); ObRoleMgr role_mgr; role_mgr.set_role(ObRoleMgr::SLAVE); ObUpsReplayRunnable replay_thread0; ASSERT_EQ(OB_SUCCESS, replay_thread0.init(log_dir, log_file_max_id, &role_mgr, true)); replay_thread0.start(); usleep(1000); ASSERT_EQ(lf_pos, table_mgr.data_len_); ASSERT_EQ(0, memcmp(table_mgr.log_data_, lf, lf_pos)); table_mgr.cond_.signal(); usleep(1000); ASSERT_EQ(lf_pos, table_mgr.data_len_); ASSERT_EQ(0, memcmp(table_mgr.log_data_, lf, lf_pos)); table_mgr.cond_.signal(); usleep(1000); replay_thread0.stop(); replay_thread0.wait(); // test replay error ObUpsReplayRunnable replay_thread1; //replay_thread1.ups_main_ = &test_main; ASSERT_EQ(OB_SUCCESS, replay_thread1.init(log_dir, log_file_max_id, &role_mgr, true)); replay_thread1.start(); usleep(1000); ASSERT_EQ(lf_pos, table_mgr.data_len_); ASSERT_EQ(0, memcmp(table_mgr.log_data_, lf, lf_pos)); table_mgr.is_error_ = true; table_mgr.cond_.signal(); usleep(1000); ASSERT_EQ(lf_pos, table_mgr.data_len_); ASSERT_EQ(0, memcmp(table_mgr.log_data_, lf, lf_pos)); table_mgr.cond_.signal(); replay_thread1.wait(); table_mgr.is_error_ = false; // test open log file error ObLogWriter *log_writer2 = new ObLogWriter(); ASSERT_EQ(OB_SUCCESS, log_writer2->init(log_dir, log_file_max_size, sm4t.get_slave_mgr(), 0)); ASSERT_EQ(OB_SUCCESS, log_writer2->start_log(log_file_max_id - 10, log_max_seq)); ASSERT_EQ(OB_SUCCESS, log_writer2->write_log(OB_LOG_UPS_MUTATOR, lf, lf_pos)); ASSERT_EQ(OB_SUCCESS, log_writer2->flush_log()); ASSERT_EQ(OB_SUCCESS, log_writer2->switch_log_file(new_log_file_id)); delete log_writer2; char cmd[100]; snprintf(cmd, 100, "rm %s/%lu", log_dir, log_file_max_id - 9); system(cmd); ObUpsReplayRunnable replay_thread2; //replay_thread2.ups_main_ = &test_main; ASSERT_EQ(OB_SUCCESS, replay_thread2.init(log_dir, log_file_max_id - 10, &role_mgr, true)); replay_thread2.start(); usleep(1000); ASSERT_EQ(lf_pos, table_mgr.data_len_); ASSERT_EQ(0, memcmp(table_mgr.log_data_, lf, lf_pos)); table_mgr.cond_.signal(); replay_thread2.wait(); }