1 #include "../../include/freenet/messagelistinserter.h"
\r
2 #include "../../include/freenet/messagexml.h"
\r
3 #include "../../include/freenet/messagelistxml.h"
\r
9 MessageListInserter::MessageListInserter()
\r
14 MessageListInserter::MessageListInserter(FCPv2 *fcp):IIndexInserter<long>(fcp)
\r
19 void MessageListInserter::CheckForNeededInsert()
\r
21 // only do 1 insert at a time
\r
22 if(m_inserting.size()==0)
\r
29 previous.SetToGMTime();
\r
31 previous.Add(0,0,0,-m_daysbackward);
\r
33 // query for identities that have messages in the past X days and (we haven't inserted lists for in the past 30 minutes OR identity has a record in tmpMessageListInsert)
\r
34 sql="SELECT tblLocalIdentity.LocalIdentityID ";
\r
35 sql+="FROM tblLocalIdentity INNER JOIN tblMessageInserts ON tblLocalIdentity.LocalIdentityID=tblMessageInserts.LocalIdentityID ";
\r
36 sql+="WHERE tblMessageInserts.Day>=? AND ((tblLocalIdentity.LastInsertedMessageList<=? OR tblLocalIdentity.LastInsertedMessageList IS NULL OR tblLocalIdentity.LastInsertedMessageList='') OR tblLocalIdentity.LocalIdentityID IN (SELECT LocalIdentityID FROM tmpMessageListInsert)) ";
\r
39 SQLite3DB::Statement st=m_db->Prepare(sql);
\r
40 st.Bind(0,previous.Format("%Y-%m-%d"));
\r
41 st.Bind(1,(now-(1.0/48.0)).Format("%Y-%m-%d %H:%M:%S"));
\r
44 if(st.RowReturned())
\r
46 int localidentityid;
\r
47 st.ResultInt(0,localidentityid);
\r
48 StartInsert(localidentityid);
\r
54 const bool MessageListInserter::HandlePutFailed(FCPMessage &message)
\r
56 std::vector<std::string> idparts;
\r
57 long localidentityid;
\r
60 StringFunctions::Split(message["Identifier"],"|",idparts);
\r
61 StringFunctions::Convert(idparts[1],localidentityid);
\r
62 StringFunctions::Convert(idparts[2],index);
\r
64 if(message["Fatal"]=="true" || message["Code"]=="9")
\r
66 SQLite3DB::Statement st=m_db->Prepare("INSERT INTO tblMessageListInserts(LocalIdentityID,Day,InsertIndex,Inserted) VALUES(?,?,?,'false');");
\r
67 st.Bind(0,localidentityid);
\r
68 st.Bind(1,idparts[4]);
\r
73 RemoveFromInsertList(localidentityid);
\r
79 const bool MessageListInserter::HandlePutSuccessful(FCPMessage &message)
\r
82 std::vector<std::string> idparts;
\r
83 long localidentityid;
\r
86 StringFunctions::Split(message["Identifier"],"|",idparts);
\r
87 StringFunctions::Convert(idparts[1],localidentityid);
\r
88 StringFunctions::Convert(idparts[2],index);
\r
90 SQLite3DB::Statement st=m_db->Prepare("INSERT INTO tblMessageListInserts(LocalIdentityID,Day,InsertIndex,Inserted) VALUES(?,?,?,'true');");
\r
91 st.Bind(0,localidentityid);
\r
92 st.Bind(1,idparts[4]);
\r
97 st=m_db->Prepare("UPDATE tblLocalIdentity SET LastInsertedMessageList=? WHERE LocalIdentityID=?;");
\r
98 st.Bind(0,now.Format("%Y-%m-%d %H:%M:%S"));
\r
99 st.Bind(1,localidentityid);
\r
102 // delete any record from tmpMessageListInsert
\r
103 st=m_db->Prepare("DELETE FROM tmpMessageListInsert WHERE LocalIdentityID=?;");
\r
104 st.Bind(0,localidentityid);
\r
107 RemoveFromInsertList(localidentityid);
\r
109 m_log->WriteLog(LogFile::LOGLEVEL_DEBUG,"MessageListInserter::HandlePutSuccessful successfully inserted MessageList.");
\r
114 void MessageListInserter::Initialize()
\r
116 std::string tempval;
\r
118 m_fcpuniquename="MessageListInserter";
\r
120 Option::Instance()->Get("MessageListDaysBackward",tempval);
\r
121 StringFunctions::Convert(tempval,m_daysbackward);
\r
124 void MessageListInserter::StartInsert(const long &localidentityid)
\r
126 FCPMessage message;
\r
129 std::string privatekey;
\r
130 std::string localidentityidstr;
\r
131 MessageListXML mlxml;
\r
132 MessageXML messxml;
\r
133 std::string xmlstr;
\r
134 std::string xmlsizestr;
\r
136 std::string indexstr;
\r
139 date.SetToGMTime();
\r
140 date.Add(0,0,0,-m_daysbackward);
\r
141 StringFunctions::Convert(localidentityid,localidentityidstr);
\r
143 SQLite3DB::Statement st=m_db->Prepare("SELECT Day, InsertIndex, MessageXML, PrivateKey FROM tblMessageInserts INNER JOIN tblLocalIdentity ON tblMessageInserts.LocalIdentityID=tblLocalIdentity.LocalIdentityID WHERE tblLocalIdentity.LocalIdentityID=? AND Day>=?;");
\r
144 st.Bind(0,localidentityid);
\r
145 st.Bind(1,date.Format("%Y-%m-%d"));
\r
148 while(st.RowReturned())
\r
152 std::string xmlstr;
\r
153 std::vector<std::string> boards;
\r
155 st.ResultText(0,day);
\r
156 st.ResultInt(1,index);
\r
157 st.ResultText(2,xmlstr);
\r
158 st.ResultText(3,privatekey);
\r
160 messxml.ParseXML(xmlstr);
\r
162 mlxml.AddMessage(day,index,messxml.GetBoards());
\r
168 // get last inserted messagelist index for this day
\r
170 st=m_db->Prepare("SELECT MAX(InsertIndex) FROM tblMessageListInserts WHERE LocalIdentityID=? AND Day=?;");
\r
171 st.Bind(0,localidentityid);
\r
172 st.Bind(1,now.Format("%Y-%m-%d"));
\r
174 if(st.ResultNull(0)==false)
\r
176 st.ResultInt(0,index);
\r
179 StringFunctions::Convert(index,indexstr);
\r
181 // actually insert message
\r
182 xmlstr=mlxml.GetXML();
\r
183 StringFunctions::Convert(xmlstr.size(),xmlsizestr);
\r
185 message.SetName("ClientPut");
\r
186 message["URI"]=privatekey+m_messagebase+"|"+now.Format("%Y-%m-%d")+"|MessageList|"+indexstr+".xml";
\r
187 message["Identifier"]=m_fcpuniquename+"|"+localidentityidstr+"|"+indexstr+"|"+message["URI"];
\r
188 message["UploadFrom"]="direct";
\r
189 message["DataLength"]=xmlsizestr;
\r
190 m_fcp->SendMessage(message);
\r
191 m_fcp->SendRaw(xmlstr.c_str(),xmlstr.size());
\r
193 m_inserting.push_back(localidentityid);
\r