add counters for WriteBufferFromHTTP

Author: CheSema

src/Common/ProfileEvents.cpp

@@ -985,6 +985,9 @@ The server successfully detected this situation and will download merged part fr
     M(ReadWriteBufferFromHTTPRequestsSent, "Number of HTTP requests sent by ReadWriteBufferFromHTTP", ValueType::Number) \
     M(ReadWriteBufferFromHTTPBytes, "Total size of payload bytes received and sent by ReadWriteBufferFromHTTP. Doesn't include HTTP headers.", ValueType::Bytes) \
     \
+    M(WriteBufferFromHTTPRequestsSent, "Number of HTTP requests sent by WriteBufferFromHTTP", ValueType::Number) \
+    M(WriteBufferFromHTTPBytes, "Total size of payload bytes received and sent by WriteBufferFromHTTP. Doesn't include HTTP headers.", ValueType::Bytes) \
+    \
     M(ConcurrencyControlSlotsGranted, "Number of CPU slot granted according to guarantee of 1 thread per query and for queries with setting 'use_concurrency_control' = 0", ValueType::Number) \
     M(ConcurrencyControlSlotsDelayed, "Number of CPU slot not granted initially and required to wait for a free CPU slot", ValueType::Number) \
     M(ConcurrencyControlSlotsAcquired, "Total number of CPU slot acquired", ValueType::Number) \

src/IO/ReadWriteBufferFromHTTP.cpp

@@ -1,6 +1,4 @@
 #include "ReadWriteBufferFromHTTP.h"
-#include <sstream>
-#include <string_view>
 
 #include <IO/HTTPCommon.h>
 #include <Common/NetException.h>

src/IO/WriteBufferFromHTTP.cpp

@@ -5,6 +5,12 @@
 #include <Interpreters/Context.h>
 
 
+namespace ProfileEvents
+{
+    extern const Event WriteBufferFromHTTPRequestsSent;
+    extern const Event WriteBufferFromHTTPBytes;
+}
+
 namespace DB
 {
 
@@ -39,9 +45,16 @@ WriteBufferFromHTTP::WriteBufferFromHTTP(
 
     LOG_TRACE((getLogger("WriteBufferToHTTP")), "Sending request to {}", uri.toString());
 
+    ProfileEvents::increment(ProfileEvents::WriteBufferFromHTTPRequestsSent);
     ostr = &session->sendRequest(request);
 }
 
+void WriteBufferFromHTTP::nextImpl()
+{
+    ProfileEvents::increment(ProfileEvents::WriteBufferFromHTTPBytes, offset());
+    WriteBufferFromOStream::nextImpl();
+}
+
 void WriteBufferFromHTTP::finalizeImpl()
 {
     // Make sure the content in the buffer has been flushed

src/IO/WriteBufferFromHTTP.h

@@ -29,6 +29,9 @@ class WriteBufferFromHTTP : public WriteBufferFromOStream
                                  size_t buffer_size_ = DBMS_DEFAULT_BUFFER_SIZE,
                                  ProxyConfiguration proxy_configuration = {});
 
+    // Counts the counter WriteBufferFromHTTPBytes
+    void nextImpl() override;
+
     /// Receives response from the server after sending all data.
     void finalizeImpl() override;