static void fq_conn_free(fq_conn_s *conn_s) { if(conn_s->user) free(conn_s->user); if(conn_s->pass) free(conn_s->pass); if(conn_s->queue) free(conn_s->queue); if(conn_s->queue_type) free(conn_s->queue_type); if(conn_s->tosend) fq_msg_free(conn_s->tosend); #define SWEEP_CONN_Q(TYPE, FREE, QNAME, ENTRYNAME) do { \ TYPE tofree; \ ck_fifo_spsc_dequeue_lock(&conn_s->QNAME); \ while(ck_fifo_spsc_dequeue(&conn_s->QNAME, &tofree) == true) { \ FREE(tofree); \ } \ ck_fifo_spsc_dequeue_unlock(&conn_s->QNAME); \ ck_fifo_spsc_entry_t *garbage = NULL; \ ck_fifo_spsc_deinit(&conn_s->QNAME, &garbage); \ while (garbage != NULL) { \ ck_fifo_spsc_entry_t *gn = garbage->next; \ free(garbage); \ garbage = gn; \ } \ } while(0) SWEEP_CONN_Q(fq_msg *, fq_msg_free, q, qhead); SWEEP_CONN_Q(void *, mark_safe_free, backq, backqhead); SWEEP_CONN_Q(cmd_instr *, free, cmdq, cmdqhead); #undef SWEEP_CONN_Q if(conn_s->cleanup_hook) conn_s->cleanup_hook(conn_s); free(conn_s); }
static mtev_hook_return_t zipkin_fq_publish(void *closure, int64_t traceid, int64_t spanid, unsigned char *buf, size_t len) { if(N_L_S_ON(debugls)) { int blen; char *b64buf; b64buf = malloc(len*2); blen = mtev_b64_encode(buf,len,b64buf,len*3); b64buf[blen] = '\0'; mtevL(debugls,"%s\n", b64buf); free(b64buf); } if(zc_client != NULL) { char buff[128]; fq_msg *msg; snprintf(buff, sizeof(buff), "%s%llx", zc_routing_prefix, (long long int)traceid); msg = fq_msg_alloc(buf, len); fq_msg_id(msg,NULL); fq_msg_route(msg, buff, strlen(buff)); fq_msg_exchange(msg, zc_exchange, strlen(zc_exchange)); fq_client_publish(zc_client, msg); fq_msg_free(msg); } return MTEV_HOOK_CONTINUE; }
static void mark_safe_free(void *pptr) { if(CHECK_HOOK_REQ_PTR(pptr)) { void *ptr = UNMARKED_HOOK_REQ_PTR(pptr); free(ptr); } else fq_msg_free(pptr); }
int main(int argc, char **argv) { hrtime_t s0, s, f, f0; uint64_t cnt = 0, icnt = 0; int psize = 0, i = 0; fq_client c; fq_msg *m; char *fq_debug = getenv("FQ_DEBUG"); if(fq_debug) fq_debug_set_bits(atoi(fq_debug)); signal(SIGPIPE, SIG_IGN); fq_client_init(&c, 0, logger); if(argc < 5) { fprintf(stderr, "%s <host> <port> <user> <pass> [size [count]]\n", argv[0]); exit(-1); } fq_client_creds(c, argv[1], atoi(argv[2]), argv[3], argv[4]); fq_client_heartbeat(c, 1000); fq_client_set_backlog(c, 10000, 100); fq_client_connect(c); if(argc > 5) { psize = atoi(argv[5]); } printf("payload size -> %d\n", psize); if(argc > 6) { send_count = atoi(argv[6]); } printf("message count -> %d\n", send_count); s0 = s = fq_gethrtime(); while(i < send_count || fq_client_data_backlog(c) > 0) { if(i < send_count) { m = fq_msg_alloc_BLANK(psize); memset(m->payload, 0, psize); fq_msg_exchange(m, "maryland", 8); fq_msg_route(m, "test.prefix.boo", 15); fq_msg_id(m, NULL); fq_client_publish(c, m); cnt++; i++; fq_msg_free(m); } else usleep(100); f = fq_gethrtime(); if(f-s > 1000000000) { print_rate(c, s, f, cnt, icnt); icnt = 0; cnt = 0; s = f; } } f0 = fq_gethrtime(); print_rate(c, s0, f0, i, 0); (void) argc; return 0; }
int main(int argc, char **argv) { hrtime_t s0, s, f, f0; uint64_t cnt = 0, icnt = 0; int psize = 0, i = 0, rcvd = 0; fq_client c; fq_bind_req breq; fq_msg *m; signal(SIGPIPE, SIG_IGN); fq_client_init(&c, 0, logger); if(argc < 5) { fprintf(stderr, "%s <host> <port> <user> <pass> [size [count]]\n", argv[0]); exit(-1); } fq_client_creds(c, argv[1], atoi(argv[2]), argv[3], argv[4]); fq_client_heartbeat(c, 1000); fq_client_set_backlog(c, 10000, 100); fq_client_connect(c); memset(&breq, 0, sizeof(breq)); memcpy(breq.exchange.name, "maryland", 8); breq.exchange.len = 8; breq.peermode = 0; breq.program = (char *)"prefix:\"test.prefix.\""; fq_client_bind(c, &breq); while(breq.out__route_id == 0) usleep(100); printf("route set -> %u\n", breq.out__route_id); if(breq.out__route_id == FQ_BIND_ILLEGAL) { fprintf(stderr, "Failure to bind...\n"); exit(-1); } if(argc > 5) { psize = atoi(argv[5]); } printf("payload size -> %d\n", psize); if(argc > 6) { send_count = atoi(argv[6]); } printf("message count -> %d\n", send_count); s0 = s = fq_gethrtime(); while(i < send_count || fq_client_data_backlog(c) > 0) { if(i < send_count) { m = fq_msg_alloc_BLANK(psize); memset(m->payload, 0, psize); fq_msg_exchange(m, "maryland", 8); fq_msg_route(m, "test.prefix.foo", 15); fq_msg_id(m, NULL); fq_client_publish(c, m); cnt++; i++; fq_msg_free(m); } else usleep(100); f = fq_gethrtime(); while(m = fq_client_receive(c)) { icnt++; rcvd++; fq_msg_deref(m); } if(f-s > 1000000000) { print_rate(c, s, f, cnt, icnt); icnt = 0; cnt = 0; s = f; } } f0 = fq_gethrtime(); print_rate(c, s0, f0, i, 0); do { icnt=0; while(m = fq_client_receive(c)) { icnt++; rcvd++; fq_msg_deref(m); } } while(rcvd < send_count); f0 = fq_gethrtime(); print_rate(c, s0, f0, 0, rcvd); printf("Total received during test: %d\n", rcvd); (void) argc; return 0; }
int main(int argc, char **argv) { char *host = NULL, *exchange = NULL, *route = NULL; char *address = strdup("127.0.0.1:8765"); char *user = "******"; char *pass = "******"; int port = 0; int c = 0; if (argc == 1) { usage(argv[0]); exit(-1); } while((c = getopt(argc, argv, "ha:x:r:u:p:")) != EOF) { switch(c) { case 'h': usage(argv[0]); exit(0); break; case 'a': address = strdup(optarg); break; case 'x': exchange = strdup(optarg); break; case 'r': route = strdup(optarg); break; case 'u': user = strdup(optarg); break; case 'p': pass = strdup(optarg); break; default: usage(argv[0]); exit(-1); } } // Separate host and port in address string // We have strdup'ed address so we can mutate it host = address; for (char *p = address; *p != 0; p++) { if (*p == ':') { *p = 0; port = strtol(p+1, NULL, 10); break; } } if (!port) { printf("Illegal port specification\n"); exit(-1); } if (!exchange) { printf("Exchange argument required"); exit(-1); } if (!route) { printf("Route argument required"); exit(-1); } fq_client cli; fq_msg *m = NULL; size_t exchange_len = strlen(exchange); size_t route_len = strlen(route); signal(SIGPIPE, SIG_IGN); // ignore SIGPIPE fq_client_init(&cli, 0, logger); fq_client_creds(cli, host, port, user, pass); fq_client_heartbeat(cli, 1000); fq_client_set_backlog(cli, 10000, 100); fq_client_connect(cli); while(true) { char *line = NULL; size_t line_cap = 0; int line_len; line_len = getline(&line, &line_cap, stdin); if(line_len < 0) { // wait for queues to drain while(fq_client_data_backlog(cli) > 0) { usleep(100); } exit(0); } m = fq_msg_alloc(line, line_len); fq_msg_exchange(m, exchange, exchange_len); fq_msg_route(m, route, route_len); fq_msg_id(m, NULL); fq_client_publish(cli, m); fq_msg_free(m); } return 0; }