1 files changed, 118 insertions, 0 deletions
diff --git a/src/main/java/org/onap/dmaap/mr/tools/TraceCommand.java b/src/main/java/org/onap/dmaap/mr/tools/TraceCommand.java
new file mode 100644
index 0000000..95edf5b
--- /dev/null
+++ b/src/main/java/org/onap/dmaap/mr/tools/TraceCommand.java
@@ -0,0 +1,118 @@
+/*******************************************************************************
+ *  ============LICENSE_START=======================================================
+ *  org.onap.dmaap
+ *  ================================================================================
+ *  Copyright © 2017 AT&T Intellectual Property. All rights reserved.
+ *  ================================================================================
+ *  Licensed 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.
+ *  ============LICENSE_END=========================================================
+ *
+ *  ECOMP is a trademark and service mark of AT&T Intellectual Property.
+ *  
+ *******************************************************************************/
+package org.onap.dmaap.mr.tools;
+
+import java.io.PrintStream;
+import java.net.URI;
+import java.util.List;
+import java.util.Map;
+
+import com.att.nsa.apiClient.http.HttpTracer;
+import com.att.nsa.cmdtool.Command;
+import com.att.nsa.cmdtool.CommandNotReadyException;
+
+public class TraceCommand implements Command<MRCommandContext>
+{
+	@Override
+	public void checkReady ( MRCommandContext context ) throws CommandNotReadyException
+	{
+	}
+
+	@Override
+	public void execute ( String[] parts, MRCommandContext context, final PrintStream out ) throws CommandNotReadyException
+	{
+		if ( parts[0].equalsIgnoreCase ( "on" ))
+		{
+			context.useTracer ( new HttpTracer ()
+			{
+				@Override
+				public void outbound ( URI uri, Map<String, List<String>> headers, String method, byte[] entity )
+				{
+					out.println ( kLineBreak );
+					out.println ( ">>> " + method + " " + uri.toString() );
+					for ( Map.Entry<String,List<String>> e : headers.entrySet () )
+					{
+						final StringBuffer vals = new StringBuffer ();
+						for ( String val : e.getValue () )
+						{
+							if ( vals.length () > 0 ) vals.append ( ", " );
+							vals.append ( val );
+						}
+						out.println ( ">>> " + e.getKey () + ": " + vals.toString() );
+					}
+					if ( entity != null )
+					{
+						out.println ();
+						out.println ( new String ( entity ) );
+					}
+					out.println ( kLineBreak );
+				}
+
+				@Override
+				public void inbound ( Map<String, List<String>> headers, int statusCode, String responseLine, byte[] entity )
+				{
+					out.println ( kLineBreak );
+					out.println ( "<<< " + responseLine );
+					for ( Map.Entry<String,List<String>> e : headers.entrySet () )
+					{
+						final StringBuffer vals = new StringBuffer ();
+						for ( String val : e.getValue () )
+						{
+							if ( vals.length () > 0 ) vals.append ( ", " );
+							vals.append ( val );
+						}
+						out.println ( "<<< " + e.getKey () + ": " + vals.toString() );
+					}
+					if ( entity != null )
+					{
+						out.println ();
+						out.println ( new String ( entity ) );
+					}
+					out.println ( kLineBreak );
+				}
+			} );
+		}
+		else
+		{
+			context.noTracer ();
+		}
+	}
+
+	@Override
+	public void displayHelp ( PrintStream out )
+	{
+		out.println ( "trace on|off" );
+		out.println ( "\tWhen trace is on, HTTP interaction is printed to the console." );
+	}
+
+	@Override
+	public String[] getMatches ()
+	{
+		return new String[]
+		{
+			"trace (on)",
+			"trace (off)"
+		};
+	}
+
+	private static final String kLineBreak = "======================================================================";
+}