X-Git-Url: https://git.lttng.org/?p=lttng-tools.git;a=blobdiff_plain;f=src%2Fbin%2Flttng-relayd%2Fviewer-stream.h;h=e7d733b9e60ecb10479992b5e74042089e824cdb;hp=793fa7ccae67c0d22952222541fe707dbb5efc46;hb=80516611b6f19201b1e173fb448935aca7a9e668;hpb=2f8f53af90479595d530f8f02e71dd0b9fb810ee diff --git a/src/bin/lttng-relayd/viewer-stream.h b/src/bin/lttng-relayd/viewer-stream.h index 793fa7cca..e7d733b9e 100644 --- a/src/bin/lttng-relayd/viewer-stream.h +++ b/src/bin/lttng-relayd/viewer-stream.h @@ -1,24 +1,15 @@ +#ifndef _VIEWER_STREAM_H +#define _VIEWER_STREAM_H + /* - * Copyright (C) 2013 - Julien Desfossez - * David Goulet - * - * This program is free software; you can redistribute it and/or modify it - * under the terms of the GNU General Public License, version 2 only, as - * published by the Free Software Foundation. + * Copyright (C) 2013 Julien Desfossez + * Copyright (C) 2013 David Goulet + * Copyright (C) 2015 Mathieu Desnoyers * - * This program is distributed in the hope that it will be useful, but WITHOUT - * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or - * FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License for - * more details. + * SPDX-License-Identifier: GPL-2.0-only * - * You should have received a copy of the GNU General Public License along with - * this program; if not, write to the Free Software Foundation, Inc., 51 - * Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA. */ -#ifndef _VIEWER_STREAM_H -#define _VIEWER_STREAM_H - #include #include #include @@ -27,61 +18,77 @@ #include "ctf-trace.h" #include "lttng-viewer-abi.h" +#include "stream.h" -/* Stub */ struct relay_stream; /* - * Shadow copy of the relay_stream structure for the viewer side. The only - * fields updated by the writer (streaming side) after allocation are : - * total_index_received and close_flag. Everything else is updated by the - * reader (viewer side). + * The viewer stream's lifetime is the intersection of their viewer connection's + * lifetime and the duration during which at least: + * a) their input source is still active + * b) they still have data left to send to the client. + * + * This means that both the sessiond/consumerd connection or the viewer + * connection may tear down (and unpublish) a relay_viewer_stream. + * + * Viewer stream updates are protected by their associated stream's lock. */ struct relay_viewer_stream { - uint64_t stream_handle; - uint64_t session_id; - int read_fd; - int index_read_fd; - char *path_name; - char *channel_name; - uint64_t last_sent_index; - uint64_t total_index_received; - uint64_t tracefile_count; - uint64_t tracefile_count_current; - /* Stop after reading this tracefile. */ - uint64_t tracefile_count_last; - struct lttng_ht_node_u64 stream_n; - struct rcu_head rcu_node; - struct ctf_trace *ctf_trace; + struct urcu_ref ref; + + /* Back ref to stream. */ + struct relay_stream *stream; + + struct { + struct fs_handle *handle; + struct lttng_trace_chunk *trace_chunk; + } stream_file; + /* index file from which to read the index data. */ + struct lttng_index_file *index_file; /* - * This lock blocks only when the writer is about to start overwriting - * a file currently read by the reader. + * Last seen rotation count in stream. * - * This is nested INSIDE the viewer_stream_rotation_lock. - */ - pthread_mutex_t overwrite_lock; - /* Information telling us if the stream is a metadata stream. */ - unsigned int metadata_flag:1; - /* - * Information telling us that the stream is closed in write, so - * we don't expect new indexes and we can read up to EOF. + * Sampled on every change to the viewer stream trace chunk, + * this allows the live server to determine if it saw the latest + * rotation that occurred on the receiving end. */ - unsigned int close_write_flag:1; + uint64_t last_seen_rotation_count; + + char *path_name; + char *channel_name; + + uint64_t current_tracefile_id; + /* - * If the streaming side closes a FD in use in the viewer side, - * it sets this flag to inform that it is a normal error. + * Counts the number of sent indexes. The "tag" associated + * with an index to send is the current index_received_seqcount, + * because we increment index_received_seqcount after sending + * each index. This index_received_seqcount counter can also be + * updated when catching up with the producer. */ - unsigned int abort_flag:1; + uint64_t index_sent_seqcount; + /* Indicates if this stream has been sent to a viewer client. */ - unsigned int sent_flag:1; + bool sent_flag; + /* For metadata stream, how much metadata has been sent. */ + uint64_t metadata_sent; + + struct lttng_ht_node_u64 stream_n; + struct rcu_head rcu_node; }; struct relay_viewer_stream *viewer_stream_create(struct relay_stream *stream, + struct lttng_trace_chunk *viewer_trace_chunk, enum lttng_viewer_seek seek_t); -struct relay_viewer_stream *viewer_stream_find_by_id(uint64_t id); -void viewer_stream_destroy(struct relay_viewer_stream *stream); -void viewer_stream_delete(struct relay_viewer_stream *stream); -int viewer_stream_rotate(struct relay_viewer_stream *vstream, - struct relay_stream *stream); + +struct relay_viewer_stream *viewer_stream_get_by_id(uint64_t id); +bool viewer_stream_get(struct relay_viewer_stream *vstream); +void viewer_stream_put(struct relay_viewer_stream *vstream); +int viewer_stream_rotate(struct relay_viewer_stream *vstream); +bool viewer_stream_is_tracefile_seq_readable(struct relay_viewer_stream *vstream, + uint64_t seq); +void print_viewer_streams(void); +void viewer_stream_close_files(struct relay_viewer_stream *vstream); +void viewer_stream_sync_tracefile_array_tail(struct relay_viewer_stream *vstream); #endif /* _VIEWER_STREAM_H */