X-Git-Url: http://git.efficios.com/?p=lttng-tools.git;a=blobdiff_plain;f=src%2Fbin%2Flttng-relayd%2Fstream.h;h=ae75dacc3a061e1d77e16445f1f3d0371baf0db9;hp=e385032cb71f98d4994dfad241e915f61e0cb28a;hb=a8f9f353edcfac878ed6977381e3030608bf4e92;hpb=ce4d40839ac3beef1a58730d3636a522497bc60f

diff --git a/src/bin/lttng-relayd/stream.h b/src/bin/lttng-relayd/stream.h
index e385032cb..ae75dacc3 100644
--- a/src/bin/lttng-relayd/stream.h
+++ b/src/bin/lttng-relayd/stream.h
@@ -31,6 +31,11 @@
 #include "stream-fd.h"
 #include "tracefile-array.h"
 
+struct relay_stream_chunk_id {
+	bool is_set;
+	uint64_t value;
+};
+
 /*
  * Represents a stream in the relay
  */
@@ -47,8 +52,12 @@ struct relay_stream {
 	 * inside the ctf_trace lock.
 	 */
 	pthread_mutex_t lock;
-	uint64_t prev_seq;		/* previous data sequence number encountered. */
-	uint64_t last_net_seq_num;	/* seq num to encounter before closing. */
+	/* previous data sequence number written to disk. */
+	uint64_t prev_data_seq;
+	/* previous index sequence number written to disk. */
+	uint64_t prev_index_seq;
+	/* seq num to encounter before closing. */
+	uint64_t last_net_seq_num;
 
 	/* FD on which to write the stream data. */
 	struct stream_fd *stream_fd;
@@ -63,6 +72,11 @@ struct relay_stream {
 	uint64_t tracefile_size_current;
 	uint64_t tracefile_count;
 
+	/*
+	 * Position in the tracefile where we have the full index also on disk.
+	 */
+	uint64_t pos_after_last_complete_data_index;
+
 	/*
 	 * Counts the number of received indexes. The "tag" associated
 	 * with an index is taken before incrementing this seqcount.
@@ -129,12 +143,33 @@ struct relay_stream {
 	struct lttng_ht_node_u64 node;
 	bool in_stream_ht;		/* is stream in stream hash table. */
 	struct rcu_head rcu_node;	/* For call_rcu teardown. */
+	/*
+	 * When we have written the data and index corresponding to this
+	 * seq_num, rotate the tracefile (session rotation). The path_name is
+	 * already up-to-date.
+	 * This is set to -1ULL when no rotation is pending.
+	 *
+	 * Always access with stream lock held.
+	 */
+	uint64_t rotate_at_seq_num;
+	/*
+	 * This is the id of the chunk where we are writing to if no rotation is
+	 * pending (rotate_at_seq_num == -1ULL). If a rotation is pending, this
+	 * is the chunk_id we will have after the rotation. It must be updated
+	 * atomically with rotate_at_seq_num.
+	 *
+	 * Always access with stream lock held.
+	 *
+	 * This attribute is not set if the stream is created by a pre-2.11
+	 * consumer.
+	 */
+	struct relay_stream_chunk_id current_chunk_id;
 };
 
 struct relay_stream *stream_create(struct ctf_trace *trace,
 	uint64_t stream_handle, char *path_name,
 	char *channel_name, uint64_t tracefile_size,
-	uint64_t tracefile_count);
+	uint64_t tracefile_count, const struct relay_stream_chunk_id *chunk_id);
 
 struct relay_stream *stream_get_by_id(uint64_t stream_id);
 bool stream_get(struct relay_stream *stream);