diff options
Diffstat (limited to 'cpp/src/IceBox/Admin.cpp')
| -rw-r--r-- | cpp/src/IceBox/Admin.cpp | 218 |
1 files changed, 218 insertions, 0 deletions
diff --git a/cpp/src/IceBox/Admin.cpp b/cpp/src/IceBox/Admin.cpp new file mode 100644 index 00000000000..5bc96d0f084 --- /dev/null +++ b/cpp/src/IceBox/Admin.cpp @@ -0,0 +1,218 @@ +// ********************************************************************** +// +// Copyright (c) 2003-2011 ZeroC, Inc. All rights reserved. +// +// This copy of Ice is licensed to you under the terms described in the +// ICE_LICENSE file included in this distribution. +// +// ********************************************************************** + +#include <Ice/Application.h> +#include <Ice/SliceChecksums.h> +#include <IceUtil/Options.h> +#include <IceBox/IceBox.h> + +using namespace std; +using namespace Ice; + +class Client : public Application +{ +public: + + void usage(); + virtual int run(int, char*[]); +}; + +//COMPILERFIX: Borland C++ 2010 doesn't support wmain for console applications. +#if defined(_WIN32 ) && !defined(__BCPLUSPLUS__) + +int +wmain(int argc, wchar_t* argv[]) + +#else + +int +main(int argc, char* argv[]) + +#endif +{ + Client app; + return app.main(argc, argv); +} + +void +Client::usage() +{ + cerr << "Usage: " << appName() << " [options] [command...]\n"; + cerr << + "Options:\n" + "-h, --help Show this message.\n" + "-v, --version Display the Ice version.\n" + "\n" + "Commands:\n" + "start SERVICE Start a service.\n" + "stop SERVICE Stop a service.\n" + "shutdown Shutdown the server.\n" + ; +} + +int +Client::run(int argc, char* argv[]) +{ + IceUtilInternal::Options opts; + opts.addOpt("h", "help"); + opts.addOpt("v", "version"); + + vector<string> commands; + try + { + commands = opts.parse(argc, (const char**)argv); + } + catch(const IceUtilInternal::BadOptException& e) + { + cerr << e.reason << endl; + usage(); + return EXIT_FAILURE; + } + + if(opts.isSet("help")) + { + usage(); + return EXIT_SUCCESS; + } + if(opts.isSet("version")) + { + cout << ICE_STRING_VERSION << endl; + return EXIT_SUCCESS; + } + + if(commands.empty()) + { + usage(); + return EXIT_FAILURE; + } + + + ObjectPrx base = communicator()->propertyToProxy("IceBoxAdmin.ServiceManager.Proxy"); + + if(base == 0) + { + // + // The old deprecated way to retrieve the service manager proxy + // + + PropertiesPtr properties = communicator()->getProperties(); + + Identity managerIdentity; + managerIdentity.category = properties->getPropertyWithDefault("IceBox.InstanceName", "IceBox"); + managerIdentity.name = "ServiceManager"; + + string managerProxy; + if(properties->getProperty("Ice.Default.Locator").empty()) + { + string managerEndpoints = properties->getProperty("IceBox.ServiceManager.Endpoints"); + if(managerEndpoints.empty()) + { + cerr << appName() << ": property `IceBoxAdmin.ServiceManager.Proxy' is not set" << endl; + return EXIT_FAILURE; + } + + managerProxy = "\"" + communicator()->identityToString(managerIdentity) + "\" :" + managerEndpoints; + } + else + { + string managerAdapterId = properties->getProperty("IceBox.ServiceManager.AdapterId"); + if(managerAdapterId.empty()) + { + cerr << appName() << ": property `IceBoxAdmin.ServiceManager.Proxy' is not set" << endl; + return EXIT_FAILURE; + } + + managerProxy = "\"" + communicator()->identityToString(managerIdentity) + "\" @" + managerAdapterId; + } + + base = communicator()->stringToProxy(managerProxy); + } + + IceBox::ServiceManagerPrx manager = IceBox::ServiceManagerPrx::checkedCast(base); + if(!manager) + { + cerr << appName() << ": `" << base << "' is not an IceBox::ServiceManager" << endl; + return EXIT_FAILURE; + } + + Ice::SliceChecksumDict serverChecksums = manager->getSliceChecksums(); + Ice::SliceChecksumDict localChecksums = Ice::sliceChecksums(); + for(Ice::SliceChecksumDict::const_iterator p = localChecksums.begin(); p != localChecksums.end(); ++p) + { + Ice::SliceChecksumDict::const_iterator q = serverChecksums.find(p->first); + if(q == serverChecksums.end()) + { + cerr << appName() << ": server is using unknown Slice type `" << q->first << "'" << endl; + } + else if(p->second != q->second) + { + cerr << appName() << ": server is using a different Slice definition of `" << q->first << "'" << endl; + } + } + + + vector<string>::const_iterator r; + for(r = commands.begin(); r != commands.end(); ++r) + { + if((*r) == "shutdown") + { + manager->shutdown(); + } + else if((*r) == "start") + { + if(++r == commands.end()) + { + cerr << appName() << ": no service name specified." << endl; + return EXIT_FAILURE; + } + + try + { + manager->startService(*r); + } + catch(const IceBox::NoSuchServiceException&) + { + cerr << appName() << ": unknown service `" << *r << "'" << endl; + } + catch(const IceBox::AlreadyStartedException&) + { + cerr << appName() << ": service already started." << endl; + } + } + else if((*r) == "stop") + { + if(++r == commands.end()) + { + cerr << appName() << ": no service name specified." << endl; + return EXIT_FAILURE; + } + + try + { + manager->stopService(*r); + } + catch(const IceBox::NoSuchServiceException&) + { + cerr << appName() << ": unknown service `" << *r << "'" << endl; + } + catch(const IceBox::AlreadyStoppedException&) + { + cerr << appName() << ": service already stopped." << endl; + } + } + else + { + cerr << appName() << ": unknown command `" << *r << "'" << endl; + usage(); + return EXIT_FAILURE; + } + } + + return EXIT_SUCCESS; +} |