X-Git-Url: https://git.lttng.org/?p=lttng-tools.git;a=blobdiff_plain;f=include%2Flttng%2Faction%2Faction-internal.h;h=12029e42f5d9ade44dde8f68eef6399eb2a48719;hp=57035d869d5dd47e435b8bdb448278aa114914ab;hb=7f4d5b07cf7be895b38b69073389a4fcc318ec29;hpb=10615eee4887bd869ac85885afcffd030da350be

diff --git a/include/lttng/action/action-internal.h b/include/lttng/action/action-internal.h
index 57035d869..12029e42f 100644
--- a/include/lttng/action/action-internal.h
+++ b/include/lttng/action/action-internal.h
@@ -12,12 +12,14 @@
 #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/payload-view.h>
+#include <common/payload.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);
 typedef int (*action_serialize_cb)(struct lttng_action *action,
@@ -27,6 +29,8 @@ 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);
 
 struct lttng_action {
 	struct urcu_ref ref;
@@ -35,6 +39,21 @@ struct lttng_action {
 	action_serialize_cb serialize;
 	action_equal_cb equal;
 	action_destroy_cb destroy;
+	action_get_rate_policy_cb get_rate_policy;
+
+	/* 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.
+	 */
+	uint64_t execution_failure_counter;
 };
 
 struct lttng_action_comm {
@@ -48,7 +67,8 @@ 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);
 
 LTTNG_HIDDEN
 bool lttng_action_validate(struct lttng_action *action);
@@ -61,10 +81,6 @@ 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);
@@ -78,4 +94,16 @@ 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);
+
 #endif /* LTTNG_ACTION_INTERNAL_H */