You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
133 lines
3.3 KiB
133 lines
3.3 KiB
#include "handlers.h"
|
|
|
|
const char* VERSION = "0.0.1";
|
|
const char* STATS_FORMAT =
|
|
"STAT connections %d\r\n"
|
|
"STAT requests %d\r\n"
|
|
"STAT hits %d\r\n"
|
|
"STAT misses %d\r\n"
|
|
"STAT hit_ratio %2.4f\r\n"
|
|
"STAT backlog %d\r\n"
|
|
"%s"
|
|
"END\r\n";
|
|
|
|
void handle_stats(KCLIST* tokens, FILE* client){
|
|
char out[1024];
|
|
int total = hits + misses;
|
|
float hit_ratio = 0;
|
|
if(hits){
|
|
hit_ratio = (float)hits / (float)(hits + misses);
|
|
}
|
|
int size = queue_size(&requests);
|
|
char* status = kcdbstatus(db);
|
|
KCLIST* stats = kclistnew();
|
|
tokenize(stats, status, "\n");
|
|
char status_buf[1024];
|
|
strcpy(status_buf, "");
|
|
int stat_count = kclistcount(stats);
|
|
int i;
|
|
for(i = 0; i < stat_count; ++i){
|
|
size_t part_size;
|
|
const char* part = kclistget(stats, i, &part_size);
|
|
KCLIST* parts = kclistnew();
|
|
tokenize(parts, (char*)part, "\t");
|
|
char buf[128];
|
|
if(kclistcount(parts) == 2){
|
|
sprintf(buf, "STAT %s %s\r\n", kclistget(parts, 0, &part_size), kclistget(parts, 1, &part_size));
|
|
}
|
|
kclistdel(parts);
|
|
strcat(status_buf, buf);
|
|
}
|
|
sprintf(out, STATS_FORMAT, connections, total, hits, misses, hit_ratio, size, status_buf);
|
|
fputs(out, client);
|
|
}
|
|
|
|
void handle_version(KCLIST* tokens, FILE* client){
|
|
char out[1024];
|
|
sprintf(out, "VERSION %s\r\n", VERSION);
|
|
fputs(out, client);
|
|
}
|
|
|
|
void handle_flush(KCLIST* tokens, FILE* client){
|
|
if(kcdbclear(db)){
|
|
fputs("OK\r\n", client);
|
|
}
|
|
}
|
|
|
|
void handle_delete(KCLIST* tokens, FILE* client){
|
|
if(kclistcount(tokens)){
|
|
char* key;
|
|
list_shift(tokens, &key);
|
|
if(key != NULL){
|
|
if(kcdbremove(db, key, strlen(key))){
|
|
fputs("DELETED\r\n", client);
|
|
}
|
|
free(key);
|
|
}
|
|
}
|
|
}
|
|
|
|
void handle_get(KCLIST* tokens, FILE* client){
|
|
if(kclistcount(tokens)){
|
|
char* key;
|
|
list_shift(tokens, &key);
|
|
if(key == NULL){
|
|
return;
|
|
}
|
|
char out[1024];
|
|
char* result_buffer;
|
|
size_t result_size;
|
|
result_buffer = kcdbget(db, key, strlen(key), &result_size);
|
|
if(result_buffer){
|
|
if(strcmp(result_buffer, "0") == 0){
|
|
++misses;
|
|
sprintf(out, "VALUE %s 0 0\r\n\r\nEND\r\n", key);
|
|
} else{
|
|
++hits;
|
|
sprintf(out, "VALUE %s 0 %d\r\n%s\r\nEND\r\n", key, (int)strlen(result_buffer), result_buffer);
|
|
}
|
|
kcfree(result_buffer);
|
|
} else{
|
|
++misses;
|
|
sprintf(out, "VALUE %s 0 2\r\n{}\r\nEND\r\n", key);
|
|
kcdbset(db, key, strlen(key), "0", 1);
|
|
queue_add(&requests, key);
|
|
}
|
|
fputs(out, client);
|
|
free(key);
|
|
} else{
|
|
fputs("INVALID GET COMMAND: GET <KEY>\r\n", client);
|
|
}
|
|
}
|
|
|
|
int handle_command(char* buffer, FILE* client){
|
|
int status = 0;
|
|
KCLIST* tokens = kclistnew();
|
|
tokenize(tokens, buffer, " ");
|
|
char* command;
|
|
list_shift(tokens, &command);
|
|
if(command != NULL){
|
|
if(strncmp(command, "get", 3) == 0){
|
|
handle_get(tokens, client);
|
|
} else if(strcmp(command, "stats") == 0){
|
|
handle_stats(tokens, client);
|
|
} else if(strcmp(command, "flush_all") == 0){
|
|
handle_flush(tokens, client);
|
|
} else if(strcmp(command, "delete") == 0){
|
|
handle_delete(tokens, client);
|
|
} else if(strcmp(command, "version") == 0){
|
|
handle_version(tokens, client);
|
|
} else if(strcmp(command, "quit") == 0){
|
|
status = -1;
|
|
} else{
|
|
char out[1024];
|
|
sprintf(out, "UNKNOWN COMMAND: %s\r\n", command);
|
|
fputs(out, client);
|
|
}
|
|
free(command);
|
|
}
|
|
|
|
kclistdel(tokens);
|
|
|
|
return status;
|
|
}
|