X-Git-Url: https://git.lttng.org/?p=lttng-tools.git;a=blobdiff_plain;f=include%2Flttng%2Faction%2Faction-internal.h;h=55e34163b1cb14749bd6451c51605cf766da5d51;hp=152b27eabbc38518c146ce15219b7153efdf0f2f;hb=588c4b0d153e8919b8d9159b69ce0db8fbe52af0;hpb=c0a66c84b5b2484b75798aec7543b680b4d4ab6c

diff --git a/include/lttng/action/action-internal.h b/include/lttng/action/action-internal.h
index 152b27eab..55e34163b 100644
--- a/include/lttng/action/action-internal.h
+++ b/include/lttng/action/action-internal.h
@@ -8,14 +8,18 @@
 #ifndef LTTNG_ACTION_INTERNAL_H
 #define LTTNG_ACTION_INTERNAL_H
 
-#include <lttng/action/action.h>
-#include <common/macros.h>
 #include <common/buffer-view.h>
 #include <common/dynamic-buffer.h>
-#include <common/sessiond-comm/payload-view.h>
-#include <common/sessiond-comm/payload.h>
+#include <common/macros.h>
+#include <common/payload-view.h>
+#include <common/payload.h>
+#include <lttng/lttng.h>
+#include <pthread.h>
 #include <stdbool.h>
 #include <sys/types.h>
+#include <urcu/ref.h>
+
+struct lttng_rate_policy;
 
 typedef bool (*action_validate_cb)(struct lttng_action *action);
 typedef void (*action_destroy_cb)(struct lttng_action *action);
@@ -26,13 +30,37 @@ typedef bool (*action_equal_cb)(const struct lttng_action *a,
 typedef ssize_t (*action_create_from_payload_cb)(
 		struct lttng_payload_view *view,
 		struct lttng_action **action);
+typedef const struct lttng_rate_policy *(*action_get_rate_policy_cb)(
+		const struct lttng_action *action);
+typedef enum lttng_action_status (*action_add_error_query_results_cb)(
+		const struct lttng_action *action,
+		struct lttng_error_query_results *results);
 
 struct lttng_action {
+	struct urcu_ref ref;
 	enum lttng_action_type type;
 	action_validate_cb validate;
 	action_serialize_cb serialize;
 	action_equal_cb equal;
 	action_destroy_cb destroy;
+	action_get_rate_policy_cb get_rate_policy;
+	action_add_error_query_results_cb add_error_query_results;
+
+	/* Internal use only. */
+
+	/* The number of time the actions was enqueued for execution. */
+	uint64_t execution_request_counter;
+	/*
+	 * The number of time the action was actually executed.
+	 * Action rate policy can impact on this number.
+	 * */
+	uint64_t execution_counter;
+	/*
+	 * The number of time the action execution failed.
+	 * An unsigned long is used to use a type which makes atomic
+	 * operations possible.
+	 */
+	unsigned long execution_failure_counter;
 };
 
 struct lttng_action_comm {
@@ -46,7 +74,9 @@ void lttng_action_init(struct lttng_action *action,
 		action_validate_cb validate,
 		action_serialize_cb serialize,
 		action_equal_cb equal,
-		action_destroy_cb destroy);
+		action_destroy_cb destroy,
+		action_get_rate_policy_cb get_rate_policy,
+		action_add_error_query_results_cb add_error_query_results);
 
 LTTNG_HIDDEN
 bool lttng_action_validate(struct lttng_action *action);
@@ -59,12 +89,44 @@ LTTNG_HIDDEN
 ssize_t lttng_action_create_from_payload(struct lttng_payload_view *view,
 		struct lttng_action **action);
 
-LTTNG_HIDDEN
-enum lttng_action_type lttng_action_get_type_const(
-		const struct lttng_action *action);
-
 LTTNG_HIDDEN
 bool lttng_action_is_equal(const struct lttng_action *a,
 		const struct lttng_action *b);
 
+LTTNG_HIDDEN
+void lttng_action_get(struct lttng_action *action);
+
+LTTNG_HIDDEN
+void lttng_action_put(struct lttng_action *action);
+
+LTTNG_HIDDEN
+const char* lttng_action_type_string(enum lttng_action_type action_type);
+
+LTTNG_HIDDEN
+void lttng_action_increase_execution_request_count(struct lttng_action *action);
+
+LTTNG_HIDDEN
+void lttng_action_increase_execution_count(struct lttng_action *action);
+
+LTTNG_HIDDEN
+void lttng_action_increase_execution_failure_count(struct lttng_action *action);
+
+LTTNG_HIDDEN
+bool lttng_action_should_execute(const struct lttng_action *action);
+
+LTTNG_HIDDEN
+enum lttng_action_status lttng_action_add_error_query_results(
+		const struct lttng_action *action,
+		struct lttng_error_query_results *results);
+
+/*
+ * For use by the various lttng_action implementation. Implements the default
+ * behavior to the generic error "execution failure counter" that all actions
+ * (except group, which passes-through) provide.
+ */
+LTTNG_HIDDEN
+enum lttng_action_status lttng_action_generic_add_error_query_results(
+		const struct lttng_action *action,
+		struct lttng_error_query_results *results);
+
 #endif /* LTTNG_ACTION_INTERNAL_H */