Content::iterator Find(const std::string& filename)
{
for (Content::iterator it = content_.begin();
it != content_.end(); ++it)
{
if ((*it)->GetFilename() == filename)
{
return it;
}
}
return content_.end();
}
void Clear()
{
for (Content::iterator it = content_.begin();
it != content_.end(); ++it)
{
delete *it;
}
}
void Remove(const std::string& filename)
{
Content::iterator it = Find(filename);
if (it != content_.end())
{
delete *it;
content_.erase(it);
}
}
PostDataStatus Store(std::string& completed,
const char* chunkData,
size_t chunkSize,
const std::string& filename,
size_t filesize)
{
boost::mutex::scoped_lock lock(mutex_);
std::set<std::string>::iterator wasDiscarded = discardedFiles_.find(filename);
if (wasDiscarded != discardedFiles_.end())
{
discardedFiles_.erase(wasDiscarded);
return PostDataStatus_Failure;
}
ChunkedFile* f;
Content::iterator it = Find(filename);
if (it == content_.end())
{
f = new ChunkedFile(filename);
// Make some room
if (content_.size() >= numPlaces_)
{
discardedFiles_.insert(content_.front()->GetFilename());
delete content_.front();
content_.pop_front();
}
content_.push_back(f);
}
else
{
f = *it;
}
f->AddChunk(chunkData, chunkSize);
if (f->GetNumBytes() > filesize)
{
Remove(filename);
}
else if (f->GetNumBytes() == filesize)
{
f->Flatten(completed);
Remove(filename);
return PostDataStatus_Success;
}
return PostDataStatus_Pending;
}