|
/**
|
||
|
|
* 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 "NodeServer.h"
|
||
|
|
#include "servant/Communicator.h"
|
||
|
|
#include "util/tc_option.h"
|
||
|
|
#include "util/tc_file.h"
|
||
|
|
#include "util/tc_config.h"
|
||
|
|
#include <iostream>
|
||
|
|
|
||
|
|
using namespace std;
|
||
|
|
using namespace tars;
|
||
|
|
|
||
|
|
NodeServer g_app;
|
||
|
|
TC_Config* g_pconf;
|
||
|
|
|
||
|
|
static string outfill(const string& s, char c = ' ', int n = 29)
|
||
|
|
{
|
||
|
|
return (s + string(abs(n - (int)s.length()), c));
|
||
|
|
}
|
||
|
|
|
||
|
|
bool getConfig(const string &sLocator,const string &sRegistryObj,string &sNodeId,string &sLocalIp, string &sConfigFile, bool bCloseCout)
|
||
|
|
{
|
||
|
|
try
|
||
|
|
{
|
||
|
|
//string sLocalIp;
|
||
|
|
string sResult;
|
||
|
|
string sTemplate;
|
||
|
|
TC_Config tConf;
|
||
|
|
CommunicatorFactory::getInstance()->getCommunicator()->setProperty("locator",sLocator);
|
||
|
|
RegistryPrx pRegistryPrx = CommunicatorFactory::getInstance()->getCommunicator()->stringToProxy<RegistryPrx>(sRegistryObj);
|
||
|
|
if( sLocalIp.empty() && pRegistryPrx->getClientIp(sLocalIp) != 0)
|
||
|
|
{
|
||
|
|
cerr<<"cannot get localip: " <<sLocalIp << endl;
|
||
|
|
return false;
|
||
|
|
}
|
||
|
|
|
||
|
|
if(sNodeId == "" )
|
||
|
|
{
|
||
|
|
sNodeId = sLocalIp;
|
||
|
|
}
|
||
|
|
|
||
|
|
pRegistryPrx->getNodeTemplate(sNodeId,sTemplate);
|
||
|
|
if(TC_Common::trim(sTemplate) == "" )
|
||
|
|
{
|
||
|
|
cerr << "cannot get node Template nodeid:"<<sNodeId <<",Template:"<<sTemplate<< endl;
|
||
|
|
return false;
|
||
|
|
}
|
||
|
|
|
||
|
|
sTemplate = TC_Common::replace(sTemplate, "${locator}",sLocator);
|
||
|
|
sTemplate = TC_Common::replace(sTemplate, "${registryObj}",sRegistryObj);
|
||
|
|
sTemplate = TC_Common::replace(sTemplate, "${localip}",sLocalIp);
|
||
|
|
//sTemplate = TC_Common::replace(sTemplate, "${nodeid}",sNodeId);
|
||
|
|
tConf.parseString(sTemplate);
|
||
|
|
if (!bCloseCout)
|
||
|
|
{
|
||
|
|
string sCloseCoutValue = tConf.get("/tars/application/server<closecout>");
|
||
|
|
if (sCloseCoutValue == "1")
|
||
|
|
{
|
||
|
|
sTemplate = TC_Common::replace(sTemplate, "closecout=1", "closecout=0");
|
||
|
|
tConf.parseString(sTemplate);
|
||
|
|
}
|
||
|
|
else if (sCloseCoutValue.empty())
|
||
|
|
{
|
||
|
|
map<string, string> m;
|
||
|
|
m["closecout"] = "0";
|
||
|
|
tConf.insertDomainParam("/tars/application/server", m,true);
|
||
|
|
}
|
||
|
|
else
|
||
|
|
{
|
||
|
|
cerr << "failed to set closeout value" << endl;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
string sConfigPath = TC_File::extractFilePath(sConfigFile);
|
||
|
|
if(!TC_File::makeDirRecursive( sConfigPath ))
|
||
|
|
{
|
||
|
|
cerr<<"cannot create dir: " <<sConfigPath << endl;
|
||
|
|
exit( 0 );
|
||
|
|
}
|
||
|
|
string sFileTemp = sConfigPath + "/config.conf.tem";
|
||
|
|
ofstream configfile(sFileTemp.c_str());
|
||
|
|
if(!configfile.good())
|
||
|
|
{
|
||
|
|
cerr << "cannot create configuration file: "<<sConfigFile<< endl;
|
||
|
|
return false;
|
||
|
|
}
|
||
|
|
configfile <<TC_Common::replace(tConf.tostr(),"\\s"," ") << endl;
|
||
|
|
configfile.close();
|
||
|
|
|
||
|
|
string sFileBak = sConfigFile+"."+TC_Common::now2str() + ".bak";
|
||
|
|
if(TC_File::isFileExist(sConfigFile))
|
||
|
|
{
|
||
|
|
TC_File::copyFile(sConfigFile,sFileBak,true);
|
||
|
|
}
|
||
|
|
|
||
|
|
TC_File::copyFile(sFileTemp,sConfigFile,true);
|
||
|
|
TC_File::removeFile(sFileTemp,false);
|
||
|
|
|
||
|
|
return true;
|
||
|
|
}
|
||
|
|
catch(exception &e)
|
||
|
|
{
|
||
|
|
cerr<< "load config erro:" <<e.what()<< endl;
|
||
|
|
}
|
||
|
|
return false;
|
||
|
|
}
|
||
|
|
|
||
|
|
void monitorNode(const string &configFile)
|
||
|
|
{
|
||
|
|
TC_Config conf;
|
||
|
|
conf.parseFile(configFile);
|
||
|
|
|
||
|
|
string nodeObj = "AdminObj@"+conf["/tars/application/server<local>"];
|
||
|
|
|
||
|
|
ServantPrx prx = CommunicatorFactory::getInstance()->getCommunicator()->stringToProxy<ServantPrx>(nodeObj);
|
||
|
|
prx->tars_ping();
|
||
|
|
}
|
||
|
|
|
||
|
|
void parseConfig(int argc, char *argv[])
|
||
|
|
{
|
||
|
|
|
||
|
|
TC_Option tOp;//consider
|
||
|
|
tOp.decode(argc, argv);
|
||
|
|
|
||
|
|
if (tOp.hasParam("nodeversion"))
|
||
|
|
{
|
||
|
|
cout << "Node:" << TARS_VERSION << "_" << NODE_VERSION << endl;
|
||
|
|
exit(0);
|
||
|
|
}
|
||
|
|
|
||
|
|
if (tOp.hasParam("monitor"))
|
||
|
|
{
|
||
|
|
try
|
||
|
|
{
|
||
|
|
string configFile = tOp.getValue("config");
|
||
|
|
|
||
|
|
monitorNode(configFile);
|
||
|
|
}
|
||
|
|
catch (exception &ex)
|
||
|
|
{
|
||
|
|
cout << "failed:" << ex.what() << endl;
|
||
|
|
exit(-1);
|
||
|
|
}
|
||
|
|
exit(0);
|
||
|
|
return;
|
||
|
|
}
|
||
|
|
|
||
|
|
string sLocator = tOp.getValue("locator");
|
||
|
|
string sNodeId = tOp.getValue("nodeid");
|
||
|
|
string sConfigFile = tOp.getValue("config");
|
||
|
|
string sRegistryObj = tOp.getValue("registryObj");
|
||
|
|
string sLocalIp = tOp.getValue("localip");
|
||
|
|
|
||
|
|
if(sConfigFile == "")
|
||
|
|
{
|
||
|
|
cerr << endl;
|
||
|
|
cerr <<"start server with locator config, for example: "<<endl;
|
||
|
|
cerr << argv[0] << " --locator=tars.tarsregistry.QueryObj@tcp -h 172.25.38.67 -p 17890" " --config=config.conf [--nodeid = 172.25.38.67 --registryObj=tars.tarsregistry.RegistryObj]" << endl;
|
||
|
|
cerr << argv[0] << " --config=config.conf" << endl;
|
||
|
|
exit(0);
|
||
|
|
}
|
||
|
|
|
||
|
|
if(!TC_File::isAbsolute(sConfigFile))
|
||
|
|
{
|
||
|
|
char sCwd[PATH_MAX];
|
||
|
|
if ( getcwd( sCwd, PATH_MAX ) == NULL )
|
||
|
|
{
|
||
|
|
TLOGERROR("cannot get the current directory:\n" << endl);
|
||
|
|
exit( 0 );
|
||
|
|
}
|
||
|
|
sConfigFile = string(sCwd) +"/"+ sConfigFile;
|
||
|
|
}
|
||
|
|
sConfigFile = TC_File::simplifyDirectory(sConfigFile);
|
||
|
|
if(sLocator != "")
|
||
|
|
{
|
||
|
|
if(sRegistryObj == "")
|
||
|
|
{
|
||
|
|
sRegistryObj = "tars.tarsregistry.RegistryObj";
|
||
|
|
}
|
||
|
|
bool bCloseOut=true;
|
||
|
|
if(tOp.hasParam("closecout"))
|
||
|
|
{
|
||
|
|
if(TC_Common::lower(tOp.getValue("closecout"))=="n")
|
||
|
|
{
|
||
|
|
bCloseOut = false;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
bool bRet = getConfig(sLocator,sRegistryObj,sNodeId, sLocalIp, sConfigFile, bCloseOut);
|
||
|
|
if(bRet == false)
|
||
|
|
{
|
||
|
|
cerr<<"reload config erro. use old config";
|
||
|
|
}
|
||
|
|
}
|
||
|
|
cout <<endl;
|
||
|
|
map<string,string> mOp = tOp.getMulti();
|
||
|
|
for(map<string,string>::const_iterator it = mOp.begin(); it != mOp.end(); ++it)
|
||
|
|
{
|
||
|
|
cout << outfill( it->first)<< it->second << endl;
|
||
|
|
}
|
||
|
|
|
||
|
|
}
|
||
|
|
|
||
|
|
int main( int argc, char* argv[] )
|
||
|
|
{
|
||
|
|
try
|
||
|
|
{
|
||
|
|
TC_Common::daemon();
|
||
|
|
|
||
|
|
parseConfig(argc,argv);
|
||
|
|
|
||
|
|
g_pconf = &g_app.getConfig();
|
||
|
|
|
||
|
|
g_app.main( argc, argv );
|
||
|
|
g_app.waitForShutdown();
|
||
|
|
}
|
||
|
|
catch ( exception& ex )
|
||
|
|
{
|
||
|
|
cout<< ex.what() << endl;
|
||
|
|
}
|
||
|
|
catch ( ... )
|
||
|
|
{
|
||
|
|
cout<< "main unknow exception cathed" << endl;
|
||
|
|
}
|
||
|
|
|
||
|
|
return 0;
|
||
|
|
}
|