src/freenet/periodicdbmaintenance.cpp

   1 #include "../../include/freenet/periodicdbmaintenance.h"\r
   2 #include "../../include/stringfunctions.h"\r
   3 \r
   4 #ifdef XMEM\r
   5         #include <xmem.h>\r
   6 #endif\r
   7 \r
   8 PeriodicDBMaintenance::PeriodicDBMaintenance()\r
   9 {\r
  10         m_check10mins.SetToGMTime();\r
  11         m_check30mins.SetToGMTime();\r
  12         m_check1hour.SetToGMTime();\r
  13         m_check6hours.SetToGMTime();\r
  14         m_check1day.SetToGMTime();\r
  15 \r
  16         // set back times so they will do their first maintenance soon ( within the next hour ) - stagger so not all run at once\r
  17         m_check30mins.Add(0,-5);\r
  18         m_check1hour.Add(0,-42);\r
  19         m_check6hours.Add(0,-1,-5);\r
  20         m_check1day.Add(0,0,-23);\r
  21 \r
  22 }\r
  23 \r
  24 void PeriodicDBMaintenance::Do10MinuteMaintenance()\r
  25 {\r
  26 \r
  27         m_log->WriteLog(LogFile::LOGLEVEL_DEBUG,"PeriodicDBMaintenance::Do10MinuteMaintenance");\r
  28 }\r
  29 \r
  30 void PeriodicDBMaintenance::Do30MinuteMaintenance()\r
  31 {\r
  32 \r
  33         m_log->WriteLog(LogFile::LOGLEVEL_DEBUG,"PeriodicDBMaintenance::Do30MinuteMaintenance");\r
  34 }\r
  35 \r
  36 void PeriodicDBMaintenance::Do1HourMaintenance()\r
  37 {\r
  38         // recalculate all trust levels - this is CPU instensive\r
  39         // TODO - will probably have to change this to do 1 identity at a time as this locks that database for the duration\r
  40         m_db->Execute("UPDATE tblIdentity SET PeerMessageTrust=(SELECT PeerMessageTrust FROM vwCalculatedPeerTrust WHERE TargetIdentityID=IdentityID), PeerTrustListTrust=(SELECT PeerTrustListTrust FROM vwCalculatedPeerTrust WHERE TargetIdentityID=IdentityID);");\r
  41 \r
  42         m_log->WriteLog(LogFile::LOGLEVEL_DEBUG,"PeriodicDBMaintenance::Do1HourMaintenance");\r
  43 }\r
  44 \r
  45 void PeriodicDBMaintenance::Do6HourMaintenance()\r
  46 {\r
  47 \r
  48         m_log->WriteLog(LogFile::LOGLEVEL_DEBUG,"PeriodicDBMaintenance::Do6HourMaintenance");\r
  49 }\r
  50 \r
  51 void PeriodicDBMaintenance::Do1DayMaintenance()\r
  52 {\r
  53         DateTime date;\r
  54 \r
  55         // delete all puzzles 2 or more days old\r
  56         date.SetToGMTime();\r
  57         date.Add(0,0,0,-2);\r
  58         m_db->Execute("DELETE FROM tblIntroductionPuzzleInserts WHERE Day<='"+date.Format("%Y-%m-%d")+"';");\r
  59         m_db->Execute("DELETE FROM tblIntroductionPuzzleRequests WHERE Day<='"+date.Format("%Y-%m-%d")+"';");\r
  60 \r
  61         m_log->WriteLog(LogFile::LOGLEVEL_DEBUG,"PeriodicDBMaintenance::Do1DayMaintenance");\r
  62 \r
  63         // delete all identities we've never seen and were added more than 20 days ago\r
  64         date.SetToGMTime();\r
  65         date.Add(0,0,0,-20);\r
  66         m_db->Execute("DELETE FROM tblIdentity WHERE LastSeen IS NULL AND DateAdded<'"+date.Format("%Y-%m-%d")+"';");\r
  67 \r
  68         // delete old identity requests - we don't need them anymore\r
  69         date.SetToGMTime();\r
  70         date.Add(0,0,0,-2);\r
  71         m_db->Execute("DELETE FROM tblIdentityRequests WHERE Day<'"+date.Format("%Y-%m-%d")+"';");\r
  72 \r
  73         // delete old local identity inserts - we don't need them anymore\r
  74         date.SetToGMTime();\r
  75         date.Add(0,0,0,-2);\r
  76         m_db->Execute("DELETE FROM tblLocalIdentityInserts WHERE Day<'"+date.Format("%Y-%m-%d")+"';");\r
  77 \r
  78         // delete old message list inserts/requests - we don't need them anymore\r
  79         date.SetToGMTime();\r
  80         date.Add(0,0,0,-2);\r
  81         m_db->Execute("DELETE FROM tblMessageListInserts WHERE Day<'"+date.Format("%Y-%m-%d")+"';");\r
  82         m_db->Execute("DELETE FROM tblMessageListRequests WHERE Day<'"+date.Format("%Y-%m-%d")+"';");\r
  83 \r
  84         // delete old trust list inserts/requests - we don't need them anymore\r
  85         date.SetToGMTime();\r
  86         date.Add(0,0,0,-2);\r
  87         m_db->Execute("DELETE FROM tblTrustListInserts WHERE Day<'"+date.Format("%Y-%m-%d")+"';");\r
  88         m_db->Execute("DELETE FROM tblTrustListRequests WHERE Day<'"+date.Format("%Y-%m-%d")+"';");\r
  89 \r
  90         // delete trust lists from identities we aren't trusting anymore\r
  91         m_db->Execute("DELETE FROM tblPeerTrust WHERE IdentityID NOT IN (SELECT IdentityID FROM tblIdentity WHERE (LocalTrustListTrust>=(SELECT OptionValue FROM tblOption WHERE Option='MinLocalTrustListTrust')) AND (PeerTrustListTrust IS NULL OR PeerTrustListTrust>=(SELECT OptionValue FROM tblOption WHERE Option='MinPeerTrustListTrust')));");\r
  92 \r
  93         // remove identityid from messages where the identity has been deleted\r
  94         m_db->Execute("UPDATE tblMessage SET IdentityID=NULL WHERE IdentityID NOT IN (SELECT IdentityID FROM tblIdentity);");\r
  95 \r
  96         // try to re-attach messages from identities that were previously deleted, but have been since re-added\r
  97         // first get the names from messages that have a NULL IdentityID\r
  98         SQLite3DB::Statement st=m_db->Prepare("SELECT FromName FROM tblMessage WHERE IdentityID IS NULL GROUP BY FromName;");\r
  99         st.Step();\r
 100         while(st.RowReturned())\r
 101         {\r
 102                 std::string name="";\r
 103                 std::string publickey="";\r
 104                 int identityid=0;\r
 105                 st.ResultText(0,name);\r
 106 \r
 107                 std::vector<std::string> parts;\r
 108                 StringFunctions::Split(name,"@",parts);\r
 109 \r
 110                 // find identities with this name\r
 111                 SQLite3DB::Statement st2=m_db->Prepare("SELECT IdentityID,PublicKey FROM tblIdentity WHERE Name=?;");\r
 112                 st2.Bind(0,name);\r
 113                 st2.Step();\r
 114                 while(st2.RowReturned())\r
 115                 {\r
 116                         publickey="";\r
 117                         identityid=0;\r
 118                         st2.ResultText(1,publickey);\r
 119                         // check if public key matches 2nd part\r
 120                         if(parts.size()>1 && publickey.find(parts[1])==4)\r
 121                         {\r
 122                                 // we have the identity - so update the messages table with the identityid\r
 123                                 st2.ResultInt(0,identityid);\r
 124 \r
 125                                 SQLite3DB::Statement st3=m_db->Prepare("UPDATE tblMessage SET IdentityID=? WHERE Name=? AND IdentityID IS NULL;");\r
 126                                 st3.Bind(0,identityid);\r
 127                                 st3.Bind(1,name);\r
 128                                 st3.Step();\r
 129                         }\r
 130                         st2.Step();\r
 131                 }\r
 132 \r
 133                 st.Step();\r
 134         }\r
 135 \r
 136 }\r
 137 \r
 138 void PeriodicDBMaintenance::Process()\r
 139 {\r
 140         DateTime now;\r
 141         now.SetToGMTime();\r
 142         \r
 143         if(m_check10mins<=(now-((1.0/1440.0)*10.0)))\r
 144         {\r
 145                 Do10MinuteMaintenance();\r
 146                 m_check10mins=now;\r
 147         }\r
 148         if(m_check30mins<=(now-((1.0/1440.0)*30.0)))\r
 149         {\r
 150                 Do30MinuteMaintenance();\r
 151                 m_check30mins=now;\r
 152         }\r
 153         if(m_check1hour<=(now-(1.0/24.0)))\r
 154         {\r
 155                 Do1HourMaintenance();\r
 156                 m_check1hour=now;\r
 157         }\r
 158         if(m_check6hours<=(now-(1.0/4.0)))\r
 159         {\r
 160                 Do6HourMaintenance();\r
 161                 m_check6hours=now;\r
 162         }\r
 163         if(m_check1day<=(now-(1.0)))\r
 164         {\r
 165                 Do1DayMaintenance();\r
 166                 m_check1day=now;\r
 167         }\r
 168 \r
 169 }\r
 170 \r
 171 void PeriodicDBMaintenance::RegisterWithThread(FreenetMasterThread *thread)\r
 172 {\r
 173         thread->RegisterPeriodicProcessor(this);\r
 174 }\r