1 #ifndef _iindexrequester_
\r
2 #define _iindexrequester_
\r
4 #include "../idatabase.h"
\r
5 #include "../ilogger.h"
\r
6 #include "../datetime.h"
\r
7 #include "../option.h"
\r
8 #include "../stringfunctions.h"
\r
9 #include "ifreenetregistrable.h"
\r
10 #include "ifcpconnected.h"
\r
11 #include "ifcpmessagehandler.h"
\r
12 #include "iperiodicprocessor.h"
\r
18 template <class IDTYPE>
\r
19 class IIndexRequester:public IFreenetRegistrable,public IFCPConnected,public IFCPMessageHandler,public IPeriodicProcessor,public IDatabase,public ILogger
\r
23 IIndexRequester(FCPv2 *fcp);
\r
24 virtual ~IIndexRequester() {}
\r
26 virtual void FCPConnected();
\r
27 virtual void FCPDisconnected();
\r
28 virtual const bool HandleMessage(FCPMessage &message);
\r
30 virtual void Process();
\r
32 virtual void RegisterWithThread(FreenetMasterThread *thread);
\r
35 void InitializeIIndexRequester();
\r
36 virtual void Initialize()=0; // initialize m_maxrequests and m_fcpuniquename
\r
37 virtual void PopulateIDList()=0;
\r
38 virtual void StartRequest(const IDTYPE &id)=0;
\r
39 virtual const bool HandleAllData(FCPMessage &message)=0;
\r
40 virtual const bool HandleGetFailed(FCPMessage &message)=0;
\r
41 virtual void RemoveFromRequestList(const IDTYPE id);
\r
43 DateTime m_tempdate;
\r
44 std::string m_messagebase;
\r
45 std::map<IDTYPE,bool> m_ids; // map of all ids we know and whether we have requested file from them yet
\r
46 std::vector<IDTYPE> m_requesting; // list of ids we are currently requesting from
\r
48 // these MUST be populated by child class
\r
50 std::string m_fcpuniquename;
\r
54 template <class IDTYPE>
\r
55 IIndexRequester<IDTYPE>::IIndexRequester()
\r
57 InitializeIIndexRequester();
\r
60 template <class IDTYPE>
\r
61 IIndexRequester<IDTYPE>::IIndexRequester(FCPv2 *fcp):IFCPConnected(fcp)
\r
63 InitializeIIndexRequester();
\r
66 template <class IDTYPE>
\r
67 void IIndexRequester<IDTYPE>::FCPConnected()
\r
69 // make sure variables have been initialized by the derived class
\r
70 if(m_maxrequests==-1)
\r
72 m_log->WriteLog(LogFile::LOGLEVEL_ERROR,"IIndexRequester<IDTYPE>::FCPConnected maxrequests not initialized correctly!");
\r
74 if(m_fcpuniquename=="")
\r
76 m_log->WriteLog(LogFile::LOGLEVEL_ERROR,"IIndexRequester<IDTYPE>::FCPConnected fcpuniquename not initialized correctly!");
\r
78 if(m_fcpuniquename.find("|")!=std::string::npos)
\r
80 m_log->WriteLog(LogFile::LOGLEVEL_ERROR,"IIndexRequester<IDTYPE>::FCPConnected fcpuniquename contains | character! This is not a valid character!");
\r
83 m_requesting.clear();
\r
87 template <class IDTYPE>
\r
88 void IIndexRequester<IDTYPE>::FCPDisconnected()
\r
93 template <class IDTYPE>
\r
94 const bool IIndexRequester<IDTYPE>::HandleMessage(FCPMessage &message)
\r
97 if(message["Identifier"].find(m_fcpuniquename)==0)
\r
99 if(message.GetName()=="DataFound")
\r
104 if(message.GetName()=="AllData")
\r
106 return HandleAllData(message);
\r
109 if(message.GetName()=="GetFailed")
\r
111 return HandleGetFailed(message);
\r
114 if(message.GetName()=="IdentifierCollision")
\r
116 // remove one of the ids from the requesting list
\r
118 std::vector<std::string> idparts;
\r
119 StringFunctions::Split(message["Identifier"],"|",idparts);
\r
120 StringFunctions::Convert(idparts[1],id);
\r
121 RemoveFromRequestList(id);
\r
129 template <class IDTYPE>
\r
130 void IIndexRequester<IDTYPE>::InitializeIIndexRequester()
\r
133 m_fcpuniquename="";
\r
135 Option::Instance()->Get("MessageBase",m_messagebase);
\r
136 m_tempdate.SetToGMTime();
\r
139 template <class IDTYPE>
\r
140 void IIndexRequester<IDTYPE>::Process()
\r
142 // max is the smaller of the config value or the total number of ids we will request from
\r
143 typename std::map<IDTYPE,bool>::size_type max=m_maxrequests>m_ids.size() ? m_ids.size() : m_maxrequests;
\r
145 // try to keep up to max requests going
\r
146 if(m_requesting.size()<max)
\r
148 typename std::map<IDTYPE,bool>::iterator i=m_ids.begin();
\r
150 while(i!=m_ids.end() && (*i).second==true)
\r
157 StartRequest((*i).first);
\r
161 // we requested from all ids in the list, repopulate the list
\r
165 // special case - if there were 0 ids on the list when we started then we will never get a chance to repopulate the list
\r
166 // this will recheck for ids every minute
\r
169 if(m_ids.size()==0 && m_tempdate<(now-(1.0/1440.0)))
\r
177 template <class IDTYPE>
\r
178 void IIndexRequester<IDTYPE>::RegisterWithThread(FreenetMasterThread *thread)
\r
180 thread->RegisterFCPConnected(this);
\r
181 thread->RegisterFCPMessageHandler(this);
\r
182 thread->RegisterPeriodicProcessor(this);
\r
185 template <class IDTYPE>
\r
186 void IIndexRequester<IDTYPE>::RemoveFromRequestList(const IDTYPE id)
\r
188 typename std::vector<IDTYPE>::iterator i=m_requesting.begin();
\r
189 while(i!=m_requesting.end() && (*i)!=id)
\r
193 if(i!=m_requesting.end())
\r
195 m_requesting.erase(i);
\r
199 #endif // _iindexrequester_
\r