| 65 | |
| 66 | |
| 67 | int main(int argc, char** argv) |
| 68 | { |
| 69 | // Register log tools. |
| 70 | add(Owned<tool::Tool>(new tool::Benchmark())); |
| 71 | add(Owned<tool::Tool>(new tool::Initialize())); |
| 72 | add(Owned<tool::Tool>(new tool::Read())); |
| 73 | add(Owned<tool::Tool>(new tool::Replica())); |
| 74 | |
| 75 | if (argc < 2) { |
| 76 | usage(argv[0]); |
| 77 | return 1; |
| 78 | } |
| 79 | |
| 80 | if (!strcmp(argv[1], "help")) { |
| 81 | if (argc == 2) { |
| 82 | usage(argv[0]); |
| 83 | return 0; |
| 84 | } |
| 85 | |
| 86 | // 'mesos-log help command' => 'mesos-log command --help' |
| 87 | argv[1] = argv[2]; |
| 88 | argv[2] = (char*) "--help"; |
| 89 | } |
| 90 | |
| 91 | string command = argv[1]; |
| 92 | |
| 93 | if (!tools.contains(command)) { |
| 94 | cerr << "Cannot find command '" << command << "'" << endl << endl; |
| 95 | usage(argv[0]); |
| 96 | return 1; |
| 97 | } |
| 98 | |
| 99 | // Execute the command. |
| 100 | Try<Nothing> execute = tools[command]->execute(argc, argv); |
| 101 | if (execute.isError()) { |
| 102 | cerr << execute.error() << endl; |
| 103 | return 1; |
| 104 | } |
| 105 | |
| 106 | return 0; |
| 107 | } |