mini-redis-cpp/Server.cpp at main · SoftGridAlex/mini-redis-cpp · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
#include "Server.h"
#include <iostream>
#include <sstream>
#include <thread>

#pragma comment(lib, "ws2_32.lib")


Server::Server(int p) {
    port = p;
    serverSocket = INVALID_SOCKET;
}

Server::~Server() {
    if (serverSocket != INVALID_SOCKET) {

        closesocket(serverSocket);

    }
    WSACleanup();
}

void Server::start() {

    WSADATA wsa;
    WSAStartup(MAKEWORD(2, 2), &wsa);

    serverSocket = socket(AF_INET, SOCK_STREAM, 0);

    sockaddr_in addr;
    addr.sin_family = AF_INET;
    addr.sin_port = htons(port);
    addr.sin_addr.s_addr = INADDR_ANY;

    bind(serverSocket, (sockaddr*)&addr, sizeof(addr));
    listen(serverSocket, 5);

    std::cout << "Serverul Mini-Redis ruleaza pe portul " << port << "...\n";

    while (true) {
        SOCKET clientSocket = accept(serverSocket, NULL, NULL);

        if (clientSocket != INVALID_SOCKET) {
            std::thread(&Server::gestioneazaClient, this, clientSocket).detach();
        }
    }
}

void Server::gestioneazaClient(SOCKET clientSocket) {
    std::string salut = "Conectat cu succes la Mini-Redis!\n";
    send(clientSocket, salut.c_str(), salut.size(), 0);

    char buffer[1024];

    while(true){

        memset(buffer, 0, sizeof(buffer));
        int primiti = recv(clientSocket, buffer, sizeof(buffer) - 1, 0);

        if (primiti <= 0) break;


        std::stringstream ss(buffer);
        std::string comanda, cheie, valoare;

        ss >> comanda;

        std::string raspuns = "";

        if (comanda == "SET") {

            ss >> cheie >> valoare;
            db.set(cheie, valoare);
            raspuns = "OK\n";
        }
        else if (comanda == "GET") {

            ss >> cheie;
            std::string val = db.get(cheie);
            raspuns = val.empty() ? "Cheia nu exista!\n" : val + "\n";

        }

        else if (comanda == "DEL") {
            ss >> cheie;
            raspuns = db.del(cheie) ? "(integer) 1\n" : "(integer) 0\n";

        }
        else if (comanda == "EXIT") {
            break;
        }

        send(clientSocket, raspuns.c_str(), raspuns.size(), 0);


    }

    closesocket(clientSocket);
}