[asterisk-commits] qwell: branch qwell/router_removal r390471 - in /team/qwell/router_removal: i...

SVN commits to the Asterisk project asterisk-commits at lists.digium.com
Wed Jun 5 09:41:36 CDT 2013


Author: qwell
Date: Wed Jun  5 09:41:35 2013
New Revision: 390471

URL: http://svnview.digium.com/svn/asterisk?view=rev&rev=390471
Log:
Convert to ao2, add removal.  Still a few more things to do.

Modified:
    team/qwell/router_removal/include/asterisk/stasis_message_router.h
    team/qwell/router_removal/main/stasis_message_router.c

Modified: team/qwell/router_removal/include/asterisk/stasis_message_router.h
URL: http://svnview.digium.com/svn/asterisk/team/qwell/router_removal/include/asterisk/stasis_message_router.h?view=diff&rev=390471&r1=390470&r2=390471
==============================================================================
--- team/qwell/router_removal/include/asterisk/stasis_message_router.h (original)
+++ team/qwell/router_removal/include/asterisk/stasis_message_router.h Wed Jun  5 09:41:35 2013
@@ -47,9 +47,12 @@
 
 /*!
  * \brief Create a new message router object.
+ *
  * \param topic Topic to subscribe route to.
+ *
  * \return New \ref stasis_message_router.
  * \return \c NULL on error.
+ *
  * \since 12
  */
 struct stasis_message_router *stasis_message_router_create(
@@ -59,6 +62,7 @@
  * \brief Unsubscribe the router from the upstream topic.
  *
  * \param router Router to unsubscribe.
+ *
  * \since 12
  */
 void stasis_message_router_unsubscribe(struct stasis_message_router *router);
@@ -71,6 +75,7 @@
  * vs. stasis_message_router_unsubscribe().
  *
  * \param router Router to unsubscribe.
+ *
  * \since 12
  */
 void stasis_message_router_unsubscribe_and_join(
@@ -80,6 +85,7 @@
  * \brief Returns whether \a router has received its final message.
  *
  * \param router Router.
+ *
  * \return True (non-zero) if stasis_subscription_final_message() has been
  *         received.
  * \return False (zero) if waiting for the end.
@@ -88,10 +94,15 @@
 
 /*!
  * \brief Add a route to a message router.
+ *
  * \param router Router to add the route to.
  * \param message_type Type of message to route.
  * \param callback Callback to forard messages of \a message_type to.
  * \param data Data pointer to pass to \a callback.
+ *
+ * \retval 0 on success
+ * \retval -1 on failure
+ *
  * \since 12
  */
 int stasis_message_router_add(struct stasis_message_router *router,
@@ -100,10 +111,29 @@
 			      void *data);
 
 /*!
+ * \brief Remove a route from a message router.
+ *
+ * \param router Router to remove the route from.
+ * \param message_type Type of message to route.
+ *
+ * \retval 0 on success
+ * \retval -1 on failure
+ *
+ * \since 12
+ */
+int stasis_message_router_remove(struct stasis_message_router *router,
+			      struct stasis_message_type *message_type);
+
+/*!
  * \brief Sets the default route of a router.
+ *
  * \param router Router to set the default route of.
  * \param callback Callback to forard messages which otherwise have no home.
  * \param data Data pointer to pass to \a callback.
+ *
+ * \retval 0 on success
+ * \retval -1 on failure
+ *
  * \since 12
  */
 int stasis_message_router_set_default(struct stasis_message_router *router,

Modified: team/qwell/router_removal/main/stasis_message_router.c
URL: http://svnview.digium.com/svn/asterisk/team/qwell/router_removal/main/stasis_message_router.c?view=diff&rev=390471&r1=390470&r2=390471
==============================================================================
--- team/qwell/router_removal/main/stasis_message_router.c (original)
+++ team/qwell/router_removal/main/stasis_message_router.c Wed Jun  5 09:41:35 2013
@@ -33,8 +33,6 @@
 
 #include "asterisk/astobj2.h"
 #include "asterisk/stasis_message_router.h"
-
-#define INITIAL_ROUTES_MAX 8
 
 /*! \internal */
 struct stasis_message_route {
@@ -54,36 +52,58 @@
 	route->message_type = NULL;
 }
 
+static int route_hash(const void *obj, const int flags)
+{
+	const struct stasis_message_route *route = obj;
+
+	return ast_str_hash(stasis_message_type_name(route->message_type));
+}
+
+static int route_cmp(void *obj, void *arg, int flags)
+{
+	const struct stasis_message_route *left = obj;
+	const struct stasis_message_route *right = arg;
+
+	return (left->message_type == right->message_type) ? CMP_MATCH | CMP_STOP : 0;
+}
+
 /*! \internal */
 struct stasis_message_router {
 	/*! Subscription to the upstream topic */
 	struct stasis_subscription *subscription;
-	/*! Variable length array of the routes */
-	struct stasis_message_route **routes;
+	/*! Subscribed routes */
+	struct ao2_container *routes;
 	/*! Route of last resort */
 	struct stasis_message_route *default_route;
-	/*! Allocated length of the routes array */
-	size_t num_routes_max;
-	/*! Current size of the routes array */
-	size_t num_routes_current;
 };
 
 static void router_dtor(void *obj)
 {
 	struct stasis_message_router *router = obj;
-	size_t i;
 
 	ast_assert(!stasis_subscription_is_subscribed(router->subscription));
 	ast_assert(stasis_subscription_is_done(router->subscription));
 	router->subscription = NULL;
-	for (i = 0; i < router->num_routes_current; ++i) {
-		ao2_cleanup(router->routes[i]);
-		router->routes[i] = NULL;
-	}
-	ast_free(router->routes);
+
+	ao2_cleanup(router->routes);
 	router->routes = NULL;
+
 	ao2_cleanup(router->default_route);
 	router->default_route = NULL;
+}
+
+static struct stasis_message_route *find_route(struct stasis_message_router *router, struct stasis_message_type *message_type)
+{
+	struct stasis_message_route route;
+
+	if (message_type == NULL) {
+		return NULL;
+	}
+
+	memset(&route, 0, sizeof(route));
+	route.message_type = message_type;
+
+	return ao2_find(router->routes, &route, OBJ_POINTER);
 }
 
 static void router_dispatch(void *data,
@@ -95,28 +115,14 @@
 	RAII_VAR(struct stasis_message_router *, router_needs_cleanup, NULL, ao2_cleanup);
 	RAII_VAR(struct stasis_message_route *, route, NULL, ao2_cleanup);
 	struct stasis_message_type *type = stasis_message_type(message);
-	size_t i;
 
 	{
 		SCOPED_AO2LOCK(lock, router);
 
-		/* We don't expect many message types, so a simple loop should
-		 * be adequate, even if the complexity is O(n). Sorting the list
-		 * would be an easy way to bring that down to O(log(n)). Using a
-		 * hashtable/ao2_container could be even better. Just be sure to
-		 * profile before you optimize!
-		 */
-		route = router->default_route;
-		for (i = 0; i < router->num_routes_current; ++i) {
-			if (router->routes[i]->message_type == type) {
-				route = router->routes[i];
-				break;
+		if (!(route = find_route(router, type))) {
+			if ((route = router->default_route)) {
+				ao2_ref(route, +1);
 			}
-		}
-
-		/* Ref the route before leaving the scoped lock */
-		if (route) {
-			ao2_ref(route, +1);
 		}
 	}
 
@@ -141,10 +147,7 @@
 		return NULL;
 	}
 
-	router->num_routes_max = INITIAL_ROUTES_MAX;
-	router->routes = ast_calloc(router->num_routes_max,
-				    sizeof(*router->routes));
-	if (!router->routes) {
+	if (!(router->routes = ao2_container_alloc(7, route_hash, route_cmp))) {
 		return NULL;
 	}
 
@@ -212,31 +215,21 @@
 static int add_route(struct stasis_message_router *router,
 		     struct stasis_message_route *route)
 {
-	struct stasis_message_route **routes;
-	size_t i;
 	SCOPED_AO2LOCK(lock, router);
-
-	/* Check for route conflicts */
-	for (i = 0; i < router->num_routes_current; ++i) {
-		if (router->routes[i]->message_type == route->message_type) {
-			return -1;
-		}
-	}
-
-	/* Increase list size, if needed */
-	if (router->num_routes_current + 1 > router->num_routes_max) {
-		routes = realloc(router->routes,
-				 2 * router->num_routes_max * sizeof(*routes));
-		if (!routes) {
-			return -1;
-		}
-		router->routes = routes;
-		router->num_routes_max *= 2;
-	}
-
-
-	ao2_ref(route, +1);
-	router->routes[router->num_routes_current++] = route;
+	RAII_VAR(struct stasis_message_route *, existing_route, NULL, ao2_cleanup);
+
+	if ((existing_route = find_route(router, route->message_type))) {
+		return -1;
+	}
+
+	ao2_link(router->routes, route);
+	return 0;
+}
+
+static int remove_route(struct stasis_message_router *router,
+			struct stasis_message_route *route)
+{
+	ao2_unlink(router->routes, route);
 	return 0;
 }
 
@@ -255,6 +248,19 @@
 	return add_route(router, route);
 }
 
+int stasis_message_router_remove(struct stasis_message_router *router,
+			      struct stasis_message_type *message_type)
+{
+	SCOPED_AO2LOCK(lock, router);
+	RAII_VAR(struct stasis_message_route *, route, NULL, ao2_cleanup);
+
+	if (!(route = find_route(router, message_type))) {
+		return -1;
+	}
+
+	return remove_route(router, route);
+}
+
 int stasis_message_router_set_default(struct stasis_message_router *router,
 				      stasis_subscription_cb callback,
 				      void *data)




More information about the asterisk-commits mailing list