Switch to storing in sqlite

server/src/server.cpp

@@ -2,21 +2,18 @@
 #include "NetConst.hpp"
 #include "sql.hpp"
 #include "values/DnaManager.hpp"
-#include <iostream>
 
+#include <sqlite3.h>
+
+#include <iostream>
 #include <map>
 
 // use pthread rw_lock to lock db so you can safy clone .db file
 
-typedef std::vector<NetUnit> NetList;
-typedef std::vector<NetList> UserList;
-
-std::map<uint64_t, UserList> data;
-
 // When a new client connected:
 void call(int sock, sockaddr_in newSocketInfo)
 {
-  std::cout << "new User" << std::endl;
+  printf("new User");
 
   int64_t conf, id;
   Message message;
@@ -36,21 +33,41 @@ void call(int sock, sockaddr_in newSocketInfo)
     return;
   }
 
-  auto found = data.find(id);
-  if (found == data.end())
+  bool ok = true;
+
+  sqlite3 *db;
+  sqlite3_stmt *stmt;
+
+  int rc = sqlite3_open(DB_NAME, &db);
+  if (rc)
   {
-    UserList list;
-    data.insert(std::pair<uint64_t, UserList>(id, list));
-    found = data.find(id);
+    fprintf(stderr, "Can't open database: %s\n", sqlite3_errmsg(db));
+    ok = false;
   }
 
-  while (true)
+  sqlite3_prepare_v2(db, max_gen, -1, &stmt, NULL);
+  sqlite3_bind_int64(stmt, 1, id);
+  int64_t gen = 0;
+  while (sqlite3_step(stmt) != SQLITE_DONE)
   {
-    size_t gen = found->second.size();
+    int num_cols = sqlite3_column_count(stmt);
+   
+    int type = sqlite3_column_type(stmt, 0);
+    
+    if(type == SQLITE_NULL) break;
+    gen = sqlite3_column_int64(stmt, 0);
+    gen++;
+   
+  
+  }
+  sqlite3_finalize(stmt);
 
+  sqlite3_prepare_v2(db, insert_data, -1, &stmt, NULL);
+
+  while (ok)
+  {
     message.mess = Mess::REQ_SEND_GEN;
     message.data = gen;
-
     TcpSocket::sendt(sock, &message, sizeof(Message));
     TcpSocket::recvt(sock, &message, sizeof(Message));
 
@@ -59,16 +76,28 @@ void call(int sock, sockaddr_in newSocketInfo)
       break;
     }
 
-    NetList list;
+    std::vector<NetUnit> list;
     list.resize(NUM_PER_GEN);
 
     TcpSocket::recvt(sock, list.data(), NUM_PER_GEN * sizeof(NetUnit));
-    found->second.push_back(list);
+    
+    for(size_t i = 0; i < list.size(); i++){
+      sqlite3_bind_int64(stmt, 1, id);
+      sqlite3_bind_int64(stmt, 2, gen);
+      sqlite3_bind_int64(stmt, 3, list[i].hash);
+      sqlite3_bind_int64(stmt, 4, list[i].index);
+      sqlite3_bind_int64(stmt, 5, list[i].liked);
+      sqlite3_step(stmt);
+      sqlite3_reset(stmt);
+    }
+
+    gen++;
   }
 
-  printf("id: %ld, size: %ld\n", id, found->second.size());
+  sqlite3_finalize(stmt);
 
-  std::cout << "del USER" << std::endl;
+  sqlite3_close(db);
+  printf("del USER");
   TcpSocket::closet(sock);
 }