void print_stats(struct db *db) {
int kcounter, vcounter;
double kavg, kdev, vavg, vdev;
stddev_get(db->tree->keys_stddev, &kcounter, &kavg, &kdev);
stddev_get(db->tree->values_stddev, &vcounter, &vavg, &vdev);
log_info("Items: %i(%i) Avg/stddev: Keys=%.3f/%.3f Values=%.3f/%.3f",
kcounter, vcounter,
kavg, kdev, vavg, vdev);
}
void base_print_stats(struct base *base)
{
log_info(base->db, "Stats: %s", "");
log_info(base->db, "%u/%u logs in use",
(unsigned)(log_get_number(logs_newest(base->logs)) -
log_get_number(logs_oldest(base->logs)) + 1),
base->max_open_logs);
uint64_t counter;
double avg, dev;
stddev_get(&base->used_size, &counter, &avg, &dev);
log_info(base->db, "Item stats: %9.1f bytes average, %7.1f bytes deviation, "
"%llu items", avg, dev, (unsigned long long)counter);
unsigned long allocated, wasted;
itree_mem_stats(base->itree, &allocated, &wasted);
log_info(base->db, "Tree memory: %8.1f MB committed, "
"%8.1f MB in use, committed/used ratio of %.3f",
(float)(allocated) / (1024*1024.),
(float)(allocated - wasted) / (1024*1024.),
(float)allocated / (float)(allocated - wasted));
log_info(base->db, "Disk space: %9.1f MB committed, %8.1f MB in use, "
"committed/used ratio of %.3f",
(float)base->disk_size.sum / (1024*1024.),
(float)base->used_size.sum / (1024*1024.),
(float)base->disk_size.sum / (float)base->used_size.sum);
}
int log_is_unused(struct log *log)
{
uint64_t count;
stddev_get(&log->used_size, &count, NULL, NULL);
return count == 0;
}
int main(int argc, const char *argv[])
{
int base_src_port = 65500;
int polling = 0;
int busy_poll = 0;
int packet_timestamp = 0;
static struct option long_options[] = {
{"src-port", required_argument, 0, 's'},
{"polling", no_argument, 0, 'p'},
{"busy-poll", required_argument, 0, 'b'},
{"timestamp", no_argument, 0, 't'},
{NULL, 0, 0, 0}};
const char *optstring = optstring_from_long_options(long_options);
optind = 1;
while (1) {
int option_index = 0;
int arg = getopt_long(argc, (char **)argv, optstring,
long_options, &option_index);
if (arg == -1) {
break;
}
switch (arg) {
case 's':
base_src_port = atoi(optarg);
break;
case 'p':
polling = 1;
break;
case 'b':
busy_poll = atoi(optarg);
break;
case 't':
packet_timestamp = 1;
break;
default:
FATAL("Unknown option %c: %s", arg, argv[optind]);
}
}
int thread_num = argc - optind;
if (thread_num < 1) {
FATAL("Usage: %s [--polling] [--busy-poll=<>] [--src-port=<>] "
"[--timestamp] [target:port...]",
argv[0]);
}
struct net_addr *target_addrs =
calloc(thread_num, sizeof(struct net_addr));
int t;
for (t = 0; t < thread_num; t++) {
const char *target_addr_str = argv[optind + t];
parse_addr(&target_addrs[t], target_addr_str);
fprintf(stderr, "[*] Sending to %s, polling=%i, src_port=%i\n",
addr_to_str(&target_addrs[t]), polling,
base_src_port + t);
}
struct state *array_of_states =
calloc(thread_num, sizeof(struct state));
for (t = 0; t < thread_num; t++) {
struct state *state = &array_of_states[t];
state->target_addr = &target_addrs[t];
state->polling = polling;
state->busy_poll = busy_poll;
if (base_src_port > 0) {
state->src_port = base_src_port + t;
}
thread_spawn(thread_loop, state);
}
while (1) {
struct timeval timeout = NSEC_TIMEVAL(MSEC_NSEC(1000UL));
while (1) {
int r = select(0, NULL, NULL, NULL, &timeout);
if (r != 0) {
continue;
}
if (TIMEVAL_NSEC(&timeout) == 0) {
break;
}
}
for (t = 0; t < thread_num; t++) {
struct state *state = &array_of_states[t];
uint64_t pps;
double avg, stddev;
double avg_packet, stddev_packet;
pthread_spin_lock(&state->lock);
stddev_get(&state->stddev, &pps, &avg, &stddev);
int min = state->stddev.min;
stddev_get(&state->stddev_packet, NULL, &avg_packet,
&stddev_packet);
stddev_init(&state->stddev);
stddev_init(&state->stddev_packet);
pthread_spin_unlock(&state->lock);
printf("pps=%6lu avg=%7.3fus dev=%7.3fus min=%6.3fus ",
pps, avg / 1000., stddev / 1000.,
(double)min / 1000.);
if (packet_timestamp) {
printf("(packet=%5.3fus/%5.3f) ",
avg_packet / 1000.,
stddev_packet / 1000.);
}
}
printf("\n");
}
return 0;
}
