71 lines
2.4 KiB
C++
71 lines
2.4 KiB
C++
#pragma once
|
|
#if defined(NO_INTELLISENSE) && !defined(FDBSERVER_EXEC_HELPER_ACTOR_G_H)
|
|
#define FDBSERVER_EXEC_HELPER_ACTOR_G_H
|
|
#include "fdbserver/FDBExecHelper.actor.g.h"
|
|
#elif !defined(FDBSERVER_EXEC_HELPER_ACTOR_H)
|
|
#define FDBSERVER_EXEC_HELPER_ACTOR_H
|
|
|
|
#include <string>
|
|
#include <vector>
|
|
#include <map>
|
|
#include "flow/Arena.h"
|
|
#include "flow/flow.h"
|
|
#include "flow/actorcompiler.h"
|
|
|
|
// execute/snapshot command takes two arguments: <param1> <param2>
|
|
// param1 - represents the command type/name
|
|
// param2 - takes a binary path followed by a set of arguments in the following
|
|
// format <binary-path>:<key1=val1>,<key2=val2>...
|
|
// this class will abstract the format and give functions to get various pieces
|
|
// of information
|
|
class ExecCmdValueString {
|
|
public: // ctor & dtor
|
|
ExecCmdValueString() {}
|
|
explicit ExecCmdValueString(StringRef cmdValueString);
|
|
|
|
public: // interfaces
|
|
StringRef getBinaryPath();
|
|
VectorRef<StringRef> getBinaryArgs();
|
|
StringRef getBinaryArgValue(StringRef key);
|
|
void setCmdValueString(StringRef cmdValueString);
|
|
StringRef getCmdValueString(void);
|
|
|
|
public: // helper functions
|
|
void dbgPrint();
|
|
|
|
private: // functions
|
|
void parseCmdValue();
|
|
|
|
private: // data
|
|
Standalone<StringRef> cmdValueString;
|
|
Standalone<VectorRef<StringRef>> binaryArgs;
|
|
StringRef binaryPath;
|
|
std::map<StringRef, StringRef> keyValueMap;
|
|
};
|
|
|
|
// FIXME: move this function to a common location
|
|
// spawns a process pointed by `binPath` and the arguments provided at `paramList`,
|
|
// if the process spawned takes more than `maxWaitTime` then it will be killed
|
|
// if isSync is set to true then the process will be synchronously executed
|
|
ACTOR Future<int> spawnProcess(std::string binPath, std::vector<std::string> paramList, double maxWaitTime, bool isSync);
|
|
|
|
// helper to run all the work related to running the exec command
|
|
ACTOR Future<int> execHelper(ExecCmdValueString* execArg, std::string folder, std::string role);
|
|
|
|
// returns true if the execUID op is in progress
|
|
bool isExecOpInProgress(UID execUID);
|
|
// adds the execUID op to the list of ops in progress
|
|
void setExecOpInProgress(UID execUID);
|
|
// clears the execUID op from the list of ops in progress
|
|
void clearExecOpInProgress(UID execUID);
|
|
|
|
|
|
// registers a non-stopped TLog instance
|
|
void registerTLog(UID uid);
|
|
// unregisters a stopped TLog instance
|
|
void unregisterTLog(UID uid);
|
|
// checks if there is any non-stopped TLog instance
|
|
bool isTLogInSameNode();
|
|
|
|
#endif
|