6222cfc023e2da85fbe38ba2468502304d33ad6e
[asterisk/asterisk.git] / tests / test_sched.c
1 /*
2  * Asterisk -- An open source telephony toolkit.
3  *
4  * Copyright (C) 2009, Digium, Inc.
5  *
6  * Russell Bryant <russell@digium.com>
7  *
8  * See http://www.asterisk.org for more information about
9  * the Asterisk project. Please do not directly contact
10  * any of the maintainers of this project for assistance;
11  * the project provides a web site, mailing lists and IRC
12  * channels for your use.
13  *
14  * This program is free software, distributed under the terms of
15  * the GNU General Public License Version 2. See the LICENSE file
16  * at the top of the source tree.
17  */
18
19 /*! \file
20  *
21  * \brief ast_sched performance test module
22  *
23  * \author Russell Bryant <russell@digium.com>
24  */
25
26 #include "asterisk.h"
27
28 #include <inttypes.h>
29
30 ASTERISK_FILE_VERSION(__FILE__, "$Revision$")
31
32 #include "asterisk/module.h"
33 #include "asterisk/cli.h"
34 #include "asterisk/utils.h"
35 #include "asterisk/sched.h"
36
37 static int sched_cb(const void *data)
38 {
39         return 0;
40 }
41
42 static char *handle_cli_sched_test(struct ast_cli_entry *e, int cmd, struct ast_cli_args *a)
43 {
44         struct sched_context *con;
45         struct timeval start;
46         unsigned int num, i;
47         int *sched_ids = NULL;
48
49         switch (cmd) {
50         case CLI_INIT:
51                 e->command = "sched test";
52                 e->usage = ""
53                         "Usage: sched test <num>\n"
54                         "";
55                 return NULL;
56         case CLI_GENERATE:
57                 return NULL;
58         }
59
60         if (a->argc != e->args + 1) {
61                 return CLI_SHOWUSAGE;
62         }
63
64         if (sscanf(a->argv[e->args], "%u", &num) != 1) {
65                 return CLI_SHOWUSAGE;
66         }
67
68         if (!(con = sched_context_create())) {
69                 ast_cli(a->fd, "Test failed - could not create scheduler context\n");
70                 return CLI_FAILURE;
71         }
72
73         if (!(sched_ids = ast_malloc(sizeof(*sched_ids) * num))) {
74                 ast_cli(a->fd, "Test failed - memory allocation failure\n");
75                 goto return_cleanup;
76         }
77
78         ast_cli(a->fd, "Testing ast_sched_add() performance - timing how long it takes "
79                         "to add %u entries at random time intervals from 0 to 60 seconds\n", num);
80
81         start = ast_tvnow();
82
83         for (i = 0; i < num; i++) {
84                 int when = abs(ast_random()) % 60000;
85                 if ((sched_ids[i] = ast_sched_add(con, when, sched_cb, NULL)) == -1) {
86                         ast_cli(a->fd, "Test failed - sched_add returned -1\n");
87                         goto return_cleanup;
88                 }
89         }
90
91         ast_cli(a->fd, "Test complete - %" PRIi64 " us\n", ast_tvdiff_us(ast_tvnow(), start));
92
93         ast_cli(a->fd, "Testing ast_sched_del() performance - timing how long it takes "
94                         "to delete %u entries with random time intervals from 0 to 60 seconds\n", num);
95
96         start = ast_tvnow();
97
98         for (i = 0; i < num; i++) {
99                 if (ast_sched_del(con, sched_ids[i]) == -1) {
100                         ast_cli(a->fd, "Test failed - sched_del returned -1\n");
101                         goto return_cleanup;
102                 }
103         }
104
105         ast_cli(a->fd, "Test complete - %" PRIi64 " us\n", ast_tvdiff_us(ast_tvnow(), start));
106
107 return_cleanup:
108         sched_context_destroy(con);
109         if (sched_ids) {
110                 ast_free(sched_ids);
111         }
112
113         return CLI_SUCCESS;
114 }
115
116 static struct ast_cli_entry cli_sched[] = {
117         AST_CLI_DEFINE(handle_cli_sched_test, "Test ast_sched add/del performance"),
118 };
119
120 static int unload_module(void)
121 {
122         ast_cli_unregister_multiple(cli_sched, ARRAY_LEN(cli_sched));
123         return 0;
124 }
125
126 static int load_module(void)
127 {
128         ast_cli_register_multiple(cli_sched, ARRAY_LEN(cli_sched));
129         return AST_MODULE_LOAD_SUCCESS;
130 }
131
132 AST_MODULE_INFO_STANDARD(ASTERISK_GPL_KEY, "ast_sched performance test module");