X-Git-Url: https://git.pterodactylus.net/?a=blobdiff_plain;f=src%2Ffreenet%2Fperiodicdbmaintenance.cpp;h=2ac1e07e7b33d4d4f257c4c781aaa925f0f72281;hb=8a0a83a78390f22f99d4487cda2569909dfbc28e;hp=63863f80c74ca0879beb89c7e3829ac104dfc7c7;hpb=4430e7762844c66428b6f822288beb71b7f82b95;p=fms.git

diff --git a/src/freenet/periodicdbmaintenance.cpp b/src/freenet/periodicdbmaintenance.cpp
index 63863f8..2ac1e07 100644
--- a/src/freenet/periodicdbmaintenance.cpp
+++ b/src/freenet/periodicdbmaintenance.cpp
@@ -1,5 +1,6 @@
 #include "../../include/freenet/periodicdbmaintenance.h"
 #include "../../include/stringfunctions.h"
+#include "../../include/option.h"
 
 #ifdef XMEM
 	#include <xmem.h>
@@ -7,6 +8,8 @@
 
 PeriodicDBMaintenance::PeriodicDBMaintenance()
 {
+	std::string tempval;
+
 	m_check10mins.SetToGMTime();
 	m_check30mins.SetToGMTime();
 	m_check1hour.SetToGMTime();
@@ -19,6 +22,10 @@ PeriodicDBMaintenance::PeriodicDBMaintenance()
 	m_check6hours.Add(0,-1,-5);
 	m_check1day.Add(0,0,-23);
 
+	tempval="180";
+	Option::Instance()->Get("DeleteMessagesOlderThan",tempval);
+	StringFunctions::Convert(tempval,m_deletemessagesolderthan);
+
 }
 
 void PeriodicDBMaintenance::Do10MinuteMaintenance()
@@ -36,8 +43,58 @@ void PeriodicDBMaintenance::Do30MinuteMaintenance()
 void PeriodicDBMaintenance::Do1HourMaintenance()
 {
 	// recalculate all trust levels - this is CPU instensive
-	// TODO - will probably have to change this to do 1 identity at a time as this locks that database for the duration
-	m_db->Execute("UPDATE tblIdentity SET PeerMessageTrust=(SELECT PeerMessageTrust FROM vwCalculatedPeerTrust WHERE TargetIdentityID=IdentityID), PeerTrustListTrust=(SELECT PeerTrustListTrust FROM vwCalculatedPeerTrust WHERE TargetIdentityID=IdentityID);");
+	// do 1 identity at a time as doing it with 1 UPDATE statement locks that database for the duration
+	SQLite3DB::Statement st=m_db->Prepare("SELECT TargetIdentityID,PeerMessageTrust,PeerTrustListTrust FROM vwCalculatedPeerTrust;");
+	SQLite3DB::Statement upd=m_db->Prepare("UPDATE tblIdentity SET PeerMessageTrust=?, PeerTrustListTrust=? WHERE IdentityID=?");
+	st.Step();
+	while(st.RowReturned())
+	{
+		int identityid=0;
+		int trust=0;
+		
+		st.ResultInt(0,identityid);
+
+		upd.Bind(0,identityid);
+		if(st.ResultNull(1)==false)
+		{
+			trust=0;
+			st.ResultInt(1,trust);
+			upd.Bind(0,trust);
+		}
+		else
+		{
+			upd.Bind(0);
+		}
+		if(st.ResultNull(2)==false)
+		{
+			trust=0;
+			st.ResultInt(2,trust);
+			upd.Bind(1,trust);
+		}
+		else
+		{
+			upd.Bind(1);
+		}
+		upd.Bind(2,identityid);
+		upd.Step();
+		upd.Reset();
+
+		st.Step();
+	}
+
+	// set null peer trust for identities without a calculated trust
+	st=m_db->Prepare("SELECT IdentityID FROM tblIdentity WHERE IdentityID NOT IN (SELECT TargetIdentityID FROM vwCalculatedPeerTrust);");
+	upd=m_db->Prepare("UPDATE tblIdentity SET PeerMessageTrust=NULL, PeerTrustListTrust=NULL WHERE IdentityID=?;");
+	st.Step();
+	while(st.RowReturned())
+	{
+		int identityid=0;
+		st.ResultInt(0,identityid);
+		upd.Bind(0,identityid);
+		upd.Step();
+		upd.Reset();
+		st.Step();
+	}
 
 	m_log->WriteLog(LogFile::LOGLEVEL_DEBUG,"PeriodicDBMaintenance::Do1HourMaintenance");
 }
@@ -45,6 +102,34 @@ void PeriodicDBMaintenance::Do1HourMaintenance()
 void PeriodicDBMaintenance::Do6HourMaintenance()
 {
 
+	// if we remove a board and the reply boardid is still set to it, we need to replace it with a boardid that does exist
+	SQLite3DB::Statement st=m_db->Prepare("SELECT MessageID FROM tblMessage WHERE ReplyBoardID NOT IN (SELECT BoardID FROM tblBoard);");
+	SQLite3DB::Statement st2=m_db->Prepare("SELECT BoardID FROM tblMessageBoard WHERE MessageID=?;");
+	SQLite3DB::Statement upd=m_db->Prepare("UPDATE tblMessage SET ReplyBoardID=? WHERE MessageID=?;");
+	st.Step();
+	while(st.RowReturned())
+	{
+		// find a valid boardid for the message
+		int messageid=0;
+		int boardid=0;
+
+		st.ResultInt(0,messageid);
+
+		st2.Bind(0,messageid);
+		st2.Step();
+		if(st2.RowReturned())
+		{
+			st2.ResultInt(0,boardid);
+			upd.Bind(0,boardid);
+			upd.Bind(1,messageid);
+			upd.Step();
+			upd.Reset();
+		}
+		st2.Reset();
+		
+		st.Step();
+	}
+
 	m_log->WriteLog(LogFile::LOGLEVEL_DEBUG,"PeriodicDBMaintenance::Do6HourMaintenance");
 }
 
@@ -147,6 +232,13 @@ void PeriodicDBMaintenance::Do1DayMaintenance()
 	st.Bind(0,date.Format("%Y-%m-%d %H:%M:%S"));
 	st.Step();
 
+	// delete old messages
+	date.SetToGMTime();
+	date.Add(0,0,0,-m_deletemessagesolderthan);
+	st=m_db->Prepare("DELETE FROM tblMessage WHERE MessageDate<?;");
+	st.Bind(0,date.Format("%Y-%m-%d"));
+	st.Step();
+
 }
 
 void PeriodicDBMaintenance::Process()