X-Git-Url: http://git.indexdata.com/?a=blobdiff_plain;f=src%2Fmetaproxy_prog.cpp;h=a4ec7f0792aaa2ae2c8268ee94fd853ad69c8d53;hb=9181e947fbaaaaed1cbf53f38f13b56bc762366c;hp=5306b9da80961de345294a22a38bb7b2d7af2657;hpb=21f827642aea20d64c3a121b8d3fdc225d30e544;p=metaproxy-moved-to-github.git diff --git a/src/metaproxy_prog.cpp b/src/metaproxy_prog.cpp index 5306b9d..a4ec7f0 100644 --- a/src/metaproxy_prog.cpp +++ b/src/metaproxy_prog.cpp @@ -1,5 +1,5 @@ /* This file is part of Metaproxy. - Copyright (C) 2005-2011 Index Data + Copyright (C) 2005-2013 Index Data Metaproxy is free software; you can redistribute it and/or modify it under the terms of the GNU General Public License as published by the Free @@ -29,8 +29,8 @@ Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA #include #include -#include "router_flexml.hpp" -#include "factory_static.hpp" +#include +#include #if HAVE_UNISTD_H #include @@ -44,53 +44,86 @@ Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA namespace mp = metaproxy_1; -mp::RouterFleXML *routerp = 0; +mp::RouterXML *routerp = 0; + +static void set_log_prefix(void) +{ +#if HAVE_UNISTD_H + char str[80]; + + sprintf(str, "%lld", (long long) getpid()); + yaz_log_init_prefix(str); +#endif +} #if HAVE_UNISTD_H static pid_t process_group = 0; +static void sig_usr1_handler(int s) +{ + yaz_log(YLOG_LOG, "metaproxy received SIGUSR1"); + routerp->stop(); +} + static void sig_term_handler(int s) { + yaz_log(YLOG_LOG, "metaproxy received SIGTERM"); + yaz_log(YLOG_LOG, "metaproxy stop"); kill(-process_group, SIGTERM); /* kill all children processes as well */ _exit(0); } #endif -static void handler_debug(void *data) +static void work_common(void *data) { + set_log_prefix(); #if HAVE_UNISTD_H process_group = getpgid(0); // save process group ID - + signal(SIGTERM, sig_term_handler); + signal(SIGUSR1, sig_usr1_handler); #endif - routerp = (mp::RouterFleXML*) data; + routerp = (mp::RouterXML*) data; routerp->start(); mp::Package pack; - pack.router(*routerp).move(); /* should never exit */ + pack.router(*routerp).move(); + yaz_log(YLOG_LOG, "metaproxy stop"); /* only for graceful stop */ +#if HAVE_UNISTD_H + kill(-process_group, SIGTERM); /* kill all children processes as well */ +#endif + _exit(0); } - -static void handler_normal(void *data) + +static void work_debug(void *data) +{ + work_common(data); +} + +static void work_normal(void *data) { #if HAVE_UNISTD_H /* make the current working process group leader */ setpgid(0, 0); #endif - handler_debug(data); + work_common(data); } static int sc_main( yaz_sc_t s, int argc, char **argv) { + bool test_config = false; const char *fname = 0; int ret; char *arg; unsigned mode = 0; const char *pidfile = 0; const char *uid = 0; - - while ((ret = options("c{config}:Dh{help}l:p:u:V{version}w:X", + + set_log_prefix(); + + while ((ret = options("c{config}:Dh{help}l:m:p:tu:v:V{version}w:X", argv, argc, &arg)) != -2) { switch (ret) @@ -105,10 +138,13 @@ static int sc_main( std::cerr << "metaproxy\n" " -h|--help help\n" " -V|--version version\n" + " -v level\n" " -c|--config f config filename\n" " -D daemon and keepalive operation\n" " -l f log file f\n" + " -m logformat log time format (strftime)\n" " -p f pid file f\n" + " -t test configuration\n" " -u id change uid to id\n" " -w dir changes working directory to dir\n" " -X debug mode (no fork/daemon mode)\n" @@ -116,18 +152,27 @@ static int sc_main( " -install install windows service\n" " -remove remove windows service\n" #endif - + << std::endl; break; case 'l': yaz_log_init_file(arg); break; + case 'm': + yaz_log_time_format(arg); + break; case 'p': pidfile = arg; break; + case 't': + test_config = true; + break; case 'u': uid = arg; break; + case 'v': + yaz_log_init_level(yaz_log_mask_str(arg)); + break; case 'V': std::cout << VERSION; #ifdef VERSION_SHA1 @@ -143,7 +188,7 @@ static int sc_main( #else chdir(arg) #endif - ) + ) { std::cerr << "chdir " << arg << " failed" << std::endl; return 1; @@ -161,21 +206,23 @@ static int sc_main( std::cerr << "No configuration given; use -h for help\n"; return 1; } - - yaz_log(YLOG_LOG, "Metaproxy start " VERSION + + yaz_log(YLOG_LOG, "metaproxy %s " VERSION #ifdef VERSION_SHA1 - " " VERSION_SHA1 + " " VERSION_SHA1 #endif - ); - + , test_config ? "test" : "start" + ); + + yaz_log_xml_errors(0, YLOG_LOG); xmlDocPtr doc = xmlReadFile(fname, - NULL, + NULL, XML_PARSE_XINCLUDE + XML_PARSE_NOBLANKS + XML_PARSE_NSCLEAN + XML_PARSE_NONET ); - + if (!doc) { - yaz_log (YLOG_FATAL,"XML parsing failed"); + yaz_log(YLOG_FATAL,"XML parsing failed"); return 1; } // and perform Xinclude then @@ -185,7 +232,7 @@ static int sc_main( yaz_log(YLOG_FATAL, "XInclude processing failed"); return 1; } - WRBUF base_path = wrbuf_alloc(); + mp::wrbuf base_path; const char *last_p = strrchr(fname, #ifdef WIN32 '\\' @@ -195,21 +242,23 @@ static int sc_main( ); if (last_p) wrbuf_write(base_path, fname, last_p - fname); - + else + wrbuf_puts(base_path, "."); ret = 0; try { - mp::FactoryStatic factory; - mp::RouterFleXML *router = - new mp::RouterFleXML(doc, factory, false, wrbuf_cstr(base_path)); - wrbuf_destroy(base_path); - - yaz_sc_running(s); - - yaz_daemon("metaproxy", mode, mode == YAZ_DAEMON_DEBUG ? - handler_debug : handler_normal, router, pidfile, uid); + mp::RouterXML *router = + new mp::RouterXML(doc, test_config, wrbuf_cstr(base_path)); + if (!test_config) + { + + yaz_sc_running(s); + + yaz_daemon("metaproxy", mode, mode == YAZ_DAEMON_DEBUG ? + work_debug : work_normal, router, pidfile, uid); + } } catch (std::logic_error &e) { - yaz_log (YLOG_FATAL,"std::logic error: %s" , e.what() ); + yaz_log(YLOG_FATAL,"std::logic error: %s" , e.what() ); ret = 1; } catch (std::runtime_error &e) { @@ -220,8 +269,9 @@ static int sc_main( yaz_log(YLOG_FATAL, "Unknown Exception"); ret = 1; } - wrbuf_destroy(base_path); xmlFreeDoc(doc); + if (test_config) + yaz_log(YLOG_LOG, "metaproxy test exit code %d", ret); return ret; } @@ -241,8 +291,6 @@ int main(int argc, char **argv) exit(ret); } - - /* * Local variables: * c-basic-offset: 4