wbtiger
/
foundationdb
mirror of https://github.com/apple/foundationdb.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity
foundationdb/fdbserver/workloads/RandomSelector.actor.cpp

575 lines
22 KiB
C++
Raw Blame History

/*
* RandomSelector.actor.cpp
*
* This source file is part of the FoundationDB open source project
*
* Copyright 2013-2022 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.
*/
#include "fdbclient/NativeAPI.actor.h"
#include "fdbserver/TesterInterface.actor.h"
#include "fdbclient/ReadYourWrites.h"
#include "fdbserver/workloads/workloads.actor.h"
#include "flow/actorcompiler.h" // This must be the last #include.
struct RandomSelectorWorkload : TestWorkload {
static constexpr auto NAME = "RandomSelector";
int minOperationsPerTransaction, maxOperationsPerTransaction, maxKeySpace, maxOffset, minInitialAmount,
maxInitialAmount;
double testDuration;
bool fail;
std::vector<Future<Void>> clients;
PerfIntCounter transactions, retries;
RandomSelectorWorkload(WorkloadContext const& wcx)
: TestWorkload(wcx), transactions("Transactions"), retries("Retries") {
minOperationsPerTransaction = getOption(options, "minOperationsPerTransaction"_sr, 10);
maxOperationsPerTransaction = getOption(options, "minOperationsPerTransaction"_sr, 50);
maxKeySpace = getOption(options, "maxKeySpace"_sr, 20);
maxOffset = getOption(options, "maxOffset"_sr, 5);
minInitialAmount = getOption(options, "minInitialAmount"_sr, 5);
maxInitialAmount = getOption(options, "maxInitialAmount"_sr, 10);
testDuration = getOption(options, "testDuration"_sr, 10.0);
fail = false;
}
Future<Void> setup(Database const& cx) override { return randomSelectorSetup(cx->clone(), this); }
Future<Void> start(Database const& cx) override {
clients.push_back(timeout(randomSelectorClient(cx->clone(), this), testDuration, Void()));
return delay(testDuration);
}
Future<bool> check(Database const& cx) override {
clients.clear();
return !fail;
}
void getMetrics(std::vector<PerfMetric>& m) override {
m.push_back(transactions.getMetric());
m.push_back(retries.getMetric());
}
ACTOR Future<Void> randomSelectorSetup(Database cx, RandomSelectorWorkload* self) {
state Value myValue = StringRef(format("%d", deterministicRandom()->randomInt(0, 10000000)));
state Transaction tr(cx);
state std::string clientID;
clientID = format("%08d", self->clientId);
loop {
try {
for (int i = 0; i < self->maxOffset; i++) {
tr.set(StringRef(clientID + "a/" + format("%010d", i)), myValue);
tr.set(StringRef(clientID + "c/" + format("%010d", i)), myValue);
tr.set(StringRef(clientID + "e/" + format("%010d", i)), myValue);
}
wait(tr.commit());
break;
} catch (Error& e) {
wait(tr.onError(e));
tr.reset();
}
}
return Void();
}
ACTOR Future<Void> randomSelectorClient(Database cx, RandomSelectorWorkload* self) {
state int i;
state int j;
state std::string clientID;
state std::string myKeyA;
state std::string myKeyB;
state std::string myValue;
state std::string myRandomIDKey;
state bool onEqualA;
state bool onEqualB;
state int offsetA;
state int offsetB;
state int randomLimit;
state int randomByteLimit;
state Reverse reverse = Reverse::False;
state Error error;
clientID = format("%08d", self->clientId);
loop {
state Transaction tr(cx);
loop {
try {
tr.clear(KeyRangeRef(StringRef(clientID + "b/"), StringRef(clientID + "c/")));
tr.clear(KeyRangeRef(StringRef(clientID + "d/"), StringRef(clientID + "e/")));
for (i = 0;
i < deterministicRandom()->randomInt(self->minInitialAmount, self->maxInitialAmount + 1);
i++) {
myKeyA = format("%010d", deterministicRandom()->randomInt(0, self->maxKeySpace + 1));
myValue = format("%d", deterministicRandom()->randomInt(0, 10000000));
tr.set(StringRef(clientID + "b/" + myKeyA), myValue);
tr.set(StringRef(clientID + "d/" + myKeyA), myValue);
//TraceEvent("RYOWInit").detail("Key",myKeyA).detail("Value",myValue);
}
wait(tr.commit());
tr.reset();
break;
} catch (Error& e) {
wait(tr.onError(e));
}
}
state ReadYourWritesTransaction trRYOW(cx);
try {
for (i = 0; i < deterministicRandom()->randomInt(self->minOperationsPerTransaction,
self->maxOperationsPerTransaction + 1);
i++) {
j = deterministicRandom()->randomInt(0, 16);
if (j < 3) {
myKeyA = format("%010d", deterministicRandom()->randomInt(0, self->maxKeySpace + 1));
myValue = format("%d", deterministicRandom()->randomInt(0, 10000000));
//TraceEvent("RYOWset").detail("Key",myKeyA).detail("Value",myValue);
trRYOW.set(StringRef(clientID + "b/" + myKeyA), myValue);
loop {
try {
tr.set(StringRef(clientID + "d/" + myKeyA), myValue);
wait(tr.commit());
tr.reset();
break;
} catch (Error& e) {
wait(tr.onError(e));
}
}
} else if (j < 4) {
myKeyA = format("%010d", deterministicRandom()->randomInt(0, self->maxKeySpace + 1));
//TraceEvent("RYOWclear").detail("Key",myKeyA);
trRYOW.clear(StringRef(clientID + "b/" + myKeyA));
loop {
try {
tr.clear(StringRef(clientID + "d/" + myKeyA));
wait(tr.commit());
tr.reset();
break;
} catch (Error& e) {
wait(tr.onError(e));
}
}
} else if (j < 5) {
int a = deterministicRandom()->randomInt(1, self->maxKeySpace + 1);
int b = deterministicRandom()->randomInt(1, self->maxKeySpace + 1);
myKeyA = format("%010d", std::min(a, b) - 1);
myKeyB = format("%010d", std::max(a, b));
//TraceEvent("RYOWclearRange").detail("KeyA",myKeyA).detail("KeyB",myKeyB);
trRYOW.clear(
KeyRangeRef(StringRef(clientID + "b/" + myKeyA), StringRef(clientID + "b/" + myKeyB)));
loop {
try {
tr.clear(KeyRangeRef(StringRef(clientID + "d/" + myKeyA),
StringRef(clientID + "d/" + myKeyB)));
wait(tr.commit());
tr.reset();
break;
} catch (Error& e) {
wait(tr.onError(e));
}
}
} else if (j < 6) {
myKeyA = format("%010d", deterministicRandom()->randomInt(0, self->maxKeySpace + 1));
state Optional<Value> getTest1;
Optional<Value> getTest = wait(trRYOW.get(StringRef(clientID + "b/" + myKeyA)));
getTest1 = getTest;
loop {
try {
Optional<Value> getTest2 = wait(tr.get(StringRef(clientID + "d/" + myKeyA)));
if ((getTest1.present() && (!getTest2.present() || getTest1.get() != getTest2.get())) ||
(!getTest1.present() && getTest2.present())) {
TraceEvent(SevError, "RanSelTestFailure")
.detail("Reason", "The get results did not match")
.detail("KeyA", myKeyA)
.detail("RYOW", getTest1.present() ? printable(getTest1.get()) : "Not Present")
.detail("Regular",
getTest2.present() ? printable(getTest2.get()) : "Not Present");
self->fail = true;
}
tr.reset();
break;
} catch (Error& e) {
wait(tr.onError(e));
tr.reset();
}
}
} else if (j < 7) {
myKeyA = format("%010d", deterministicRandom()->randomInt(0, self->maxKeySpace + 1));
myRandomIDKey = format("%010d", deterministicRandom()->randomInt(0, 1000000000));
myValue = format("%d", deterministicRandom()->randomInt(0, 10000000));
//TraceEvent("RYOWadd").detail("Key",myKeyA).detail("Value", "\\x01");
trRYOW.atomicOp(StringRef(clientID + "b/" + myKeyA), myValue, MutationRef::AddValue);
loop {
try {
tr.set(StringRef(clientID + "z/" + myRandomIDKey), StringRef());
tr.atomicOp(StringRef(clientID + "d/" + myKeyA), myValue, MutationRef::AddValue);
wait(tr.commit());
tr.reset();
break;
} catch (Error& e) {
error = e;
wait(tr.onError(e));
if (error.code() == error_code_commit_unknown_result) {
Optional<Value> thing = wait(tr.get(StringRef(clientID + "z/" + myRandomIDKey)));
if (thing.present())
break;
}
}
}
} else if (j < 8) {
myKeyA = format("%010d", deterministicRandom()->randomInt(0, self->maxKeySpace + 1));
myRandomIDKey = format("%010d", deterministicRandom()->randomInt(0, 1000000000));
myValue = format("%d", deterministicRandom()->randomInt(0, 10000000));
//TraceEvent("RYOWappendIfFits").detail("Key",myKeyA).detail("Value", myValue);
trRYOW.atomicOp(StringRef(clientID + "b/" + myKeyA), myValue, MutationRef::AppendIfFits);
loop {
try {
tr.set(StringRef(clientID + "z/" + myRandomIDKey), StringRef());
tr.atomicOp(StringRef(clientID + "d/" + myKeyA), myValue, MutationRef::AppendIfFits);
wait(tr.commit());
tr.reset();
break;
} catch (Error& e) {
error = e;
wait(tr.onError(e));
if (error.code() == error_code_commit_unknown_result) {
Optional<Value> thing = wait(tr.get(StringRef(clientID + "z/" + myRandomIDKey)));
if (thing.present())
break;
}
}
}
} else if (j < 9) {
myKeyA = format("%010d", deterministicRandom()->randomInt(0, self->maxKeySpace + 1));
myRandomIDKey = format("%010d", deterministicRandom()->randomInt(0, 1000000000));
myValue = format("%d", deterministicRandom()->randomInt(0, 10000000));
//TraceEvent("RYOWand").detail("Key",myKeyA).detail("Value", myValue);
trRYOW.atomicOp(StringRef(clientID + "b/" + myKeyA), myValue, MutationRef::And);
loop {
try {
tr.set(StringRef(clientID + "z/" + myRandomIDKey), StringRef());
tr.atomicOp(StringRef(clientID + "d/" + myKeyA), myValue, MutationRef::And);
wait(tr.commit());
tr.reset();
break;
} catch (Error& e) {
error = e;
wait(tr.onError(e));
if (error.code() == error_code_commit_unknown_result) {
Optional<Value> thing = wait(tr.get(StringRef(clientID + "z/" + myRandomIDKey)));
if (thing.present())
break;
}
}
}
} else if (j < 10) {
myKeyA = format("%010d", deterministicRandom()->randomInt(0, self->maxKeySpace + 1));
myRandomIDKey = format("%010d", deterministicRandom()->randomInt(0, 1000000000));
myValue = format("%d", deterministicRandom()->randomInt(0, 10000000));
//TraceEvent("RYOWor").detail("Key",myKeyA).detail("Value", myValue);
trRYOW.atomicOp(StringRef(clientID + "b/" + myKeyA), myValue, MutationRef::Or);
loop {
try {
tr.set(StringRef(clientID + "z/" + myRandomIDKey), StringRef());
tr.atomicOp(StringRef(clientID + "d/" + myKeyA), myValue, MutationRef::Or);
wait(tr.commit());
tr.reset();
break;
} catch (Error& e) {
error = e;
wait(tr.onError(e));
if (error.code() == error_code_commit_unknown_result) {
Optional<Value> thing = wait(tr.get(StringRef(clientID + "z/" + myRandomIDKey)));
if (thing.present())
break;
}
}
}
} else if (j < 11) {
myKeyA = format("%010d", deterministicRandom()->randomInt(0, self->maxKeySpace + 1));
myRandomIDKey = format("%010d", deterministicRandom()->randomInt(0, 1000000000));
myValue = format("%d", deterministicRandom()->randomInt(0, 10000000));
//TraceEvent("RYOWxor").detail("Key",myKeyA).detail("Value", myValue);
trRYOW.atomicOp(StringRef(clientID + "b/" + myKeyA), myValue, MutationRef::Xor);
loop {
try {
tr.set(StringRef(clientID + "z/" + myRandomIDKey), StringRef());
tr.atomicOp(StringRef(clientID + "d/" + myKeyA), myValue, MutationRef::Xor);
wait(tr.commit());
tr.reset();
break;
} catch (Error& e) {
error = e;
wait(tr.onError(e));
if (error.code() == error_code_commit_unknown_result) {
Optional<Value> thing = wait(tr.get(StringRef(clientID + "z/" + myRandomIDKey)));
if (thing.present())
break;
}
}
}
} else if (j < 12) {
myKeyA = format("%010d", deterministicRandom()->randomInt(0, self->maxKeySpace + 1));
myRandomIDKey = format("%010d", deterministicRandom()->randomInt(0, 1000000000));
myValue = format("%d", deterministicRandom()->randomInt(0, 10000000));
//TraceEvent("RYOWmax").detail("Key",myKeyA).detail("Value", myValue);
trRYOW.atomicOp(StringRef(clientID + "b/" + myKeyA), myValue, MutationRef::Max);
loop {
try {
tr.set(StringRef(clientID + "z/" + myRandomIDKey), StringRef());
tr.atomicOp(StringRef(clientID + "d/" + myKeyA), myValue, MutationRef::Max);
wait(tr.commit());
tr.reset();
break;
} catch (Error& e) {
error = e;
wait(tr.onError(e));
if (error.code() == error_code_commit_unknown_result) {
Optional<Value> thing = wait(tr.get(StringRef(clientID + "z/" + myRandomIDKey)));
if (thing.present())
break;
}
}
}
} else if (j < 13) {
myKeyA = format("%010d", deterministicRandom()->randomInt(0, self->maxKeySpace + 1));
myRandomIDKey = format("%010d", deterministicRandom()->randomInt(0, 1000000000));
myValue = format("%d", deterministicRandom()->randomInt(0, 10000000));
//TraceEvent("RYOWmin").detail("Key",myKeyA).detail("Value", myValue);
trRYOW.atomicOp(StringRef(clientID + "b/" + myKeyA), myValue, MutationRef::Min);
loop {
try {
tr.set(StringRef(clientID + "z/" + myRandomIDKey), StringRef());
tr.atomicOp(StringRef(clientID + "d/" + myKeyA), myValue, MutationRef::Min);
wait(tr.commit());
tr.reset();
break;
} catch (Error& e) {
error = e;
wait(tr.onError(e));
if (error.code() == error_code_commit_unknown_result) {
Optional<Value> thing = wait(tr.get(StringRef(clientID + "z/" + myRandomIDKey)));
if (thing.present())
break;
}
}
}
} else if (j < 14) {
myKeyA = format("%010d", deterministicRandom()->randomInt(0, self->maxKeySpace + 1));
myRandomIDKey = format("%010d", deterministicRandom()->randomInt(0, 1000000000));
myValue = format("%d", deterministicRandom()->randomInt(0, 10000000));
//TraceEvent("RYOWbytemin").detail("Key",myKeyA).detail("Value", myValue);
trRYOW.atomicOp(StringRef(clientID + "b/" + myKeyA), myValue, MutationRef::ByteMin);
loop {
try {
tr.set(StringRef(clientID + "z/" + myRandomIDKey), StringRef());
tr.atomicOp(StringRef(clientID + "d/" + myKeyA), myValue, MutationRef::ByteMin);
wait(tr.commit());
tr.reset();
break;
} catch (Error& e) {
error = e;
wait(tr.onError(e));
if (error.code() == error_code_commit_unknown_result) {
Optional<Value> thing = wait(tr.get(StringRef(clientID + "z/" + myRandomIDKey)));
if (thing.present())
break;
}
}
}
} else if (j < 15) {
myKeyA = format("%010d", deterministicRandom()->randomInt(0, self->maxKeySpace + 1));
myRandomIDKey = format("%010d", deterministicRandom()->randomInt(0, 1000000000));
myValue = format("%d", deterministicRandom()->randomInt(0, 10000000));
//TraceEvent("RYOWbytemax").detail("Key",myKeyA).detail("Value", myValue);
trRYOW.atomicOp(StringRef(clientID + "b/" + myKeyA), myValue, MutationRef::ByteMax);
loop {
try {
tr.set(StringRef(clientID + "z/" + myRandomIDKey), StringRef());
tr.atomicOp(StringRef(clientID + "d/" + myKeyA), myValue, MutationRef::ByteMax);
wait(tr.commit());
tr.reset();
break;
} catch (Error& e) {
error = e;
wait(tr.onError(e));
if (error.code() == error_code_commit_unknown_result) {
Optional<Value> thing = wait(tr.get(StringRef(clientID + "z/" + myRandomIDKey)));
if (thing.present())
break;
}
}
}
} else {
int a = deterministicRandom()->randomInt(1, self->maxKeySpace + 1);
int b = deterministicRandom()->randomInt(1, self->maxKeySpace + 1);
myKeyA = format("%010d", std::min(a, b) - 1);
myKeyB = format("%010d", std::max(a, b));
onEqualA = deterministicRandom()->randomInt(0, 2) != 0;
onEqualB = deterministicRandom()->randomInt(0, 2) != 0;
offsetA = deterministicRandom()->randomInt(-1 * self->maxOffset / 2, self->maxOffset / 2);
offsetB = deterministicRandom()->randomInt(-1 * self->maxOffset / 2, self->maxOffset / 2);
randomLimit = deterministicRandom()->randomInt(0, 2 * self->maxOffset + self->maxKeySpace);
randomByteLimit =
deterministicRandom()->randomInt(0, (self->maxOffset + self->maxKeySpace) * 512);
reverse.set(deterministicRandom()->coinflip());
//TraceEvent("RYOWgetRange").detail("KeyA", myKeyA).detail("KeyB", myKeyB).detail("OnEqualA",onEqualA).detail("OnEqualB",onEqualB).detail("OffsetA",offsetA).detail("OffsetB",offsetB).detail("RandomLimit",randomLimit).detail("RandomByteLimit", randomByteLimit).detail("Reverse", reverse);
state RangeResult getRangeTest1;
RangeResult getRangeTest =
wait(trRYOW.getRange(KeySelectorRef(StringRef(clientID + "b/" + myKeyA), onEqualA, offsetA),
KeySelectorRef(StringRef(clientID + "b/" + myKeyB), onEqualB, offsetB),
randomLimit,
Snapshot::False,
reverse));
getRangeTest1 = getRangeTest;
loop {
try {
RangeResult getRangeTest2 = wait(
tr.getRange(KeySelectorRef(StringRef(clientID + "d/" + myKeyA), onEqualA, offsetA),
KeySelectorRef(StringRef(clientID + "d/" + myKeyB), onEqualB, offsetB),
randomLimit,
Snapshot::False,
reverse));
bool fail = false;
if (getRangeTest1.size() != getRangeTest2.size()) {
TraceEvent(SevError, "RanSelTestFailure")
.detail("Reason", "The getRange results did not match sizes")
.detail("Size1", getRangeTest1.size())
.detail("Size2", getRangeTest2.size())
.detail("Limit", randomLimit)
.detail("ByteLimit", randomByteLimit)
.detail("Bytes1", getRangeTest1.expectedSize())
.detail("Bytes2", getRangeTest2.expectedSize())
.detail("Reverse", reverse);
fail = true;
self->fail = true;
}
for (int k = 0; k < std::min(getRangeTest1.size(), getRangeTest2.size()); k++) {
if (getRangeTest1[k].value != getRangeTest2[k].value) {
std::string keyA = printable(getRangeTest1[k].key);
std::string valueA = printable(getRangeTest1[k].value);
std::string keyB = printable(getRangeTest2[k].key);
std::string valueB = printable(getRangeTest2[k].value);
TraceEvent(SevError, "RanSelTestFailure")
.detail("Reason", "The getRange results did not match contents")
.detail("KeyA", keyA)
.detail("ValueA", valueA)
.detail("KeyB", keyB)
.detail("ValueB", valueB)
.detail("Reverse", reverse);
fail = true;
self->fail = true;
}
}
if (fail) {
std::string outStr1 = "";
for (int k = 0; k < getRangeTest1.size(); k++) {
outStr1 = outStr1 + printable(getRangeTest1[k].key) + " " +
format("%d", getRangeTest1[k].value.size()) + " ";
}
std::string outStr2 = "";
for (int k = 0; k < getRangeTest2.size(); k++) {
outStr2 = outStr2 + printable(getRangeTest2[k].key) + " " +
format("%d", getRangeTest2[k].value.size()) + " ";
}
TraceEvent("RanSelTestLog").detail("RYOW", outStr1).detail("Normal", outStr2);
}
tr.reset();
break;
} catch (Error& e) {
wait(tr.onError(e));
}
}
}
}
wait(trRYOW.commit());
++self->transactions;
state Transaction finalTransaction(cx);
loop {
try {
state RangeResult finalTest1 = wait(finalTransaction.getRange(
KeyRangeRef(StringRef(clientID + "b/"), StringRef(clientID + "c/")), self->maxKeySpace));
RangeResult finalTest2 = wait(finalTransaction.getRange(
KeyRangeRef(StringRef(clientID + "d/"), StringRef(clientID + "e/")), self->maxKeySpace));
if (finalTest1.size() != finalTest2.size()) {
TraceEvent(SevError, "RanSelTestFailure")
.detail("Reason", "The final results did not match sizes");
self->fail = true;
}
for (int k = 0; k < finalTest1.size(); k++)
if (finalTest1[k].value != finalTest2[k].value) {
TraceEvent(SevError, "RanSelTestFailure")
.detail("Reason", "The final results did not match contents")
.detail("KeyA", printable(finalTest1[k].key))
.detail("ValueA", printable(finalTest1[k].value))
.detail("KeyB", printable(finalTest2[k].key))
.detail("ValueB", printable(finalTest2[k].value))
.detail("Reverse", reverse);
self->fail = true;
}
break;
} catch (Error& e) {
wait(finalTransaction.onError(e));
}
}
} catch (Error& e) {
wait(trRYOW.onError(e));
++self->retries;
}
}
}
};
WorkloadFactory<RandomSelectorWorkload> RandomSelectorWorkloadFactory;
Reference in New Issue View Git Blame Copy Permalink