Merge pull request #24 from zzc20001/main

delete_function

Author: stepbystepcode

main.cc

@@ -24,7 +24,7 @@ int main() {
         request_processing(req, std::move(callback));
     });
     drogon::app().registerHandler("/api/newfriends", [](const HttpRequestPtr &req, std::function<void(const HttpResponsePtr &)> &&callback) {
-        friend_request(req, std::move(callback));
+        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));

msg_controller.cc

@@ -83,7 +83,7 @@ void check(const HttpRequestPtr &req, std::function<void(const HttpResponsePtr &
     callback(res);
 }
 
-void friend_request(const HttpRequestPtr &req, std::function<void(const HttpResponsePtr &)> &&callback)
+void friend_operation(const HttpRequestPtr &req, std::function<void(const HttpResponsePtr &)> &&callback)
 {
     std::string authHeader = req->getHeader("Authorization");
     if (authHeader.substr(0, 7) == "Bearer ")
@@ -94,7 +94,10 @@ void friend_request(const HttpRequestPtr &req, std::function<void(const HttpResp
         {
             std::string sender = jwtDecrypt(bearerToken);
             std::string receiver = req->getParameter("username");
+            std::string operation = req->getParameter("operation");
+            if(operation=="add")
             sql_addrequest(sender, receiver);
+            else sql_delete_operation(sender,receiver);
         }
         catch (const std::exception &e)
         {
@@ -116,9 +119,9 @@ void request_processing(const HttpRequestPtr &req, std::function<void(const Http
         try
         {
             std::string receiver = jwtDecrypt(bearerToken);
-            std::string sender =req->getParameter("username");
+            std::string sender = req->getParameter("username");
             std::string attitude = req->getParameter("info");
-            sql_process_request(sender,receiver, attitude);
+            sql_process_request(sender, receiver, attitude);
         }
         catch (const std::exception &e)
         {
@@ -129,40 +132,51 @@ 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) {
+void info(const HttpRequestPtr &req, std::function<void(const HttpResponsePtr &)> &&callback)
+{
     auto body = req->getBody();
     Json::Value req_json, res_json;
     Json::Reader reader;
     std::string bodyStr(body);
-    if (!reader.parse(bodyStr, req_json)) {
+    if (!reader.parse(bodyStr, req_json))
+    {
         callback(HttpResponse::newHttpResponse());
         return;
     }
     std::string me, who_send_me;
     Json::FastWriter writer;
     std::string authHeader = req->getHeader("Authorization");
-    if (authHeader.substr(0, 7) == "Bearer ") {
+    if (authHeader.substr(0, 7) == "Bearer ")
+    {
         std::string bearerToken = authHeader.substr(7);
         // 在此处使用Bearer Token进行身份验证
-        try {
+        try
+        {
             me = jwtDecrypt(bearerToken);
-        } catch (const std::exception &e) {
+        }
+        catch (const std::exception &e)
+        {
             std::cerr << e.what() << '\n';
             std::cout << "Wrong token" << std::endl;
         }
-    } else {
+    }
+    else
+    {
         // 连接没有Authorization头部Bearer Token
         std::cout << "No Authorization" << std::endl;
     }
 
     auto res = HttpResponse::newHttpResponse();
     res->addHeader("Access-Control-Allow-Origin", "*");
-    if(req_json["person"].asString()==""){
-        res->setBody(writer.write(get_chat_info(me,"")));
+    if (req_json["person"].asString() == "")
+    {
+        res->setBody(writer.write(get_chat_info(me, "")));
         callback(res);
-    }else{
+    }
+    else
+    {
         who_send_me = req_json["person"].asString();
-        res->setBody(writer.write(get_chat_info(me,who_send_me)));
+        res->setBody(writer.write(get_chat_info(me, who_send_me)));
         callback(res);
     }
 }

msg_controller.h

@@ -6,5 +6,5 @@ void chat(const HttpRequestPtr &req, std::function<void(const HttpResponsePtr &)
 void info(const HttpRequestPtr &req, std::function<void(const HttpResponsePtr &)> &&callback);
 void check(const HttpRequestPtr &req, std::function<void(const HttpResponsePtr &)> &&callback);
 void request_processing(const HttpRequestPtr &req, std::function<void(const HttpResponsePtr &)> &&callback);
-void friend_request(const HttpRequestPtr &req, std::function<void(const HttpResponsePtr &)> &&callback);
+void friend_operation(const HttpRequestPtr &req, std::function<void(const HttpResponsePtr &)> &&callback);
 #endif

mysql.cc

@@ -7,6 +7,50 @@
 #include "jdbc/mysql_driver.h"
 #include "json/json.h"
 
+void sql_delete_operation(std::string sender, std::string receiver)
+{
+    std::vector<std::string> usernamelist;
+    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 readdata = "SELECT friends FROM users WHERE username = ?";
+    std::string Sfriendlist, Rfriendlist;
+    std::string changedata = "UPDATE users SET friends=? WHERE username =?";
+    sql::PreparedStatement *readdatament = con->prepareStatement(readdata);
+    readdatament->setString(1, sender);
+    sql::ResultSet *SresultSet = readdatament->executeQuery();
+    if (SresultSet->next())
+    {
+        Sfriendlist = SresultSet->getString("friends");
+    }
+    int pos = Sfriendlist.find(receiver);
+    if (pos != std::string::npos && pos != 0)
+        Sfriendlist.erase(pos - 1, receiver.length() + 1);
+    else if (pos == 0)
+        Sfriendlist.erase(pos, receiver.length() + 1);
+    std::cout << Sfriendlist << "\n";
+    sql::PreparedStatement *updateStatement = con->prepareStatement(changedata);
+    updateStatement->setString(1, Sfriendlist);
+    updateStatement->setString(2, sender);
+    updateStatement->execute();
+    readdatament->setString(1, receiver);
+    sql::ResultSet *RresultSet = readdatament->executeQuery();
+    if (RresultSet->next())
+    {
+        Rfriendlist = RresultSet->getString("friends");
+    }
+    pos = Rfriendlist.find(sender);
+    if (pos != std::string::npos && pos != 0)
+        Rfriendlist.erase(pos - 1, sender.length() + 1);
+    else if (pos == 0)
+        Rfriendlist.erase(pos, sender.length() + 1);
+    std::cout << Rfriendlist << "\n";
+    updateStatement->setString(1, Rfriendlist);
+    updateStatement->setString(2, receiver);
+    updateStatement->execute();
+}
 void sql_process_request(std::string sender, std::string receiver, std::string attitude)
 {
     std::vector<std::string> usernamelist;
@@ -91,7 +135,6 @@ void sql_process_request(std::string sender, std::string receiver, std::string a
             updateS->execute();
         }
     }
-
     return;
 }
 

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);
 Json::Value get_chat_info(std::string,std::string="");
+void sql_delete_operation(std::string,std::string);
 //bool sql_check(std::string );
 #endif