Allowing application to request logoutDisconnect in a session and spe…

…cify a custom text

artio-core/src/main/java/uk/co/real_logic/artio/session/DirectSessionProxy.java

@@ -267,9 +267,13 @@ private boolean notNullOrEmpty(final String string)
         return string != null && string.length() > 0;
     }
 
-    public long sendLogout(final int msgSeqNo, final int sequenceIndex, final int lastMsgSeqNumProcessed)
+    public long sendLogout(
+        final int msgSeqNo,
+        final int sequenceIndex,
+        final int lastMsgSeqNumProcessed,
+        final byte[] text)
     {
-        return sendLogout(msgSeqNo, null, 0, sequenceIndex, lastMsgSeqNumProcessed);
+        return sendLogout(msgSeqNo, text, text == null ? 0 : text.length, sequenceIndex, lastMsgSeqNumProcessed);
     }
 
     public long sendLogout(

artio-core/src/main/java/uk/co/real_logic/artio/session/Session.java

@@ -587,6 +587,16 @@ public Reply<ThrottleConfigurationStatus> throttleMessagesAt(
             id, throttleWindowInMs, throttleLimitOfMessages);
     }
 
+    /**
+     * Default version of {@link Session#logoutAndDisconnect(byte[])}
+     *
+     * @return the position within the Aeron stream where the disconnect is encoded.
+     */
+    public long logoutAndDisconnect()
+    {
+        return logoutAndDisconnect(APPLICATION_DISCONNECT);
+    }
+
     /**
      * Send a logout message and immediately disconnect the session. You should normally use
      * the <code>startLogout</code> method and not this one.
@@ -596,20 +606,27 @@ public Reply<ThrottleConfigurationStatus> throttleMessagesAt(
      * message. This should only be used when you want to rapidly disconnect the session and are willing
      * to take the risk that the logout message is not received.
      *
+     * @param text value to be assigned to tag 50 of logout message.
+     *
      * @return the position within the Aeron stream where the disconnect is encoded.
      * @see Session#startLogout()
      */
-    public long logoutAndDisconnect()
+    public long logoutAndDisconnect(final byte[] text)
     {
-        return logoutAndDisconnect(APPLICATION_DISCONNECT);
+        return logoutAndDisconnect(APPLICATION_DISCONNECT, text);
     }
 
     long logoutAndDisconnect(final DisconnectReason reason)
+    {
+        return logoutAndDisconnect(reason, null);
+    }
+
+    long logoutAndDisconnect(final DisconnectReason reason, final byte[] text)
     {
         long position = NO_OPERATION;
         if (state() != DISCONNECTED)
         {
-            position = trySendLogout();
+            position = trySendLogout(text);
             if (position < 0)
             {
                 state(LOGGING_OUT_AND_DISCONNECTING);
@@ -2237,10 +2254,15 @@ private void incNextHeartbeatTime()
     }
 
     private long trySendLogout()
+    {
+        return trySendLogout(null);
+    }
+
+    private long trySendLogout(final byte[] text)
     {
         final int sentSeqNum = newSentSeqNum();
         final long position = (logoutRejectReason == NO_LOGOUT_REJECT_REASON) ?
-            proxy.sendLogout(sentSeqNum, sequenceIndex(), lastMsgSeqNumProcessed) :
+            proxy.sendLogout(sentSeqNum, sequenceIndex(), lastMsgSeqNumProcessed, text) :
             proxy.sendLogout(sentSeqNum, sequenceIndex(), logoutRejectReason, lastMsgSeqNumProcessed);
         if (position >= 0)
         {

artio-core/src/main/java/uk/co/real_logic/artio/session/SessionProxy.java

@@ -54,7 +54,7 @@ long sendLogon(
         CancelOnDisconnectOption cancelOnDisconnectOption,
         int cancelOnDisconnectTimeoutWindowInMs);
 
-    long sendLogout(int msgSeqNo, int sequenceIndex, int lastMsgSeqNumProcessed);
+    long sendLogout(int msgSeqNo, int sequenceIndex, int lastMsgSeqNumProcessed, byte[] text);
 
     long sendLogout(
         int msgSeqNo, int sequenceIndex, int rejectReason, int lastMsgSeqNumProcessed);

artio-core/src/test/java/uk/co/real_logic/artio/session/AbstractSessionTest.java

@@ -366,7 +366,10 @@ public void shouldReplyToValidLogoutWhenBackPressured()
 
     private void backPressureLogout()
     {
-        when(sessionProxy.sendLogout(anyInt(), eq(SEQUENCE_INDEX), anyInt())).thenReturn(BACK_PRESSURED, POSITION);
+        when(sessionProxy.sendLogout(anyInt(),
+          eq(SEQUENCE_INDEX),
+          anyInt(),
+          any())).thenReturn(BACK_PRESSURED, POSITION);
 
         backPressureDisconnect();
     }
@@ -659,7 +662,7 @@ public void shouldLogoutOnLogonWithLowSequenceNumberAndWithAsyncProxy()
     private void verifyLogoutOnlyOnce()
     {
         verify(sessionProxy, times(1)).sendLogout(
-            anyInt(), eq(SEQUENCE_INDEX), eq(NO_LAST_MSG_SEQ_NUM_PROCESSED));
+            anyInt(), eq(SEQUENCE_INDEX), eq(NO_LAST_MSG_SEQ_NUM_PROCESSED), any());
     }
 
     @Test
@@ -1276,7 +1279,12 @@ protected void givenActive()
 
     public void verifyLogout(final int msgSeqNo, final VerificationMode times)
     {
-        verify(sessionProxy, times).sendLogout(msgSeqNo, SEQUENCE_INDEX, NO_LAST_MSG_SEQ_NUM_PROCESSED);
+        verifyLogout(msgSeqNo, times, null);
+    }
+
+    public void verifyLogout(final int msgSeqNo, final VerificationMode times, final byte[] text)
+    {
+        verify(sessionProxy, times).sendLogout(msgSeqNo, SEQUENCE_INDEX, NO_LAST_MSG_SEQ_NUM_PROCESSED, text);
     }
 
     public void assertState(final SessionState state)

artio-core/src/test/java/uk/co/real_logic/artio/session/AcceptorSessionTest.java

@@ -96,6 +96,17 @@ public void shouldBeActivatedBySuccessfulLogin()
         assertState(ACTIVE);
     }
 
+    @Test
+    public void shouldRequestLogoutWithCustomText()
+    {
+        shouldBeActivatedBySuccessfulLogin();
+        final byte[] text = "custom text".getBytes();
+
+        session.logoutAndDisconnect(text);
+
+        verifyLogout(2, times(1), text);
+    }
+
     @Test
     public void shouldRequestResendIfHighSeqNoLogon()
     {

artio-system-tests/src/test/java/uk/co/real_logic/artio/system_tests/ExternallyControlledSystemTest.java

@@ -408,13 +408,18 @@ public long sendLogon(
             return acceptingSessionWriter.send(logon, adjustedMsgSeqNo);
         }
 
-        public long sendLogout(final int msgSeqNo, final int sequenceIndex, final int lastMsgSeqNumProcessed)
+        public long sendLogout(
+            final int msgSeqNo, final int sequenceIndex, final int lastMsgSeqNumProcessed, final byte[] text)
         {
             sentLogouts++;
 
             final int adjustedMsgSeqNo = msgSeqNo + sequenceNumberAdjustment;
             final HeaderEncoder header = logout.header();
             setupHeader(header, adjustedMsgSeqNo);
+            if (text != null)
+            {
+                logout.text(text);
+            }
 
             if (send)
             {

artio-system-tests/src/test/java/uk/co/real_logic/artio/system_tests/GatewayToGatewaySystemTest.java

@@ -480,6 +480,21 @@ public void initiatorSessionCanBeDisconnected()
         assertSequenceIndicesAre(0);
     }
 
+    @Test(timeout = TEST_TIMEOUT_IN_MS)
+    public void initiatorSessionCanBeDisconnectedCustomLogoutMessage()
+    {
+        acquireAcceptingSession();
+
+        String customLogoutMessage = "custom logout message";
+        logoutSession(initiatingSession, customLogoutMessage);
+
+        assertSessionsDisconnected();
+        testSystem.await("logout text does not match", () ->
+          acceptingOtfAcceptor.lastReceivedMessage().get(58).equals(customLogoutMessage));
+
+        assertSequenceIndicesAre(0);
+    }
+
     @Test(timeout = TEST_TIMEOUT_IN_MS)
     public void acceptorSessionCanBeDisconnected()
     {
@@ -567,7 +582,7 @@ public void sequenceNumbersShouldResetOverDisconnects()
         messagesCanBeExchanged();
         assertSequenceFromInitToAcceptAt(2, 2);
 
-        logoutSession(initiatingSession);
+        logoutSession(initiatingSession, null);
 
         assertSequenceIndicesAre(0);
         assertSessionsDisconnected();

artio-system-tests/src/test/java/uk/co/real_logic/artio/system_tests/SystemTestUtil.java

@@ -576,13 +576,18 @@ public static long disconnectSession(final Session session)
         return position;
     }
 
-    public static long logoutSession(final Session session)
+    public static long logoutSession(final Session session, final String text)
     {
-        final long position = session.startLogout();
+        final long position = session.logoutAndDisconnect(text == null ? null : text.getBytes());
         assertThat(position, greaterThan(0L));
         return position;
     }
 
+    public static long logoutSession(final Session session)
+    {
+        return logoutSession(session, null);
+    }
+
     public static long logoutSession(final TestSystem testSystem, final Session session)
     {
         return testSystem.awaitSend(session::startLogout);