LivestatusQuery#ExecuteCommandHelper(): take CheckResultProducer::Ptr

Author: Al2Klimov

lib/livestatus/livestatusquery.cpp

@@ -570,7 +570,7 @@ void LivestatusQuery::ExecuteGetHelper(const Stream::Ptr& stream)
 	SendResponse(stream, LivestatusErrorOK, result.str());
 }
 
-void LivestatusQuery::ExecuteCommandHelper(const Stream::Ptr& stream)
+void LivestatusQuery::ExecuteCommandHelper(const CheckResultProducer::Ptr& producer, const Stream::Ptr& stream)
 {
 	{
 		std::unique_lock<std::mutex> lock(l_QueryMutex);
@@ -580,7 +580,7 @@ void LivestatusQuery::ExecuteCommandHelper(const Stream::Ptr& stream)
 
 	Log(LogNotice, "LivestatusQuery")
 		<< "Executing command: " << m_Command;
-	ExternalCommandProcessor::Execute(m_Command);
+	ExternalCommandProcessor::Execute(producer, m_Command);
 	SendResponse(stream, LivestatusErrorOK, "");
 }
 

lib/livestatus/livestatusquery.hpp

@@ -5,6 +5,7 @@
 
 #include "livestatus/filter.hpp"
 #include "livestatus/aggregator.hpp"
+#include "remote/crproducer.hpp"
 #include "base/object.hpp"
 #include "base/array.hpp"
 #include "base/stream.hpp"
@@ -76,7 +77,7 @@ class LivestatusQuery final : public Object
 	static String QuoteStringPython(const String& str);
 
 	void ExecuteGetHelper(const Stream::Ptr& stream);
-	void ExecuteCommandHelper(const Stream::Ptr& stream);
+	void ExecuteCommandHelper(const CheckResultProducer::Ptr& producer, const Stream::Ptr& stream);
 	void ExecuteErrorHelper(const Stream::Ptr& stream);
 
 	void SendResponse(const Stream::Ptr& stream, int code, const String& data);