Files
doris/be/src/agent/utils.h
Adonis Ling e412dd12e8 [chore](build) Use include-what-you-use to optimize includes (PART II) (#18761)
Currently, there are some useless includes in the codebase. We can use a tool named include-what-you-use to optimize these includes. By using a strict include-what-you-use policy, we can get lots of benefits from it.
2023-04-19 23:11:48 +08:00

90 lines
2.7 KiB
C++

// Licensed to the Apache Software Foundation (ASF) under one
// or more contributor license agreements. See the NOTICE file
// distributed with this work for additional information
// regarding copyright ownership. The ASF licenses this file
// to you 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
#include <butil/macros.h>
#include <map>
#include <string>
#include "common/status.h"
namespace doris {
class TConfirmUnusedRemoteFilesRequest;
class TConfirmUnusedRemoteFilesResult;
class TFinishTaskRequest;
class TMasterInfo;
class TMasterResult;
class TReportRequest;
class MasterServerClient {
public:
static MasterServerClient* create(const TMasterInfo& master_info);
static MasterServerClient* instance();
~MasterServerClient() = default;
// Report finished task to the master server
//
// Input parameters:
// * request: The information of finished task
//
// Output parameters:
// * result: The result of report task
Status finish_task(const TFinishTaskRequest& request, TMasterResult* result);
// Report tasks/olap tablet/disk state to the master server
//
// Input parameters:
// * request: The information to report
//
// Output parameters:
// * result: The result of report task
Status report(const TReportRequest& request, TMasterResult* result);
Status confirm_unused_remote_files(const TConfirmUnusedRemoteFilesRequest& request,
TConfirmUnusedRemoteFilesResult* result);
private:
MasterServerClient(const TMasterInfo& master_info);
DISALLOW_COPY_AND_ASSIGN(MasterServerClient);
// Not owner. Reference to the ExecEnv::_master_info
const TMasterInfo& _master_info;
};
class AgentUtils {
public:
AgentUtils() = default;
virtual ~AgentUtils() = default;
// Execute shell cmd
virtual bool exec_cmd(const std::string& command, std::string* errmsg,
bool redirect_stderr = true);
// Write a map to file by json format
virtual bool write_json_to_file(const std::map<std::string, std::string>& info,
const std::string& path);
private:
DISALLOW_COPY_AND_ASSIGN(AgentUtils);
}; // class AgentUtils
} // namespace doris