diff options
Diffstat (limited to 'qpid/cpp/src')
-rw-r--r-- | qpid/cpp/src/qpid/client/ConnectionHandler.cpp | 12 | ||||
-rw-r--r-- | qpid/cpp/src/qpid/client/StateManager.cpp | 12 | ||||
-rw-r--r-- | qpid/cpp/src/qpid/client/StateManager.h | 1 |
3 files changed, 21 insertions, 4 deletions
diff --git a/qpid/cpp/src/qpid/client/ConnectionHandler.cpp b/qpid/cpp/src/qpid/client/ConnectionHandler.cpp index 3d338af920..9b2f662c8e 100644 --- a/qpid/cpp/src/qpid/client/ConnectionHandler.cpp +++ b/qpid/cpp/src/qpid/client/ConnectionHandler.cpp @@ -147,11 +147,15 @@ void ConnectionHandler::close() fail("Connection closed before it was established"); break; case OPEN: - setState(CLOSING); - proxy.close(200, OK); - waitFor(FINISHED); + if (setState(CLOSING, OPEN)) { + proxy.close(200, OK); + waitFor(FINISHED);//FINISHED = CLOSED or FAILED + } + //else, state was changed from open after we checked, can only + //change to failed or closed, so nothing to do break; - // Nothing to do for CLOSING, CLOSED, FAILED or NOT_STARTED + + // Nothing to do if already CLOSING, CLOSED, FAILED or if NOT_STARTED } } diff --git a/qpid/cpp/src/qpid/client/StateManager.cpp b/qpid/cpp/src/qpid/client/StateManager.cpp index 9239950a3f..5462e0fed4 100644 --- a/qpid/cpp/src/qpid/client/StateManager.cpp +++ b/qpid/cpp/src/qpid/client/StateManager.cpp @@ -60,6 +60,18 @@ void StateManager::setState(int s) stateLock.notifyAll(); } +bool StateManager::setState(int s, int expected) +{ + Monitor::ScopedLock l(stateLock); + if (state == expected) { + state = s; + stateLock.notifyAll(); + return true; + } else { + return false; + } +} + int StateManager::getState() const { Monitor::ScopedLock l(stateLock); diff --git a/qpid/cpp/src/qpid/client/StateManager.h b/qpid/cpp/src/qpid/client/StateManager.h index b01664a0c1..3c8412dfa7 100644 --- a/qpid/cpp/src/qpid/client/StateManager.h +++ b/qpid/cpp/src/qpid/client/StateManager.h @@ -36,6 +36,7 @@ class StateManager public: StateManager(int initial); void setState(int state); + bool setState(int state, int expected); int getState() const ; void waitForStateChange(int current); void waitFor(std::set<int> states); |