mirror of
https://github.com/asterisk/asterisk.git
synced 2025-10-23 04:58:48 +00:00
Remove dispatch object allocation from Stasis publishing
While looking for areas for performance improvement, I realized that an unused feature in Stasis was negatively impacting performance. When a message is sent to a subscriber, a dispatch object is allocated for the dispatch, containing the topic the message was published to, the subscriber the message is being sent to, and the message itself. The topic is actually unused by any subscriber in Asterisk today. And the subscriber is associated with the taskprocessor the message is being dispatched to. First, this patch removes the unused topic parameter from Stasis subscription callbacks. Second, this patch introduces the concept of taskprocessor local data, data that may be set on a taskprocessor and provided along with the data pointer when a task is pushed using the ast_taskprocessor_push_local() call. This allows the task to have both data specific to that taskprocessor, in addition to data specific to that invocation. With those two changes, the dispatch object can be removed completely, and the message is simply refcounted and sent directly to the taskprocessor. Review: https://reviewboard.asterisk.org/r/2884/ git-svn-id: https://origsvn.digium.com/svn/asterisk/branches/12@400181 65c4cc65-6c06-0410-ace0-fbb531ad65f3
This commit is contained in:
109
main/stasis.c
109
main/stasis.c
@@ -249,7 +249,6 @@ static void subscription_dtor(void *obj)
|
||||
* \param message Message to send.
|
||||
*/
|
||||
static void subscription_invoke(struct stasis_subscription *sub,
|
||||
struct stasis_topic *topic,
|
||||
struct stasis_message *message)
|
||||
{
|
||||
/* Notify that the final message has been received */
|
||||
@@ -260,7 +259,7 @@ static void subscription_invoke(struct stasis_subscription *sub,
|
||||
}
|
||||
|
||||
/* Since sub is mostly immutable, no need to lock sub */
|
||||
sub->callback(sub->data, sub, topic, message);
|
||||
sub->callback(sub->data, sub, message);
|
||||
|
||||
/* Notify that the final message has been processed */
|
||||
if (stasis_subscription_final_message(sub, message)) {
|
||||
@@ -301,6 +300,9 @@ struct stasis_subscription *internal_stasis_subscribe(
|
||||
if (!sub->mailbox) {
|
||||
return NULL;
|
||||
}
|
||||
ast_taskprocessor_set_local(sub->mailbox, sub);
|
||||
/* Taskprocessor has a reference */
|
||||
ao2_ref(sub, +1);
|
||||
}
|
||||
|
||||
ao2_ref(topic, +1);
|
||||
@@ -327,6 +329,13 @@ struct stasis_subscription *stasis_subscribe(
|
||||
return internal_stasis_subscribe(topic, callback, data, 1);
|
||||
}
|
||||
|
||||
static int sub_cleanup(void *data)
|
||||
{
|
||||
struct stasis_subscription *sub = data;
|
||||
ao2_cleanup(sub);
|
||||
return 0;
|
||||
}
|
||||
|
||||
struct stasis_subscription *stasis_unsubscribe(struct stasis_subscription *sub)
|
||||
{
|
||||
/* The subscription may be the last ref to this topic. Hold
|
||||
@@ -349,6 +358,11 @@ struct stasis_subscription *stasis_unsubscribe(struct stasis_subscription *sub)
|
||||
/* Now let everyone know about the unsubscribe */
|
||||
send_subscription_unsubscribe(topic, sub);
|
||||
|
||||
/* When all that's done, remove the ref the mailbox has on the sub */
|
||||
if (sub->mailbox) {
|
||||
ast_taskprocessor_push(sub->mailbox, sub_cleanup, sub);
|
||||
}
|
||||
|
||||
/* Unsubscribing unrefs the subscription */
|
||||
ao2_cleanup(sub);
|
||||
return NULL;
|
||||
@@ -475,93 +489,39 @@ static int topic_remove_subscription(struct stasis_topic *topic, struct stasis_s
|
||||
return ast_vector_remove_elem_unordered(topic->subscribers, sub);
|
||||
}
|
||||
|
||||
/*!
|
||||
* \internal
|
||||
* \brief Information needed to dispatch a message to a subscription
|
||||
*/
|
||||
struct dispatch {
|
||||
/*! Topic message was published to */
|
||||
struct stasis_topic *topic;
|
||||
/*! The message itself */
|
||||
struct stasis_message *message;
|
||||
/*! Subscription receiving the message */
|
||||
struct stasis_subscription *sub;
|
||||
};
|
||||
|
||||
static void dispatch_dtor(struct dispatch *dispatch)
|
||||
{
|
||||
ao2_cleanup(dispatch->topic);
|
||||
ao2_cleanup(dispatch->message);
|
||||
ao2_cleanup(dispatch->sub);
|
||||
|
||||
ast_free(dispatch);
|
||||
}
|
||||
|
||||
static struct dispatch *dispatch_create(struct stasis_topic *topic, struct stasis_message *message, struct stasis_subscription *sub)
|
||||
{
|
||||
struct dispatch *dispatch;
|
||||
|
||||
ast_assert(topic != NULL);
|
||||
ast_assert(message != NULL);
|
||||
ast_assert(sub != NULL);
|
||||
|
||||
dispatch = ast_malloc(sizeof(*dispatch));
|
||||
if (!dispatch) {
|
||||
return NULL;
|
||||
}
|
||||
|
||||
dispatch->topic = topic;
|
||||
ao2_ref(topic, +1);
|
||||
|
||||
dispatch->message = message;
|
||||
ao2_ref(message, +1);
|
||||
|
||||
dispatch->sub = sub;
|
||||
ao2_ref(sub, +1);
|
||||
|
||||
return dispatch;
|
||||
}
|
||||
|
||||
/*!
|
||||
* \brief Dispatch a message to a subscriber
|
||||
* \param data \ref dispatch object
|
||||
* \return 0
|
||||
*/
|
||||
static int dispatch_exec(void *data)
|
||||
static int dispatch_exec(struct ast_taskprocessor_local *local)
|
||||
{
|
||||
struct dispatch *dispatch = data;
|
||||
struct stasis_subscription *sub = local->local_data;
|
||||
struct stasis_message *message = local->data;
|
||||
|
||||
subscription_invoke(dispatch->sub, dispatch->topic, dispatch->message);
|
||||
dispatch_dtor(dispatch);
|
||||
subscription_invoke(sub, message);
|
||||
ao2_cleanup(message);
|
||||
|
||||
return 0;
|
||||
}
|
||||
|
||||
static void dispatch_message(struct stasis_subscription *sub,
|
||||
struct stasis_topic *publisher_topic, struct stasis_message *message)
|
||||
struct stasis_message *message)
|
||||
{
|
||||
if (sub->mailbox) {
|
||||
struct dispatch *dispatch;
|
||||
|
||||
dispatch = dispatch_create(publisher_topic, message, sub);
|
||||
if (!dispatch) {
|
||||
ao2_bump(message);
|
||||
if (ast_taskprocessor_push_local(sub->mailbox, dispatch_exec, message) != 0) {
|
||||
/* Push failed; ugh. */
|
||||
ast_log(LOG_DEBUG, "Dropping dispatch\n");
|
||||
return;
|
||||
}
|
||||
|
||||
if (ast_taskprocessor_push(sub->mailbox, dispatch_exec, dispatch) != 0) {
|
||||
/* Push failed; just delete the dispatch.
|
||||
*/
|
||||
ast_log(LOG_DEBUG, "Dropping dispatch\n");
|
||||
dispatch_dtor(dispatch);
|
||||
ao2_cleanup(message);
|
||||
}
|
||||
} else {
|
||||
/* Dispatch directly */
|
||||
subscription_invoke(sub, publisher_topic, message);
|
||||
subscription_invoke(sub, message);
|
||||
}
|
||||
}
|
||||
|
||||
void stasis_forward_message(struct stasis_topic *_topic, struct stasis_topic *publisher_topic, struct stasis_message *message)
|
||||
void stasis_publish(struct stasis_topic *_topic, struct stasis_message *message)
|
||||
{
|
||||
size_t i;
|
||||
/* The topic may be unref'ed by the subscription invocation.
|
||||
@@ -571,23 +531,18 @@ void stasis_forward_message(struct stasis_topic *_topic, struct stasis_topic *pu
|
||||
SCOPED_AO2LOCK(lock, topic);
|
||||
|
||||
ast_assert(topic != NULL);
|
||||
ast_assert(publisher_topic != NULL);
|
||||
ast_assert(message != NULL);
|
||||
|
||||
for (i = 0; i < ast_vector_size(topic->subscribers); ++i) {
|
||||
struct stasis_subscription *sub = ast_vector_get(topic->subscribers, i);
|
||||
struct stasis_subscription *sub =
|
||||
ast_vector_get(topic->subscribers, i);
|
||||
|
||||
ast_assert(sub != NULL);
|
||||
|
||||
dispatch_message(sub, publisher_topic, message);
|
||||
dispatch_message(sub, message);
|
||||
}
|
||||
}
|
||||
|
||||
void stasis_publish(struct stasis_topic *topic, struct stasis_message *message)
|
||||
{
|
||||
stasis_forward_message(topic, topic, message);
|
||||
}
|
||||
|
||||
/*!
|
||||
* \brief Forwarding information
|
||||
*
|
||||
@@ -748,7 +703,7 @@ static void send_subscription_unsubscribe(struct stasis_topic *topic,
|
||||
stasis_publish(topic, msg);
|
||||
|
||||
/* Now we have to dispatch to the subscription itself */
|
||||
dispatch_message(sub, topic, msg);
|
||||
dispatch_message(sub, msg);
|
||||
}
|
||||
|
||||
struct topic_pool_entry {
|
||||
|
Reference in New Issue
Block a user