vbatts/cri-o
1
0
Fork 0
Code Issues Pull requests Releases Wiki Activity

conmon: Track the number of bytes written to the container log file

Browse source 
Signed-off-by: Mrunal Patel <mrunalp@gmail.com>
This commit is contained in:
Mrunal Patel 2017-09-25 15:35:25 -07:00
parent 48d0706a49
commit 82899bdb4e
1 changed files with 35 additions and 9 deletions
Download patch file Download diff file Expand all files Collapse all files

36
conmon/conmon.c
View file

@ -285,11 +285,13 @@ const char *stdpipe_name(stdpipe_t pipe)
* line in buf, and will partially write the final line of the log if buf is * line in buf, and will partially write the final line of the log if buf is
* not terminated by a newline. * not terminated by a newline.
*/ */
int write_k8s_log(int fd, stdpipe_t pipe, const char *buf, ssize_t buflen) static int write_k8s_log(int fd, stdpipe_t pipe, const char *buf, ssize_t buflen)
{ {
char tsbuf[TSBUFLEN]; char tsbuf[TSBUFLEN];
static stdpipe_t trailing_line = NO_PIPE; static stdpipe_t trailing_line = NO_PIPE;
writev_buffer_t bufv = {0}; writev_buffer_t bufv = {0};
static int64_t bytes_written = 0;
int64_t bytes_to_be_written = 0;
/* /*
* Use the same timestamp for every line of the log in this buffer. * Use the same timestamp for every line of the log in this buffer.
@ -303,6 +305,8 @@ int write_k8s_log(int fd, stdpipe_t pipe, const char *buf, ssize_t buflen)
while (buflen > 0) { while (buflen > 0) {
const char *line_end = NULL; const char *line_end = NULL;
ptrdiff_t line_len = 0; ptrdiff_t line_len = 0;
bool insert_newline = FALSE;
bool insert_timestamp = FALSE;
/* Find the end of the line, or alternatively the end of the buffer. */ /* Find the end of the line, or alternatively the end of the buffer. */
line_end = memchr(buf, '\n', buflen); line_end = memchr(buf, '\n', buflen);
@ -310,12 +314,15 @@ int write_k8s_log(int fd, stdpipe_t pipe, const char *buf, ssize_t buflen)
line_end = &buf[buflen-1]; line_end = &buf[buflen-1];
line_len = line_end - buf + 1; line_len = line_end - buf + 1;
bytes_to_be_written = line_len;
if (trailing_line != pipe) {
/* /*
* Write the (timestamp, stream) tuple if there isn't any trailing * Write the (timestamp, stream) tuple if there isn't any trailing
* output from the previous line (or if there is trailing output but * output from the previous line (or if there is trailing output but
* the current buffer being printed is from a different pipe). * the current buffer being printed is from a different pipe).
*/ */
if (trailing_line != pipe) { insert_timestamp = TRUE;
bytes_to_be_written += (TSBUFLEN - 1);
/* /*
* If there was a trailing line from a different pipe, prepend a * If there was a trailing line from a different pipe, prepend a
* newline to split it properly. This technically breaks the flow * newline to split it properly. This technically breaks the flow
@ -323,9 +330,24 @@ int write_k8s_log(int fd, stdpipe_t pipe, const char *buf, ssize_t buflen)
* wasn't one output) but without modifying the file in a * wasn't one output) but without modifying the file in a
* non-append-only way there's not much we can do. * non-append-only way there's not much we can do.
*/ */
if ((trailing_line != NO_PIPE && if (trailing_line != NO_PIPE) {
writev_buffer_append_segment(fd, &bufv, "\n", -1) < 0) || insert_newline = TRUE;
writev_buffer_append_segment(fd, &bufv, tsbuf, -1) < 0) { bytes_to_be_written += 1;
}
}
/* Output a newline */
if (insert_newline) {
if (writev_buffer_append_segment(fd, &bufv, "\n", -1) < 0) {
nwarn("failed to write newline to log");
goto next;
}
}
/* Output a timestamp */
if (insert_timestamp) {
if (writev_buffer_append_segment(fd, &bufv, tsbuf, -1) < 0) {
nwarn("failed to write (timestamp, stream) to log"); nwarn("failed to write (timestamp, stream) to log");
goto next; goto next;
} }
@ -337,6 +359,8 @@ int write_k8s_log(int fd, stdpipe_t pipe, const char *buf, ssize_t buflen)
goto next; goto next;
} }
bytes_written += bytes_to_be_written;
/* If we did not output a full line, then we are a trailing_line. */ /* If we did not output a full line, then we are a trailing_line. */
trailing_line = (*line_end == '\n') ? NO_PIPE : pipe; trailing_line = (*line_end == '\n') ? NO_PIPE : pipe;
@ -350,6 +374,8 @@ next:
nwarn("failed to flush buffer to log"); nwarn("failed to flush buffer to log");
} }
ninfo("Total bytes written: %"PRId64"", bytes_written);
return 0; return 0;
} }