get_chat_info->get_my_info

Author: stepbystepcode

main.cc

@@ -25,13 +25,6 @@ int main()
                                   { friend_operation(req, std::move(callback)); });
     drogon::app().registerHandler("/api/info", [](const HttpRequestPtr &req, std::function<void(const HttpResponsePtr &)> &&callback)
                                   { info(req, std::move(callback)); });
-
-
-    // drogon::app().registerHandler("/api/file/tree", [](const HttpRequestPtr &req, std::function<void(const HttpResponsePtr &)> &&callback)
-    //                               { genTree(req, std::move(callback)); });
-    // drogon::app().registerHandler("/api/file/cat", [](const HttpRequestPtr &req, std::function<void(const HttpResponsePtr &)> &&callback)
-    //                               { catFile(req, std::move(callback)); });
-
     drogon::app().registerHandler("/api/file/save", [](const HttpRequestPtr &req, std::function<void(const HttpResponsePtr &)> &&callback)
                                   { saveFile(req, std::move(callback)); });
     drogon::app().registerHandler("/api/avatar", [](const HttpRequestPtr &req, std::function<void(const HttpResponsePtr &)> &&callback)
@@ -41,13 +34,10 @@ int main()
     drogon::app().registerHandler("/api/file/get", [](const HttpRequestPtr &req, std::function<void(const HttpResponsePtr &)> &&callback)
                                   { getPicture(req, std::move(callback)); },
                                   {Get});
-
     drogon::app().registerHandler("/api/file/commands", [](const HttpRequestPtr &req, std::function<void(const HttpResponsePtr &)> &&callback)
                                   { commandsCtrl(req, std::move(callback)); });
-
     drogon::app().registerHandler("/api/file/lock", [](const HttpRequestPtr &req, std::function<void(const HttpResponsePtr &)> &&callback)
                                   {add_lock(req, std::move(callback));});
-
     drogon::app().setUploadPath("./uploads").run();
     return 0;
 }

msg_controller.cc

@@ -63,7 +63,7 @@ void check(const HttpRequestPtr &req, std::function<void(const HttpResponsePtr &
     }
     else
     {
-        //res->setBody("No Authorization");
+        res_json["message"] = "No Authorization";
         res_json["code"] = 401;
 
     }
@@ -153,38 +153,17 @@ void request_processing(const HttpRequestPtr &req, std::function<void(const Http
 // get chat info
 void info(const HttpRequestPtr &req, std::function<void(const HttpResponsePtr &)> &&callback)
 {
-    auto body = req->getBody();
-    Json::Value req_json, res_json;
+    Json::Value res_json;
     Json::Reader reader;
-    std::string bodyStr(body);
-    if (!reader.parse(bodyStr, req_json))
-    {
-        callback(HttpResponse::newHttpResponse());
-        return;
-    }
     std::string me, who_send_me;
     Json::FastWriter writer;
     auto res = HttpResponse::newHttpResponse();
     res->addHeader("Access-Control-Allow-Origin", "*");
-
     if (jwtVerify(req))
     {
         res_json["code"]=200;
         me = jwtDecrypt(req->getHeader("Authorization").substr(7));
-        if (req_json["person"].asString() == "")
-        {
-            res_json["message"] = get_chat_info(me, "");
-            
-            //res->setBody(writer.write(get_chat_info(me, "")));
-        }
-        else
-        {
-            who_send_me = req_json["person"].asString();
-            res_json["message"] = get_chat_info(me, who_send_me);
-
-            //who_send_me = req_json["person"].asString();
-            //res->setBody(writer.write(get_chat_info(me, who_send_me)));
-        }
+        res_json["message"] = get_my_info(me);
     }
     else
     {

mysql.cc

@@ -304,10 +304,9 @@ void sql_add(std::string username, std::string passwd, int avatar)
     }
 }
 
-Json::Value get_chat_info(std::string me, std::string who_send_me)
+Json::Value get_my_info(std::string me)
 {
     Json::Value json;
-    std::string send[2] = {me, who_send_me};
 
     try
     {
@@ -316,14 +315,11 @@ Json::Value get_chat_info(std::string me, std::string who_send_me)
         sql::Connection *con;
         con = driver->connect("tcp://8.130.48.157:3306", "root", "abc.123");
         con->setSchema("flypen");
-
-        for (int i = 0; i < 2; i++)
-        {
-            if (!send[i].empty())
+            if (!me.empty())
             {
                 std::string sql = "SELECT * FROM users WHERE username = ? LIMIT 1";
                 sql::PreparedStatement *prepStmt = con->prepareStatement(sql);
-                prepStmt->setString(1, send[i]);
+                prepStmt->setString(1, me);
 
                 sql::ResultSet *res = prepStmt->executeQuery();
 
@@ -377,10 +373,10 @@ Json::Value get_chat_info(std::string me, std::string who_send_me)
                     Json::StreamWriterBuilder builder;
                     std::string userJson = Json::writeString(builder, user);
 
-                    json[send[i]] = user;
+                    json[me] = user;
                 }
             }
-        }
+        
     }
     catch (sql::SQLException &e)
     {
@@ -486,7 +482,7 @@ Json::Value sql_find_my_msg(std::string me, std::string connect_type)
                 updateStmt->setInt(1, id);
                 updateStmt->executeUpdate();
                 delete updateStmt;
-                std::cout<<"change"<<std::endl;
+                // std::cout<<"change"<<std::endl;
             }
 
             Json::Value item;

mysql.h

@@ -14,7 +14,7 @@ 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);
 int lockcheck(std::string filename);
-Json::Value get_chat_info(std::string,std::string="");
+Json::Value get_my_info(std::string);
 void sql_delete_operation(std::string,std::string);
 void set_avatar(std::string person, int avatar);
 int sql_findexist(std::string receiver);