程式師世界 >> 編程語言 >> JAVA編程 >> JAVA綜合教程 >> 上傳單個文件到多台機器工具

上傳單個文件到多台機器工具

編輯：JAVA綜合教程

上傳單個文件到多台機器工具

使用示例：
./mooon_upload -h=192.168.10.11,192.168.10.12 -p=6000 -u=root -P='root123' -s=./abc -d=/tmp/
表示將本地的文件./abc上傳到兩台機器192.168.10.11和192.168.10.12的/tmp/目錄

#include "mooon/net/libssh2.h"
#include "mooon/sys/stop_watch.h"
#include "mooon/utils/args_parser.h"
#include "mooon/utils/print_color.h"
#include "mooon/utils/string_utils.h"
#include "mooon/utils/tokener.h"
#include <fstream>
#include <iostream>
// 逗號分隔的遠程主機列表
STRING_ARG_DEFINE(h, "", "remote hosts");
// 遠程主機的sshd端口號
INTEGER_ARG_DEFINE(uint16_t, p, 22, 10, 65535, "remote hosts port");
// 用戶名
STRING_ARG_DEFINE(u, "root", "remote host user");
// 密碼
STRING_ARG_DEFINE(P, "", "remote host password");
// 被上傳的文件路徑
STRING_ARG_DEFINE(s, "", "the source file uploaded");
// 文件上傳後存放的目錄路徑
STRING_ARG_DEFINE(d, "", "the directory to store");
// 連接超時，單位為秒
INTEGER_ARG_DEFINE(uint16_t, t, 10, 1, 65535, "timeout seconds to remote host");
// 結果信息
struct ResultInfo
{
bool success; // 為true表示執行成功
std::string ip; // 遠程host的IP地址
uint32_t seconds; // 運行花費的時長，精確到秒
ResultInfo()
: success(false), seconds(0)
{
}
std::string str() const
{
std::string tag = success? "SUCCESS": "FAILURE";
return mooon::utils::CStringUtils::format_string("[%s %s]: %u seconds", ip.c_str(), tag.c_str(), seconds);
}
};
inline std::ostream& operator <<(std::ostream& out, const struct ResultInfo& result)
{
std::string tag = result.success? "SUCCESS": "FAILURE";
out << "["PRINT_COLOR_YELLOW << result.ip << PRINT_COLOR_NONE" " << tag << "] " << result.seconds << " seconds";
return out;
}
int main(int argc, char* argv[])
{
// 解析命令行參數
std::string errmsg;
if (!mooon::utils::parse_arguments(argc, argv, &errmsg))
{
fprintf(stderr, "parameter error: %s\n", errmsg.c_str());
exit(1);
}
uint16_t port = mooon::argument::p->value();
std::string source = mooon::argument::s->value();
std::string directory = mooon::argument::d->value();
std::string hosts = mooon::argument::h->value();
std::string user = mooon::argument::u->value();
std::string password = mooon::argument::P->value();
mooon::utils::CStringUtils::trim(source);
mooon::utils::CStringUtils::trim(directory);
mooon::utils::CStringUtils::trim(hosts);
mooon::utils::CStringUtils::trim(user);
mooon::utils::CStringUtils::trim(password);
// 檢查參數（-s）
if (source.empty())
{
fprintf(stderr, "parameter[-s]'s value not set\n");
exit(1);
}
// 檢查參數（-d）
if (directory.empty())
{
fprintf(stderr, "parameter[-d]'s value not set\n");
exit(1);
}
// 檢查參數（-h）
if (hosts.empty())
{
// 嘗試從環境變量取值
const char* hosts_ = getenv("HOSTS");
if (NULL == hosts_)
{
fprintf(stderr, "parameter[-h]'s value not set\n");
exit(1);
}
hosts= hosts_;
mooon::utils::CStringUtils::trim(hosts);
if (hosts.empty())
{
fprintf(stderr, "parameter[-h]'s value not set\n");
exit(1);
}
}
// 檢查參數（-u）
if (user.empty())
{
fprintf(stderr, "parameter[-u]'s value not set\n");
exit(1);
}
// 檢查參數（-P）
if (password.empty())
{
fprintf(stderr, "parameter[-P]'s value not set\n");
exit(1);
}
std::vector<std::string> hosts_ip;
const std::string& remote_hosts_ip = hosts;
int num_remote_hosts_ip = mooon::utils::CTokener::split(&hosts_ip, remote_hosts_ip, ",", true);
if (0 == num_remote_hosts_ip)
{
fprintf(stderr, "parameter[-h] error\n");
exit(1);
}
std::string remote_filepath = directory + std::string("/") + mooon::utils::CStringUtils::extract_filename(source);
std::vector<struct ResultInfo> results(num_remote_hosts_ip);
for (int i=0; i<num_remote_hosts_ip; ++i)
{
bool color = true;
const std::string& remote_host_ip = hosts_ip[i];
results[i].ip = remote_host_ip;
results[i].success = false;
fprintf(stdout, "["PRINT_COLOR_YELLOW"%s"PRINT_COLOR_NONE"]\n", remote_host_ip.c_str());
fprintf(stdout, PRINT_COLOR_GREEN);
mooon::sys::CStopWatch stop_watch;
try
{
int file_size = 0;
mooon::net::CLibssh2 libssh2(remote_host_ip, port, user, password, mooon::argument::t->value());
libssh2.upload(source, remote_filepath, &file_size);
fprintf(stdout, "["PRINT_COLOR_YELLOW"%s"PRINT_COLOR_NONE"] SUCCESS: %d bytes\n", remote_host_ip.c_str(), file_size);
results[i].success = true;
}
catch (mooon::sys::CSyscallException& ex)
{
if (color)
fprintf(stdout, PRINT_COLOR_NONE); // color = true;
fprintf(stderr, "["PRINT_COLOR_RED"%s"PRINT_COLOR_NONE"] failed: %s\n", remote_host_ip.c_str(), ex.str().c_str());
}
catch (mooon::utils::CException& ex)
{
if (color)
fprintf(stdout, PRINT_COLOR_NONE); // color = true;
fprintf(stderr, "["PRINT_COLOR_RED"%s"PRINT_COLOR_NONE"] failed: %s\n", remote_host_ip.c_str(), ex.str().c_str());
}
results[i].seconds = stop_watch.get_elapsed_microseconds() / 1000000;
std::cout << std::endl;
}
// 輸出總結
std::cout << std::endl;
std::cout << "================================" << std::endl;
int num_success = 0; // 成功的個數
int num_failure = 0; // 失敗的個數
for (std::vector<struct ResultInfo>::size_type i=0; i<results.size(); ++i)
{
const struct ResultInfo& result_info = results[i];
std::cout << result_info << std::endl;
if (result_info.success)
++num_success;
else
++num_failure;
}
std::cout << "SUCCESS: " << num_success << ", FAILURE: " << num_failure << std::endl;
return 0;
}