foundationdb/fdbrpc/AsyncFileNonDurable.actor.h

845 lines
35 KiB
C++

/*
* AsyncFileNonDurable.actor.h
*
* This source file is part of the FoundationDB open source project
*
* Copyright 2013-2018 Apple Inc. and the FoundationDB project authors
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
#pragma once
// When actually compiled (NO_INTELLISENSE), include the generated version of this file. In intellisense use the source
// version.
#if defined(NO_INTELLISENSE) && !defined(FLOW_ASYNCFILENONDURABLE_ACTOR_G_H)
#define FLOW_ASYNCFILENONDURABLE_ACTOR_G_H
#include "fdbrpc/AsyncFileNonDurable.actor.g.h"
#elif !defined(FLOW_ASYNCFILENONDURABLE_ACTOR_H)
#define FLOW_ASYNCFILENONDURABLE_ACTOR_H
#include "flow/flow.h"
#include "fdbrpc/IAsyncFile.h"
#include "flow/ActorCollection.h"
#include "fdbrpc/simulator.h"
#include "fdbrpc/TraceFileIO.h"
#include "fdbrpc/RangeMap.h"
#include "flow/actorcompiler.h" // This must be the last #include.
#undef max
#undef min
ACTOR Future<Void> sendOnProcess(ISimulator::ProcessInfo* process, Promise<Void> promise, TaskPriority taskID);
ACTOR Future<Void> sendErrorOnProcess(ISimulator::ProcessInfo* process,
Promise<Void> promise,
Error e,
TaskPriority taskID);
ACTOR template <class T>
Future<T> sendErrorOnShutdown(Future<T> in) {
choose {
when(wait(success(g_simulator.getCurrentProcess()->shutdownSignal.getFuture()))) {
throw io_error().asInjectedFault();
}
when(T rep = wait(in)) { return rep; }
}
}
class AsyncFileDetachable final : public IAsyncFile, public ReferenceCounted<AsyncFileDetachable> {
private:
Reference<IAsyncFile> file;
Future<Void> shutdown;
public:
explicit AsyncFileDetachable(Reference<IAsyncFile> file) : file(file) { shutdown = doShutdown(this); }
ACTOR Future<Void> doShutdown(AsyncFileDetachable* self) {
wait(success(g_simulator.getCurrentProcess()->shutdownSignal.getFuture()));
self->file = Reference<IAsyncFile>();
return Void();
}
ACTOR static Future<Reference<IAsyncFile>> open(Future<Reference<IAsyncFile>> wrappedFile) {
choose {
when(wait(success(g_simulator.getCurrentProcess()->shutdownSignal.getFuture()))) {
throw io_error().asInjectedFault();
}
when(Reference<IAsyncFile> f = wait(wrappedFile)) { return makeReference<AsyncFileDetachable>(f); }
}
}
void addref() override { ReferenceCounted<AsyncFileDetachable>::addref(); }
void delref() override { ReferenceCounted<AsyncFileDetachable>::delref(); }
Future<int> read(void* data, int length, int64_t offset) override {
if (!file.getPtr() || g_simulator.getCurrentProcess()->shutdownSignal.getFuture().isReady())
return io_error().asInjectedFault();
return sendErrorOnShutdown(file->read(data, length, offset));
}
Future<Void> write(void const* data, int length, int64_t offset) override {
if (!file.getPtr() || g_simulator.getCurrentProcess()->shutdownSignal.getFuture().isReady())
return io_error().asInjectedFault();
return sendErrorOnShutdown(file->write(data, length, offset));
}
Future<Void> truncate(int64_t size) override {
if (!file.getPtr() || g_simulator.getCurrentProcess()->shutdownSignal.getFuture().isReady())
return io_error().asInjectedFault();
return sendErrorOnShutdown(file->truncate(size));
}
Future<Void> sync() override {
if (!file.getPtr() || g_simulator.getCurrentProcess()->shutdownSignal.getFuture().isReady())
return io_error().asInjectedFault();
return sendErrorOnShutdown(file->sync());
}
Future<int64_t> size() const override {
if (!file.getPtr() || g_simulator.getCurrentProcess()->shutdownSignal.getFuture().isReady())
return io_error().asInjectedFault();
return sendErrorOnShutdown(file->size());
}
int64_t debugFD() const override {
if (!file.getPtr())
throw io_error().asInjectedFault();
return file->debugFD();
}
std::string getFilename() const override {
if (!file.getPtr())
throw io_error().asInjectedFault();
return file->getFilename();
}
};
// An async file implementation which wraps another async file and will randomly destroy sectors that it is writing when
// killed This is used to simulate a power failure which prevents all written data from being persisted to disk
class AsyncFileNonDurable final : public IAsyncFile, public ReferenceCounted<AsyncFileNonDurable> {
public:
UID id;
std::string filename;
// An approximation of the size of the file; .size() should be used instead of this variable in most cases
mutable int64_t approximateSize;
// The address of the machine that opened the file
NetworkAddress openedAddress;
bool aio;
private:
// The wrapped IAsyncFile
Reference<IAsyncFile> file;
// The maximum amount of time a write is delayed before being passed along to the underlying file
double maxWriteDelay;
// Modifications which haven't been pushed to file, mapped by the location in the file that is being modified.
// Be sure to update minSizeAfterPendingModifications when modifying pendingModifications.
RangeMap<uint64_t, Future<Void>> pendingModifications;
// The size of the file after the set of pendingModifications completes,
// (the set pending at the time of reading this member). Must be updated in
// lockstep with any inserts into the pendingModifications map. Tracking
// this variable is necessary so that we can know the range of the file a
// truncate is modifying, so we can insert it into the pendingModifications
// map. Until minSizeAfterPendingModificationsIsExact is true, this is only a lower bound.
mutable int64_t minSizeAfterPendingModifications = 0;
mutable bool minSizeAfterPendingModificationsIsExact = false;
// Will be blocked whenever kill is running
Promise<Void> killed;
Promise<Void> killComplete;
// Used by sync (and kill) to force writes which have not yet been passed along.
// If true is sent, then writes will be durable. If false, then they may not be durable.
Promise<bool> startSyncPromise;
// The performance parameters of the simulated disk
Reference<DiskParameters> diskParameters;
// Set to true the first time sync is called on the file
bool hasBeenSynced;
// Used to describe what corruption is allowed by the file as well as the type of corruption being used on a
// particular page
enum KillMode { NO_CORRUPTION = 0, DROP_ONLY = 1, FULL_CORRUPTION = 2 };
// Limits what types of corruption are applied to writes from this file
KillMode killMode;
ActorCollection
reponses; // cannot call getResult on this actor collection, since the actors will be on different processes
AsyncFileNonDurable(const std::string& filename,
Reference<IAsyncFile> file,
Reference<DiskParameters> diskParameters,
NetworkAddress openedAddress,
bool aio)
: openedAddress(openedAddress), pendingModifications(uint64_t(-1)), approximateSize(0), reponses(false),
aio(aio) {
// This is only designed to work in simulation
ASSERT(g_network->isSimulated());
this->id = deterministicRandom()->randomUniqueID();
//TraceEvent("AsyncFileNonDurable_Create", id).detail("Filename", filename);
this->file = file;
this->filename = filename;
this->diskParameters = diskParameters;
maxWriteDelay = 5.0;
hasBeenSynced = false;
killMode = (KillMode)deterministicRandom()->randomInt(1, 3);
//TraceEvent("AsyncFileNonDurable_CreateEnd", id).detail("Filename", filename).backtrace();
}
public:
static std::map<std::string, Future<Void>> filesBeingDeleted;
// Creates a new AsyncFileNonDurable which wraps the provided IAsyncFile
ACTOR static Future<Reference<IAsyncFile>> open(std::string filename,
std::string actualFilename,
Future<Reference<IAsyncFile>> wrappedFile,
Reference<DiskParameters> diskParameters,
bool aio) {
state ISimulator::ProcessInfo* currentProcess = g_simulator.getCurrentProcess();
state TaskPriority currentTaskID = g_network->getCurrentTask();
state Future<Void> shutdown = success(currentProcess->shutdownSignal.getFuture());
//TraceEvent("AsyncFileNonDurableOpenBegin").detail("Filename", filename).detail("Addr", g_simulator.getCurrentProcess()->address);
wait(g_simulator.onMachine(currentProcess));
try {
wait(success(wrappedFile) || shutdown);
if (shutdown.isReady())
throw io_error().asInjectedFault();
state Reference<IAsyncFile> file = wrappedFile.get();
// If we are in the process of deleting a file, we can't let someone else modify it at the same time. We
// therefore block the creation of new files until deletion is complete
state std::map<std::string, Future<Void>>::iterator deletedFile = filesBeingDeleted.find(filename);
if (deletedFile != filesBeingDeleted.end()) {
//TraceEvent("AsyncFileNonDurableOpenWaitOnDelete1").detail("Filename", filename);
wait(deletedFile->second || shutdown);
//TraceEvent("AsyncFileNonDurableOpenWaitOnDelete2").detail("Filename", filename);
if (shutdown.isReady())
throw io_error().asInjectedFault();
}
state Reference<AsyncFileNonDurable> nonDurableFile(
new AsyncFileNonDurable(filename, file, diskParameters, currentProcess->address, aio));
// Causes the approximateSize member to be set
state Future<int64_t> sizeFuture = nonDurableFile->size();
wait(success(sizeFuture) || shutdown);
if (shutdown.isReady())
throw io_error().asInjectedFault();
//TraceEvent("AsyncFileNonDurableOpenComplete").detail("Filename", filename);
wait(g_simulator.onProcess(currentProcess, currentTaskID));
return nonDurableFile;
} catch (Error& e) {
state Error err = e;
std::string currentFilename =
(wrappedFile.isReady() && !wrappedFile.isError()) ? wrappedFile.get()->getFilename() : actualFilename;
currentProcess->machine->openFiles.erase(currentFilename);
//TraceEvent("AsyncFileNonDurableOpenError").error(e, true).detail("Filename", filename).detail("Address", currentProcess->address).detail("Addr", g_simulator.getCurrentProcess()->address);
wait(g_simulator.onProcess(currentProcess, currentTaskID));
throw err;
}
}
~AsyncFileNonDurable() override {
//TraceEvent("AsyncFileNonDurable_Destroy", id).detail("Filename", filename);
}
void addref() override { ReferenceCounted<AsyncFileNonDurable>::addref(); }
void delref() override {
if (delref_no_destroy()) {
ASSERT(filesBeingDeleted.count(filename) == 0);
//TraceEvent("AsyncFileNonDurable_StartDelete", id).detail("Filename", filename);
Future<Void> deleteFuture = deleteFile(this);
if (!deleteFuture.isReady())
filesBeingDeleted[filename] = deleteFuture;
}
}
// Passes along reads straight to the underlying file, waiting for any outstanding changes that could affect the
// results
Future<int> read(void* data, int length, int64_t offset) override { return read(this, data, length, offset); }
// Writes data to the file. Writes are delayed a random amount of time before being
// passed to the underlying file
Future<Void> write(void const* data, int length, int64_t offset) override {
//TraceEvent("AsyncFileNonDurable_Write", id).detail("Filename", filename).detail("Offset", offset).detail("Length", length);
if (length == 0) {
TraceEvent(SevWarnAlways, "AsyncFileNonDurable_EmptyModification", id).detail("Filename", filename);
return Void();
}
debugFileSet("AsyncFileNonDurableWrite", filename, data, offset, length);
Promise<Void> writeStarted;
Promise<Future<Void>> writeEnded;
writeEnded.send(write(this, writeStarted, writeEnded.getFuture(), data, length, offset));
return writeStarted.getFuture();
}
// Truncates the file. Truncates are delayed a random amount of time before being
// passed to the underlying file
Future<Void> truncate(int64_t size) override {
//TraceEvent("AsyncFileNonDurable_Truncate", id).detail("Filename", filename).detail("Offset", size);
debugFileTruncate("AsyncFileNonDurableTruncate", filename, size);
Promise<Void> truncateStarted;
Promise<Future<Void>> truncateEnded;
truncateEnded.send(truncate(this, truncateStarted, truncateEnded.getFuture(), size));
return truncateStarted.getFuture();
}
// Fsyncs the file. This allows all delayed modifications to the file to complete before
// syncing the underlying file
Future<Void> sync() override {
//TraceEvent("AsyncFileNonDurable_Sync", id).detail("Filename", filename);
Future<Void> syncFuture = sync(this, true);
reponses.add(syncFuture);
return syncFuture;
}
// Passes along size requests to the underlying file, augmenting with any writes past the end of the file
Future<int64_t> size() const override { return size(this); }
int64_t debugFD() const override { return file->debugFD(); }
std::string getFilename() const override { return file->getFilename(); }
// Forces a non-durable sync (some writes are not made or made incorrectly)
// This is used when the file should 'die' without first completing its operations
//(e.g. to simulate power failure)
Future<Void> kill() {
TraceEvent("AsyncFileNonDurable_Kill", id).detail("Filename", filename);
TEST(true); // AsyncFileNonDurable was killed
return sync(this, false);
}
private:
// Returns a future that is used to ensure the waiter ends up on the main thread
Future<Void> returnToMainThread() {
Promise<Void> p;
Future<Void> f = p.getFuture();
g_network->onMainThread(std::move(p), g_network->getCurrentTask());
return f;
}
// Gets existing modifications that overlap the specified range. Optionally inserts a new modification into the map
std::vector<Future<Void>> getModificationsAndInsert(int64_t offset,
int64_t length,
bool insertModification = false,
Future<Void> value = Void()) {
auto modification = RangeMapRange<uint64_t>(offset, length >= 0 ? offset + length : uint64_t(-1));
auto priorModifications = pendingModifications.intersectingRanges(modification);
// Aggregate existing modifications in this range
std::vector<Future<Void>> modificationFutures;
for (auto itr = priorModifications.begin(); itr != priorModifications.end(); ++itr) {
if (itr.value().isValid() && (!itr.value().isReady() || itr.value().isError())) {
modificationFutures.push_back(itr.value());
}
}
// Add the modification if we are doing a write or truncate
if (insertModification)
pendingModifications.insert(modification, value);
return modificationFutures;
}
// Checks if the file is killed. If so, then the current sync is completed if running and then an error is thrown
ACTOR static Future<Void> checkKilled(AsyncFileNonDurable const* self, std::string context) {
if (self->killed.isSet()) {
//TraceEvent("AsyncFileNonDurable_KilledInCheck", self->id).detail("In", context).detail("Filename", self->filename);
wait(self->killComplete.getFuture());
TraceEvent("AsyncFileNonDurable_KilledFileOperation", self->id)
.detail("In", context)
.detail("Filename", self->filename);
TEST(true); // AsyncFileNonDurable operation killed
throw io_error().asInjectedFault();
}
return Void();
}
// Passes along reads straight to the underlying file, waiting for any outstanding changes that could affect the
// results
ACTOR Future<int> onRead(AsyncFileNonDurable* self, void* data, int length, int64_t offset) {
wait(checkKilled(self, "Read"));
vector<Future<Void>> priorModifications = self->getModificationsAndInsert(offset, length);
wait(waitForAll(priorModifications));
state Future<int> readFuture = self->file->read(data, length, offset);
wait(success(readFuture) || self->killed.getFuture());
// throws if we were killed
wait(checkKilled(self, "ReadEnd"));
debugFileCheck("AsyncFileNonDurableRead", self->filename, data, offset, length);
// if(g_simulator.getCurrentProcess()->rebooting)
//TraceEvent("AsyncFileNonDurable_ReadEnd", self->id).detail("Filename", self->filename);
return readFuture.get();
}
ACTOR Future<int> read(AsyncFileNonDurable* self, void* data, int length, int64_t offset) {
state ISimulator::ProcessInfo* currentProcess = g_simulator.getCurrentProcess();
state TaskPriority currentTaskID = g_network->getCurrentTask();
wait(g_simulator.onMachine(currentProcess));
try {
state int rep = wait(self->onRead(self, data, length, offset));
wait(g_simulator.onProcess(currentProcess, currentTaskID));
return rep;
} catch (Error& e) {
state Error err = e;
wait(g_simulator.onProcess(currentProcess, currentTaskID));
throw err;
}
}
// Delays writes a random amount of time before passing them through to the underlying file.
// If a kill interrupts the delay, then the output could be the correct write, part of the write,
// or none of the write. It may also corrupt parts of sectors which have not been written correctly
ACTOR Future<Void> write(AsyncFileNonDurable* self,
Promise<Void> writeStarted,
Future<Future<Void>> ownFuture,
void const* data,
int length,
int64_t offset) {
state ISimulator::ProcessInfo* currentProcess = g_simulator.getCurrentProcess();
state TaskPriority currentTaskID = g_network->getCurrentTask();
wait(g_simulator.onMachine(currentProcess));
state double delayDuration = deterministicRandom()->random01() * self->maxWriteDelay;
state Standalone<StringRef> dataCopy(StringRef((uint8_t*)data, length));
state Future<bool> startSyncFuture = self->startSyncPromise.getFuture();
try {
//TraceEvent("AsyncFileNonDurable_Write", self->id).detail("Delay", delayDuration).detail("Filename", self->filename).detail("WriteLength", length).detail("Offset", offset);
wait(checkKilled(self, "Write"));
Future<Void> writeEnded = wait(ownFuture);
std::vector<Future<Void>> priorModifications =
self->getModificationsAndInsert(offset, length, true, writeEnded);
self->minSizeAfterPendingModifications = std::max(self->minSizeAfterPendingModifications, offset + length);
if (BUGGIFY_WITH_PROB(0.001))
priorModifications.push_back(
delay(deterministicRandom()->random01() * FLOW_KNOBS->MAX_PRIOR_MODIFICATION_DELAY) ||
self->killed.getFuture());
else
priorModifications.push_back(waitUntilDiskReady(self->diskParameters, length) ||
self->killed.getFuture());
wait(waitForAll(priorModifications));
self->approximateSize = std::max(self->approximateSize, length + offset);
self->reponses.add(sendOnProcess(currentProcess, writeStarted, currentTaskID));
} catch (Error& e) {
self->reponses.add(sendErrorOnProcess(currentProcess, writeStarted, e, currentTaskID));
throw;
}
//TraceEvent("AsyncFileNonDurable_WriteDoneWithPreviousMods", self->id).detail("Delay", delayDuration).detail("Filename", self->filename).detail("WriteLength", length).detail("Offset", offset);
// Wait a random amount of time or until a sync/kill is issued
state bool saveDurable = true;
choose {
when(wait(delay(delayDuration))) {}
when(bool durable = wait(startSyncFuture)) { saveDurable = durable; }
}
debugFileCheck("AsyncFileNonDurableWriteAfterWait", self->filename, dataCopy.begin(), offset, length);
// In AIO mode, only page-aligned writes are supported
ASSERT(!self->aio || (offset % 4096 == 0 && length % 4096 == 0));
// Non-durable writes should introduce errors at the page level and corrupt at the sector level
// Otherwise, we can perform the entire write at once
int diskPageLength = saveDurable ? length : 4096;
int diskSectorLength = saveDurable ? length : 512;
vector<Future<Void>> writeFutures;
for (int writeOffset = 0; writeOffset < length;) {
// Number of bytes until the next diskPageLength file offset within the write or the end of the write.
int pageLength = diskPageLength;
if (!self->aio && !saveDurable) {
// If not in AIO mode, and the save is not durable, then we can't perform the entire write all at once
// and the first and last pages touched by the write could be partial.
pageLength = std::min<int64_t>((int64_t)length - writeOffset,
diskPageLength - ((offset + writeOffset) % diskPageLength));
}
// choose a random action to perform on this page write (write correctly, corrupt, or don't write)
KillMode pageKillMode = (KillMode)deterministicRandom()->randomInt(0, self->killMode + 1);
for (int pageOffset = 0; pageOffset < pageLength;) {
// Number of bytes until the next diskSectorLength file offset within the write or the end of the write.
int sectorLength = diskSectorLength;
if (!self->aio && !saveDurable) {
// If not in AIO mode, and the save is not durable, then we can't perform the entire write all at
// once and the first and last sectors touched by the write could be partial.
sectorLength =
std::min<int64_t>((int64_t)length - (writeOffset + pageOffset),
diskSectorLength - ((offset + writeOffset + pageOffset) % diskSectorLength));
}
// If saving durable, then perform the write correctly. Otherwise, perform the write correcly with a
// probability of 1/3. If corrupting the write, then this sector will be written correctly with a 1/4
// chance
if (saveDurable || pageKillMode == NO_CORRUPTION ||
(pageKillMode == FULL_CORRUPTION && deterministicRandom()->random01() < 0.25)) {
// if (!saveDurable) TraceEvent(SevInfo, "AsyncFileNonDurableWrite", self->id).detail("Filename",
// self->filename).detail("Offset", offset+writeOffset+pageOffset).detail("Length", sectorLength);
writeFutures.push_back(self->file->write(
dataCopy.begin() + writeOffset + pageOffset, sectorLength, offset + writeOffset + pageOffset));
}
// If the write is not durable, then the write will either be corrupted or not written at all. If
// corrupted, there is 1/4 chance that a given sector will not be written
else if (pageKillMode == FULL_CORRUPTION && deterministicRandom()->random01() < 0.66667) {
// The incorrect part of the write can be the rightmost bytes (side = 0), the leftmost bytes (side =
// 1), or the entire write (side = 2)
int side = deterministicRandom()->randomInt(0, 3);
// There is a 1/2 chance that a bad write will have garbage written into its bad portion
// The chance is increased to 1 if the entire write is bad
bool garbage = side == 2 || deterministicRandom()->random01() < 0.5;
int64_t goodStart = 0;
int64_t goodEnd = sectorLength;
int64_t badStart = 0;
int64_t badEnd = sectorLength;
if (side == 0) {
goodEnd = deterministicRandom()->randomInt(0, sectorLength);
badStart = goodEnd;
} else if (side == 1) {
badEnd = deterministicRandom()->randomInt(0, sectorLength);
goodStart = badEnd;
} else
goodEnd = 0;
// Write randomly generated bytes, if required
if (garbage && badStart != badEnd) {
uint8_t* badData = const_cast<uint8_t*>(&dataCopy.begin()[badStart + writeOffset + pageOffset]);
for (int i = 0; i < badEnd - badStart; i += sizeof(uint32_t)) {
uint32_t val = deterministicRandom()->randomUInt32();
memcpy(&badData[i], &val, std::min(badEnd - badStart - i, (int64_t)sizeof(uint32_t)));
}
writeFutures.push_back(self->file->write(dataCopy.begin() + writeOffset + pageOffset,
sectorLength,
offset + writeOffset + pageOffset));
debugFileSet("AsyncFileNonDurableBadWrite",
self->filename,
dataCopy.begin() + writeOffset + pageOffset,
offset + writeOffset + pageOffset,
sectorLength);
} else if (goodStart != goodEnd)
writeFutures.push_back(
self->file->write(dataCopy.begin() + goodStart + writeOffset + pageOffset,
goodEnd - goodStart,
goodStart + offset + writeOffset + pageOffset));
TraceEvent("AsyncFileNonDurable_BadWrite", self->id)
.detail("Offset", offset + writeOffset + pageOffset)
.detail("Length", sectorLength)
.detail("GoodStart", goodStart)
.detail("GoodEnd", goodEnd)
.detail("HasGarbage", garbage)
.detail("Side", side)
.detail("Filename", self->filename);
TEST(true); // AsyncFileNonDurable bad write
} else {
TraceEvent("AsyncFileNonDurable_DroppedWrite", self->id)
.detail("Offset", offset + writeOffset + pageOffset)
.detail("Length", sectorLength)
.detail("Filename", self->filename);
TEST(true); // AsyncFileNonDurable dropped write
}
pageOffset += sectorLength;
}
writeOffset += pageLength;
}
wait(waitForAll(writeFutures));
//TraceEvent("AsyncFileNonDurable_WriteDone", self->id).detail("Delay", delayDuration).detail("Filename", self->filename).detail("WriteLength", length).detail("Offset", offset);
return Void();
}
// Delays truncates a random amount of time before passing them through to the underlying file.
// If a kill interrupts the delay, then the truncate may or may not be performed
ACTOR Future<Void> truncate(AsyncFileNonDurable* self,
Promise<Void> truncateStarted,
Future<Future<Void>> ownFuture,
int64_t size) {
state ISimulator::ProcessInfo* currentProcess = g_simulator.getCurrentProcess();
state TaskPriority currentTaskID = g_network->getCurrentTask();
wait(g_simulator.onMachine(currentProcess));
state double delayDuration = deterministicRandom()->random01() * self->maxWriteDelay;
state Future<bool> startSyncFuture = self->startSyncPromise.getFuture();
try {
//TraceEvent("AsyncFileNonDurable_Truncate", self->id).detail("Delay", delayDuration).detail("Filename", self->filename);
wait(checkKilled(self, "Truncate"));
state Future<Void> truncateEnded = wait(ownFuture);
// Need to know the size of the file directly before this truncate
// takes effect to see what range it modifies.
if (!self->minSizeAfterPendingModificationsIsExact) {
wait(success(self->size()));
}
ASSERT(self->minSizeAfterPendingModificationsIsExact);
int64_t beginModifiedRange = std::min(size, self->minSizeAfterPendingModifications);
self->minSizeAfterPendingModifications = size;
std::vector<Future<Void>> priorModifications =
self->getModificationsAndInsert(beginModifiedRange, /*through end of file*/ -1, true, truncateEnded);
if (BUGGIFY_WITH_PROB(0.001))
priorModifications.push_back(
delay(deterministicRandom()->random01() * FLOW_KNOBS->MAX_PRIOR_MODIFICATION_DELAY) ||
self->killed.getFuture());
else
priorModifications.push_back(waitUntilDiskReady(self->diskParameters, 0) || self->killed.getFuture());
wait(waitForAll(priorModifications));
self->approximateSize = size;
self->reponses.add(sendOnProcess(currentProcess, truncateStarted, currentTaskID));
} catch (Error& e) {
self->reponses.add(sendErrorOnProcess(currentProcess, truncateStarted, e, currentTaskID));
throw;
}
// Wait a random amount of time or until a sync/kill is issued
state bool saveDurable = true;
choose {
when(wait(delay(delayDuration))) {}
when(bool durable = wait(startSyncFuture)) { saveDurable = durable; }
}
if (g_network->check_yield(TaskPriority::DefaultYield)) {
wait(delay(0, TaskPriority::DefaultYield));
}
// If performing a durable truncate, then pass it through to the file. Otherwise, pass it through with a 1/2
// chance
if (saveDurable || self->killMode == NO_CORRUPTION || deterministicRandom()->random01() < 0.5)
wait(self->file->truncate(size));
else {
TraceEvent("AsyncFileNonDurable_DroppedTruncate", self->id).detail("Size", size);
TEST(true); // AsyncFileNonDurable dropped truncate
}
return Void();
}
// Waits for delayed modifications to the file to complete and then syncs the underlying file
// If durable is false, then some of the delayed modifications will not be applied or will be
// applied incorrectly
ACTOR Future<Void> onSync(AsyncFileNonDurable* self, bool durable) {
//TraceEvent("AsyncFileNonDurable_ImplSync", self->id).detail("Filename", self->filename).detail("Durable", durable);
ASSERT(durable || !self->killed.isSet()); // this file is kill()ed only once
if (durable) {
self->hasBeenSynced = true;
wait(waitUntilDiskReady(self->diskParameters, 0, true) || self->killed.getFuture());
}
wait(checkKilled(self, durable ? "Sync" : "Kill"));
if (!durable)
self->killed.send(Void());
// Get all outstanding modifications
std::vector<Future<Void>> outstandingModifications;
std::vector<RangeMapRange<uint64_t>> stillPendingModifications;
auto rangeItr = self->pendingModifications.ranges();
for (auto itr = rangeItr.begin(); itr != rangeItr.end(); ++itr) {
if (itr.value().isValid() && (!itr->value().isReady() || itr->value().isError())) {
outstandingModifications.push_back(itr->value());
if (!itr.value().isReady())
stillPendingModifications.push_back(itr->range());
}
}
Future<Void> allModifications = waitForAll(outstandingModifications);
// Clear out the pending modifications map of all completed modifications
self->pendingModifications.insert(RangeMapRange<uint64_t>(0, -1), Void());
for (auto itr = stillPendingModifications.begin(); itr != stillPendingModifications.end(); ++itr)
self->pendingModifications.insert(
*itr, success(allModifications)); // waitForAll cannot wait on the same future more than once, so wrap
// the future with success
// Signal all modifications to end their delay and reset the startSyncPromise
Promise<bool> startSyncPromise = self->startSyncPromise;
self->startSyncPromise = Promise<bool>();
// Writes will be durable in a kill with a 10% probability
state bool writeDurable = durable || deterministicRandom()->random01() < 0.1;
startSyncPromise.send(writeDurable);
// Wait for outstanding writes to complete
if (durable)
wait(allModifications);
else
wait(success(errorOr(allModifications)));
if (!durable) {
// Sometimes sync the file if writes were made durably. Before a file is first synced, it is stored in a
// temporary file and then renamed to the correct location once sync is called. By not calling sync, we
// simulate a failure to fsync the directory storing the file
if (self->hasBeenSynced && writeDurable && deterministicRandom()->random01() < 0.5) {
TEST(true); // AsyncFileNonDurable kill was durable and synced
wait(success(errorOr(self->file->sync())));
}
// Setting this promise could trigger the deletion of the AsyncFileNonDurable; after this none of its
// members should be used
//TraceEvent("AsyncFileNonDurable_ImplSyncEnd", self->id).detail("Filename", self->filename).detail("Durable", durable);
self->killComplete.send(Void());
}
// A killed file cannot be allowed to report that it successfully synced
else {
wait(checkKilled(self, "SyncEnd"));
wait(self->file->sync());
//TraceEvent("AsyncFileNonDurable_ImplSyncEnd", self->id).detail("Filename", self->filename).detail("Durable", durable);
}
return Void();
}
ACTOR Future<Void> sync(AsyncFileNonDurable* self, bool durable) {
state ISimulator::ProcessInfo* currentProcess = g_simulator.getCurrentProcess();
state TaskPriority currentTaskID = g_network->getCurrentTask();
wait(g_simulator.onMachine(currentProcess));
try {
wait(self->onSync(self, durable));
wait(g_simulator.onProcess(currentProcess, currentTaskID));
return Void();
} catch (Error& e) {
state Error err = e;
wait(g_simulator.onProcess(currentProcess, currentTaskID));
throw err;
}
}
// Passes along size requests to the underlying file, augmenting with any writes past the end of the file
ACTOR static Future<int64_t> onSize(AsyncFileNonDurable const* self) {
//TraceEvent("AsyncFileNonDurable_Size", self->id).detail("Filename", self->filename);
wait(checkKilled(self, "Size"));
state Future<int64_t> sizeFuture = self->file->size();
wait(success(sizeFuture) || self->killed.getFuture());
wait(checkKilled(self, "SizeEnd"));
// Include any modifications which extend past the end of the file
self->approximateSize = self->minSizeAfterPendingModifications =
std::max<int64_t>(sizeFuture.get(), self->minSizeAfterPendingModifications);
self->minSizeAfterPendingModificationsIsExact = true;
return self->approximateSize;
}
ACTOR static Future<int64_t> size(AsyncFileNonDurable const* self) {
state ISimulator::ProcessInfo* currentProcess = g_simulator.getCurrentProcess();
state TaskPriority currentTaskID = g_network->getCurrentTask();
wait(g_simulator.onMachine(currentProcess));
try {
state int64_t rep = wait(onSize(self));
wait(g_simulator.onProcess(currentProcess, currentTaskID));
return rep;
} catch (Error& e) {
state Error err = e;
wait(g_simulator.onProcess(currentProcess, currentTaskID));
throw err;
}
}
// Finishes all outstanding actors on an AsyncFileNonDurable and then deletes it
ACTOR Future<Void> deleteFile(AsyncFileNonDurable* self) {
state ISimulator::ProcessInfo* currentProcess = g_simulator.getCurrentProcess();
state TaskPriority currentTaskID = g_network->getCurrentTask();
state std::string filename = self->filename;
wait(g_simulator.onMachine(currentProcess));
try {
// Make sure all writes have gone through.
Promise<bool> startSyncPromise = self->startSyncPromise;
self->startSyncPromise = Promise<bool>();
startSyncPromise.send(true);
std::vector<Future<Void>> outstandingModifications;
for (auto itr = self->pendingModifications.ranges().begin();
itr != self->pendingModifications.ranges().end();
++itr)
if (itr->value().isValid() && !itr->value().isReady())
outstandingModifications.push_back(itr->value());
// Ignore errors here so that all modifications can finish
wait(waitForAllReady(outstandingModifications));
// Make sure we aren't in the process of killing the file
if (self->killed.isSet())
wait(self->killComplete.getFuture());
// Remove this file from the filesBeingDeleted map so that new files can be created with this filename
g_simulator.getMachineByNetworkAddress(self->openedAddress)->closingFiles.erase(self->getFilename());
g_simulator.getMachineByNetworkAddress(self->openedAddress)->deletingFiles.erase(self->getFilename());
AsyncFileNonDurable::filesBeingDeleted.erase(self->filename);
//TraceEvent("AsyncFileNonDurable_FinishDelete", self->id).detail("Filename", self->filename);
delete self;
wait(g_simulator.onProcess(currentProcess, currentTaskID));
return Void();
} catch (Error& e) {
state Error err = e;
wait(g_simulator.onProcess(currentProcess, currentTaskID));
throw err;
}
}
};
#include "flow/unactorcompiler.h"
#endif