Return to heavy-wq.c CVS log

Up to [ELWIX - Embedded LightWeight unIX -] / embedaddon / quagga / tests

Annotation of embedaddon/quagga/tests/heavy-wq.c, revision 1.1.1.1

1.1       misho       1: /*
                      2:  * This file is part of Quagga.
                      3:  *
                      4:  * Quagga is free software; you can redistribute it and/or modify it
                      5:  * under the terms of the GNU General Public License as published by the
                      6:  * Free Software Foundation; either version 2, or (at your option) any
                      7:  * later version.
                      8:  *
                      9:  * Quagga is distributed in the hope that it will be useful, but
                     10:  * WITHOUT ANY WARRANTY; without even the implied warranty of
                     11:  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
                     12:  * General Public License for more details.
                     13:  *
                     14:  * You should have received a copy of the GNU General Public License
                     15:  * along with Quagga; see the file COPYING.  If not, write to the Free
                     16:  * Software Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA
                     17:  * 02111-1307, USA.
                     18:  */
                     19: 
                     20: /* This programme shows the effects of 'heavy' long-running functions
                     21:  * on the cooperative threading model.
                     22:  *
                     23:  * Run it with a config file containing 'password whatever', telnet to it
                     24:  * (it defaults to port 4000) and enter the 'clear foo string' command.
                     25:  * then type whatever and observe that the vty interface is unresponsive
                     26:  * for quite a period of time, due to the clear_something command
                     27:  * taking a very long time to complete.
                     28:  */
                     29: #include <zebra.h>
                     30: 
                     31: #include "thread.h"
                     32: #include "vty.h"
                     33: #include "command.h"
                     34: #include "memory.h"
                     35: #include "log.h"
                     36: #include "workqueue.h"
                     37: #include <math.h>
                     38: 
                     39: extern struct thread_master *master;
                     40: static struct work_queue *heavy_wq;
                     41: 
                     42: struct heavy_wq_node
                     43: {
                     44:   char *str;
                     45:   int i;
                     46: };
                     47: 
                     48: enum
                     49: {
                     50:   ITERS_FIRST = 0,
                     51:   ITERS_ERR = 100,
                     52:   ITERS_LATER = 400,
                     53:   ITERS_PRINT = 10,
                     54:   ITERS_MAX = 1000,
                     55: };
                     56: 
                     57: static void
                     58: heavy_wq_add (struct vty *vty, const char *str, int i)
                     59: {
                     60:   struct heavy_wq_node *hn;
                     61: 
                     62:   if ((hn = XCALLOC (MTYPE_PREFIX_LIST, sizeof(struct heavy_wq_node))) == NULL)
                     63:     {
                     64:       zlog_err ("%s: unable to allocate hn", __func__);
                     65:       return;
                     66:     }
                     67:   
                     68:   hn->i = i;
                     69:   if (!(hn->str = XSTRDUP (MTYPE_PREFIX_LIST_STR, str)))
                     70:     {
                     71:       zlog_err ("%s: unable to xstrdup", __func__);
                     72:       XFREE (MTYPE_PREFIX_LIST, hn);
                     73:       return;
                     74:     }
                     75:   
                     76:   work_queue_add (heavy_wq, hn);
                     77:   
                     78:   return;
                     79: }
                     80: 
                     81: static void
                     82: slow_func_err (struct work_queue *wq, struct work_queue_item *item)
                     83: {
                     84:   printf ("%s: running error function\n", __func__);
                     85: }
                     86: 
                     87: static void
                     88: slow_func_del (struct work_queue *wq, void *data)
                     89: {
                     90:   struct heavy_wq_node *hn = data;
                     91:   assert (hn && hn->str);
                     92:   printf ("%s: %s\n", __func__, hn->str);
                     93:   XFREE (MTYPE_PREFIX_LIST_STR, hn->str);
                     94:   hn->str = NULL;  
                     95:   XFREE(MTYPE_PREFIX_LIST, hn);
                     96: }
                     97: 
                     98: static wq_item_status
                     99: slow_func (struct work_queue *wq, void *data)
                    100: {
                    101:   struct heavy_wq_node *hn = data;
                    102:   double x = 1;
                    103:   int j;
                    104:   
                    105:   assert (hn && hn->str);
                    106:   
                    107:   for (j = 0; j < 300; j++)
                    108:     x += sin(x)*j;
                    109:   
                    110:   if ((hn->i % ITERS_LATER) == 0)
                    111:     return WQ_RETRY_LATER;
                    112:   
                    113:   if ((hn->i % ITERS_ERR) == 0)
                    114:     return WQ_RETRY_NOW;
                    115:   
                    116:   if ((hn->i % ITERS_PRINT) == 0)
                    117:     printf ("%s did %d, x = %g\n", hn->str, hn->i, x);
                    118: 
                    119:   return WQ_SUCCESS;
                    120: }
                    121: 
                    122: static void
                    123: clear_something (struct vty *vty, const char *str)
                    124: {
                    125:   int i;
                    126:   
                    127:   /* this could be like iterating through 150k of route_table 
                    128:    * or worse, iterating through a list of peers, to bgp_stop them with
                    129:    * each having 150k route tables to process...
                    130:    */
                    131:   for (i = ITERS_FIRST; i < ITERS_MAX; i++)
                    132:     heavy_wq_add (vty, str, i);
                    133: }
                    134: 
                    135: DEFUN (clear_foo,
                    136:        clear_foo_cmd,
                    137:        "clear foo .LINE",
                    138:        "clear command\n"
                    139:        "arbitrary string\n")
                    140: {
                    141:   char *str;
                    142:   if (!argc)
                    143:     {
                    144:       vty_out (vty, "%% string argument required%s", VTY_NEWLINE);
                    145:       return CMD_WARNING;
                    146:     }
                    147:   
                    148:   str = argv_concat (argv, argc, 0);
                    149:   
                    150:   clear_something (vty, str);
                    151:   XFREE (MTYPE_TMP, str);
                    152:   return CMD_SUCCESS;
                    153: }
                    154: 
                    155: static int
                    156: heavy_wq_init ()
                    157: {
                    158:   if (! (heavy_wq = work_queue_new (master, "heavy_work_queue")))
                    159:     {
                    160:       zlog_err ("%s: could not get new work queue!", __func__);
                    161:       return -1;
                    162:     }
                    163:   
                    164:   heavy_wq->spec.workfunc = &slow_func;
                    165:   heavy_wq->spec.errorfunc = &slow_func_err;
                    166:   heavy_wq->spec.del_item_data = &slow_func_del;
                    167:   heavy_wq->spec.max_retries = 3;
                    168:   heavy_wq->spec.hold = 1000;
                    169:   
                    170:   return 0;
                    171: }
                    172: 
                    173: void
                    174: test_init()
                    175: {
                    176:   install_element (VIEW_NODE, &clear_foo_cmd);
                    177:   heavy_wq_init();
                    178: }