76 lines
2.9 KiB
C++
76 lines
2.9 KiB
C++
/*
|
|
* TSSMappingUtil.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/SystemData.h"
|
|
#include "fdbclient/KeyBackedTypes.h"
|
|
#include "fdbserver/TSSMappingUtil.actor.h"
|
|
#include "flow/actorcompiler.h" // This must be the last #include.
|
|
|
|
ACTOR Future<Void> readTSSMappingRYW(Reference<ReadYourWritesTransaction> tr,
|
|
std::map<UID, StorageServerInterface>* tssMapping) {
|
|
KeyBackedMap<UID, UID> tssMapDB = KeyBackedMap<UID, UID>(tssMappingKeys.begin);
|
|
state std::vector<std::pair<UID, UID>> uidMapping =
|
|
wait(tssMapDB.getRange(tr, UID(), Optional<UID>(), CLIENT_KNOBS->TOO_MANY));
|
|
ASSERT(uidMapping.size() < CLIENT_KNOBS->TOO_MANY);
|
|
|
|
state std::map<UID, StorageServerInterface> mapping;
|
|
for (auto& it : uidMapping) {
|
|
state UID ssId = it.first;
|
|
Optional<Value> v = wait(tr->get(serverListKeyFor(it.second)));
|
|
(*tssMapping)[ssId] = decodeServerListValue(v.get());
|
|
}
|
|
return Void();
|
|
}
|
|
|
|
ACTOR Future<Void> readTSSMapping(Transaction* tr, std::map<UID, StorageServerInterface>* tssMapping) {
|
|
state RangeResult mappingList = wait(tr->getRange(tssMappingKeys, CLIENT_KNOBS->TOO_MANY));
|
|
ASSERT(!mappingList.more && mappingList.size() < CLIENT_KNOBS->TOO_MANY);
|
|
|
|
for (auto& it : mappingList) {
|
|
state UID ssId = Codec<UID>::unpack(Tuple::unpack(it.key.removePrefix(tssMappingKeys.begin)));
|
|
UID tssId = Codec<UID>::unpack(Tuple::unpack(it.value));
|
|
Optional<Value> v = wait(tr->get(serverListKeyFor(tssId)));
|
|
(*tssMapping)[ssId] = decodeServerListValue(v.get());
|
|
}
|
|
return Void();
|
|
}
|
|
|
|
ACTOR Future<Void> removeTSSPairsFromCluster(Database cx, std::vector<std::pair<UID, UID>> pairsToRemove) {
|
|
state Reference<ReadYourWritesTransaction> tr = makeReference<ReadYourWritesTransaction>(cx);
|
|
state KeyBackedMap<UID, UID> tssMapDB = KeyBackedMap<UID, UID>(tssMappingKeys.begin);
|
|
loop {
|
|
try {
|
|
tr->setOption(FDBTransactionOptions::PRIORITY_SYSTEM_IMMEDIATE);
|
|
tr->setOption(FDBTransactionOptions::ACCESS_SYSTEM_KEYS);
|
|
for (auto& tssPair : pairsToRemove) {
|
|
// DO NOT remove server list key - that'll break a bunch of stuff. DD will eventually call
|
|
// removeStorageServer
|
|
tr->clear(serverTagKeyFor(tssPair.second));
|
|
tssMapDB.erase(tr, tssPair.first);
|
|
}
|
|
wait(tr->commit());
|
|
break;
|
|
} catch (Error& e) {
|
|
wait(tr->onError(e));
|
|
}
|
|
}
|
|
return Void();
|
|
}
|