src/client-manager/client_manager.c

/*
 * client_manager.c
 *
 *  Created on: 2009-09-11
 *      Author: dziki_rysio_997
 */

#include <stdio.h>
#include <stdlib.h>
#include <arpa/inet.h>
#include <math.h>
#include <string.h>
#include <unistd.h>

#define START '1'
#define STOP '9'
#define STATS '5'

struct sockaddr_in serv_addr;
int * fd;

struct statistics {
        unsigned long long wrongAnswerCount;
        unsigned long long correctAnswerCount;
        unsigned long long hitCount;
        unsigned long long missCount;
        unsigned long long sentSetsCount;
        unsigned long long sentGetsCount;
        unsigned long long storedCount;
        unsigned long long notStoredCount;
        double totalTime;
        double totalSetResponseTime;
        double totalGetResponseTime;
};

#define TOTAL_STATS_SIZE sizeof(struct statistics)

struct statistics stats;
struct statistics totalStats;

int main(int argc, char** argv) {

        if (argc < 3) {
                printf("Usage: %s host [host ...] port\n", argv[0]);
                return 1;
        }

        int hostCount = argc - 2;
        fd = (int*) malloc(hostCount * sizeof(int));

        char input[256];

        while (1) {

                printf("command> ");
                fflush(stdout);

                scanf("%256s", input);

                char signal;

                if (strcmp(input, "start") == 0) {
                        signal = START;
                } else if (strcmp(input, "stop") == 0) {
                        signal = STOP;
                } else if (strcmp(input, "stats") == 0) {
                        signal = STATS;
                } else if (strcmp(input, "exit") == 0) {
                        printf("exiting...\n");
                        break;
                } else {
                        printf("Available commands: start stop stats exit\n");
                        continue;
                }

                memset(&totalStats, 0, sizeof(totalStats));

                int i;
                for (i = 0; i < hostCount; i++) {

                        fd[i] = socket(AF_INET, SOCK_STREAM, 0);
                        memset(&serv_addr, 0, sizeof(serv_addr));
                        serv_addr.sin_family = AF_INET;
                        serv_addr.sin_addr.s_addr = inet_addr(argv[1 + i]);
                        serv_addr.sin_port = htons(atoi(argv[argc - 1]));

                        if (connect(fd[i], (struct sockaddr*) &serv_addr, sizeof(serv_addr))
                                        == -1)
                                printf("Connection failed.\n");
                        else
                                printf("Connected.\n");

                        if (send(fd[i], &signal, 1, 0)) {
                                printf("Sent %s signal.\n", input);
                        }

                        if (signal == STATS) { // receiving the response

                                char recBuf[TOTAL_STATS_SIZE];

                                struct statistics stats;


                                if (recv(fd[i], (void*) recBuf, TOTAL_STATS_SIZE, 0)
                                                == TOTAL_STATS_SIZE) {
                                        printf("Received stats from %s:%s\n", argv[1 + i],
                                                        argv[argc - 1]);
                                        memcpy(&stats, (void*) recBuf, TOTAL_STATS_SIZE);
                                        totalStats.sentSetsCount += stats.sentSetsCount;
                                        totalStats.storedCount += stats.storedCount;
                                        totalStats.notStoredCount += stats.notStoredCount;
                                        totalStats.sentGetsCount += stats.sentGetsCount;
                                        totalStats.hitCount += stats.hitCount;
                                        totalStats.missCount += stats.missCount;
                                        totalStats.correctAnswerCount += stats.correctAnswerCount;
                                        totalStats.wrongAnswerCount += stats.wrongAnswerCount;
                                        totalStats.totalGetResponseTime += stats.totalGetResponseTime;
                                        totalStats.totalSetResponseTime += stats.totalSetResponseTime;
                                        totalStats.totalTime = stats.totalTime;

                                }
                        }

                        close(fd[i]);
                }

                if (signal == STATS) {
                        printf("\nsent sets:%llu\n", totalStats.sentSetsCount);
                        printf("stored:%llu\n", totalStats.storedCount);
                        printf("not stored:%llu\n", totalStats.notStoredCount);
                        printf("sent gets:%llu\n", totalStats.sentGetsCount);
                        printf("hits:%llu\n", totalStats.hitCount);
                        printf("misses:%llu\n", totalStats.missCount);
                        printf("correct:%llu\n", totalStats.correctAnswerCount);
                        printf("wrong:%llu\n", totalStats.wrongAnswerCount);
                        printf("total time:%lg\n", totalStats.totalTime);
                        printf("avg set response time s:%lg\n",
                                        totalStats.totalSetResponseTime / (totalStats.storedCount
                                                        + totalStats.notStoredCount));
                        printf("avg get response time s:%lg\n",
                                        totalStats.totalGetResponseTime / (totalStats.hitCount
                                                        + totalStats.missCount));

                        if (totalStats.totalTime != 0) {
                                printf("req/s:%lg\n", (double) (totalStats.storedCount
                                                + totalStats.notStoredCount + totalStats.hitCount
                                                + totalStats.missCount) / totalStats.totalTime);
                        }
                }

        }

        free(fd);

        return 0;
}

---