| #pragma once |
| |
| #ifndef GEODE_INTEGRATION_TEST_THINCLIENTFAILOVER2_H_ |
| #define GEODE_INTEGRATION_TEST_THINCLIENTFAILOVER2_H_ |
| |
| /* |
| * Licensed to the Apache Software Foundation (ASF) under one or more |
| * contributor license agreements. See the NOTICE file distributed with |
| * this work for additional information regarding copyright ownership. |
| * The ASF licenses this file to You under the Apache License, Version 2.0 |
| * (the "License"); you may not use this file except in compliance with |
| * the License. You may obtain a copy of the License at |
| * |
| * http://www.apache.org/licenses/LICENSE-2.0 |
| * |
| * Unless required by applicable law or agreed to in writing, software |
| * distributed under the License is distributed on an "AS IS" BASIS, |
| * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
| * See the License for the specific language governing permissions and |
| * limitations under the License. |
| */ |
| |
| #include "fw_dunit.hpp" |
| |
| #include <string> |
| |
| #define ThinClientFailover2 "DistOps" |
| #define ROOT_SCOPE DISTRIBUTED_ACK |
| |
| #include "CacheHelper.hpp" |
| |
| namespace { // NOLINT(google-build-namespaces) |
| |
| using apache::geode::client::CacheableKey; |
| using apache::geode::client::CacheableString; |
| using apache::geode::client::CacheHelper; |
| |
| /* |
| * This test for client failover with client notification. |
| */ |
| bool isLocalServer = false; |
| |
| CacheHelper* cacheHelper = nullptr; |
| |
| #define CLIENT1 s1p1 |
| #define CLIENT2 s1p2 |
| #define SERVER1 s2p1 |
| #define SERVER2 s2p2 |
| static bool isLocator = false; |
| static int numberOfLocators = 0; |
| const std::string locatorsG = |
| CacheHelper::getLocatorHostPort(isLocator, isLocalServer, numberOfLocators); |
| #include "LocatorHelper.hpp" |
| void initClient(const bool isthinClient) { |
| if (cacheHelper == nullptr) { |
| cacheHelper = new CacheHelper(isthinClient); |
| } |
| ASSERT(cacheHelper, "Failed to create a CacheHelper client instance."); |
| } |
| void cleanProc() { |
| if (cacheHelper != nullptr) { |
| delete cacheHelper; |
| cacheHelper = nullptr; |
| } |
| } |
| |
| CacheHelper* getHelper() { |
| ASSERT(cacheHelper != nullptr, "No cacheHelper initialized."); |
| return cacheHelper; |
| } |
| |
| void _verifyEntry(const char* name, const char* key, const char* val, |
| bool noKey) { |
| // Verify key and value exist in this region, in this process. |
| const char* value = val ? val : ""; |
| std::string msg; |
| if (noKey) { |
| msg = |
| std::string("Verify key ") + key + " does not exist in region " + name; |
| } else if (!val) { |
| msg = std::string("Verify value for key ") + key + |
| " does not exist in region " + name; |
| } else { |
| msg = std::string("Verify value for key ") + key + " is: " + value + |
| " in region " + name; |
| } |
| LOG(msg); |
| |
| auto regPtr = getHelper()->getRegion(name); |
| ASSERT(regPtr != nullptr, "Region not found."); |
| |
| auto keyPtr = CacheableKey::create(key); |
| |
| if (noKey == false) { // need to find the key! |
| ASSERT(regPtr->containsKey(keyPtr), "Key not found in region."); |
| } |
| |
| // loop up to MAX times, testing condition |
| uint32_t MAX = 100; |
| uint32_t SLEEP = 10; // milliseconds |
| uint32_t containsKeyCnt = 0; |
| uint32_t containsValueCnt = 0; |
| uint32_t testValueCnt = 0; |
| |
| for (int i = MAX; i >= 0; i--) { |
| if (noKey) { |
| if (regPtr->containsKey(keyPtr)) { |
| containsKeyCnt++; |
| } else { |
| break; |
| } |
| ASSERT(containsKeyCnt < MAX, "Key found in region."); |
| } |
| if (val == nullptr) { |
| if (regPtr->containsValueForKey(keyPtr)) { |
| containsValueCnt++; |
| } else { |
| break; |
| } |
| ASSERT(containsValueCnt < MAX, "Value found in region."); |
| } |
| |
| if (val != nullptr) { |
| auto checkPtr = |
| std::dynamic_pointer_cast<CacheableString>(regPtr->get(keyPtr)); |
| |
| ASSERT(checkPtr != nullptr, "Value Ptr should not be null."); |
| LOG("In verify loop, get returned " + checkPtr->value() + " for key " + |
| key); |
| if (strcmp(checkPtr->value().c_str(), value) != 0) { |
| testValueCnt++; |
| } else { |
| break; |
| } |
| ASSERT(testValueCnt < MAX, "Incorrect value found."); |
| } |
| dunit::sleep(SLEEP); |
| } |
| } |
| |
| #define verifyInvalid(x, y) _verifyInvalid(x, y, __LINE__) |
| |
| void _verifyInvalid(const char* name, const char* key, int line) { |
| LOG(std::string("verifyInvalid() called from ") + std::to_string(line) + |
| "\n"); |
| _verifyEntry(name, key, nullptr, false); |
| LOG("Entry invalidated."); |
| } |
| |
| #define verifyDestroyed(x, y) _verifyDestroyed(x, y, __LINE__) |
| |
| void _verifyDestroyed(const char* name, const char* key, int line) { |
| LOG(std::string("verifyDestroyed() called from ") + std::to_string(line) + |
| "\n"); |
| _verifyEntry(name, key, nullptr, true); |
| LOG("Entry destroyed."); |
| } |
| |
| #define verifyEntry(x, y, z) _verifyEntry(x, y, z, __LINE__) |
| |
| void _verifyEntry(const char* name, const char* key, const char* val, |
| int line) { |
| LOG(std::string("verifyEntry() called from ") + std::to_string(line) + "\n"); |
| _verifyEntry(name, key, val, false); |
| LOG("Entry verified."); |
| } |
| |
| void createRegion(const char* name, bool ackMode, const char* endpoints, |
| bool clientNotificationEnabled = false) { |
| LOG("createRegion() entered."); |
| std::cout << "Creating region -- " << name << " ackMode is " << ackMode |
| << "\n" |
| << std::flush; |
| // ack, caching |
| auto regPtr = getHelper()->createRegion(name, ackMode, true, nullptr, |
| endpoints, clientNotificationEnabled); |
| ASSERT(regPtr != nullptr, "Failed to create region."); |
| LOG("Region created."); |
| } |
| void createPooledRegion(const std::string& name, bool ackMode, |
| const std::string& locators, |
| const std::string& poolname, |
| bool clientNotificationEnabled = false, |
| bool cachingEnable = true) { |
| LOG("createRegion_Pool() entered."); |
| std::cout << "Creating region -- " << name << " ackMode is " << ackMode |
| << "\n" |
| << std::flush; |
| auto regPtr = |
| getHelper()->createPooledRegion(name, ackMode, locators, poolname, |
| cachingEnable, clientNotificationEnabled); |
| ASSERT(regPtr != nullptr, "Failed to create region."); |
| LOG("Pooled Region created."); |
| } |
| void createEntry(const char* name, const char* key, const char* value) { |
| LOG("createEntry() entered."); |
| std::cout << "Creating entry -- key: " << key << " value: " << value |
| << " in region " << name << "\n" |
| << std::flush; |
| // Create entry, verify entry is correct |
| auto keyPtr = CacheableKey::create(key); |
| auto valPtr = CacheableString::create(value); |
| |
| auto regPtr = getHelper()->getRegion(name); |
| ASSERT(regPtr != nullptr, "Region not found."); |
| |
| ASSERT(!regPtr->containsKey(keyPtr), |
| "Key should not have been found in region."); |
| ASSERT(!regPtr->containsValueForKey(keyPtr), |
| "Value should not have been found in region."); |
| |
| // regPtr->create( keyPtr, valPtr ); |
| regPtr->put(keyPtr, valPtr); |
| LOG("Created entry."); |
| |
| verifyEntry(name, key, value); |
| LOG("Entry created."); |
| } |
| |
| void updateEntry(const char* name, const char* key, const char* value) { |
| LOG("updateEntry() entered."); |
| std::cout << "Updating entry -- key: " << key << " value: " << value |
| << " in region " << name << "\n" |
| << std::flush; |
| // Update entry, verify entry is correct |
| auto keyPtr = CacheableKey::create(key); |
| auto valPtr = CacheableString::create(value); |
| |
| auto regPtr = getHelper()->getRegion(name); |
| ASSERT(regPtr != nullptr, "Region not found."); |
| |
| ASSERT(regPtr->containsKey(keyPtr), "Key should have been found in region."); |
| // ASSERT( regPtr->containsValueForKey( keyPtr ), "Value should have been |
| // found in region." ); |
| |
| regPtr->put(keyPtr, valPtr); |
| LOG("Put entry."); |
| |
| verifyEntry(name, key, value); |
| LOG("Entry updated."); |
| } |
| |
| void doNetsearch(const char* name, const char* key, const char* value) { |
| LOG("doNetsearch() entered."); |
| std::cout << "Netsearching for entry -- key: " << key |
| << " expecting value: " << value << " in region " << name << "\n" |
| << std::flush; |
| // Get entry created in Process A, verify entry is correct |
| auto keyPtr = CacheableKey::create(key); |
| |
| auto regPtr = getHelper()->getRegion(name); |
| std::cout << "netsearch region " << regPtr->getName() << "\n" << std::flush; |
| fflush(stdout); |
| ASSERT(regPtr != nullptr, "Region not found."); |
| |
| /*NIL: Changed the asserion due to the change in invalidate. |
| Now we create new entery for every invalidate event received or |
| localInvalidate call |
| so expect containsKey to returns true insted of false earlier. */ |
| ASSERT(regPtr->containsKey(keyPtr), "Key should have been found in region."); |
| ASSERT(!regPtr->containsValueForKey(keyPtr), |
| "Value should not have been found in region."); |
| |
| auto checkPtr = std::dynamic_pointer_cast<CacheableString>( |
| regPtr->get(keyPtr)); // force a netsearch |
| |
| if (checkPtr != nullptr) { |
| LOG("checkPtr is not null"); |
| LOG(std::string("In net search, get returned ") + checkPtr->value() + |
| " for key " + key); |
| } else { |
| LOG("checkPtr is nullptr"); |
| } |
| verifyEntry(name, key, value); |
| LOG("Netsearch complete."); |
| } |
| |
| void invalidateEntry(const char* name, const char* key) { |
| LOG("invalidateEntry() entered."); |
| std::cout << "Invalidating entry -- key: " << key << " in region " << name << "\n" << std::flush; |
| // Invalidate entry, verify entry is invalidated |
| auto keyPtr = CacheableKey::create(key); |
| |
| auto regPtr = getHelper()->getRegion(name); |
| ASSERT(regPtr != nullptr, "Region not found."); |
| |
| ASSERT(regPtr->containsKey(keyPtr), "Key should have been found in region."); |
| // ASSERT( regPtr->containsValueForKey( keyPtr ), "Value should have been |
| // found in region." ); |
| |
| regPtr->invalidate(keyPtr); |
| LOG("Invalidate entry."); |
| |
| verifyInvalid(name, key); |
| LOG("Entry invalidated."); |
| } |
| |
| void destroyEntry(const char* name, const char* key) { |
| LOG("destroyEntry() entered."); |
| std::cout << "Destroying entry -- key: " << key << " in region " << name << "\n" << std::flush; |
| // Destroy entry, verify entry is destroyed |
| auto keyPtr = CacheableKey::create(key); |
| |
| auto regPtr = getHelper()->getRegion(name); |
| ASSERT(regPtr != nullptr, "Region not found."); |
| |
| ASSERT(regPtr->containsKey(keyPtr), "Key should have been found in region."); |
| |
| regPtr->destroy(keyPtr); |
| LOG("Destroy entry."); |
| |
| verifyDestroyed(name, key); |
| LOG("Entry destroyed."); |
| } |
| |
| void destroyRegion(const char* name) { |
| LOG("destroyRegion() entered."); |
| auto regPtr = getHelper()->getRegion(name); |
| regPtr->destroyRegion(); |
| LOG("Region destroyed."); |
| } |
| |
| const char* keys[] = {"Key-1", "Key-2", "Key-3", "Key-4", "Key-5"}; |
| const char* vals[] = {"Value-1", "Value-2", "Value-3", "Value-4", "Value-5"}; |
| const char* nvals[] = {"New Value-1", "New Value-2", "New Value-3", |
| "New Value-4", "New Value-5"}; |
| |
| const char* regionName = "DistRegionAck"; |
| |
| const bool USE_ACK = true; |
| const bool NO_ACK = false; |
| |
| DUNIT_TASK_DEFINITION(SERVER1, CreateServer1) |
| { |
| if (isLocalServer) CacheHelper::initServer(1); |
| LOG("SERVER1 started"); |
| } |
| END_TASK_DEFINITION |
| |
| DUNIT_TASK_DEFINITION(CLIENT1, StepOne_Pool_Locator) |
| { |
| initClient(true); |
| createPooledRegion(regionName, USE_ACK, locatorsG, "__TEST_POOL1__", true); |
| auto regPtr = getHelper()->getRegion(regionName); |
| regPtr->registerAllKeys(false, false, false); |
| LOG("StepOne complete."); |
| } |
| END_TASK_DEFINITION |
| |
| DUNIT_TASK_DEFINITION(CLIENT2, StepTwo_Pool_Locator) |
| { |
| initClient(true); |
| createPooledRegion(regionName, USE_ACK, locatorsG, "__TEST_POOL1__", true); |
| auto regPtr = getHelper()->getRegion(regionName); |
| regPtr->registerAllKeys(false, false, false); |
| LOG("StepTwo complete."); |
| } |
| END_TASK_DEFINITION |
| |
| DUNIT_TASK_DEFINITION(CLIENT1, StepThree) |
| { |
| createEntry(regionName, keys[0], vals[0]); |
| createEntry(regionName, keys[4], vals[4]); |
| LOG("StepThree complete."); |
| } |
| END_TASK_DEFINITION |
| |
| DUNIT_TASK_DEFINITION(CLIENT2, StepFour) |
| { |
| doNetsearch(regionName, keys[0], vals[0]); |
| createEntry(regionName, keys[1], vals[1]); |
| LOG("StepFour complete."); |
| } |
| END_TASK_DEFINITION |
| |
| DUNIT_TASK_DEFINITION(SERVER2, CreateServer2) |
| { |
| if (isLocalServer) CacheHelper::initServer(2); |
| LOG("SERVER2 started"); |
| } |
| END_TASK_DEFINITION |
| |
| DUNIT_TASK_DEFINITION(SERVER1, CloseServer1) |
| { |
| if (isLocalServer) { |
| CacheHelper::closeServer(1); |
| LOG("SERVER1 stopped"); |
| } |
| } |
| END_TASK_DEFINITION |
| |
| DUNIT_TASK_DEFINITION(CLIENT1, StepFive) |
| { |
| auto reg0 = getHelper()->getRegion(regionName); |
| auto vec0 = reg0->serverKeys(); |
| ASSERT(vec0.size() == 3, "Should have 3 keys in first region."); |
| |
| std::string key0, key1, key2; |
| key0 = vec0[0]->toString().c_str(); |
| key1 = vec0[1]->toString().c_str(); |
| key2 = vec0[2]->toString().c_str(); |
| ASSERT(key0 != key1, "The two keys should be different in first region."); |
| ASSERT(key0 != key2, "The two keys should be different in first region."); |
| ASSERT(key1 != key2, "The two keys should be different in first region."); |
| ASSERT(key0 == keys[0] || key0 == keys[1] || key0 == keys[4], |
| "Unexpected key in first region."); |
| ASSERT(key1 == keys[0] || key1 == keys[1] || key1 == keys[4], |
| "Unexpected key in first region."); |
| ASSERT(key2 == keys[0] || key2 == keys[1] || key2 == keys[4], |
| "Unexpected key in first region."); |
| |
| doNetsearch(regionName, keys[1], vals[1]); |
| updateEntry(regionName, keys[0], nvals[0]); |
| verifyEntry(regionName, keys[0], nvals[0]); |
| LOG("StepFive complete."); |
| } |
| END_TASK_DEFINITION |
| DUNIT_TASK_DEFINITION(CLIENT2, StepSix) |
| { |
| // verifyInvalid( regionNames[0], keys[0] ); |
| // verifyInvalid( regionNames[1], keys[2] ); |
| // This step get the value from region, if key is inavalidate in region, |
| // value |
| // will get from server. |
| doNetsearch(regionName, keys[0], nvals[0]); |
| updateEntry(regionName, keys[1], nvals[1]); |
| LOG("StepSix complete."); |
| } |
| END_TASK_DEFINITION |
| |
| DUNIT_TASK_DEFINITION(CLIENT1, StepSeven) |
| { |
| verifyEntry(regionName, keys[1], nvals[1]); |
| LOG("StepSeven complete."); |
| } |
| END_TASK_DEFINITION |
| DUNIT_TASK_DEFINITION(CLIENT1, StepEight) |
| { |
| destroyEntry(regionName, keys[0]); |
| LOG("StepEight complete."); |
| } |
| END_TASK_DEFINITION |
| |
| DUNIT_TASK_DEFINITION(CLIENT2, StepNine) |
| { |
| verifyDestroyed(regionName, keys[0]); |
| destroyEntry(regionName, keys[1]); |
| LOG("StepNine complete."); |
| } |
| END_TASK_DEFINITION |
| DUNIT_TASK_DEFINITION(CLIENT1, StepTen) |
| { |
| verifyDestroyed(regionName, keys[1]); |
| LOG("StepTen complete."); |
| } |
| END_TASK_DEFINITION |
| DUNIT_TASK_DEFINITION(CLIENT1, CloseCache1) |
| { cleanProc(); } |
| END_TASK_DEFINITION |
| |
| DUNIT_TASK_DEFINITION(CLIENT2, CloseCache2) |
| { cleanProc(); } |
| END_TASK_DEFINITION |
| DUNIT_TASK_DEFINITION(SERVER2, CloseServer2) |
| { |
| if (isLocalServer) { |
| CacheHelper::closeServer(2); |
| LOG("SERVER2 stopped"); |
| } |
| } |
| END_TASK_DEFINITION |
| |
| void runThinClientFailover2() { |
| CALL_TASK(CreateLocator1); |
| CALL_TASK(CreateServer1_With_Locator); |
| |
| CALL_TASK(StepOne_Pool_Locator); |
| CALL_TASK(StepTwo_Pool_Locator); |
| |
| CALL_TASK(StepThree); |
| CALL_TASK(StepFour); |
| |
| CALL_TASK(CreateServer2_With_Locator); |
| |
| CALL_TASK(CloseServer1); // FailOver |
| CALL_TASK(StepFive); |
| CALL_TASK(StepSix); |
| CALL_TASK(StepSeven); |
| CALL_TASK(StepEight); |
| CALL_TASK(StepNine); |
| CALL_TASK(StepTen); |
| CALL_TASK(CloseCache1); |
| CALL_TASK(CloseCache2); |
| CALL_TASK(CloseServer2); |
| |
| CALL_TASK(CloseLocator1); |
| } |
| |
| } // namespace |
| |
| #endif // GEODE_INTEGRATION_TEST_THINCLIENTFAILOVER2_H_ |