mirror of
https://github.com/FairRootGroup/FairMQ.git
synced 2025-10-13 00:31:14 +00:00
83 lines
2.2 KiB
C++
83 lines
2.2 KiB
C++
/********************************************************************************
|
|
* Copyright (C) 2017 GSI Helmholtzzentrum fuer Schwerionenforschung GmbH *
|
|
* *
|
|
* This software is distributed under the terms of the *
|
|
* GNU Lesser General Public Licence (LGPL) version 3, *
|
|
* copied verbatim in the file "LICENSE" *
|
|
********************************************************************************/
|
|
|
|
#ifndef FAIR_MQ_TOOLS_PROCESS_H
|
|
#define FAIR_MQ_TOOLS_PROCESS_H
|
|
|
|
#include <boost/process.hpp>
|
|
|
|
#include <string>
|
|
|
|
namespace fair
|
|
{
|
|
namespace mq
|
|
{
|
|
namespace tools
|
|
{
|
|
|
|
/**
|
|
* Result type for execute function. Holds captured stdout output and exit code.
|
|
*/
|
|
struct execute_result
|
|
{
|
|
std::string console_out;
|
|
int exit_code;
|
|
};
|
|
|
|
/**
|
|
* Execute given command in forked process and capture stdout output
|
|
* and exit code.
|
|
*
|
|
* @param[in] cmd Command to execute
|
|
* @param[in] log_prefix How to prefix each captured output line with
|
|
* @return Captured stdout output and exit code
|
|
*/
|
|
inline execute_result execute(std::string cmd, std::string prefix = "")
|
|
{
|
|
execute_result result;
|
|
std::stringstream out;
|
|
|
|
// print full line thread-safe
|
|
std::stringstream printCmd;
|
|
printCmd << prefix << cmd << "\n";
|
|
std::cout << printCmd.str() << std::flush;
|
|
|
|
out << prefix << cmd << std::endl;
|
|
|
|
// Execute command and capture stdout, add prefix line by line
|
|
boost::process::ipstream stdout;
|
|
boost::process::child c(cmd, boost::process::std_out > stdout);
|
|
std::string line;
|
|
while (getline(stdout, line))
|
|
{
|
|
// print full line thread-safe
|
|
std::stringstream printLine;
|
|
printLine << prefix << line << "\n";
|
|
std::cout << printLine.str() << std::flush;
|
|
|
|
out << prefix << line << "\n";
|
|
}
|
|
|
|
c.wait();
|
|
|
|
// Capture exit code
|
|
result.exit_code = c.exit_code();
|
|
out << prefix << " Exit code: " << result.exit_code << std::endl;
|
|
|
|
result.console_out = out.str();
|
|
|
|
// Return result
|
|
return result;
|
|
}
|
|
|
|
} /* namespace tools */
|
|
} /* namespace mq */
|
|
} /* namespace fair */
|
|
|
|
#endif /* FAIR_MQ_TOOLS_PROCESS_H */
|