update

Author: stepbystepcode

msg_controller.cc

@@ -136,7 +136,7 @@ void request_processing(const HttpRequestPtr &req, std::function<void(const Http
     }
     auto res = HttpResponse::newHttpResponse();
     res->setBody("Success");
-
+}
 void info(const HttpRequestPtr &req, std::function<void(const HttpResponsePtr &)> &&callback) {
     auto body = req->getBody();
     Json::Value req_json, res_json;
@@ -165,8 +165,12 @@ void info(const HttpRequestPtr &req, std::function<void(const HttpResponsePtr &)
 
     auto res = HttpResponse::newHttpResponse();
     res->addHeader("Access-Control-Allow-Origin", "*");
-    who_send_me = req_json["person"].asString();
-    auto output = writer.write(get_chat_info(me,who_send_me));
-    res->setBody(output);
-    callback(res);
+    if(req_json["person"].asString()==""){
+        res->setBody(writer.write(get_chat_info(me)));
+        callback(res);
+    }else{
+        who_send_me = req_json["person"].asString();
+        res->setBody(writer.write(get_chat_info(me,who_send_me)));
+        callback(res);
+    }
 }

mysql.cc

@@ -230,6 +230,36 @@ void sql_add(std::string username, std::string passwd, int avatar) {
 
 //     return result;
 // }
+Json::Value get_chat_info(std::string me){
+  Json::Value json;
+  try {
+    sql::mysql::MySQL_Driver *driver;
+    driver = sql::mysql::get_mysql_driver_instance();
+    sql::Connection *con;
+    con = driver->connect("tcp://8.130.48.157:3306", "root", "abc.123");
+    con->setSchema("flypen");
+
+    std::string sql = "SELECT * FROM users WHERE username = ? LIMIT 1";
+    sql::PreparedStatement *prepStmt = con->prepareStatement(sql);
+    prepStmt->setString(1, me);
+    
+    sql::ResultSet *res = prepStmt->executeQuery();
+    if (res->next()) {
+      Json::Value user;
+      int avatar=res->getInt("avatar");
+      std::string friends=res->getString("friends");
+      std::string req=res->getString("req");
+      user["avatar"] = avatar;
+      user["friends"] = friends; 
+      user["req"] = req;
+      
+      json[me] = user;
+    }
+  } catch (sql::SQLException &e) {
+    std::cerr << "SQL Exception: " << e.what() << std::endl;
+  }
+  return json;
+}
 Json::Value get_chat_info(std::string me,std::string who_send_me){
   Json::Value json;
   try {

mysql.h

@@ -12,5 +12,6 @@ void sql_addrequest(std::string send,std::string receiver);
 void sql_process_request(std::string ,std::string,std::string);
 Json::Value sql_find_my_msg(std::string,std::string);
 Json::Value get_chat_info(std::string,std::string);
+Json::Value get_chat_info(std::string);
 //bool sql_check(std::string );
 #endif

user_controller.cc

@@ -17,9 +17,10 @@ void Handle(const HttpRequestPtr &req, std::function<void(const HttpResponsePtr
     Json::FastWriter writer;
     std::string msg = handler(req_json);
     res_json["msg"] = msg;
-    if (msg=="Login Success")
+    if (msg=="Login Success"){
         res_json["token"] = jwtGen(req_json);
-    else
+        res_json["username"] = req_json["username"].asString();
+    }else
         res_json["token"] = "";
     auto output = writer.write(res_json);
     auto res = HttpResponse::newHttpResponse();