From: Zdeněk Šustr <sustr4@cesnet.cz>
Date: Mon, 6 Feb 2012 14:45:39 +0000 (+0000)
Subject: Allow specifying output file through a cmdline argument
X-Git-Tag: gridsite-core_R_1_7_17~129
X-Git-Url: http://scientific.zcu.cz/git/?a=commitdiff_plain;h=1e70e6bfb84050914b97403a76bcf60a3332c4a2;p=jra1mw.git

Allow specifying output file through a cmdline argument
flush that output file after every message (used in testes)
---

diff --git a/org.glite.lb.logger-msg/examples/cmsclient.cpp b/org.glite.lb.logger-msg/examples/cmsclient.cpp
index d2bdd51..2709a5a 100644
--- a/org.glite.lb.logger-msg/examples/cmsclient.cpp
+++ b/org.glite.lb.logger-msg/examples/cmsclient.cpp
@@ -49,6 +49,7 @@ using namespace std;
 class SimpleAsyncConsumer : public ExceptionListener,
 							public MessageListener,
 							public DefaultTransportListener {
+
 private:
 
 	Connection* connection;
@@ -59,6 +60,7 @@ private:
 	bool clientAck;
 	std::string brokerURI;
 	std::string destURI;
+	FILE * outputfile;
 
 public:
 
@@ -150,7 +152,8 @@ public:
 				message->acknowledge();
 			}
 
-			printf( "Message #%d Received: %s\n", count, text.c_str() );
+			fprintf(outputfile, "Message #%d Received: %s\n", count, text.c_str() );
+			fflush(outputfile);
 		} catch (CMSException& e) {
 			e.printStackTrace();
 		}
@@ -159,16 +162,20 @@ public:
 	// If something bad happens you see it here as this class is also been
 	// registered as an ExceptionListener with the connection.
 	virtual void onException( const CMSException& ex AMQCPP_UNUSED ) {
-		printf("CMS Exception occurred.  Shutting down client.\n");
+		fprintf(stderr, "CMS Exception occurred.  Shutting down client.\n");
 		//exit(1);
 	}
 
 	virtual void transportInterrupted() {
-		std::cout << "The Connection's Transport has been Interrupted." << std::endl;
+		fprintf(stderr, "The Connection's Transiort has been Interrupted.\n");
 	}
 
 	virtual void transportResumed() {
-		std::cout << "The Connection's Transport has been Restored." << std::endl;
+		fprintf(stderr, "The Connection's Transport has been Restored.\n");
+	}
+
+	virtual void setOutfile(FILE * setout) {
+		outputfile=setout;
 	}
 
 private:
@@ -212,16 +219,29 @@ private:
 
 ////////////////////////////////////////////////////////////////////////////////
 int main(int argc AMQCPP_UNUSED, char* argv[] AMQCPP_UNUSED) {
-	if (argc <= 2) {
-		printf("Usage: %s <broker> <topic>\n", argv[0]);
+	char *argbrokerURI = NULL;
+	char *argdestURI = NULL;
+	char *outputfile = NULL;
+	FILE * outfile;
+	int i;
+
+
+	for(i=1; i<argc; i++) 
+		if((!strcmp(argv[i],"-o")) && (i<argc)) outputfile = argv[++i];
+		else 	if(!argbrokerURI) argbrokerURI=argv[i];
+			else argdestURI=argv[i];
+	if (!(outfile = fopen (outputfile,"w"))) outfile = stdout;
+
+	if (!argdestURI) {
+		printf("Usage: %s [-o <output_file>] <broker> <topic>\n", argv[0]);
 		return 1;
 	}
 
 	activemq::library::ActiveMQCPP::initializeLibrary();
 
-	std::cout << "=====================================================\n";
-	std::cout << "Starting the example:" << std::endl;
-	std::cout << "-----------------------------------------------------\n";
+	fprintf(outfile, "=====================================================\n");
+	fprintf(outfile, "Starting the example:\n");
+	fprintf(outfile, "-----------------------------------------------------\n");
 
 	// Set the URI to point to the IPAddress of your broker.
 	// add any optional params to the url to enable things like
@@ -240,7 +260,7 @@ int main(int argc AMQCPP_UNUSED, char* argv[] AMQCPP_UNUSED) {
 	//	tcp://127.0.0.1:61613?wireFormat=stomp	 use stomp instead
 	//
 	std::string brokerURI = "failover:(tcp://";
-	brokerURI += argv[1];
+	brokerURI += argbrokerURI;
 	brokerURI += ")";
 
 //	std::string brokerURI =
@@ -257,7 +277,7 @@ int main(int argc AMQCPP_UNUSED, char* argv[] AMQCPP_UNUSED) {
 	// customize where the consumer listens, to have the consumer
 	// use a topic or queue set the 'useTopics' flag.
 	//============================================================
-	std::string destURI = argv[2]; //?consumer.prefetchSize=1";
+	std::string destURI = argdestURI; //?consumer.prefetchSize=1";
 
 	//============================================================
 	// set to true to use topics instead of queues
@@ -275,19 +295,23 @@ int main(int argc AMQCPP_UNUSED, char* argv[] AMQCPP_UNUSED) {
 	// Create the consumer
 	SimpleAsyncConsumer consumer( brokerURI, destURI, useTopics, clientAck );
 
+	consumer.setOutfile(outfile);
+
 	// Start it up and it will listen forever.
 	consumer.runConsumer();
 
 	// Wait to exit.
-	std::cout << "Press 'q' to quit" << std::endl;
+	fprintf(stdout, "Press 'q' to quit\n");
 	while( std::cin.get() != 'q') {}
 
 	// All CMS resources should be closed before the library is shutdown.
 	consumer.close();
 
-	std::cout << "-----------------------------------------------------\n";
-	std::cout << "Finished with the example." << std::endl;
-	std::cout << "=====================================================\n";
+	fprintf(outfile, "-----------------------------------------------------\n");
+	fprintf(outfile, "Finished with the example.\n");
+	fprintf(outfile, "=====================================================\n");
+
+	if (outputfile) fclose(outfile);
 
 	activemq::library::ActiveMQCPP::shutdownLibrary();
 }