123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133 |
- /* vim:expandtab:shiftwidth=2:tabstop=2:smarttab:
- *
- * Gearmand client and server library.
- *
- * Copyright (C) 2011 Data Differential, http://datadifferential.com/
- * Copyright (C) 2008 Brian Aker, Eric Day
- * All rights reserved.
- *
- * Redistribution and use in source and binary forms, with or without
- * modification, are permitted provided that the following conditions are
- * met:
- *
- * * Redistributions of source code must retain the above copyright
- * notice, this list of conditions and the following disclaimer.
- *
- * * Redistributions in binary form must reproduce the above
- * copyright notice, this list of conditions and the following disclaimer
- * in the documentation and/or other materials provided with the
- * distribution.
- *
- * * The names of its contributors may not be used to endorse or
- * promote products derived from this software without specific prior
- * written permission.
- *
- * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
- * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
- * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
- * A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
- * OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
- * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
- * LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
- * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
- * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
- * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
- * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
- *
- */
- #include "gear_config.h"
- #include <iostream>
- #include <cstdlib>
- #include <string>
- #include <libgearman/gearman.h>
- #include <boost/program_options.hpp>
- #ifndef __INTEL_COMPILER
- #pragma GCC diagnostic ignored "-Wold-style-cast"
- #endif
- int main(int args, char *argv[])
- {
- std::string text_to_echo;
- std::string host;
- in_port_t port;
- boost::program_options::options_description desc("Options");
- desc.add_options()
- ("help", "Options related to the program.")
- ("host,h", boost::program_options::value<std::string>(&host)->default_value("localhost"),"Connect to the host")
- ("port,p", boost::program_options::value<in_port_t>(&port)->default_value(GEARMAN_DEFAULT_TCP_PORT), "Port number use for connection")
- ("text", boost::program_options::value<std::string>(&text_to_echo), "Text used for echo")
- ;
- boost::program_options::positional_options_description text_options;
- text_options.add("text", -1);
- boost::program_options::variables_map vm;
- try
- {
- boost::program_options::store(boost::program_options::command_line_parser(args, argv).
- options(desc).positional(text_options).run(), vm);
- boost::program_options::notify(vm);
- }
- catch(std::exception &e)
- {
- std::cout << e.what() << std::endl;
- return EXIT_FAILURE;
- }
- if (vm.count("help"))
- {
- std::cout << desc << std::endl;
- return EXIT_SUCCESS;
- }
- if (text_to_echo.empty())
- {
- while(std::cin.good())
- {
- char buffer[1024];
- std::cin.read(buffer, sizeof(buffer));
- text_to_echo.append(buffer, std::cin.gcount());
- }
- if (text_to_echo.empty())
- {
- std::cerr << "No text was provided for --text or via stdin" << std::endl;
- std::cerr << desc << std::endl;
- return EXIT_FAILURE;
- }
- }
- gearman_client_st client;
- if (gearman_client_create(&client) == NULL)
- {
- std::cerr << "Memory allocation failure on client creation" << std::endl;
- return EXIT_FAILURE;
- }
- gearman_return_t ret;
- ret= gearman_client_add_server(&client, host.c_str(), port);
- if (ret != GEARMAN_SUCCESS)
- {
- std::cerr << gearman_client_error(&client) << std::endl;
- return EXIT_FAILURE;
- }
- ret= gearman_client_echo(&client, text_to_echo.c_str(), text_to_echo.size());
- gearman_client_free(&client);
- if (ret != GEARMAN_SUCCESS)
- {
- std::cerr << gearman_client_error(&client) << std::endl;
- return EXIT_FAILURE;
- }
- std::cout << text_to_echo;
- return EXIT_SUCCESS;
- }
|