Adding upstream version 14.2.21.upstream/14.2.21upstream

Signed-off-by: Daniel Baumann <daniel.baumann@progress-linux.org>

1 files changed, 160 insertions, 0 deletions

diff --git a/src/test/messenger/simple_client.cc b/src/test/messenger/simple_client.cc
new file mode 100644
index 00000000..ba7ed2b0
--- /dev/null
+++ b/src/test/messenger/simple_client.cc
@@ -0,0 +1,160 @@
+// -*- mode:C++; tab-width:8; c-basic-offset:2; indent-tabs-mode:t -*-
+// vim: ts=8 sw=2 smarttab
+/*
+ * Ceph - scalable distributed file system
+ *
+ * Copyright (C) 2013 CohortFS, LLC
+ *
+ * This is free software; you can redistribute it and/or
+ * modify it under the terms of the GNU Lesser General Public
+ * License version 2.1, as published by the Free Software
+ * Foundation.  See file COPYING.
+ *
+ */
+
+#include <sys/types.h>
+
+#include <iostream>
+#include <string>
+
+using namespace std;
+
+#include "common/config.h"
+#include "msg/msg_types.h"
+#include "msg/Messenger.h"
+#include "messages/MPing.h"
+#include "common/Timer.h"
+#include "common/ceph_argparse.h"
+#include "global/global_init.h"
+#include "perfglue/heap_profiler.h"
+#include "common/address_helper.h"
+#include "message_helper.h"
+#include "simple_dispatcher.h"
+
+#define dout_subsys ceph_subsys_simple_client
+
+void usage(ostream& out)
+{
+  out << "usage: simple_client [options]\n"
+"options:\n"
+"  --addr X\n"
+"  --port X\n"
+"  --msgs X\n"
+"  --dsize X\n"
+    ;
+}
+
+
+int main(int argc, const char **argv)
+{
+	vector<const char*> args;
+	Messenger* messenger;
+	SimpleDispatcher *dispatcher;
+	std::vector<const char*>::iterator arg_iter;
+	std::string val;
+	entity_addr_t dest_addr;
+	ConnectionRef conn;
+	int r = 0;
+
+	std::string addr = "localhost";
+	std::string port = "1234";
+
+	int n_msgs = 50;
+	int n_dsize = 0;
+
+	struct timespec ts;
+	ts.tv_sec = 1;
+	ts.tv_nsec = 0;
+
+	argv_to_vec(argc, argv, args);
+
+	auto cct = global_init(NULL, args, CEPH_ENTITY_TYPE_ANY,
+			       CODE_ENVIRONMENT_UTILITY,
+			       CINIT_FLAG_NO_DEFAULT_CONFIG_FILE);
+
+	for (arg_iter = args.begin(); arg_iter != args.end();) {
+	  if (ceph_argparse_witharg(args, arg_iter, &val, "--addr",
+				    (char*) NULL)) {
+	    addr = val;
+	  } else if (ceph_argparse_witharg(args, arg_iter, &val, "--port",
+				    (char*) NULL)) {
+	    port = val;
+	  } else if (ceph_argparse_witharg(args, arg_iter, &val, "--msgs",
+				    (char*) NULL)) {
+	    n_msgs = atoi(val.c_str());;
+	  } else if (ceph_argparse_witharg(args, arg_iter, &val, "--dsize",
+				    (char*) NULL)) {
+	    n_dsize = atoi(val.c_str());;
+	  } else {
+	    ++arg_iter;
+	  }
+	};
+
+	if (!args.empty()) {
+	  cerr << "What is this? -- " << args[0] << std::endl;
+	  usage(cerr);
+	  exit(1);
+	}
+
+	cout  << "simple_client starting " <<
+	  "dest addr " << addr << " " <<
+	  "dest port " << port << " " <<
+	  "initial msgs (pipe depth) " << n_msgs << " " <<
+	  "data buffer size " << n_dsize << std::endl;
+
+	messenger = Messenger::create(g_ceph_context, g_conf().get_val<std::string>("ms_type"),
+				      entity_name_t::MON(-1),
+				      "client",
+				      getpid(), 0);
+
+	// enable timing prints
+	messenger->set_magic(MSG_MAGIC_TRACE_CTR);
+	messenger->set_default_policy(Messenger::Policy::lossy_client(0));
+
+	string dest_str = "tcp://";
+	dest_str += addr;
+	dest_str += ":";
+	dest_str += port;
+	entity_addr_from_url(&dest_addr, dest_str.c_str());
+	entity_addrvec_t dest_addrs(dest_addr);
+
+	dispatcher = new SimpleDispatcher(messenger);
+	messenger->add_dispatcher_head(dispatcher);
+
+	dispatcher->set_active(); // this side is the pinger
+
+	r = messenger->start();
+	if (r < 0)
+		goto out;
+
+	conn = messenger->connect_to_mon(dest_addrs);
+
+	// do stuff
+	time_t t1, t2;
+
+	t1 = time(NULL);
+
+	int msg_ix;
+	Message *m;
+	for (msg_ix = 0; msg_ix < n_msgs; ++msg_ix) {
+	  /* add a data payload if asked */
+	  if (! n_dsize) {
+	    m = new MPing();
+	  } else {
+	    m = new_simple_ping_with_data("simple_client", n_dsize);
+	  }
+	  conn->send_message(m);
+	}
+
+	// do stuff
+	while (conn->is_connected()) {
+	  nanosleep(&ts, NULL);
+	}
+
+	t2 = time(NULL);
+	cout << "Processed " << dispatcher->get_dcount() + n_msgs
+	     << " round-trip messages in " << t2-t1 << "s"
+	     << std::endl;
+out:
+	return r;
+}