|
/**
|
||
|
|
* Tencent is pleased to support the open source community by making Tars available.
|
||
|
|
*
|
||
|
|
* Copyright (C) 2016THL A29 Limited, a Tencent company. All rights reserved.
|
||
|
|
*
|
||
|
|
* Licensed under the BSD 3-Clause License (the "License"); you may not use this file except
|
||
|
|
* in compliance with the License. You may obtain a copy of the License at
|
||
|
|
*
|
||
|
|
* https://opensource.org/licenses/BSD-3-Clause
|
||
|
|
*
|
||
|
|
* 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.
|
||
|
|
*/
|
||
|
|
|
||
|
|
#include "tars_patch.h"
|
||
|
|
#include "util/tc_option.h"
|
||
|
|
#include "servant/Application.h"
|
||
|
|
|
||
|
|
#include <iostream>
|
||
|
|
|
||
|
|
using namespace tars;
|
||
|
|
|
||
|
|
void usage(int argc, char *argv[])
|
||
|
|
{
|
||
|
|
cout << "Usage:" << argv[0] << " --server=[server] --remote=[server-dir] --local=[local] [--remove]" << endl;
|
||
|
|
cout << "server: PatchObj@tcp -h ip -p port -t timeout" << endl;
|
||
|
|
cout << "remote: directory or file sync from server" << endl;
|
||
|
|
cout << "local: directory sync to local" << endl;
|
||
|
|
cout << "remove: remove local dir data" << endl;
|
||
|
|
exit(0);
|
||
|
|
}
|
||
|
|
|
||
|
|
class PatchClientInterface : public TarsPatchNotifyInterface
|
||
|
|
{
|
||
|
|
public:
|
||
|
|
virtual void onDownload(const FileInfo &fi)
|
||
|
|
{
|
||
|
|
cout << "------------------------------------" << endl;
|
||
|
|
cout << "downloading [" << fi.path << "] [" << fi.size*1.0/1024 << "K]" << endl;
|
||
|
|
}
|
||
|
|
virtual void onDownloading(const FileInfo &fi, size_t pos, const string &dest)
|
||
|
|
{
|
||
|
|
cout << "downloading [" << fi.path << " ->] [" << dest << "] [" << pos*100/fi.size << "/100%]" << endl;
|
||
|
|
}
|
||
|
|
virtual void onDownloadOK(const FileInfo &fi, const string &dest)
|
||
|
|
{
|
||
|
|
cout << "download ok [" << fi.path << "] -> [" << dest << "]" << endl;
|
||
|
|
}
|
||
|
|
virtual void onListFile()
|
||
|
|
{
|
||
|
|
cout << "listing files" << endl;
|
||
|
|
}
|
||
|
|
virtual void onListFileOK(const vector<FileInfo> &vf)
|
||
|
|
{
|
||
|
|
cout << "list ok total [" << vf.size() << "] files" << endl;
|
||
|
|
}
|
||
|
|
virtual void onRemoveLocalDirectory(const string &sDir)
|
||
|
|
{
|
||
|
|
cout << "------------------------------------" << endl;
|
||
|
|
cout << "remove local directory [" << sDir << "]" << endl;
|
||
|
|
}
|
||
|
|
virtual void onRemoveLocalDirectoryOK(const string &sDir)
|
||
|
|
{
|
||
|
|
cout << "remove local directory ok [" << sDir << "]"<< endl;
|
||
|
|
cout << "------------------------------------" << endl;
|
||
|
|
}
|
||
|
|
virtual void onSetExecutable(const FileInfo &fi)
|
||
|
|
{
|
||
|
|
cout << "executable [" << fi.path << "]" << endl;
|
||
|
|
}
|
||
|
|
virtual void onDownloadAllOK(const vector<FileInfo> &vf, time_t timeBegin, time_t timeEnd)
|
||
|
|
{
|
||
|
|
cout << "------------------------------------" << endl;
|
||
|
|
cout << "download all files succ." << endl;
|
||
|
|
}
|
||
|
|
|
||
|
|
virtual void onReportTime(const string & sFile, const time_t timeBegin, const time_t timeEnd)
|
||
|
|
{
|
||
|
|
cout << "------------------------------------" << endl;
|
||
|
|
cout << "download '" << sFile << "', begin:" << timeBegin << ", end:" << timeEnd << endl;
|
||
|
|
}
|
||
|
|
};
|
||
|
|
|
||
|
|
typedef TC_AutoPtr<PatchClientInterface> PatchClientInterfacePtr;
|
||
|
|
|
||
|
|
int main(int argc, char *argv[])
|
||
|
|
{
|
||
|
|
|
||
|
|
if(argc < 2)
|
||
|
|
{
|
||
|
|
usage(argc, argv);
|
||
|
|
}
|
||
|
|
|
||
|
|
try
|
||
|
|
{
|
||
|
|
|
||
|
|
TC_Option option;
|
||
|
|
option.decode(argc, argv);
|
||
|
|
|
||
|
|
//远程目录的相对路径
|
||
|
|
string server = option.getValue("server");
|
||
|
|
if(server.empty())
|
||
|
|
{
|
||
|
|
cout << "server must not be empty." << endl;
|
||
|
|
return 0;
|
||
|
|
}
|
||
|
|
|
||
|
|
CommunicatorFactory::getInstance()->getCommunicator()->setProperty("sync-invoke-timeout", "60000");
|
||
|
|
PatchPrx pPtr;
|
||
|
|
CommunicatorFactory::getInstance()->getCommunicator()->stringToProxy(server, pPtr);
|
||
|
|
|
||
|
|
//远程目录的相对路径
|
||
|
|
string remote_dir = option.getValue("remote");
|
||
|
|
if(remote_dir.empty())
|
||
|
|
{
|
||
|
|
cout << "remote_dir must not be empty." << endl;
|
||
|
|
return 0;
|
||
|
|
}
|
||
|
|
|
||
|
|
//本地同步路径
|
||
|
|
string local_dir = option.getValue("local");
|
||
|
|
if(local_dir.empty())
|
||
|
|
{
|
||
|
|
cout << "local_dir must not be empty." << endl;
|
||
|
|
return 0;
|
||
|
|
}
|
||
|
|
|
||
|
|
//是否先清空本地目录
|
||
|
|
bool bRemove = option.hasParam("remove");
|
||
|
|
|
||
|
|
cout << "connecting to patch server" << endl;
|
||
|
|
TarsPatch op;
|
||
|
|
op.init(pPtr, remote_dir, local_dir);
|
||
|
|
cout << "connect ok to patch server" << endl;
|
||
|
|
op.setRemove(bRemove);
|
||
|
|
|
||
|
|
PatchClientInterfacePtr p = new PatchClientInterface();
|
||
|
|
op.download(p);
|
||
|
|
}
|
||
|
|
catch(exception &ex)
|
||
|
|
{
|
||
|
|
cout << ex.what() << endl;
|
||
|
|
}
|
||
|
|
|
||
|
|
return 0;
|
||
|
|
}
|
||
|
|
|
||
|
|
|