X-Git-Url: https://git.pterodactylus.net/?a=blobdiff_plain;f=src%2Ffreenet%2Fmessagelistinserter.cpp;h=04ab9f30056d7b26bc61a54856d9bc08f88df9b3;hb=4e96d123460d6363cf7274e36bd9357768eb86ad;hp=aa1e814ee545cb94dca900a6a74883f75e330ac1;hpb=221236a4d3aac4144529d418ce368db5c98facb0;p=fms.git

diff --git a/src/freenet/messagelistinserter.cpp b/src/freenet/messagelistinserter.cpp
index aa1e814..04ab9f3 100644
--- a/src/freenet/messagelistinserter.cpp
+++ b/src/freenet/messagelistinserter.cpp
@@ -15,7 +15,7 @@ MessageListInserter::MessageListInserter()
 	Initialize();
 }
 
-MessageListInserter::MessageListInserter(FCPv2 *fcp):IIndexInserter<long>(fcp)
+MessageListInserter::MessageListInserter(FCPv2::Connection *fcp):IIndexInserter<long>(fcp)
 {
 	Initialize();
 }
@@ -60,7 +60,7 @@ void MessageListInserter::CheckForNeededInsert()
 
 }
 
-const bool MessageListInserter::HandlePutFailed(FCPMessage &message)
+const bool MessageListInserter::HandlePutFailed(FCPv2::Message &message)
 {
 	std::vector<std::string> idparts;
 	long localidentityid;
@@ -97,7 +97,7 @@ const bool MessageListInserter::HandlePutFailed(FCPMessage &message)
 
 }
 
-const bool MessageListInserter::HandlePutSuccessful(FCPMessage &message)
+const bool MessageListInserter::HandlePutSuccessful(FCPv2::Message &message)
 {
 	Poco::DateTime now;
 	std::vector<std::string> idparts;
@@ -124,10 +124,19 @@ const bool MessageListInserter::HandlePutSuccessful(FCPMessage &message)
 		st.Bind(1,localidentityid);
 		st.Step();
 
-		// delete any record from tmpMessageListInsert
-		st=m_db->Prepare("DELETE FROM tmpMessageListInsert WHERE LocalIdentityID=?;");
+		// delete only a single record from tmpMessageListInsert
+		st=m_db->Prepare("SELECT MessageListInsertID FROM tmpMessageListInsert WHERE LocalIdentityID=?;");
 		st.Bind(0,localidentityid);
 		st.Step();
+		if(st.RowReturned())
+		{
+			int id=-1;
+			st.ResultInt(0,id);
+
+			st=m_db->Prepare("DELETE FROM tmpMessageListInsert WHERE MessageListInsertID=?;");
+			st.Bind(0,id);
+			st.Step();
+		}
 
 		RemoveFromInsertList(localidentityid);
 
@@ -153,7 +162,7 @@ void MessageListInserter::Initialize()
 
 const bool MessageListInserter::StartInsert(const long &localidentityid)
 {
-	FCPMessage message;
+	FCPv2::Message message;
 	Poco::DateTime date;
 	Poco::DateTime now;
 	std::string privatekey;
@@ -253,10 +262,10 @@ const bool MessageListInserter::StartInsert(const long &localidentityid)
 		message["Identifier"]=m_fcpuniquename+"|"+localidentityidstr+"|"+indexstr+"|"+message["URI"];
 		message["UploadFrom"]="direct";
 		message["DataLength"]=xmlsizestr;
-		m_fcp->SendMessage(message);
-		m_fcp->SendRaw(xmlstr.c_str(),xmlstr.size());
+		m_fcp->Send(message);
+		m_fcp->Send(std::vector<char>(xmlstr.begin(),xmlstr.end()));
 
-		message.Reset();
+		message.Clear();
 		message.SetName("ClientPutComplexDir");
 		message["URI"]="USK"+privatekey.substr(3)+m_messagebase+"|"+Poco::DateTimeFormatter::format(now,"%Y.%m.%d")+"|MessageList/0/";
 		message["Identifier"]=m_fcpuniquename+"USK|"+message["URI"];
@@ -264,8 +273,8 @@ const bool MessageListInserter::StartInsert(const long &localidentityid)
 		message["Files.0.Name"]="MessageList.xml";
 		message["Files.0.UploadFrom"]="direct";
 		message["Files.0.DataLength"]=xmlsizestr;
-		m_fcp->SendMessage(message);
-		m_fcp->SendRaw(xmlstr.c_str(),xmlstr.size());
+		m_fcp->Send(message);
+		m_fcp->Send(std::vector<char>(xmlstr.begin(),xmlstr.end()));
 
 		m_inserting.push_back(localidentityid);
 		m_lastinsertedxml[localidentityid]=xmlstr;