relpipe/relpipe-in-posixmq.cpp: comparison src/relpipe-in-posixmq.cpp

equal deleted inserted replaced

-:65abb0376a0d
+:12e975f807ed
 *
 * You should have received a copy of the GNU General Public License
 * along with this program. If not, see <http://www.gnu.org/licenses/>.
 */
 #include <cstdlib>
+#include <csignal>
 #include <vector>
 #include <memory>
 #include <regex>
 #include <algorithm>
 #include <unistd.h>
 using namespace relpipe::cli;
 using namespace relpipe::writer;
 using namespace relpipe::in::posixmq;
+static std::shared_ptr<PosixMQCommand> command = nullptr;
+void finish(int sig) {
+	if (command) command->finish(sig);
+}
 int main(int argc, char** argv) {
 	setlocale(LC_ALL, "");
 	CLI::untieStdIO();
 	CLI cli(argc, argv);
 	int resultCode = CLI::EXIT_CODE_UNEXPECTED_ERROR;
 	try {
+		signal(SIGHUP, finish);
+		signal(SIGINT, finish);
 		CLIParser cliParser;
 		Configuration configuration = cliParser.parse(cli.arguments());
-		PosixMQCommand command;
+		command.reset(new PosixMQCommand());
 		std::shared_ptr<RelationalWriter> writer(Factory::create(std::cout));
-		command.process(writer, configuration);
+		writer->setBufferingMode(BufferingMode::ENVIRONMENT, BufferingMode::RECORD);
+		command->process(writer, configuration);
 		resultCode = CLI::EXIT_CODE_SUCCESS;
 	} catch (RelpipeCLIException e) {
 		fwprintf(stderr, L"Caught CLI exception: %ls\n", e.getMessage().c_str());
 		fwprintf(stderr, L"Debug: Input stream: eof=%ls, lastRead=%d\n", (cin.eof() ? L"true" : L"false"), cin.gcount());
 		resultCode = e.getExitCode();