/* 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 #include #include #include #include #include #include "util/string.hpp" #include "gearmand/error.hpp" #ifndef __INTEL_COMPILER #pragma GCC diagnostic ignored "-Wold-style-cast" #endif int main(int args, char *argv[]) { gearmand::error::init(argv[0]); in_port_t port; std::string text_to_echo; std::string host; std::string identifier; int timeout; int count; boost::program_options::options_description desc("Options"); desc.add_options() ("help", "Options related to the program.") ("verbose", "Send status to stdout") ("host,h", boost::program_options::value(&host)->default_value("localhost"),"Connect to the host") ("identifier", boost::program_options::value(&identifier),"Assign identifier") ("port,p", boost::program_options::value(&port)->default_value(GEARMAN_DEFAULT_TCP_PORT), "Port number use for connection") ("timeout,u", boost::program_options::value(&timeout)->default_value(-1), "Timeout in milliseconds") ("count", boost::program_options::value(&count)->default_value(1), "Number of times to send the job") ("text", boost::program_options::value(&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 (vm.count("verbose") == 0) { close(STDOUT_FILENO); } 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; } if (timeout >= 0) { gearman_client_set_timeout(&client, timeout); } gearman_return_t ret= gearman_client_add_server(&client, host.c_str(), port); if (ret != GEARMAN_SUCCESS) { gearmand::error::message(gearman_client_error(&client)); return EXIT_FAILURE; } if (identifier.empty() == false) { if (gearman_failed(gearman_client_set_identifier(&client, identifier.c_str(), identifier.size()))) { gearmand::error::message(gearman_client_error(&client)); return EXIT_FAILURE; } } int exit_code= EXIT_SUCCESS; do { size_t result_size; char *result; result= (char *)gearman_client_do(&client, "reverse", NULL, text_to_echo.c_str(), text_to_echo.size(), &result_size, &ret); if (ret == GEARMAN_WORK_DATA) { std::cout.write(result, result_size); free(result); continue; } else if (ret == GEARMAN_WORK_STATUS) { uint32_t numerator; uint32_t denominator; gearman_client_do_status(&client, &numerator, &denominator); std::clog << "Status: " << numerator << "/" << denominator << std::endl; continue; } else if (ret == GEARMAN_SUCCESS) { std::cout.write(result, result_size); free(result); } else if (ret == GEARMAN_WORK_FAIL) { gearmand::error::message("Work failed"); exit_code= EXIT_FAILURE; break; } else { gearmand::error::message(gearman_client_error(&client)); exit_code= EXIT_FAILURE; break; } --count; } while (count); gearman_client_free(&client); return exit_code; }