X-Git-Url: https://git.lttng.org/?p=lttng-tools.git;a=blobdiff_plain;f=src%2Fbin%2Flttng-relayd%2Findex.h;h=4364f12ffdb096179e6e736dd821388c4c4bb028;hp=e7f9cdbd07d98ebafacc1dfb122fb525e18778f3;hb=ce0b1d61919f37517a6212f7af2afe0fa1b1dcb0;hpb=5c32d0aeb3f1e338ff61f9b5c865a01763a3ec4b

diff --git a/src/bin/lttng-relayd/index.h b/src/bin/lttng-relayd/index.h
index e7f9cdbd0..4364f12ff 100644
--- a/src/bin/lttng-relayd/index.h
+++ b/src/bin/lttng-relayd/index.h
@@ -1,57 +1,73 @@
+#ifndef _RELAY_INDEX_H
+#define _RELAY_INDEX_H
+
 /*
- * Copyright (C) 2013 - Julien Desfossez <jdesfossez@efficios.com>
- *                      David Goulet <dgoulet@efficios.com>
- *
- * 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 <jdesfossez@efficios.com>
+ * Copyright (C) 2013 David Goulet <dgoulet@efficios.com>
+ * Copyright (C) 2015 Mathieu Desnoyers <mathieu.desnoyers@efficios.com>
  *
- * 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 _RELAY_INDEX_H
-#define _RELAY_INDEX_H
-
 #include <inttypes.h>
 #include <pthread.h>
 
 #include <common/hashtable/hashtable.h>
 #include <common/index/index.h>
 
+struct relay_stream;
+struct relay_connection;
+struct lttcomm_relayd_index;
+
 struct relay_index {
-	/* FD on which to write the index data. */
-	int fd;
 	/*
-	 * When destroying this object, this fd is checked and if valid, close it
-	 * so this is basically a lazy close of the previous fd corresponding to
-	 * the same stream id. This is used for the rotate file feature.
+	 * index lock nests inside stream lock.
 	 */
-	int to_close_fd;
+	struct urcu_ref ref;		/* Reference from getters. */
+	struct relay_stream *stream;	/* Back ref to stream */
+
+	pthread_mutex_t lock;
+	/*
+	 * index file on which to write the index data. May differ from
+	 * stream->index_file due to tracefile rotation.
+	 */
+	struct lttng_index_file *index_file;
 
 	/* Index packet data. This is the data that is written on disk. */
 	struct ctf_packet_index index_data;
+	/* Data + padding size of this packet, filled by the data thread. */
+	uint64_t total_size;
+
+	bool has_index_data;
+	bool flushed;
+	bool in_hash_table;
 
-	/* key1 = stream_id, key2 = net_seq_num */
-	struct lttng_ht_node_two_u64 index_n;
-	struct rcu_head rcu_node;
-	pthread_mutex_t mutex;
+	/*
+	 * Node within indexes_ht that corresponds to this struct
+	 * relay_index. Indexed by net_seq_num, which is unique for this
+	 * index across the stream.
+	 */
+	struct lttng_ht_node_u64 index_n;
+	struct rcu_head rcu_node;	/* For call_rcu teardown. */
 };
 
-struct relay_index *relay_index_create(uint64_t stream_id,
+struct relay_index *relay_index_get_by_id_or_create(struct relay_stream *stream,
 		uint64_t net_seq_num);
-struct relay_index *relay_index_find(uint64_t stream_id, uint64_t net_seq_num);
-void relay_index_add(struct relay_index *index, struct relay_index **_index);
-int relay_index_write(int fd, struct relay_index *index);
-void relay_index_free(struct relay_index *index);
-void relay_index_free_safe(struct relay_index *index);
-void relay_index_delete(struct relay_index *index);
-void relay_index_destroy_by_stream_id(uint64_t stream_id);
+void relay_index_put(struct relay_index *index);
+int relay_index_set_file(struct relay_index *index,
+		struct lttng_index_file *index_file,
+		uint64_t data_offset);
+int relay_index_set_data(struct relay_index *index,
+		const struct ctf_packet_index *data);
+int relay_index_try_flush(struct relay_index *index);
+
+void relay_index_close_all(struct relay_stream *stream);
+void relay_index_close_partial_fd(struct relay_stream *stream);
+uint64_t relay_index_find_last(struct relay_stream *stream);
+int relay_index_switch_all_files(struct relay_stream *stream);
+int relay_index_set_control_data(struct relay_index *index,
+		const struct lttcomm_relayd_index *data,
+		unsigned int minor_version);
 
 #endif /* _RELAY_INDEX_H */