Skip to content

Commit

Permalink
fix comments
Browse files Browse the repository at this point in the history
  • Loading branch information
Nivras committed Dec 14, 2021
1 parent 686ae90 commit aed13a1
Show file tree
Hide file tree
Showing 4 changed files with 8 additions and 9 deletions.
7 changes: 3 additions & 4 deletions src/storage/admin/AdminTask.h
Original file line number Diff line number Diff line change
Expand Up @@ -71,7 +71,6 @@ class AdminTask {
virtual int8_t getPriority() { return static_cast<int8_t>(ctx_.pri_); }

virtual void finish() {
isRunning_ = false;
finish(rc_);
}

Expand All @@ -80,7 +79,7 @@ class AdminTask {
ctx_.jobId_,
ctx_.taskId_,
apache::thrift::util::enumNameSafe(rc).c_str());
isRunning_ = false;
running_ = false;
nebula::meta::cpp2::StatsItem statsItem;
ctx_.onFinish_(rc, statsItem);
}
Expand Down Expand Up @@ -113,7 +112,7 @@ class AdminTask {
rc_.compare_exchange_strong(suc, nebula::cpp2::ErrorCode::E_USER_CANCEL);
}

virtual bool isRunning() { return isRunning_; }
virtual bool isRunning() { return running_; }

virtual bool isCanceled() { return canceled_; }

Expand All @@ -122,7 +121,7 @@ class AdminTask {
public:
std::atomic<size_t> unFinishedSubTask_;
SubTaskQueue subtasks_;
std::atomic<bool> isRunning_{false};
std::atomic<bool> running_{false};

protected:
StorageEnv* env_;
Expand Down
6 changes: 3 additions & 3 deletions src/storage/admin/AdminTaskManager.cpp
Original file line number Diff line number Diff line change
Expand Up @@ -264,7 +264,7 @@ void AdminTaskManager::schedule() {
continue;
}

task->isRunning_ = true;
task->running_ = true;
auto errOrSubTasks = task->genSubTasks();
if (!nebula::ok(errOrSubTasks)) {
LOG(ERROR) << folly::sformat(
Expand Down Expand Up @@ -369,7 +369,7 @@ bool AdminTaskManager::isFinished(JobID jobID, TaskID taskID) {
return iter->second->unFinishedSubTask_ == 0;
}

void AdminTaskManager::cancelJobs(GraphSpaceID spaceId) {
void AdminTaskManager::cancelTasks(GraphSpaceID spaceId) {
auto it = tasks_.begin();
while (it != tasks_.end()) {
if (it->second->getSpaceId() == spaceId) {
Expand All @@ -395,7 +395,7 @@ int32_t AdminTaskManager::runningJobCnt(GraphSpaceID spaceId) {
}

void AdminTaskManager::waitCancelJobs(GraphSpaceID spaceId) {
cancelJobs(spaceId);
cancelTasks(spaceId);
while (runningJobCnt(spaceId) != 0) {
usleep(1000 * 100);
}
Expand Down
2 changes: 1 addition & 1 deletion src/storage/admin/AdminTaskManager.h
Original file line number Diff line number Diff line change
Expand Up @@ -53,7 +53,7 @@ class AdminTaskManager : public meta::AdminTaskStatusReporter {
nebula::cpp2::ErrorCode cancelJob(JobID jobId);
nebula::cpp2::ErrorCode cancelTask(JobID jobId, TaskID taskId = -1);

void cancelJobs(GraphSpaceID spaceId);
void cancelTasks(GraphSpaceID spaceId);
int32_t runningJobCnt(GraphSpaceID spaceId);
void waitCancelJobs(GraphSpaceID spaceId) override;

Expand Down
2 changes: 1 addition & 1 deletion src/storage/admin/StatsTask.cpp
Original file line number Diff line number Diff line change
Expand Up @@ -75,7 +75,7 @@ nebula::cpp2::ErrorCode StatsTask::genSubTask(GraphSpaceID spaceId,
std::unordered_map<EdgeType, std::string> edges) {
if (UNLIKELY(canceled_)) {
LOG(ERROR) << "Stats task is canceled";
return nebula::cpp2::ErrorCode::SUCCEEDED;
return nebula::cpp2::ErrorCode::E_USER_CANCEL;
}

auto vIdLenRet = env_->schemaMan_->getSpaceVidLen(spaceId);
Expand Down

0 comments on commit aed13a1

Please sign in to comment.