55 lines
1.6 KiB
C++
55 lines
1.6 KiB
C++
/**
|
||
* @file WorkflowTool.h
|
||
* @brief 工作流辅助工具(支持脚本生成jar submit)
|
||
* @author huojidong
|
||
*/
|
||
#ifndef PAI_HOME_WORKFLOWENGINE_WORKFLOWTOOLS_H
|
||
#define PAI_HOME_WORKFLOWENGINE_WORKFLOWTOOLS_H
|
||
|
||
#include <string>
|
||
|
||
namespace pai {
|
||
namespace workflow {
|
||
|
||
class CWorkflowTool
|
||
{
|
||
public:
|
||
CWorkflowTool();
|
||
~CWorkflowTool();
|
||
public:
|
||
/**
|
||
* @brief 生成一个jar包
|
||
* @param jsonfilepath 传入的json路径（绝对路径）
|
||
* @param jarfilename 输入的jar包文件名(可以不填)
|
||
*/
|
||
bool GenerateJarFile(const std::string &jsonfilepath,const std::string&jarfilename,const std::string &destjarpath);
|
||
/**
|
||
* @brief 得到要存放Jar包文件路径
|
||
*/
|
||
std::string GetDestJarPath()const {return destJarPath;}
|
||
/**
|
||
* @brief 设置要存放Jar包的路径
|
||
*/
|
||
void SetDestJarPath(std::string inputJarPath){inputJarPath.append("/");destJarPath=inputJarPath;}
|
||
private:
|
||
/**
|
||
*@brief 将json文件拷贝到临时目录的之下（吻合compiler下的GeneratorJobFile函数）
|
||
*@param strFilePath json路径
|
||
*@param strCompileTmpDir 临时文件夹路径
|
||
*/
|
||
bool SaveJsonToLocalCompileTmp(const std::string& strFilePath,const std::string& strCompileTmpDir);
|
||
/**
|
||
*@brief 将生成的jar文件拷贝到PAI_HOME下的res目录下（脚本执行的当前目录）
|
||
*@param strCompileTmpDir 临时目录
|
||
*@param jarfilename 目标jar包文件名
|
||
*@param jardestpath jar包存储路径
|
||
*/
|
||
bool CopyJarFileToDestFilePath(const std::string&strCompileTmpDir,const std::string& jarfilename);
|
||
private:
|
||
std::string destJarPath;//目标路径地址
|
||
};
|
||
|
||
}
|
||
}
|
||
#endif
|