74 lines
2.4 KiB
C++
74 lines
2.4 KiB
C++
/*
|
|
* RestoreWorker.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
|
|
#if defined(NO_INTELLISENSE) && !defined(FDBSERVER_RESTOREWORKER_G_H)
|
|
#define FDBSERVER_RESTOREWORKER_G_H
|
|
#include "fdbserver/RestoreWorker.actor.g.h"
|
|
#elif !defined(FDBSERVER_RESTOREWORKER_H)
|
|
#define FDBSERVER_RESTOREWORKER_H
|
|
|
|
|
|
#include "fdbclient/Tuple.h"
|
|
#include "flow/flow.h"
|
|
#include "flow/Stats.h"
|
|
#include "fdbrpc/fdbrpc.h"
|
|
#include "fdbrpc/IAsyncFile.h"
|
|
#include <cstdint>
|
|
#include <cstdarg>
|
|
|
|
#include "fdbclient/RestoreWorkerInterface.actor.h"
|
|
#include "fdbserver/RestoreUtil.h"
|
|
#include "fdbserver/RestoreCommon.actor.h"
|
|
#include "fdbserver/RestoreRoleCommon.actor.h"
|
|
#include "fdbserver/RestoreLoader.actor.h"
|
|
#include "fdbserver/RestoreApplier.actor.h"
|
|
|
|
|
|
// Each restore worker (a process) is assigned for a role.
|
|
// MAYBE Later: We will support multiple restore roles on a worker
|
|
struct RestoreWorkerData : NonCopyable, public ReferenceCounted<RestoreWorkerData> {
|
|
UID workerID;
|
|
std::map<UID, RestoreWorkerInterface> workerInterfaces; // UID is worker's node id, RestoreWorkerInterface is worker's communication workerInterface
|
|
|
|
// Restore Roles
|
|
Optional<RestoreLoaderInterface> loaderInterf;
|
|
Optional<RestoreApplierInterface> applierInterf;
|
|
|
|
uint32_t inProgressFlag = 0; // To avoid race between duplicate message delivery that invokes the same actor multiple times
|
|
|
|
UID id() const { return workerID; };
|
|
|
|
RestoreWorkerData() = default;
|
|
|
|
~RestoreWorkerData() {
|
|
printf("[Exit] Worker:%s RestoreWorkerData is deleted\n", workerID.toString().c_str());
|
|
}
|
|
|
|
std::string describeNode() {
|
|
std::stringstream ss;
|
|
ss << "RestoreWorker workerID:" << workerID.toString();
|
|
return ss.str();
|
|
}
|
|
};
|
|
|
|
#include "flow/unactorcompiler.h"
|
|
#endif // FDBSERVER_RESTOREWORKER_H
|