talksik/uWebSockets
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Update benchmarks to compile with uSockets 0.2.0

Browse Source
This commit is contained in:
Alex Hultman
2019-06-12 11:24:01 +02:00
parent afe63d6f25
commit c5c0242830
2 changed files with 67 additions and 67 deletions
Download Patch File Download Diff File Expand all files Collapse all files
benchmarks/load_test.c
+33 -33
View File
@@ -1,6 +1,6 @@
/* This is a simple yet efficient WebSocket server benchmark much like WRK */ /* This is a simple yet efficient WebSocket server benchmark much like WRK */
#include <libusockets_new.h> #include <libusockets.h>
int SSL; int SSL;
#include <stdio.h> #include <stdio.h>
@@ -32,36 +32,36 @@ struct http_socket {
}; };
/* We don't need any of these */ /* We don't need any of these */
void on_wakeup(struct us_loop *loop) { void on_wakeup(struct us_loop_t *loop) {
} }
void on_pre(struct us_loop *loop) { void on_pre(struct us_loop_t *loop) {
} }
/* This is not HTTP POST, it is merely an event emitted post loop iteration */ /* This is not HTTP POST, it is merely an event emitted post loop iteration */
void on_post(struct us_loop *loop) { void on_post(struct us_loop_t *loop) {
} }
void next_connection(struct us_new_socket_t *s) { void next_connection(struct us_socket_t *s) {
/* We could wait with this until properly upgraded */ /* We could wait with this until properly upgraded */
if (--connections) { if (--connections) {
us_new_socket_context_connect(SSL, us_new_socket_context(SSL, s), host, port, 0, sizeof(struct http_socket)); us_socket_context_connect(SSL, us_socket_context(SSL, s), host, port, 0, sizeof(struct http_socket));
} else { } else {
printf("Running benchmark now...\n"); printf("Running benchmark now...\n");
us_new_socket_timeout(SSL, s, LIBUS_TIMEOUT_GRANULARITY); us_socket_timeout(SSL, s, LIBUS_TIMEOUT_GRANULARITY);
} }
} }
struct us_new_socket_t *on_http_socket_writable(struct us_new_socket_t *s) { struct us_socket_t *on_http_socket_writable(struct us_socket_t *s) {
struct http_socket *http_socket = (struct http_socket *) us_new_socket_ext(SSL, s); struct http_socket *http_socket = (struct http_socket *) us_socket_ext(SSL, s);
/* Are we still not upgraded yet? */ /* Are we still not upgraded yet? */
if (http_socket->upgrade_offset < sizeof(request) - 1) { if (http_socket->upgrade_offset < sizeof(request) - 1) {
http_socket->upgrade_offset += us_new_socket_write(SSL, s, request + http_socket->upgrade_offset, sizeof(request) - 1 - http_socket->upgrade_offset, 0); http_socket->upgrade_offset += us_socket_write(SSL, s, request + http_socket->upgrade_offset, sizeof(request) - 1 - http_socket->upgrade_offset, 0);
/* Now we should be */ /* Now we should be */
if (http_socket->upgrade_offset == sizeof(request) - 1) { if (http_socket->upgrade_offset == sizeof(request) - 1) {
@@ -69,30 +69,30 @@ struct us_new_socket_t *on_http_socket_writable(struct us_new_socket_t *s) {
} }
} else { } else {
/* Stream whatever is remaining of the request */ /* Stream whatever is remaining of the request */
http_socket->offset += us_new_socket_write(SSL, s, (char *) web_socket_request + http_socket->offset, sizeof(web_socket_request) - http_socket->offset, 0); http_socket->offset += us_socket_write(SSL, s, (char *) web_socket_request + http_socket->offset, sizeof(web_socket_request) - http_socket->offset, 0);
} }
return s; return s;
} }
struct us_new_socket_t *on_http_socket_close(struct us_new_socket_t *s) { struct us_socket_t *on_http_socket_close(struct us_socket_t *s) {
printf("Closed!\n"); printf("Closed!\n");
return s; return s;
} }
struct us_new_socket_t *on_http_socket_end(struct us_new_socket_t *s) { struct us_socket_t *on_http_socket_end(struct us_socket_t *s) {
return us_new_socket_close(SSL, s); return us_socket_close(SSL, s);
} }
struct us_new_socket_t *on_http_socket_data(struct us_new_socket_t *s, char *data, int length) { struct us_socket_t *on_http_socket_data(struct us_socket_t *s, char *data, int length) {
/* Get socket extension and the socket's context's extension */ /* Get socket extension and the socket's context's extension */
struct http_socket *http_socket = (struct http_socket *) us_new_socket_ext(SSL, s); struct http_socket *http_socket = (struct http_socket *) us_socket_ext(SSL, s);
//struct http_context *http_context = (struct http_context *) us_new_socket_context_ext(SSL, us_new_socket_context(SSL, s)); //struct http_context *http_context = (struct http_context *) us_socket_context_ext(SSL, us_socket_context(SSL, s));
/* We treat all data events as a response */ /* We treat all data events as a response */
http_socket->offset = us_new_socket_write(SSL, s, (char *) web_socket_request, sizeof(web_socket_request), 0); http_socket->offset = us_socket_write(SSL, s, (char *) web_socket_request, sizeof(web_socket_request), 0);
/* */ /* */
responses++; responses++;
@@ -100,14 +100,14 @@ struct us_new_socket_t *on_http_socket_data(struct us_new_socket_t *s, char *dat
return s; return s;
} }
struct us_new_socket_t *on_http_socket_open(struct us_new_socket_t *s, int is_client, char *ip, int ip_length) { struct us_socket_t *on_http_socket_open(struct us_socket_t *s, int is_client, char *ip, int ip_length) {
struct http_socket *http_socket = (struct http_socket *) us_new_socket_ext(SSL, s); struct http_socket *http_socket = (struct http_socket *) us_socket_ext(SSL, s);
/* Reset offsets */ /* Reset offsets */
http_socket->offset = 0; http_socket->offset = 0;
/* Send an upgrade request */ /* Send an upgrade request */
http_socket->upgrade_offset = us_new_socket_write(SSL, s, request, sizeof(request) - 1, 0); http_socket->upgrade_offset = us_socket_write(SSL, s, request, sizeof(request) - 1, 0);
if (http_socket->upgrade_offset == sizeof(request) - 1) { if (http_socket->upgrade_offset == sizeof(request) - 1) {
next_connection(s); next_connection(s);
} }
@@ -115,12 +115,12 @@ struct us_new_socket_t *on_http_socket_open(struct us_new_socket_t *s, int is_cl
return s; return s;
} }
struct us_new_socket_t *on_http_socket_timeout(struct us_new_socket_t *s) { struct us_socket_t *on_http_socket_timeout(struct us_socket_t *s) {
/* Print current statistics */ /* Print current statistics */
printf("Msg/sec: %f\n", ((float)responses) / LIBUS_TIMEOUT_GRANULARITY); printf("Msg/sec: %f\n", ((float)responses) / LIBUS_TIMEOUT_GRANULARITY);
responses = 0; responses = 0;
us_new_socket_timeout(SSL, s, LIBUS_TIMEOUT_GRANULARITY); us_socket_timeout(SSL, s, LIBUS_TIMEOUT_GRANULARITY);
return s; return s;
} }
@@ -140,22 +140,22 @@ int main(int argc, char **argv) {
SSL = atoi(argv[4]); SSL = atoi(argv[4]);
/* Create the event loop */ /* Create the event loop */
struct us_loop *loop = us_create_loop(1, on_wakeup, on_pre, on_post, 0); struct us_loop_t *loop = us_create_loop(0, on_wakeup, on_pre, on_post, 0);
/* Create a socket context for HTTP */ /* Create a socket context for HTTP */
struct us_new_socket_context_options_t options = {}; struct us_socket_context_options_t options = {};
struct us_new_socket_context_t *http_context = us_new_create_socket_context(SSL, loop, 0, options); struct us_socket_context_t *http_context = us_create_socket_context(SSL, loop, 0, options);
/* Set up event handlers */ /* Set up event handlers */
us_new_socket_context_on_open(SSL, http_context, on_http_socket_open); us_socket_context_on_open(SSL, http_context, on_http_socket_open);
us_new_socket_context_on_data(SSL, http_context, on_http_socket_data); us_socket_context_on_data(SSL, http_context, on_http_socket_data);
us_new_socket_context_on_writable(SSL, http_context, on_http_socket_writable); us_socket_context_on_writable(SSL, http_context, on_http_socket_writable);
us_new_socket_context_on_close(SSL, http_context, on_http_socket_close); us_socket_context_on_close(SSL, http_context, on_http_socket_close);
us_new_socket_context_on_timeout(SSL, http_context, on_http_socket_timeout); us_socket_context_on_timeout(SSL, http_context, on_http_socket_timeout);
us_new_socket_context_on_end(SSL, http_context, on_http_socket_end); us_socket_context_on_end(SSL, http_context, on_http_socket_end);
/* Start making HTTP connections */ /* Start making HTTP connections */
us_new_socket_context_connect(SSL, http_context, host, port, 0, sizeof(struct http_socket)); us_socket_context_connect(SSL, http_context, host, port, 0, sizeof(struct http_socket));
us_loop_run(loop); us_loop_run(loop);
} }
benchmarks/scale_test.c
+34 -34
View File
@@ -1,6 +1,6 @@
/* This is a scalability test for testing million(s) of pinging connections */ /* This is a scalability test for testing million(s) of pinging connections */
#include <libusockets_new.h> #include <libusockets.h>
int SSL; int SSL;
#include <stdio.h> #include <stdio.h>
@@ -42,20 +42,20 @@ struct http_socket {
}; };
/* We don't need any of these */ /* We don't need any of these */
void on_wakeup(struct us_loop *loop) { void on_wakeup(struct us_loop_t *loop) {
} }
void on_pre(struct us_loop *loop) { void on_pre(struct us_loop_t *loop) {
} }
/* This is not HTTP POST, it is merely an event emitted post loop iteration */ /* This is not HTTP POST, it is merely an event emitted post loop iteration */
void on_post(struct us_loop *loop) { void on_post(struct us_loop_t *loop) {
} }
void next_connection(struct us_new_socket_t *s) { void next_connection(struct us_socket_t *s) {
/* We could wait with this until properly upgraded */ /* We could wait with this until properly upgraded */
if (--connections/* > BATCH_CONNECT*/) { if (--connections/* > BATCH_CONNECT*/) {
/* Swap address */ /* Swap address */
@@ -63,37 +63,37 @@ void next_connection(struct us_new_socket_t *s) {
char buf[16]; char buf[16];
sprintf(buf, "127.0.0.%d", address); sprintf(buf, "127.0.0.%d", address);
us_new_socket_context_connect(SSL, us_new_socket_context(SSL, s), buf, port, 0, sizeof(struct http_socket)); us_socket_context_connect(SSL, us_socket_context(SSL, s), buf, port, 0, sizeof(struct http_socket));
} }
} }
struct us_new_socket_t *on_http_socket_writable(struct us_new_socket_t *s) { struct us_socket_t *on_http_socket_writable(struct us_socket_t *s) {
struct http_socket *http_socket = (struct http_socket *) us_new_socket_ext(SSL, s); struct http_socket *http_socket = (struct http_socket *) us_socket_ext(SSL, s);
/* Are we still not upgraded yet? */ /* Are we still not upgraded yet? */
if (http_socket->upgrade_offset < sizeof(request) - 1) { if (http_socket->upgrade_offset < sizeof(request) - 1) {
http_socket->upgrade_offset += us_new_socket_write(SSL, s, request + http_socket->upgrade_offset, sizeof(request) - 1 - http_socket->upgrade_offset, 0); http_socket->upgrade_offset += us_socket_write(SSL, s, request + http_socket->upgrade_offset, sizeof(request) - 1 - http_socket->upgrade_offset, 0);
/* Now we should be */ /* Now we should be */
if (http_socket->upgrade_offset == sizeof(request) - 1) { if (http_socket->upgrade_offset == sizeof(request) - 1) {
next_connection(s); next_connection(s);
/* Make sure to send ping */ /* Make sure to send ping */
us_new_socket_timeout(SSL, s, WEBSOCKET_PING_INTERVAL); us_socket_timeout(SSL, s, WEBSOCKET_PING_INTERVAL);
} }
} else { } else {
/* Stream whatever is remaining of the request */ /* Stream whatever is remaining of the request */
http_socket->offset += us_new_socket_write(SSL, s, (char *) web_socket_request + http_socket->offset, sizeof(web_socket_request) - http_socket->offset, 0); http_socket->offset += us_socket_write(SSL, s, (char *) web_socket_request + http_socket->offset, sizeof(web_socket_request) - http_socket->offset, 0);
if (http_socket->offset == sizeof(web_socket_request)) { if (http_socket->offset == sizeof(web_socket_request)) {
/* Reset timeout if we managed to */ /* Reset timeout if we managed to */
us_new_socket_timeout(SSL, s, WEBSOCKET_PING_INTERVAL); us_socket_timeout(SSL, s, WEBSOCKET_PING_INTERVAL);
} }
} }
return s; return s;
} }
struct us_new_socket_t *on_http_socket_close(struct us_new_socket_t *s) { struct us_socket_t *on_http_socket_close(struct us_socket_t *s) {
closed_connections++; closed_connections++;
if (closed_connections % 1000 == 0) { if (closed_connections % 1000 == 0) {
@@ -103,17 +103,17 @@ struct us_new_socket_t *on_http_socket_close(struct us_new_socket_t *s) {
return s; return s;
} }
struct us_new_socket_t *on_http_socket_end(struct us_new_socket_t *s) { struct us_socket_t *on_http_socket_end(struct us_socket_t *s) {
return us_new_socket_close(SSL, s); return us_socket_close(SSL, s);
} }
// should never get a response! // should never get a response!
struct us_new_socket_t *on_http_socket_data(struct us_new_socket_t *s, char *data, int length) { struct us_socket_t *on_http_socket_data(struct us_socket_t *s, char *data, int length) {
return s; return s;
} }
struct us_new_socket_t *on_http_socket_open(struct us_new_socket_t *s, int is_client, char *ip, int ip_length) { struct us_socket_t *on_http_socket_open(struct us_socket_t *s, int is_client, char *ip, int ip_length) {
struct http_socket *http_socket = (struct http_socket *) us_new_socket_ext(SSL, s); struct http_socket *http_socket = (struct http_socket *) us_socket_ext(SSL, s);
/* Display number of opened connections */ /* Display number of opened connections */
opened_connections++; opened_connections++;
@@ -122,26 +122,26 @@ struct us_new_socket_t *on_http_socket_open(struct us_new_socket_t *s, int is_cl
} }
/* Send an upgrade request */ /* Send an upgrade request */
http_socket->upgrade_offset = us_new_socket_write(SSL, s, request, sizeof(request) - 1, 0); http_socket->upgrade_offset = us_socket_write(SSL, s, request, sizeof(request) - 1, 0);
if (http_socket->upgrade_offset == sizeof(request) - 1) { if (http_socket->upgrade_offset == sizeof(request) - 1) {
next_connection(s); next_connection(s);
/* Make sure to send ping */ /* Make sure to send ping */
us_new_socket_timeout(SSL, s, WEBSOCKET_PING_INTERVAL); us_socket_timeout(SSL, s, WEBSOCKET_PING_INTERVAL);
} }
return s; return s;
} }
// here we should send a message as ping (part of the test) // here we should send a message as ping (part of the test)
struct us_new_socket_t *on_http_socket_timeout(struct us_new_socket_t *s) { struct us_socket_t *on_http_socket_timeout(struct us_socket_t *s) {
struct http_socket *http_socket = (struct http_socket *) us_new_socket_ext(SSL, s); struct http_socket *http_socket = (struct http_socket *) us_socket_ext(SSL, s);
/* Send ping here */ /* Send ping here */
http_socket->offset = us_new_socket_write(SSL, s, (char *) web_socket_request, sizeof(web_socket_request), 0); http_socket->offset = us_socket_write(SSL, s, (char *) web_socket_request, sizeof(web_socket_request), 0);
if (http_socket->offset == sizeof(web_socket_request)) { if (http_socket->offset == sizeof(web_socket_request)) {
/* Reset timeout if we managed to */ /* Reset timeout if we managed to */
us_new_socket_timeout(SSL, s, WEBSOCKET_PING_INTERVAL); us_socket_timeout(SSL, s, WEBSOCKET_PING_INTERVAL);
} }
return s; return s;
@@ -162,23 +162,23 @@ int main(int argc, char **argv) {
SSL = atoi(argv[4]); SSL = atoi(argv[4]);
/* Create the event loop */ /* Create the event loop */
struct us_loop *loop = us_create_loop(1, on_wakeup, on_pre, on_post, 0); struct us_loop_t *loop = us_create_loop(0, on_wakeup, on_pre, on_post, 0);
/* Create a socket context for HTTP */ /* Create a socket context for HTTP */
struct us_new_socket_context_options_t options = {}; struct us_socket_context_options_t options = {};
struct us_new_socket_context_t *http_context = us_new_create_socket_context(SSL, loop, 0, options); struct us_socket_context_t *http_context = us_create_socket_context(SSL, loop, 0, options);
/* Set up event handlers */ /* Set up event handlers */
us_new_socket_context_on_open(SSL, http_context, on_http_socket_open); us_socket_context_on_open(SSL, http_context, on_http_socket_open);
us_new_socket_context_on_data(SSL, http_context, on_http_socket_data); us_socket_context_on_data(SSL, http_context, on_http_socket_data);
us_new_socket_context_on_writable(SSL, http_context, on_http_socket_writable); us_socket_context_on_writable(SSL, http_context, on_http_socket_writable);
us_new_socket_context_on_close(SSL, http_context, on_http_socket_close); us_socket_context_on_close(SSL, http_context, on_http_socket_close);
us_new_socket_context_on_timeout(SSL, http_context, on_http_socket_timeout); us_socket_context_on_timeout(SSL, http_context, on_http_socket_timeout);
us_new_socket_context_on_end(SSL, http_context, on_http_socket_end); us_socket_context_on_end(SSL, http_context, on_http_socket_end);
/* Start making HTTP connections */ /* Start making HTTP connections */
for (int i = 0; i < BATCH_CONNECT; i++) { for (int i = 0; i < BATCH_CONNECT; i++) {
us_new_socket_context_connect(SSL, http_context, host, port, 0, sizeof(struct http_socket)); us_socket_context_connect(SSL, http_context, host, port, 0, sizeof(struct http_socket));
} }
us_loop_run(loop); us_loop_run(loop);