adding state to AuthenticationContext

Signed-off-by: Emelia Lei <[email protected]>

Author: emelialei88

src/groups/mqb/mqba/mqba_authenticator.cpp

@@ -85,6 +85,10 @@ int Authenticator::onAuthenticationRequest(
     response.status().code()     = 0;
     response.lifetimeMs()        = 10 * 60 * 1000;
 
+    context->testAndSwapState(
+        mqbnet::AuthenticationContext::State::e_AUTHENTICATING,
+        mqbnet::AuthenticationContext::State::e_AUTHENTICATED);
+
     int rc = sendAuthenticationMessage(errorDescription,
                                        authenticationResponse,
                                        context);

src/groups/mqb/mqbnet/mqbnet_authenticationcontext.cpp

@@ -30,10 +30,12 @@ namespace mqbnet {
 AuthenticationContext::AuthenticationContext(
     InitialConnectionContext* initialConnectionContext,
     bool                      isReversed,
+    State                     state,
     ConnectionType::Enum      connectionType,
     bslma::Allocator*         basicAllocator)
 : d_initialConnectionContext_p(initialConnectionContext)
 , d_authenticationMessage(basicAllocator)
+, d_state(state)
 , d_isReversed(isReversed)
 , d_connectionType(connectionType)
 {
@@ -54,6 +56,12 @@ AuthenticationContext& AuthenticationContext::setAuthenticationMessage(
     return *this;
 }
 
+int AuthenticationContext::testAndSwapState(State compareValue,
+                                            State swapValue)
+{
+    return d_state.testAndSwap(compareValue, swapValue);
+}
+
 AuthenticationContext& AuthenticationContext::setIsReversed(bool value)
 {
     d_isReversed = value;

src/groups/mqb/mqbnet/mqbnet_authenticationcontext.h

@@ -40,10 +40,19 @@ namespace mqbnet {
 
 /// VST for the context associated with an connection being authenticated.
 class AuthenticationContext {
+  public:
+    // TYPES
+    enum State {
+        e_IDLE = 0,
+        e_AUTHENTICATING,
+        e_AUTHENTICATED,
+    };
+
   private:
     // DATA
     InitialConnectionContext*           d_initialConnectionContext_p;
     bmqp_ctrlmsg::AuthenticationMessage d_authenticationMessage;
+    bsls::AtomicInt                     d_state;
     bool                                d_isReversed;
     ConnectionType::Enum                d_connectionType;
 
@@ -63,6 +72,7 @@ class AuthenticationContext {
     AuthenticationContext(
         InitialConnectionContext* initialConnectionContext,
         bool                      isReversed,
+        State                     state          = State::e_IDLE,
         ConnectionType::Enum      connectionType = ConnectionType::e_UNKNOWN,
         bslma::Allocator*         basicAllocator = 0);
 
@@ -74,6 +84,8 @@ class AuthenticationContext {
     AuthenticationContext& setIsReversed(bool value);
     AuthenticationContext& setConnectionType(ConnectionType::Enum value);
 
+    int testAndSwapState(State compareValue, State swapValue);
+
     // ACCESSORS
     InitialConnectionContext* initialConnectionContext() const;
     const bmqp_ctrlmsg::AuthenticationMessage& authenticationMessage() const;