StarMirror/tars2node
1
0
Fork 0
mirror of https://github.com/tars-node/tars2node.git synced 2024-03-22 13:10:56 +08:00
Code Issues Projects Releases Wiki Activity

version: 20180713

Browse Source
This commit is contained in:
Super Zheng 2018-07-27 12:33:14 +08:00
parent 3c4c8533da
commit deaeeb550b
35 changed files with 14819 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

52
CMakeLists.txt Normal file
View File

@ -0,0 +1,52 @@
cmake_minimum_required(VERSION 2.6)
project(tars2node)
set(CMAKE_VERBOSE_MAKEFILE off)
# version
set(TARS_VERSION "1.1.0")
set(PARSER_VERSION "${TARS_VERSION}")
set(GENERATOR_VERSION "20180713")
# namespace
set(IDL_NAMESPACE Tars)
string(TOLOWER ${IDL_NAMESPACE} TC_NAMESPACE)
set(GRAMMAR_NAME Grammar)
# idl
set(RPC_MODULE_PATH "@tars/rpc")
set(STREAM_MODULE_PATH "@tars/stream")
set(IDL_TYPE "Tars")
set(PROTOCOL_NAME "Tup")
# flag
set(CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS} -std=c++11 -g -O2 -Wall -Wno-sign-compare -Wno-unused-result")
set(CMAKE_C_FLAGS "${CMAKE_C_FLAGS} -g -O2 -Wall")
# define
add_definitions(-DTARS_VERSION="${TARS_VERSION}")
add_definitions(-DPARSER_VERSION="${PARSER_VERSION}")
add_definitions(-DGENERATOR_VERSION="${GENERATOR_VERSION}")
add_definitions(-DTC_NAMESPACE=${TC_NAMESPACE})
add_definitions(-DIDL_NAMESPACE=${IDL_NAMESPACE})
add_definitions(-DGRAMMAR_NAME=${GRAMMAR_NAME})
add_definitions(-DRPC_MODULE_PATH="${RPC_MODULE_PATH}")
add_definitions(-DSTREAM_MODULE_PATH="${STREAM_MODULE_PATH}")
add_definitions(-DEXECUTE_FILENAME="${PROJECT_NAME}")
add_definitions(-DIDL_TYPE="${IDL_TYPE}")
add_definitions(-DPROTOCOL_NAME="${PROTOCOL_NAME}")
# include
include_directories(${PROJECT_SOURCE_DIR}/third_partly/dirent/include)
include_directories(${PROJECT_SOURCE_DIR}/third_partly/tarsgrammar)
include_directories(${PROJECT_SOURCE_DIR}/third_partly/tarsparse)
include_directories(${PROJECT_SOURCE_DIR}/third_partly/util/include)
include_directories(${PROJECT_SOURCE_DIR}/include)
# source
aux_source_directory(${PROJECT_SOURCE_DIR}/third_partly/tarsgrammar DIR_SRCS)
aux_source_directory(${PROJECT_SOURCE_DIR}/third_partly/tarsparse DIR_SRCS)
aux_source_directory(${PROJECT_SOURCE_DIR}/third_partly/util/src DIR_SRCS)
aux_source_directory(${PROJECT_SOURCE_DIR}/src DIR_SRCS)
add_executable(${PROJECT_NAME} ${DIR_SRCS})

40
README.md Normal file
View File

@ -0,0 +1,40 @@
# Tars2Node
`tars2node` 可以将 Tars IDL 定义文件转换为 JavaScript 语言所使用的版本，结合 [@tars/stream](https://www.npmjs.com/package/@tars/stream) 模块执行编解码操作。
如您有 Tars RPC 需求可结合 [@tars/rpc](https://www.npmjs.com/package/@tars/rpc) 模块进行调用。
## 用法
直接使用 build 目录中的 `tars2node` 预编译程序即可（Linux 平台）。
```bash
tars2node [OPTIONS] tarsfile
```
## 选项
> Options:
> --version print tars2node version
> --rpc-path=[DIRECTORY] specify the path of rpc module.
> --stream-path=[DIRECTORY[] specify the path of stream module.
> --allow-reserved-namespace allow you to use reserved word as a namespace.
> --dir=DIRECTORY generate source file to DIRECTORY.
> --relative use current path.
> --tarsBase=DIRECTORY where to search tars files.
> --r generate source all tars files.
> --r-minimal minimize the dependent members.
> --r-reserved list of names(split by ",") that should be keeped.
> --client just for client side source file.
> --server just for server side source file.
> --dts generate d.ts file.
> --use-string-represent use string represent long type.
> --string-binary-encoding get string raw bytes <BinBuffer>.
> --optimize=[0|s] support "s" to reduce code size, default is 0.
## 从源码编译
1. 安装对应平台的 build-essential
2. 安装 [CMake](https://cmake.org/)
3. 在源码 build 目录下执行 `cmake ../ && make`

BIN
build/tars2node Executable file
View File

Binary file not shown.

273
include/code_generator.h Normal file
View File

@ -0,0 +1,273 @@
/**
* 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.
*/
#ifndef _CODEGENERATOR_H
#define _CODEGENERATOR_H
#include "parse.h"
#include "util/tc_file.h"
#include "util/tc_encoder.h"
#include <set>
#ifndef TAB
#define TAB g_parse->getTab()
#endif
#ifndef INC_TAB
#define INC_TAB g_parse->incTab()
#endif
#ifndef DEL_TAB
#define DEL_TAB g_parse->delTab()
#endif
#define TO_LOWER_STRING(str) TC_Common::lower(str)
#define DEFINE_STRING(str) string(CSTR(str))
#define CSTR(str) #str
#define IDL_NAMESPACE_STR DEFINE_STRING(IDL_NAMESPACE)
#define GET_CONST_GRAMMAR_PTR_V(name, ptr) \
ptr->getConst##name##Ptr()
#define GET_CONST_GRAMMAR_PTR_BASE(name, ptr) \
GET_CONST_GRAMMAR_PTR_V(name, ptr)
#define GET_CONST_GRAMMAR_PTR(ptr) \
GET_CONST_GRAMMAR_PTR_BASE(GRAMMAR_NAME, ptr)
#define CONST_GRAMMAR_V(name, val) \
Const##name::val
#define CONST_GRAMMAR_BASE(name, val) \
CONST_GRAMMAR_V(name, val)
#define CONST_GRAMMAR(val) \
CONST_GRAMMAR_BASE(GRAMMAR_NAME, val)
#define PROTOCOL_V(space, protocol, type) \
space + "Stream." + protocol + "." + TC_Common::upper(protocol) + "_" + type
#define PROTOCOL_SIMPLE PROTOCOL_V(IDL_NAMESPACE_STR, PROTOCOL_NAME, "SIMPLE")
#define PROTOCOL_COMPLEX PROTOCOL_V(IDL_NAMESPACE_STR, PROTOCOL_NAME, "COMPLEX")
#define PROTOCOL_VAR TO_LOWER_STRING(PROTOCOL_NAME)
using namespace TC_NAMESPACE;
class CodeGenerator
{
public:
CodeGenerator()
: uiNameIndex(0),
_sRpcPath(RPC_MODULE_PATH),
_sStreamPath(STREAM_MODULE_PATH),
_sToPath("./"),
_bClient(false),
_bServer(false),
_bRecursive(false),
_bUseSpecialPath(false),
_bUseStringRepresent(false),
_bStringBinaryEncoding(false),
_bMinimalMembers(false),
_bDTS(false),
_iOptimizeLevel(O0) {}
void createFile(const string& file, const bool bEntry = true);
void setRpcPath(const string& sPath) { _sRpcPath = sPath; }
void setStreamPath(const string& sPath) { _sStreamPath = sPath; }
void setEnableClient(bool bEnable) { _bClient = bEnable; }
void setEnableServer(bool bEnable) { _bServer = bEnable; }
void setTargetPath(const string& sPath) { _sToPath = sPath + "/"; }
void setRecursive(bool bEnable) { _bRecursive = bEnable; }
void setUseSpecialPath(bool bEnable) { _bUseSpecialPath = bEnable; }
void setUseStringRepresent(bool bEnable) { _bUseStringRepresent = bEnable; }
void setStringBinaryEncoding(bool bEnable) { _bStringBinaryEncoding = bEnable; }
void setMinimalMembers(bool bEnable) { _bMinimalMembers = bEnable; }
void setDependent(set<string>& deps) { _depMembers = deps; }
void setEnableDTS(bool bEnable) { _bDTS = bEnable; }
void setOptimize(int iLevel) { _iOptimizeLevel = iLevel; }
enum OPTIMIZE_LEVEL {O0 = 0, Os};
private:
struct ImportFileType
{
enum TYPE_T {EN_ENUM = 10000, EN_ENUM_VALUE, EN_STRUCT};
int iType;
string sNamespace;
string sTypeName;
string sName;
};
struct ImportFile
{
string sFile;
string sModule;
map<string, ImportFileType> mapVars;
};
uint32_t uiNameIndex;
map<string, ImportFile> _mapFiles;
set<string> _depMembers;
void scan(const string & sFile, bool bNotPrefix);
string makeName();
string findName(const string & sNamespace, const string & sName);
private:
string toFunctionName(const TypeIdPtr & pPtr, const string &sAction);
string getDataType(const TypePtr& pPtr);
string getClassName(const TypePtr& pPtr);
string getDtsType(const TypePtr &pPtr, const bool bStream = true);
string getDefault(const TypeIdPtr & pPtr, const string &sDefault, const string & sNamespace);
string getDefault(const TypeIdPtr & pPtr, const string &sDefault, const string & sNamespace, const bool bGlobal);
private:
string generateJS(const StructPtr & pPtr, const string &sNamespace, bool &bNeedAssert);
string generateJS(const ConstPtr &pPtr, const string &sNamespace, bool &bNeedStream);
string generateJS(const EnumPtr &pPtr, const string &sNamespace);
string generateJS(const NamespacePtr &pPtr, bool &bNeedStream, bool &bNeedAssert);
bool generateJS(const ContextPtr &pPtr);
private:
string generateJSProxy(const NamespacePtr &nPtr, bool &bNeedRpc, bool &bNeedStream);
string generateJSProxy(const NamespacePtr &nPtr, const InterfacePtr &pPtr);
string generateJSProxy(const NamespacePtr &nPtr, const InterfacePtr &pPtr, const OperationPtr &oPtr);
bool generateJSProxy(const ContextPtr &pPtr);
private:
string generateJSServer(const NamespacePtr &pPtr, bool &bNeedStream, bool &bNeedRpc, bool &bNeedAssert);
string generatePing(const NamespacePtr &nPtr, const InterfacePtr &pPtr);
string generateAsync(const NamespacePtr &nPtr, const InterfacePtr &pPtr, const OperationPtr &oPtr);
string generateDispatch(const NamespacePtr &nPtr, const InterfacePtr &pPtr, const OperationPtr &oPtr);
string generateJSServer(const NamespacePtr &nPtr, const InterfacePtr &pPtr, const OperationPtr &oPtr);
string generateJSServer(const InterfacePtr &pPtr, const NamespacePtr &nPtr);
bool generateJSServer(const ContextPtr &pPtr);
private:
string generateJSServerImp(const NamespacePtr &nPtr, const InterfacePtr &pPtr, const OperationPtr &oPtr);
string generateJSServerImp(const NamespacePtr &nPtr, const InterfacePtr &pPtr);
string generateJSServerImp(const ContextPtr &cPtr, const NamespacePtr &nPtr);
void generateJSServerImp(const ContextPtr &cPtr);
private:
string generateDTS(const StructPtr &pPtr, const string &sNamespace);
string generateDTS(const ConstPtr &pPtr, const string &sNamespace, bool &bNeedStream);
string generateDTS(const EnumPtr &pPtr, const string &sNamespace);
string generateDTS(const NamespacePtr &pPtr, bool &bNeedStream);
string generateDTS(const NamespacePtr &pPtr, const string &sContent);
void generateDTS(const ContextPtr &cPtr);
private:
string generateDTSServer(const NamespacePtr &pPtr, bool &bNeedStream, bool &bNeedRpc);
string generateDTSServer(const NamespacePtr &nPtr, const InterfacePtr &pPtr);
void generateDTSServer(const ContextPtr &cPtr);
private:
string generateDTSProxyInfo();
string generateDTSProxy(const InterfacePtr &pPtr);
string generateDTSProxy(const NamespacePtr &pPtr, bool &bNeedStream, bool &bNeedRpc);
void generateDTSProxy(const ContextPtr &pPtr);
private:
void makeUTF8File(const string & sFileName, const string & sFileContent);
bool isSimple(const TypePtr & pPtr) const;
bool isBinBuffer(const TypePtr & pPtr) const;
bool isRawOrString(const TypePtr & pPtr) const;
bool isDependent(const string & sNamespace, const string & sName) const;
string getRealFileInfo(const string & sPath);
string printHeaderRemark(const string & sTypeName);
private:
string _sRpcPath;
string _sStreamPath;
string _sToPath;
bool _bClient;
bool _bServer;
bool _bRecursive;
bool _bUseSpecialPath;
bool _bUseStringRepresent;
bool _bStringBinaryEncoding;
bool _bMinimalMembers;
bool _bEntry;
string _sIdlFile;
bool _bDTS;
int _iOptimizeLevel;
};
#endif

90
src/code_generator.cpp Normal file
View File

@ -0,0 +1,90 @@
/**
* 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 "code_generator.h"
string CodeGenerator::printHeaderRemark(const string &sTypeName)
{
ostringstream s;
s << "// **********************************************************************" << endl;
s << "// Parsed By " << IDL_NAMESPACE_STR << "Parser(" << PARSER_VERSION << "), Generated By " << EXECUTE_FILENAME << "(" << GENERATOR_VERSION << ")" << endl;
s << "// " << IDL_NAMESPACE_STR << "Parser Maintained By <" << TC_Common::upper(IDL_NAMESPACE_STR) << "> and " << EXECUTE_FILENAME << " Maintained By <superzheng>" << endl;
s << "// Generated from \"" << TC_File::extractFileName(_sIdlFile) << "\" by " <<
(_bEntry ? sTypeName : (_bMinimalMembers ? "Minimal" : "Relation")) << " Mode" << endl;
s << "// **********************************************************************" << endl;
s << endl;
s << "/* eslint-disable */" << endl;
s << endl;
return s.str();
}
void CodeGenerator::createFile(const string &file, const bool bEntry)
{
_sIdlFile = getRealFileInfo(file);
_bEntry = bEntry;
g_parse->parse(_sIdlFile);
vector<ContextPtr> contexts = g_parse->getContexts();
for(size_t i = 0; i < contexts.size(); i++)
{
if (_sIdlFile == contexts[i]->getFileName())
{
scan(_sIdlFile, true); //分析枚举值、结构体所在的文件
if (!_bClient && !_bServer)
{
generateJS(contexts[i]); //生成当前文件的编解码文件
if(_bDTS) generateDTS(contexts[i]); //生成 typescript 描述文件
}
if (_bClient)
{
if(!generateJSProxy(contexts[i])) return; //生成当前文件的客户端代理类文件
if(_bDTS) generateDTSProxy(contexts[i]); //生成客户端 typescript 描述文件
}
if (_bServer)
{
if(!generateJSServer(contexts[i])) return; //生成当前文件的服务端代理类文件
if(_bDTS) generateDTSServer(contexts[i]); //生成服务端 typescript 描述文件
generateJSServerImp(contexts[i]); //生成当前文件的服务端实现类文件
}
vector<string> files = contexts[i]->getIncludes();
for (size_t ii = 0; _bRecursive && ii < files.size(); ii++)
{
CodeGenerator node;
node.setRpcPath(_sRpcPath);
node.setStreamPath(_sStreamPath);
node.setTargetPath(_sToPath);
node.setRecursive(_bRecursive);
node.setUseSpecialPath(_bUseSpecialPath);
node.setUseStringRepresent(_bUseStringRepresent);
node.setStringBinaryEncoding(_bStringBinaryEncoding);
node.setMinimalMembers(_bMinimalMembers);
node.setDependent(_depMembers);
node.setEnableDTS(_bDTS);
node.createFile(files[ii], false);
}
}
}
}

38
src/file_util.cpp Normal file
View File

@ -0,0 +1,38 @@
/**
* 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 "code_generator.h"
void CodeGenerator::makeUTF8File(const string& sFileName, const string& sFileContent)
{
try {
char header[3] = {(char)(0xef), (char)(0xbb), (char)(0xbf)};
string sData(header, 3);
sData += TC_Encoder::gbk2utf8(sFileContent);
TC_File::save2file(sFileName, sData.c_str());
} catch (...) {
cout << "Convert GBK to UTF8 failed, current encoding is GBK.";
TC_File::save2file(sFileName, sFileContent.c_str());
}
}
string CodeGenerator::getRealFileInfo(const string& sPath)
{
return TC_File::extractFilePath(sPath) + TC_File::excludeFileExt(TC_File::extractFileName(sPath)) + "." + TC_Common::lower(IDL_TYPE);
}

398
src/gen_js.cpp Normal file
View File

@ -0,0 +1,398 @@
/**
* 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 "code_generator.h"
string CodeGenerator::generateJS(const EnumPtr &pPtr, const string &sNamespace)
{
ostringstream s;
s << TAB << sNamespace << "." << pPtr->getId() << " = {" << endl;
INC_TAB;
//成员变量
int nenum = -1;
bool bDependent = false;
vector<TypeIdPtr>& member = pPtr->getAllMemberPtr();
for (size_t i = 0; i < member.size(); i++)
{
bDependent |= isDependent(sNamespace, member[i]->getId());
if (member[i]->hasDefault())
{
nenum = TC_Common::strto<int>(member[i]->def());
}
else
{
nenum++;
}
s << TAB << "\"" << member[i]->getId() << "\" : " << TC_Common::tostr(nenum) << "," << endl;
}
s << TAB << "\"_classname\" : \"" << sNamespace << "." << pPtr->getId() << "\"" << endl;
DEL_TAB;
s << TAB << "};" << endl;
//函数
s << TAB << sNamespace << "." << pPtr->getId() << "._write = function(os, tag, val) { return os.writeInt32(tag, val); };" << endl;
s << TAB << sNamespace << "." << pPtr->getId() << "._read = function(is, tag, def) { return is.readInt32(tag, true, def); };" << endl;
if (!_bMinimalMembers || _bEntry || bDependent || isDependent(sNamespace, pPtr->getId())) {
return s.str();
} else {
return "";
}
}
string CodeGenerator::generateJS(const ConstPtr &pPtr, const string &sNamespace, bool &bNeedStream)
{
if (_bMinimalMembers && !_bEntry && !isDependent(sNamespace, pPtr->getTypeIdPtr()->getId()))
{
return "";
}
ostringstream s;
if (_bStringBinaryEncoding && GET_CONST_GRAMMAR_PTR(pPtr)->t == CONST_GRAMMAR(STRING))
{
bNeedStream = true;
}
s << TAB << sNamespace << "." << pPtr->getTypeIdPtr()->getId() << " = " << getDefault(pPtr->getTypeIdPtr(), GET_CONST_GRAMMAR_PTR(pPtr)->v, sNamespace, false) << ";" << endl;
return s.str();
}
string CodeGenerator::generateJS(const StructPtr & pPtr, const string &sNamespace, bool &bNeedAssert)
{
if (_bMinimalMembers && !_bEntry && !isDependent(sNamespace, pPtr->getId()))
{
return "";
}
ostringstream s;
vector<TypeIdPtr>& member = pPtr->getAllMemberPtr();
s << TAB << sNamespace << "." << pPtr->getId() << " = function() {"<< endl;
INC_TAB;
for (size_t i = 0; i < member.size(); i++)
{
s << TAB << "this." << member[i]->getId() << " = " << getDefault(member[i], member[i]->def(), sNamespace) << ";" << endl;
}
{
s << TAB << "this._classname = \"" << sNamespace << "." << pPtr->getId() << "\";" << endl;
}
DEL_TAB;
s << TAB << "};" << endl;
s << TAB << sNamespace << "." << pPtr->getId() << "._classname = \"" << sNamespace << "." << pPtr->getId() << "\";" << endl;
string sProto = TC_Common::replace(pPtr->getSid(), "::", ".");
s << TAB << sProto << "._write = function (os, tag, value) { os.writeStruct(tag, value); };" << endl;
s << TAB << sProto << "._read = function (is, tag, def) { return is.readStruct(tag, true, def); };" << endl;
//_readFrom
s << TAB << sNamespace << "." << pPtr->getId() << "._readFrom = function (is) {" << endl;
INC_TAB;
s << TAB << "var tmp = new " << sNamespace << "." << pPtr->getId() << ";" << endl;
for (size_t i = 0; i < member.size(); i++)
{
string sFuncName = toFunctionName(member[i], "read");
s << TAB << "tmp." << member[i]->getId() << " = is." << sFuncName << "(" << member[i]->getTag()
<< ", " << (member[i]->isRequire()?"true":"false") << ", ";
if (isSimple(member[i]->getTypePtr()))
{
s << getDefault(member[i], member[i]->def(), sNamespace)
<< (isRawOrString(member[i]->getTypePtr()) ? ", 1" : "");
}
else
{
s << getDataType(member[i]->getTypePtr());
}
s << ");" << endl;
}
s << TAB << "return tmp;" << endl;
DEL_TAB;
s << TAB << "};" << endl;
//_writeTo
s << TAB << sNamespace << "." << pPtr->getId() << ".prototype._writeTo = function (os) {" << endl;
INC_TAB;
for (size_t i = 0; i < member.size(); i++)
{
string sFuncName = toFunctionName(member[i], "write");
s << TAB << "os." << sFuncName << "(" << member[i]->getTag() << ", this." << member[i]->getId()
<< (isRawOrString(member[i]->getTypePtr()) ? ", 1" : "") << ");" << endl;
}
DEL_TAB;
s << TAB << "};" << endl;
/*
* Size Optimize:
* Remove <mutil_map> support.
* Remove toBinBuffer, readFromObject, toObject, new, create members.
*/
if (_iOptimizeLevel == Os)
{
return s.str();
}
//_equal
vector<string> key = pPtr->getKey();
s << TAB << sNamespace << "." << pPtr->getId() << ".prototype._equal = function (" << (key.size() > 0 ? "anItem" : "") << ") {" << endl;
INC_TAB;
if (key.size() > 0)
{
s << TAB << "return ";
for (size_t i = 0; i < key.size(); i++)
{
for (size_t ii =0; ii < member.size(); ii++)
{
if (key[i] != member[ii]->getId())
{
continue;
}
if (isSimple(member[i]->getTypePtr()))
{
s << (i==0?"":TAB + TAB) << "this." << key[i] << " === " << "anItem." << key[i];
}
else
{
s << (i==0?"":TAB + TAB) << "this._equal(" << "anItem)";
}
}
if (i != key.size() - 1)
{
s << " && " << endl;
}
}
s << ";" << endl;
}
else
{
bNeedAssert = true;
s << TAB << "assert.fail(\"this structure not define key operation\");" << endl;
}
DEL_TAB;
s << TAB << "};" << endl;
//_genKey
s << TAB << sNamespace << "." << pPtr->getId() << ".prototype._genKey = function () {" << endl;
INC_TAB;
s << TAB << "if (!this._proto_struct_name_) {" << endl;
INC_TAB;
s << TAB << "this._proto_struct_name_ = \"STRUCT\" + Math.random();" << endl;
DEL_TAB;
s << TAB << "}" << endl;
s << TAB << "return this._proto_struct_name_;" << endl;
DEL_TAB;
s << TAB << "};" << endl;
//toObject
s << TAB << sNamespace << "." << pPtr->getId() << ".prototype.toObject = function() { "<< endl;
INC_TAB;
s << TAB << "return {" << endl;
for (size_t i = 0; i < member.size(); i++)
{
INC_TAB;
if (i > 0 && i < member.size()) {
s << "," << endl;
}
if (isSimple(member[i]->getTypePtr())) {
s << TAB << "\"" << member[i]->getId() << "\" : this." << member[i]->getId();
}
else {
s << TAB << "\"" << member[i]->getId() << "\" : this." << member[i]->getId() << ".toObject()";
}
DEL_TAB;
}
s << endl;
s << TAB << "};" << endl;
DEL_TAB;
s << TAB << "};" << endl;
//readFromJson
s << TAB << sNamespace << "." << pPtr->getId() << ".prototype.readFromObject = function(json) { "<< endl;
INC_TAB;
for (size_t i = 0; i < member.size(); i++)
{
if (isSimple(member[i]->getTypePtr())) {
s << TAB << "json.hasOwnProperty(\"" << member[i]->getId() << "\") && (this." << member[i]->getId() << " = json." << member[i]->getId() << ");" << endl;
} else {
s << TAB << "json.hasOwnProperty(\"" << member[i]->getId() << "\") && (this." << member[i]->getId() << ".readFromObject(json." << member[i]->getId() << "));" << endl;
}
}
DEL_TAB;
s << TAB << "};" << endl;
//toBinBuffer
s << TAB << sNamespace << "." << pPtr->getId() << ".prototype.toBinBuffer = function () {" << endl;
INC_TAB;
s << TAB << "var os = new " << IDL_NAMESPACE_STR << "Stream." << IDL_TYPE << "OutputStream();" << endl;
s << TAB << "this._writeTo(os);" << endl;
s << TAB << "return os.getBinBuffer();" << endl;
DEL_TAB;
s << TAB << "};" << endl;
//new
s << TAB << sNamespace << "." << pPtr->getId() << ".new = function () {" << endl;
INC_TAB;
s << TAB << "return new " << sNamespace << "." << pPtr->getId() << "();" << endl;
DEL_TAB;
s << TAB << "};" << endl;
//create
s << TAB << sNamespace << "." << pPtr->getId() << ".create = function (is) {" << endl;
INC_TAB;
s << TAB << "return " << sNamespace << "." << pPtr->getId() << "._readFrom(is);" << endl;
DEL_TAB;
s << TAB << "};" << endl;
return s.str();
}
string CodeGenerator::generateJS(const NamespacePtr &pPtr, bool &bNeedStream, bool &bNeedAssert)
{
ostringstream sstr;
vector<StructPtr> ss(pPtr->getAllStructPtr());
for (size_t last = 0; last != ss.size() && ss.size() != 0;)
{
last = ss.size();
for (vector<StructPtr>::iterator iter=ss.begin(); iter!=ss.end();)
{
string str = generateJS(*iter, pPtr->getId(), bNeedAssert);
if (!str.empty()) {
sstr << str << endl;
iter = ss.erase(iter);
} else {
iter++;
}
}
}
ostringstream cstr;
vector<ConstPtr> &cs = pPtr->getAllConstPtr();
for (size_t i = 0; i < cs.size(); i++)
{
cstr << generateJS(cs[i], pPtr->getId(), bNeedStream);
}
ostringstream estr;
vector<EnumPtr> & es = pPtr->getAllEnumPtr();
for (size_t i = 0; i < es.size(); i++)
{
estr << generateJS(es[i], pPtr->getId());
}
ostringstream str;
if (!estr.str().empty()) str << estr.str() << endl;
if (!cstr.str().empty()) str << cstr.str() << endl;
if (!sstr.str().empty())
{
bNeedStream = true;
str << sstr.str() << endl;
}
return str.str();
}
bool CodeGenerator::generateJS(const ContextPtr &pPtr)
{
vector<NamespacePtr> namespaces = pPtr->getNamespaces();
ostringstream istr;
set<string> setNamespace;
for(size_t i = 0; i < namespaces.size(); i++)
{
if (setNamespace.count(namespaces[i]->getId()) == 0)
{
istr << TAB << "var " << namespaces[i]->getId() << " = " << namespaces[i]->getId() << " || {};" << endl;
istr << TAB << "module.exports." << namespaces[i]->getId() << " = " << namespaces[i]->getId() << ";" << endl << endl;
setNamespace.insert(namespaces[i]->getId());
}
}
//先生成编解码体
ostringstream estr;
bool bNeedAssert = false;
bool bNeedStream = false;
for(size_t i = 0; i < namespaces.size(); i++)
{
estr << generateJS(namespaces[i], bNeedStream, bNeedAssert);
}
if (estr.str().empty())
{
return false;
}
//再生成导入模块
ostringstream ostr;
for (map<string, ImportFile>::iterator it = _mapFiles.begin(); it != _mapFiles.end(); it++)
{
if (it->second.sModule.empty()) continue;
if (estr.str().find(it->second.sModule + ".") == string::npos) continue;
ostr << "var " << it->second.sModule << " = require(\"" << it->second.sFile << "\");" << endl;
}
//生成文件内容
ostringstream sstr;
sstr << printHeaderRemark("Structure");
sstr << "\"use strict\";" << endl << endl;
if (bNeedAssert)
{
sstr << "var assert = require(\"assert\");" << endl;
}
if (bNeedStream)
{
sstr << "var " << IDL_NAMESPACE_STR << "Stream = require(\"" << _sStreamPath << "\");" << endl;
}
sstr << ostr.str() << endl;
sstr << istr.str();
sstr << estr.str() << endl;
string sFileName = TC_File::excludeFileExt(_sToPath + TC_File::extractFileName(pPtr->getFileName())) + IDL_TYPE + ".js";
TC_File::makeDirRecursive(_sToPath, 0755);
makeUTF8File(sFileName, sstr.str());
return true;
}

239
src/gen_js_dts.cpp Normal file
View File

@ -0,0 +1,239 @@
/**
* 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 "code_generator.h"
string CodeGenerator::generateDTS(const EnumPtr &pPtr, const string &sNamespace)
{
ostringstream s;
INC_TAB;
s << TAB << "enum " << pPtr->getId() << " {" << endl;
INC_TAB;
//成员变量
int nenum = -1;
bool bDependent = false;
vector<TypeIdPtr>& member = pPtr->getAllMemberPtr();
for (size_t i = 0; i < member.size(); i++)
{
bDependent |= isDependent(sNamespace, member[i]->getId());
if (member[i]->hasDefault())
{
nenum = TC_Common::strto<int>(member[i]->def());
}
else
{
nenum++;
}
s << TAB << "\"" << member[i]->getId() << "\" = " << TC_Common::tostr(nenum) << ((i < member.size() - 1) ? "," : "") << endl;
}
DEL_TAB;
s << TAB << "}" << endl;
DEL_TAB;
if (!_bMinimalMembers || _bEntry || bDependent || isDependent(sNamespace, pPtr->getId())) {
return s.str();
} else {
return "";
}
}
string CodeGenerator::generateDTS(const ConstPtr &pPtr, const string &sNamespace, bool &bNeedStream)
{
if (_bMinimalMembers && !_bEntry && !isDependent(sNamespace, pPtr->getTypeIdPtr()->getId()))
{
return "";
}
ostringstream s;
if (_bStringBinaryEncoding && GET_CONST_GRAMMAR_PTR(pPtr)->t == CONST_GRAMMAR(STRING))
{
bNeedStream = true;
}
INC_TAB;
s << TAB << "const " << pPtr->getTypeIdPtr()->getId() << ":"
<< getDtsType(pPtr->getTypeIdPtr()->getTypePtr()) << ";"
<< endl;
DEL_TAB;
return s.str();
}
string CodeGenerator::generateDTS(const StructPtr &pPtr, const string &sNamespace)
{
if (_bMinimalMembers && !_bEntry && !isDependent(sNamespace, pPtr->getId()))
{
return "";
}
string sStructName = pPtr->getId() + "$OBJ";
vector<TypeIdPtr> &member = pPtr->getAllMemberPtr();
ostringstream s;
INC_TAB;
s << TAB << "class " << pPtr->getId() << " {" << endl;
INC_TAB;
for (size_t i = 0; i < member.size(); i++)
{
s << TAB << (member[i]->getId()) << (member[i]->isRequire()?": ":"?: ") << getDtsType(member[i]->getTypePtr()) << ";" << endl;
}
if (member.size() > 0)
{
s << endl;
}
/*
* Size Optimize:
* Remove <mutil_map> support.
* Remove toBinBuffer, readFromObject, toObject, new, create members.
*/
if (_iOptimizeLevel != Os)
{
s << TAB << "toObject(): " << sStructName << ";" << endl;
s << TAB << "readFromObject(json: " << sStructName << "): void;" << endl;
s << TAB << "toBinBuffer(): " << IDL_NAMESPACE_STR << "Stream.BinBuffer;" << endl;
s << TAB << "static new(): " << pPtr->getId() << ";" << endl;
s << TAB << "static create(is: " << IDL_NAMESPACE_STR << "Stream." << IDL_TYPE << "InputStream): " << pPtr->getId() << ";" << endl;
}
DEL_TAB;
s << TAB << "}" << endl;
s << TAB << "interface " << sStructName << " {" << endl;
INC_TAB;
for (size_t i = 0; i < member.size(); i++)
{
const string &sType = getDtsType(member[i]->getTypePtr(), false);
if (!sType.empty())
{
s << TAB << (member[i]->getId()) << (member[i]->isRequire()?": ":"?: ") << sType << ";" << endl;
}
}
DEL_TAB;
s << TAB << "}" << endl;
DEL_TAB;
return s.str();
}
string CodeGenerator::generateDTS(const NamespacePtr &pPtr, bool &bNeedStream)
{
//结构
ostringstream sstr;
vector<StructPtr> ss(pPtr->getAllStructPtr());
for (size_t last = 0; last != ss.size() && ss.size() != 0;)
{
last = ss.size();
for (vector<StructPtr>::iterator iter=ss.begin(); iter!=ss.end();)
{
string str = generateDTS(*iter, pPtr->getId());
if (!str.empty())
{
sstr << str << endl;
iter = ss.erase(iter);
}
else
{
iter++;
}
}
}
//常量
ostringstream cstr;
vector<ConstPtr> &cs = pPtr->getAllConstPtr();
for (size_t i = 0; i < cs.size(); i++)
{
cstr << generateDTS(cs[i], pPtr->getId(), bNeedStream);
}
//枚举
ostringstream estr;
vector<EnumPtr> &es = pPtr->getAllEnumPtr();
for (size_t i = 0; i < es.size(); i++)
{
estr << generateDTS(es[i], pPtr->getId());
}
ostringstream kstr;
if (!estr.str().empty()) kstr << estr.str() << endl;
if (!cstr.str().empty()) kstr << cstr.str() << endl;
if (!sstr.str().empty())
{
bNeedStream = true;
kstr << sstr.str();
}
return kstr.str();
}
string CodeGenerator::generateDTS(const NamespacePtr &pPtr, const string &sContent)
{
ostringstream str;
if (!sContent.empty())
{
str << "export declare namespace " << pPtr->getId() << " {" << endl;
str << sContent;
str << "}" << endl << endl;
}
return str.str();
}
void CodeGenerator::generateDTS(const ContextPtr &pPtr)
{
vector<NamespacePtr> namespaces = pPtr->getNamespaces();
//先生成编解码体
ostringstream estr;
bool bNeedStream = false;
for (size_t i = 0; i < namespaces.size(); i++)
{
estr << generateDTS(namespaces[i], generateDTS(namespaces[i], bNeedStream));
}
if (estr.str().empty())
{
return;
}
//再生成导入模块
ostringstream ostr;
if (bNeedStream)
{
ostr << "import * as " << IDL_NAMESPACE_STR << "Stream from \"" << _sStreamPath << "\";" << endl;
}
for (map<string, ImportFile>::iterator it = _mapFiles.begin(); it != _mapFiles.end(); it++)
{
if (it->second.sModule.empty()) continue;
if (estr.str().find(it->second.sModule + ".") == string::npos) continue;
ostr << "import * as " << it->second.sModule << " from \"" << TC_File::excludeFileExt(it->second.sFile) << "\";" << endl;
}
//生成文件内容
ostringstream sstr;
sstr << printHeaderRemark("Structure");
sstr << ostr.str() << endl;
sstr << estr.str() << endl;
string sFileName = TC_File::excludeFileExt(_sToPath + TC_File::extractFileName(pPtr->getFileName())) + IDL_TYPE + ".d.ts";
TC_File::makeDirRecursive(_sToPath, 0755);
makeUTF8File(sFileName, sstr.str());
}

549
src/gen_proxy.cpp Normal file
View File

@ -0,0 +1,549 @@
/**
* 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 "code_generator.h"
#define INVOKE_RETURN(protocol, prefix, params) \
str << TAB << "return this._worker." << TC_Common::lower(protocol) << "_invoke(\"" << oPtr->getId() << "\", "; \
str << "__" << nPtr->getId() << "_" << pPtr->getId() << "$" << oPtr->getId() << "$" << prefix << "E"; \
str << "(" << sParams << params << "), arguments[arguments.length - 1], "; \
str << "__" << nPtr->getId() << "_" << pPtr->getId() << "$" << oPtr->getId() << "$IF" << ").then("; \
str << "__" << nPtr->getId() << "_" << pPtr->getId() << "$" << oPtr->getId() << "$" << prefix << "D, "; \
str << "__" << nPtr->getId() << "_" << pPtr->getId() << "$" << oPtr->getId() << "$ER);" << endl;
#define PROTOCOL_PARAMS (sParams.empty() ? "" : ", ") << "version"
struct SortOperation
{
bool operator()(const OperationPtr &o1, const OperationPtr &o2)
{
return o1->getId() < o2->getId();
}
};
string CodeGenerator::generateJSProxy(const NamespacePtr &nPtr, const InterfacePtr &pPtr, const OperationPtr &oPtr)
{
ostringstream str;
vector<ParamDeclPtr> & vParamDecl = oPtr->getAllParamDeclPtr();
bool bHasParamOut = false;
string sParams = "";
for (size_t i = 0; i < vParamDecl.size(); i++)
{
if (vParamDecl[i]->isOut())
{
continue;
}
sParams += (sParams.empty()?"":", ") + vParamDecl[i]->getTypeIdPtr()->getId();
}
//SETP01 生成函数声明（Interface = IF）
str << TAB << "var __" << nPtr->getId() << "_" << pPtr->getId() << "$" << oPtr->getId() << "$IF = {" << endl;
INC_TAB;
str << TAB << "\"name\" : \"" << oPtr->getId() << "\"," << endl;
str << TAB << "\"return\" : \"" << getClassName(oPtr->getReturnPtr()->getTypePtr()) << "\"," << endl;
str << TAB << "\"arguments\" : [";
for (size_t i = 0; i < vParamDecl.size(); i++)
{
str << (i > 0 ? ", {" : "{") << endl;
INC_TAB;
str << TAB << "\"name\" : \"" << vParamDecl[i]->getTypeIdPtr()->getId() << "\"," << endl;
str << TAB << "\"class\" : \"" << getClassName(vParamDecl[i]->getTypeIdPtr()->getTypePtr()) << "\"," << endl;
if (vParamDecl[i]->isOut())
{
bHasParamOut = true;
str << TAB << "\"direction\" : \"out\"" << endl;
}
else
{
str << TAB << "\"direction\" : \"in\"" << endl;
}
DEL_TAB;
str << TAB << "}";
}
str << "]" << endl;
DEL_TAB;
str << TAB << "};" << endl << endl;
//SETP02 生成 IDL 编码接口（IDL Encoder = IE）
str << TAB << "var __" << nPtr->getId() << "_" << pPtr->getId() << "$" << oPtr->getId() << "$IE = function (" << sParams << ") {" << endl;
INC_TAB;
str << TAB << "var os = new " << IDL_NAMESPACE_STR << "Stream." << IDL_TYPE << "OutputStream();" << endl;
for (size_t i = 0; i < vParamDecl.size(); i++)
{
if (vParamDecl[i]->isOut()) continue;
str << TAB << "os." << toFunctionName(vParamDecl[i]->getTypeIdPtr(), "write") << "("
<< (i + 1) << ", " << vParamDecl[i]->getTypeIdPtr()->getId()
<< (isRawOrString(vParamDecl[i]->getTypeIdPtr()->getTypePtr()) ? ", 1" : "") << ");" << endl;
// 写入 Dependent 列表
getDataType(vParamDecl[i]->getTypeIdPtr()->getTypePtr());
}
str << TAB << "return os.getBinBuffer();" << endl;
DEL_TAB;
str << TAB << "};" << endl << endl;
//STEP03 生成 IDL 解码函数（IDL Decoder = ID）
str << TAB << "var __" << nPtr->getId() << "_" << pPtr->getId() << "$" << oPtr->getId() << "$ID = function (data) {" << endl;
INC_TAB;
str << TAB << "try {" << endl;
INC_TAB;
if (oPtr->getReturnPtr()->getTypePtr() || bHasParamOut)
{
str << TAB << "var is = new " << IDL_NAMESPACE_STR << "Stream." << IDL_TYPE << "InputStream(data.response.sBuffer);" << endl;
}
str << TAB << "return {" << endl;
INC_TAB;
str << TAB << "\"request\" : data.request," << endl;
str << TAB << "\"response\" : {" << endl;
INC_TAB;
str << TAB << "\"costtime\" : data.request.costtime";
if (oPtr->getReturnPtr()->getTypePtr())
{
str << "," << endl;
str << TAB << "\"return\""
<< " : is." << toFunctionName(oPtr->getReturnPtr(), "read") << "(0, true, ";
if (isSimple(oPtr->getReturnPtr()->getTypePtr()))
{
str << getDefault(oPtr->getReturnPtr(), oPtr->getReturnPtr()->def(), nPtr->getId())
<< (isRawOrString(oPtr->getReturnPtr()->getTypePtr()) ? ", 1" : "");
}
else
{
str << getDataType(oPtr->getReturnPtr()->getTypePtr());
}
str << ")";
}
if (bHasParamOut)
{
str << "," << endl;
str << TAB << "\"arguments\" : {" << endl;
INC_TAB;
for (size_t i = 0; i < vParamDecl.size(); i++)
{
if (!vParamDecl[i]->isOut()) continue;
str << TAB << "\"" << vParamDecl[i]->getTypeIdPtr()->getId() << "\""
<< " : is." << toFunctionName(vParamDecl[i]->getTypeIdPtr(), "read") << "(" << (i + 1) << ", true, ";
if (isSimple(vParamDecl[i]->getTypeIdPtr()->getTypePtr()))
{
str << getDefault(vParamDecl[i]->getTypeIdPtr(), vParamDecl[i]->getTypeIdPtr()->def(), nPtr->getId())
<< (isRawOrString(vParamDecl[i]->getTypeIdPtr()->getTypePtr()) ? ", 1" : "");
}
else
{
str << getDataType(vParamDecl[i]->getTypeIdPtr()->getTypePtr());
}
str << ")";
if (i == vParamDecl.size() - 1)
{
str << endl;
}
else
{
str << "," << endl;
}
}
DEL_TAB;
str << TAB << "}";
}
str << endl;
DEL_TAB;
str << TAB << "}" << endl;
DEL_TAB;
str << TAB << "};" << endl;
DEL_TAB;
str << TAB << "} catch (e) {" << endl;
INC_TAB;
str << TAB << "throw {" << endl;
INC_TAB;
str << TAB << "\"request\" : data.request," << endl;
str << TAB << "\"response\" : {" << endl;
INC_TAB;
str << TAB << "\"costtime\" : data.request.costtime," << endl;
str << TAB << "\"error\" : {" << endl;
INC_TAB;
str << TAB << "\"code\" : " << IDL_NAMESPACE_STR << "Error.CLIENT.DECODE_ERROR," << endl;
str << TAB << "\"message\" : e.message" << endl;
DEL_TAB;
str << TAB << "}" << endl;
DEL_TAB;
str << TAB << "}" << endl;
DEL_TAB;
str << TAB << "};" << endl;
DEL_TAB;
str << TAB << "}" << endl;
DEL_TAB;
str << TAB << "};" << endl << endl;
//SETP04 生成 Protocol 编码接口（Protocol Encoder = PE）
str << TAB << "var __" << nPtr->getId() << "_" << pPtr->getId() << "$" << oPtr->getId() << "$PE = function ("
<< sParams << (sParams.empty() ? "" : ", ") << "__$PROTOCOL$VERSION) {" << endl;
INC_TAB;
str << TAB << "var " << PROTOCOL_VAR << " = new " << IDL_NAMESPACE_STR << "Stream.UniAttribute();" << endl;
str << TAB << PROTOCOL_VAR << "." << PROTOCOL_VAR << "Version = __$PROTOCOL$VERSION;" << endl;
for (size_t i = 0; i < vParamDecl.size(); i++)
{
if (vParamDecl[i]->isOut()) continue;
str << TAB << PROTOCOL_VAR << "." << toFunctionName(vParamDecl[i]->getTypeIdPtr(), "write") << "(\""
<< vParamDecl[i]->getTypeIdPtr()->getId() << "\", " << vParamDecl[i]->getTypeIdPtr()->getId()
<< (isRawOrString(vParamDecl[i]->getTypeIdPtr()->getTypePtr()) ? ", 1" : "") << ");" << endl;
// 写入 Dependent 列表
getDataType(vParamDecl[i]->getTypeIdPtr()->getTypePtr());
}
str << TAB << "return " << PROTOCOL_VAR << ";" << endl;
DEL_TAB;
str << TAB << "};" << endl << endl;
// STEP05 生成 Protocol 解码函数（Protocol Decoder = PD）
str << TAB << "var __" << nPtr->getId() << "_" << pPtr->getId() << "$" << oPtr->getId() << "$PD = function (data) {" << endl;
INC_TAB;
str << TAB << "try {" << endl;
INC_TAB;
if (oPtr->getReturnPtr()->getTypePtr() || bHasParamOut) {
str << TAB << "var " << PROTOCOL_VAR << " = data.response." << PROTOCOL_VAR << ";" << endl;
}
str << TAB << "return {" << endl;
INC_TAB;
str << TAB << "\"request\" : data.request," << endl;
str << TAB << "\"response\" : {" << endl;
INC_TAB;
str << TAB << "\"costtime\" : data.request.costtime";
if (oPtr->getReturnPtr()->getTypePtr())
{
str << "," << endl;
str << TAB << "\"return\" : " << PROTOCOL_VAR << "." << toFunctionName(oPtr->getReturnPtr(), "read") << "(\"\"";
if (!isSimple(oPtr->getReturnPtr()->getTypePtr()) && !isBinBuffer(oPtr->getReturnPtr()->getTypePtr()))
{
str << ", " << getDataType(oPtr->getReturnPtr()->getTypePtr());
}
str << ", " << getDefault(oPtr->getReturnPtr(), "", nPtr->getId(), true)
<< (isRawOrString(oPtr->getReturnPtr()->getTypePtr()) ? ", 1" : "");
str << ")";
}
if (bHasParamOut)
{
str << "," << endl;
str << TAB << "\"arguments\" : {" << endl;
INC_TAB;
for (size_t i = 0; i < vParamDecl.size(); i++)
{
if (!vParamDecl[i]->isOut()) continue;
str << TAB << "\"" << vParamDecl[i]->getTypeIdPtr()->getId() << "\" : "
<< PROTOCOL_VAR << "." << toFunctionName(vParamDecl[i]->getTypeIdPtr(), "read")
<< "(\"" << vParamDecl[i]->getTypeIdPtr()->getId() << "\"";
if (!isSimple(vParamDecl[i]->getTypeIdPtr()->getTypePtr()) && !isBinBuffer(vParamDecl[i]->getTypeIdPtr()->getTypePtr()))
{
str << ", " << getDataType(vParamDecl[i]->getTypeIdPtr()->getTypePtr());
}
str << ")";
if (i == vParamDecl.size() - 1)
{
str << endl;
}
else
{
str << "," << endl;
}
}
DEL_TAB;
str << TAB << "}";
}
str << endl;
DEL_TAB;
str << TAB << "}" << endl;
DEL_TAB;
str << TAB << "};" << endl;
DEL_TAB;
str << TAB << "} catch (e) {" << endl;
INC_TAB;
str << TAB << "throw {" << endl;
INC_TAB;
str << TAB << "\"request\" : data.request," << endl;
str << TAB << "\"response\" : {" << endl;
INC_TAB;
str << TAB << "\"costtime\" : data.request.costtime," << endl;
str << TAB << "\"error\" : {" << endl;
INC_TAB;
str << TAB << "\"code\" : " << IDL_NAMESPACE_STR << "Error.CLIENT.DECODE_ERROR," << endl;
str << TAB << "\"message\" : e.message" << endl;
DEL_TAB;
str << TAB << "}" << endl;
DEL_TAB;
str << TAB << "}" << endl;
DEL_TAB;
str << TAB << "};" << endl;
DEL_TAB;
str << TAB << "}" << endl;
DEL_TAB;
str << TAB << "};" << endl << endl;
//STEP03 生成框架调用错误处理函数（Error Response = ER）
str << TAB << "var __" << nPtr->getId() << "_" << pPtr->getId() << "$" << oPtr->getId() << "$ER = function (data) {" << endl;
INC_TAB;
str << TAB << "throw {" << endl;
INC_TAB;
str << TAB << "\"request\" : data.request," << endl;
str << TAB << "\"response\" : {" << endl;
INC_TAB;
str << TAB << "\"costtime\" : data.request.costtime," << endl;
str << TAB << "\"error\" : data.error" << endl;
DEL_TAB;
str << TAB << "}" << endl;
DEL_TAB;
str << TAB << "}" << endl;
DEL_TAB;
str << TAB << "};" << endl << endl;
//SETP04 生成函数接口
str << TAB << nPtr->getId() << "." << pPtr->getId() << "Proxy.prototype." << oPtr->getId() << " = function ("
<< sParams << ") {" << endl;
INC_TAB;
str << TAB << "var version = this._worker.version;" << endl;
str << TAB << "if (version === " << PROTOCOL_SIMPLE << " || version === " << PROTOCOL_COMPLEX << ") {" << endl;
INC_TAB;
INVOKE_RETURN(PROTOCOL_VAR, "P", PROTOCOL_PARAMS);
DEL_TAB;
str << TAB << "} else {" << endl;
INC_TAB;
INVOKE_RETURN(IDL_NAMESPACE_STR, "I", "");
DEL_TAB;
str << TAB << "}" << endl;
DEL_TAB;
str << TAB << "};" << endl;
//SETP05 绑定函数声明
str << TAB << nPtr->getId() << "." << pPtr->getId() << "Proxy." << oPtr->getId() << " = "
<< "__" << nPtr->getId() << "_" << pPtr->getId() << "$" << oPtr->getId() << "$IF;" << endl;
return str.str();
}
string CodeGenerator::generateJSProxy(const NamespacePtr &nPtr, const InterfacePtr &pPtr)
{
ostringstream str;
vector<OperationPtr> & vOperation = pPtr->getAllOperationPtr();
sort(vOperation.begin(), vOperation.end(), SortOperation());
for (size_t i = 0; i < vOperation.size(); i++)
{
str << generateJSProxy(nPtr, pPtr, vOperation[i]) << endl;
}
return str.str();
}
string CodeGenerator::generateJSProxy(const NamespacePtr &nPtr, bool &bNeedRpc, bool &bNeedStream)
{
ostringstream str;
vector<InterfacePtr> & is = nPtr->getAllInterfacePtr();
for (size_t i = 0; i < is.size(); i++)
{
str << generateJSProxy(nPtr, is[i]) << endl;
}
if (is.size() != 0)
{
bNeedRpc = true;
bNeedStream = true;
}
return str.str();
}
bool CodeGenerator::generateJSProxy(const ContextPtr &cPtr)
{
vector<NamespacePtr> namespaces = cPtr->getNamespaces();
ostringstream istr;
set<string> setNamespace;
for(size_t i = 0; i < namespaces.size(); i++)
{
if (setNamespace.count(namespaces[i]->getId()) == 0)
{
istr << TAB << "var " << namespaces[i]->getId() << " = " << namespaces[i]->getId() << " || {};" << endl;
istr << TAB << "module.exports." << namespaces[i]->getId() << " = " << namespaces[i]->getId() << ";" << endl << endl;
setNamespace.insert(namespaces[i]->getId());
}
}
set<string> setInterface;
for(size_t i = 0; i < namespaces.size(); i++)
{
vector<InterfacePtr> & is = namespaces[i]->getAllInterfacePtr();
for (size_t ii = 0; ii < is.size(); ii++)
{
if (setInterface.count(namespaces[i]->getId() + "::" + is[ii]->getId()) != 0)
{
continue;
}
setInterface.insert(namespaces[i]->getId() + "::" + is[ii]->getId());
istr << TAB << namespaces[i]->getId() << "." << is[ii]->getId() << "Proxy = function () {" << endl;
INC_TAB;
istr << TAB << "this._name = undefined;" << endl;
istr << TAB << "this._worker = undefined;" << endl;
DEL_TAB;
istr << TAB << "};" << endl << endl;
istr << TAB << namespaces[i]->getId() << "." << is[ii]->getId() << "Proxy.prototype.setTimeout = function (iTimeout) {" << endl;
INC_TAB;
istr << TAB << "this._worker.timeout = iTimeout;" << endl;
DEL_TAB;
istr << TAB << "};" << endl << endl;
istr << TAB << namespaces[i]->getId() << "." << is[ii]->getId() << "Proxy.prototype.getTimeout = function () {" << endl;
INC_TAB;
istr << TAB << "return this._worker.timeout;" << endl;
DEL_TAB;
istr << TAB << "};" << endl << endl;
istr << TAB << namespaces[i]->getId() << "." << is[ii]->getId() << "Proxy.prototype.setVersion = function (iVersion) {" << endl;
INC_TAB;
istr << TAB << "this._worker.version = iVersion;" << endl;
DEL_TAB;
istr << TAB << "};" << endl << endl;
istr << TAB << namespaces[i]->getId() << "." << is[ii]->getId() << "Proxy.prototype.getVersion = function () {" << endl;
INC_TAB;
istr << TAB << "return this._worker.version;" << endl;
DEL_TAB;
istr << TAB << "};" << endl << endl;
}
}
//先生成编解码 + 代理类
ostringstream estr;
bool bNeedAssert = false;
bool bNeedStream = false;
for(size_t i = 0; i < namespaces.size(); i++)
{
estr << generateJS(namespaces[i], bNeedStream, bNeedAssert);
}
bool bNeedRpc = false;
for(size_t i = 0; i < namespaces.size(); i++)
{
estr << generateJSProxy(namespaces[i], bNeedRpc, bNeedStream);
}
if (estr.str().empty())
{
return false;
}
//再生成导入模块
ostringstream ostr;
for (map<string, ImportFile>::iterator it = _mapFiles.begin(); it != _mapFiles.end(); it++)
{
if (it->second.sModule.empty()) continue;
if (estr.str().find(it->second.sModule + ".") == string::npos) continue;
ostr << "var " << it->second.sModule << " = require(\"" << it->second.sFile << "\");" << endl;
}
//生成文件内容
ostringstream sstr;
sstr << printHeaderRemark("Client");
sstr << "\"use strict\";" << endl << endl;
if (bNeedAssert)
{
sstr << "var assert = require(\"assert\");" << endl;
}
if (bNeedStream)
{
sstr << "var " << IDL_NAMESPACE_STR << "Stream = require(\"" << _sStreamPath << "\");" << endl;
}
if (bNeedRpc)
{
sstr << "var " << IDL_NAMESPACE_STR << "Error = require(\"" << _sRpcPath << "\").error;" << endl;
}
sstr << ostr.str() << endl;
sstr << istr.str() << endl;
sstr << estr.str() << endl;
string sFileName = TC_File::excludeFileExt(_sToPath + TC_File::extractFileName(cPtr->getFileName())) + "Proxy.js";
TC_File::makeDirRecursive(_sToPath, 0755);
makeUTF8File(sFileName, sstr.str());
return true;
}
#undef INVOKE_RETURN
#undef PROTOCOL_PARAMS

208
src/gen_proxy_dts.cpp Normal file
View File

@ -0,0 +1,208 @@
/**
* 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 "code_generator.h"
string CodeGenerator::generateDTSProxyInfo()
{
ostringstream str;
INC_TAB;
// Argument
str << TAB << "interface SharedArgumentInfo {" << endl;
INC_TAB;
str << TAB << "name: string;" << endl;
str << TAB << "class: string;" << endl;
str << TAB << "direction: 'in' | 'out';" << endl;
DEL_TAB;
str << TAB << "}" << endl;
// Function
str << TAB << "interface SharedFunctionInfo {" << endl;
INC_TAB;
str << TAB << "name: string;" << endl;
str << TAB << "return: string;" << endl;
str << TAB << "arguments: SharedArgumentInfo[];" << endl;
DEL_TAB;
str << TAB << "}" << endl;
DEL_TAB;
return str.str();
}
string CodeGenerator::generateDTSProxy(const InterfacePtr &pPtr)
{
vector<OperationPtr> &vOperation = pPtr->getAllOperationPtr();
ostringstream str;
//class
INC_TAB;
str << TAB << "class " << pPtr->getId() << "Proxy {" << endl;
INC_TAB;
str << TAB << "setTimeout(iTimeout: number): void;" << endl;
str << TAB << "getTimeout(): number;" << endl;
str << TAB << "setVersion(iVersion: number): void;" << endl;
str << TAB << "getVersion(): number;" << endl;
for (size_t i = 0; i < vOperation.size(); i++)
{
OperationPtr &oPtr = vOperation[i];
str << TAB << oPtr->getId() << "(";
vector<ParamDeclPtr> &vParamDecl = oPtr->getAllParamDeclPtr();
for (size_t j = 0; j < vParamDecl.size(); j++)
{
if(vParamDecl[j]->isOut())
{
continue;
}
str << vParamDecl[j]->getTypeIdPtr()->getId() << ": " << getDtsType(vParamDecl[j]->getTypeIdPtr()->getTypePtr());
str << ", ";
}
str << "property?: " << IDL_NAMESPACE_STR << "Rpc.InvokeProperty): Promise<" << pPtr->getId() << "$" << oPtr->getId() << "$DE>;" << endl;
str << TAB << "static " << oPtr->getId() << ": SharedFunctionInfo;" << endl;
}
DEL_TAB;
str << TAB << "}" << endl;
//interface
for (size_t i = 0; i < vOperation.size(); i++)
{
OperationPtr &oPtr = vOperation[i];
str << TAB << "interface " << pPtr->getId() << "$" << oPtr->getId() << "$DE {" << endl;
INC_TAB;
str << TAB << "request: object;" << endl;
str << TAB << "response: {" << endl;
INC_TAB;
str << TAB << "costtime: number;" << endl;
if (oPtr->getReturnPtr()->getTypePtr())
{
str << TAB << "return: " << getDtsType(oPtr->getReturnPtr()->getTypePtr()) << ";" << endl;
}
else
{
str << TAB << "return: void;" << endl;
}
vector<ParamDeclPtr> &vParamDecl = oPtr->getAllParamDeclPtr();
bool hasArgs = false;
for (size_t j = 0; j < vParamDecl.size(); j++)
{
if(vParamDecl[j]->isOut()) {
hasArgs = true;
break;
}
}
if(hasArgs)
{
str << TAB << "arguments: {" << endl;
INC_TAB;
for (size_t j = 0; j < vParamDecl.size(); j++)
{
if(!vParamDecl[j]->isOut()) {
continue;
}
str << TAB << vParamDecl[j]->getTypeIdPtr()->getId() << ": " << getDtsType(vParamDecl[j]->getTypeIdPtr()->getTypePtr()) << ";" << endl;
}
DEL_TAB;
str << TAB << "}" << endl;
}
DEL_TAB;
str << TAB << "}" << endl;
DEL_TAB;
str << TAB << "}" << endl;
}
DEL_TAB;
return str.str();
}
string CodeGenerator::generateDTSProxy(const NamespacePtr &nPtr, bool &bNeedStream, bool &bNeedRpc)
{
ostringstream str;
vector<InterfacePtr> &is = nPtr->getAllInterfacePtr();
if (is.size() > 0)
{
bNeedStream = true;
bNeedRpc = true;
str << generateDTSProxyInfo() << endl;
}
for (size_t i = 0; i < is.size(); i++)
{
str << generateDTSProxy(is[i]) << endl;
}
return str.str();
}
void CodeGenerator::generateDTSProxy(const ContextPtr &cPtr)
{
vector<NamespacePtr> namespaces = cPtr->getNamespaces();
//先生成编解码 + 代理类
ostringstream estr;
bool bNeedStream = false;
bool bNeedRpc = false;
for(size_t i = 0; i < namespaces.size(); i++)
{
ostringstream kstr;
kstr << generateDTS(namespaces[i], bNeedStream);
kstr << generateDTSProxy(namespaces[i], bNeedStream, bNeedRpc);
estr << generateDTS(namespaces[i], kstr.str());
}
if (estr.str().empty())
{
return;
}
//再生成导入模块
ostringstream ostr;
for (map<string, ImportFile>::iterator it = _mapFiles.begin(); it != _mapFiles.end(); it++)
{
if (it->second.sModule.empty()) continue;
if (estr.str().find(it->second.sModule + ".") == string::npos) continue;
ostr << "import * as " << it->second.sModule << " from \"" << TC_File::excludeFileExt(it->second.sFile) << "\";" << endl;
}
//生成文件内容
ostringstream sstr;
sstr << printHeaderRemark("Client");
if (bNeedStream)
{
sstr << "import * as " << IDL_NAMESPACE_STR << "Stream from \"" << _sStreamPath << "\";" << endl;
}
if (bNeedRpc)
{
sstr << "import * as " << IDL_NAMESPACE_STR << "Rpc from \"" << _sRpcPath << "\";" << endl;
}
sstr << ostr.str() << endl;
sstr << estr.str() << endl;
string sFileName = TC_File::excludeFileExt(_sToPath + TC_File::extractFileName(cPtr->getFileName())) + "Proxy.d.ts";
TC_File::makeDirRecursive(_sToPath, 0755);
makeUTF8File(sFileName, sstr.str());
}

381
src/gen_server.cpp Normal file
View File

@ -0,0 +1,381 @@
/**
* 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 "code_generator.h"
string CodeGenerator::generatePing(const NamespacePtr &nPtr, const InterfacePtr &pPtr)
{
ostringstream str;
str << TAB << "var __" << nPtr->getId() << "_" << pPtr->getId() << "$" << TC_Common::lower(IDL_NAMESPACE_STR) << "_ping$RE = function (_ret) {" << endl;
INC_TAB;
str << TAB << "if (this.getRequestVersion() === " << PROTOCOL_SIMPLE << " || this.getRequestVersion() === " << PROTOCOL_COMPLEX << ") {" << endl;
INC_TAB;
str << TAB << "var " << PROTOCOL_VAR << " = new " << IDL_NAMESPACE_STR << "Stream.UniAttribute();" << endl;
str << TAB << PROTOCOL_VAR << "." << PROTOCOL_VAR << "Version = this.getRequestVersion();" << endl;
str << TAB << PROTOCOL_VAR << ".writeInt32(\"\", _ret);" << endl << endl;
str << TAB << "this.doResponse(" << PROTOCOL_VAR << ".encode());" << endl;
DEL_TAB;
str << TAB << "} else {" << endl;
INC_TAB;
str << TAB << "var os = new " << IDL_NAMESPACE_STR << "Stream." << IDL_TYPE << "OutputStream();" << endl;
str << TAB << "os.writeInt32(0, _ret);" << endl << endl;
str << TAB << "this.doResponse(os.getBinBuffer());" << endl;
DEL_TAB;
str << TAB << "}" << endl;
DEL_TAB;
str << TAB << "};" << endl << endl;
str << TAB << nPtr->getId() << "." << pPtr->getId() << "Imp.prototype.__" << TC_Common::lower(IDL_NAMESPACE_STR) << "_ping = function (current) {" << endl;
INC_TAB;
str << TAB << "__" << nPtr->getId() << "_" << pPtr->getId() << "$" << TC_Common::lower(IDL_NAMESPACE_STR) << "_ping$RE.call(current, 0);" << endl << endl;
str << TAB << "return " << IDL_NAMESPACE_STR << "Error.SUCCESS;" << endl;
DEL_TAB;
str << TAB << "};" << endl;
return str.str();
}
string CodeGenerator::generateAsync(const NamespacePtr &nPtr, const InterfacePtr &pPtr, const OperationPtr &oPtr)
{
ostringstream str;
string sParams = "";
if (oPtr->getReturnPtr()->getTypePtr())
{
sParams += "_ret";
// 写入 Dependent 列表
getDataType(oPtr->getReturnPtr()->getTypePtr());
}
vector<ParamDeclPtr> & vParamDecl = oPtr->getAllParamDeclPtr();
for (size_t i = 0; i < vParamDecl.size(); i++)
{
if (!vParamDecl[i]->isOut()) continue;
sParams += (sParams.empty()?"":", ") + vParamDecl[i]->getTypeIdPtr()->getId();
}
str << TAB << "var __" << nPtr->getId() << "_" << pPtr->getId() << "$" << oPtr->getId() << "$RE = function (" << sParams << ") {" << endl;
INC_TAB;
if (sParams.empty())
{
str << TAB << "this.doResponse(new " << IDL_NAMESPACE_STR << "Stream.BinBuffer());" << endl;
DEL_TAB;
str << TAB << "};" << endl;
return str.str();
}
str << TAB << "if (this.getRequestVersion() === " << PROTOCOL_SIMPLE << " || this.getRequestVersion() === " << PROTOCOL_COMPLEX << ") {" << endl;
INC_TAB;
str << TAB << "var " << PROTOCOL_VAR << " = new " << IDL_NAMESPACE_STR << "Stream.UniAttribute();" << endl;
str << TAB << PROTOCOL_VAR << "." << PROTOCOL_VAR << "Version = this.getRequestVersion();" << endl;
if (oPtr->getReturnPtr()->getTypePtr())
{
str << TAB << PROTOCOL_VAR << "." << toFunctionName(oPtr->getReturnPtr(), "write") << "(\"\", _ret"
<< (isRawOrString(oPtr->getReturnPtr()->getTypePtr()) ? ", 1" : "") << ");" << endl;
}
for (size_t i = 0; i < vParamDecl.size(); i++)
{
if (!vParamDecl[i]->isOut()) continue;
str << TAB << PROTOCOL_VAR << "." << toFunctionName(vParamDecl[i]->getTypeIdPtr(), "write") << "(\""
<< vParamDecl[i]->getTypeIdPtr()->getId() << "\", " << vParamDecl[i]->getTypeIdPtr()->getId()
<< (isRawOrString(vParamDecl[i]->getTypeIdPtr()->getTypePtr()) ? ", 1" : "") << ");" << endl;
}
str << endl;
str << TAB << "this.doResponse(" << PROTOCOL_VAR << ".encode());" << endl;
DEL_TAB;
str << TAB << "} else {" << endl;
INC_TAB;
str << TAB << "var os = new " << IDL_NAMESPACE_STR << "Stream." << IDL_TYPE << "OutputStream();" << endl;
if (oPtr->getReturnPtr()->getTypePtr())
{
str << TAB << "os." << toFunctionName(oPtr->getReturnPtr(), "write") << "(0, _ret"
<< (isRawOrString(oPtr->getReturnPtr()->getTypePtr()) ? ", 1" : "") << ");" << endl;
}
for (size_t i = 0; i < vParamDecl.size(); i++)
{
if (!vParamDecl[i]->isOut()) continue;
str << TAB << "os." << toFunctionName(vParamDecl[i]->getTypeIdPtr(), "write") << "("
<< (i + 1) << ", " << vParamDecl[i]->getTypeIdPtr()->getId()
<< (isRawOrString(vParamDecl[i]->getTypeIdPtr()->getTypePtr()) ? ", 1" : "") << ");" << endl;
}
str << endl;
str << TAB << "this.doResponse(os.getBinBuffer());" << endl;
DEL_TAB;
str << TAB << "}" << endl;
DEL_TAB;
str << TAB << "};" << endl;
return str.str();
}
string CodeGenerator::generateDispatch(const NamespacePtr &nPtr, const InterfacePtr &pPtr, const OperationPtr &oPtr)
{
ostringstream str;
vector<ParamDeclPtr> & vParamDecl = oPtr->getAllParamDeclPtr();
str << TAB << nPtr->getId() << "." << pPtr->getId() << "Imp.prototype.__" << oPtr->getId() << " = function (current" << (vParamDecl.size() != 0 ? ", binBuffer" : "") << ") {" << endl;
INC_TAB;
ostringstream dstr;
for (size_t i = 0; i < vParamDecl.size(); i++)
{
dstr << TAB << "var " << vParamDecl[i]->getTypeIdPtr()->getId() << " = null;" << endl;
}
if (vParamDecl.size() != 0)
{
dstr << endl;
}
dstr << TAB << "if (current.getRequestVersion() === " << PROTOCOL_SIMPLE << " || current.getRequestVersion() === " << PROTOCOL_COMPLEX << ") {" << endl;
INC_TAB;
dstr << TAB << "var " << PROTOCOL_VAR << " = new " << IDL_NAMESPACE_STR << "Stream.UniAttribute();" << endl;
dstr << TAB << PROTOCOL_VAR << "." << PROTOCOL_VAR << "Version = current.getRequestVersion();" << endl;
dstr << TAB << PROTOCOL_VAR << ".decode(binBuffer);" << endl;
for (size_t i = 0; i < vParamDecl.size(); i++)
{
dstr << TAB << vParamDecl[i]->getTypeIdPtr()->getId()
<< " = " << PROTOCOL_VAR << "." << toFunctionName(vParamDecl[i]->getTypeIdPtr(), "read")
<< "(\"" << vParamDecl[i]->getTypeIdPtr()->getId() << "\"";
if (!isSimple(vParamDecl[i]->getTypeIdPtr()->getTypePtr()) && !isBinBuffer(vParamDecl[i]->getTypeIdPtr()->getTypePtr()))
{
dstr << ", " << getDataType(vParamDecl[i]->getTypeIdPtr()->getTypePtr());
}
if (vParamDecl[i]->isOut())
{
dstr << ", " << getDefault(vParamDecl[i]->getTypeIdPtr(), "", nPtr->getId(), true)
<< (isRawOrString(vParamDecl[i]->getTypeIdPtr()->getTypePtr()) ? ", 1" : "");
}
dstr << ");" << endl;
}
DEL_TAB;
dstr << TAB << "} else {" << endl;
INC_TAB;
dstr << TAB << "var is = new " << IDL_NAMESPACE_STR << "Stream." << IDL_TYPE << "InputStream(binBuffer);" << endl;
string sParams = "";
for (size_t i = 0; i < vParamDecl.size(); i++)
{
sParams += ", " + vParamDecl[i]->getTypeIdPtr()->getId();
dstr << TAB << vParamDecl[i]->getTypeIdPtr()->getId()
<< " = is." << toFunctionName(vParamDecl[i]->getTypeIdPtr(), "read") << "("
<< (i + 1) << ", " << (vParamDecl[i]->isOut() ? "false" : "true") << ", ";
if (isSimple(vParamDecl[i]->getTypeIdPtr()->getTypePtr()))
{
dstr << getDefault(vParamDecl[i]->getTypeIdPtr(), vParamDecl[i]->getTypeIdPtr()->def(), nPtr->getId())
<< (isRawOrString(vParamDecl[i]->getTypeIdPtr()->getTypePtr()) ? ", 1" : "");
}
else
{
dstr << getDataType(vParamDecl[i]->getTypeIdPtr()->getTypePtr());
}
dstr << ");" << endl;
}
DEL_TAB;
dstr << TAB << "}" << endl << endl;
if (!sParams.empty())
{
str << dstr.str();
}
str << TAB << "current.sendResponse = __" << nPtr->getId() << "_" << pPtr->getId() << "$" << oPtr->getId() << "$RE;" << endl << endl;
str << TAB << "this." << oPtr->getId() << "(current" << sParams << ");" << endl << endl;
str << TAB << "return " << IDL_NAMESPACE_STR << "Error.SUCCESS;" << endl;
DEL_TAB;
str << TAB << "};" << endl;
return str.str();
}
string CodeGenerator::generateJSServer(const NamespacePtr &nPtr, const InterfacePtr &pPtr, const OperationPtr &oPtr)
{
ostringstream str;
str << TAB << nPtr->getId() << "." << pPtr->getId() << "Imp.prototype." << oPtr->getId() << " = function () {" << endl;
INC_TAB;
str << TAB << "assert.fail(\"" << oPtr->getId() << " function not implemented\");" << endl;
DEL_TAB;
str << TAB << "};" << endl;
return str.str();
}
string CodeGenerator::generateJSServer(const InterfacePtr &pPtr, const NamespacePtr &nPtr)
{
ostringstream str;
vector<OperationPtr> & vOperation = pPtr->getAllOperationPtr();
//生成类
str << TAB << nPtr->getId() << "." << pPtr->getId() << "Imp = function () { " << endl;
INC_TAB;
str << TAB << "this._name = undefined;" << endl;
str << TAB << "this._worker = undefined;" << endl;
DEL_TAB;
str << TAB << "};" << endl << endl;
//生成初始化函数
str << TAB << nPtr->getId() << "." << pPtr->getId() << "Imp.prototype.initialize = function () {};" << endl << endl;
//生成分发函数
str << TAB << nPtr->getId() << "." << pPtr->getId() << "Imp.prototype.onDispatch = function (current, funcName, binBuffer) { " << endl;
INC_TAB;
str << TAB << "if (\"__\" + funcName in this) {" << endl;
INC_TAB;
str << TAB << "return this[\"__\" + funcName](current, binBuffer);" << endl;
DEL_TAB;
str << TAB << "} else {" << endl;
INC_TAB;
str << TAB << "return " << IDL_NAMESPACE_STR << "Error.SERVER.FUNC_NOT_FOUND;" << endl;
DEL_TAB;
str << TAB << "}" << endl;
DEL_TAB;
str << TAB << "};" << endl << endl;
//生成 PING 方法
str << generatePing(nPtr, pPtr) << endl;
//生成接口函数
for (size_t i = 0; i < vOperation.size(); i++)
{
str << generateJSServer(nPtr, pPtr, vOperation[i]) << endl;
str << generateAsync(nPtr, pPtr, vOperation[i]) << endl;
str << generateDispatch(nPtr, pPtr, vOperation[i]) << endl;
}
str << endl;
return str.str();
}
string CodeGenerator::generateJSServer(const NamespacePtr &pPtr, bool &bNeedStream, bool &bNeedRpc, bool &bNeedAssert)
{
ostringstream str;
vector<InterfacePtr> & is = pPtr->getAllInterfacePtr();
for (size_t i = 0; i < is.size(); i++)
{
str << generateJSServer(is[i], pPtr) << endl;
}
if (is.size() != 0)
{
bNeedRpc = true;
bNeedStream = true;
bNeedAssert = true;
}
return str.str();
}
bool CodeGenerator::generateJSServer(const ContextPtr &pPtr)
{
vector<NamespacePtr> namespaces = pPtr->getNamespaces();
ostringstream istr;
set<string> setNamespace;
for(size_t i = 0; i < namespaces.size(); i++)
{
if (setNamespace.count(namespaces[i]->getId()) == 0)
{
istr << TAB << "var " << namespaces[i]->getId() << " = " << namespaces[i]->getId() << " || {};" << endl;
istr << TAB << "module.exports." << namespaces[i]->getId() << " = " << namespaces[i]->getId() << ";" << endl << endl;
setNamespace.insert(namespaces[i]->getId());
}
}
//生成编解码 + 服务类
ostringstream estr;
bool bNeedAssert = false;
bool bNeedStream = false;
for(size_t i = 0; i < namespaces.size(); i++)
{
estr << generateJS(namespaces[i], bNeedStream, bNeedAssert);
}
bool bNeedRpc = false;
for(size_t i = 0; i < namespaces.size(); i++)
{
estr << generateJSServer(namespaces[i], bNeedStream, bNeedRpc, bNeedAssert) << endl;
}
if (estr.str().empty())
{
return false;
}
//再生成导入模块
ostringstream ostr;
if (bNeedAssert)
{
ostr << TAB << "var assert = require(\"assert\");" << endl;
}
if (bNeedStream)
{
ostr << TAB << "var " << IDL_NAMESPACE_STR << "Stream = require(\"" << _sStreamPath << "\");" << endl;
}
if (bNeedRpc)
{
ostr << TAB << "var " << IDL_NAMESPACE_STR << "Error = require(\"" << _sRpcPath << "\").error;" << endl;
}
for (map<string, ImportFile>::iterator it = _mapFiles.begin(); it != _mapFiles.end(); it++)
{
if (it->second.sModule.empty()) continue;
if (estr.str().find(it->second.sModule + ".") == string::npos) continue;
ostr << TAB << "var " << it->second.sModule << " = require(\"" << it->second.sFile << "\");" << endl;
}
ostringstream str;
str << printHeaderRemark("Server");
str << "\"use strict\";" << endl << endl;
str << ostr.str() << endl;
str << istr.str();
str << estr.str() << endl;
string sFileName = TC_File::excludeFileExt(_sToPath + TC_File::extractFileName(pPtr->getFileName())) + ".js";
TC_File::makeDirRecursive(_sToPath, 0755);
makeUTF8File(sFileName, str.str());
return true;
}

149
src/gen_server_dts.cpp Normal file
View File

@ -0,0 +1,149 @@
/**
* 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 "code_generator.h"
string CodeGenerator::generateDTSServer(const NamespacePtr &nPtr, const InterfacePtr &pPtr)
{
vector<OperationPtr> &vOperation = pPtr->getAllOperationPtr();
ostringstream str;
//class
INC_TAB;
str << TAB << "class " << pPtr->getId() << "Imp {" << endl;
INC_TAB;
str << TAB << "initialize(): Promise<any> | void;" << endl;
str << TAB << "protected onDispatch(current: " << IDL_NAMESPACE_STR << "Rpc." << IDL_TYPE << "Current, funcName: string, binBuffer: " << IDL_NAMESPACE_STR << "Stream.BinBuffer): number" << endl;
for (size_t i = 0; i < vOperation.size(); i++)
{
OperationPtr &oPtr = vOperation[i];
str << TAB << oPtr->getId() << "(current: " << pPtr->getId() << "$" << oPtr->getId() << "$CUR";
vector<ParamDeclPtr> &vParamDecl = oPtr->getAllParamDeclPtr();
for (size_t j = 0; j < vParamDecl.size(); j++)
{
str << ", " << vParamDecl[j]->getTypeIdPtr()->getId() << ": " << getDtsType(vParamDecl[j]->getTypeIdPtr()->getTypePtr());
}
str << "): void;" << endl;
}
DEL_TAB;
str << TAB << "}" << endl;
//interface
for (size_t i = 0; i < vOperation.size(); i++)
{
OperationPtr &oPtr = vOperation[i];
str << TAB << "interface " << pPtr->getId() << "$" << oPtr->getId() << "$CUR extends " << IDL_NAMESPACE_STR << "Rpc." << IDL_TYPE << "Current {" <<endl;
INC_TAB;
str << TAB;
if (oPtr->getReturnPtr()->getTypePtr())
{
str << "sendResponse(ret: " << getDtsType(oPtr->getReturnPtr()->getTypePtr());
vector<ParamDeclPtr> &vParamDecl = oPtr->getAllParamDeclPtr();
for (size_t j = 0; j < vParamDecl.size(); j++)
{
if(!vParamDecl[j]->isOut()) {
continue;
}
str << ", " << vParamDecl[j]->getTypeIdPtr()->getId() << ": " << getDtsType(vParamDecl[j]->getTypeIdPtr()->getTypePtr()) ;
}
str << "): void;" << endl;
}
else
{
str << "sendResponse(): void;" << endl;
}
DEL_TAB;
str << TAB << "}" << endl;
}
DEL_TAB;
return str.str();
}
string CodeGenerator::generateDTSServer(const NamespacePtr &pPtr, bool &bNeedStream, bool &bNeedRpc)
{
vector<InterfacePtr> &is = pPtr->getAllInterfacePtr();
ostringstream str;
for (size_t i = 0; i < is.size(); i++)
{
str << generateDTSServer(pPtr, is[i]) << endl;
}
if (is.size() != 0)
{
bNeedRpc = true;
bNeedStream = true;
}
return str.str();
}
void CodeGenerator::generateDTSServer(const ContextPtr &pPtr)
{
vector<NamespacePtr> namespaces = pPtr->getNamespaces();
//生成编解码 + 服务类
ostringstream estr;
bool bNeedStream = false;
bool bNeedRpc = false;
for(size_t i = 0; i < namespaces.size(); i++)
{
ostringstream kstr;
kstr << generateDTS(namespaces[i], bNeedStream);
kstr << generateDTSServer(namespaces[i], bNeedStream, bNeedRpc);
estr << generateDTS(namespaces[i], kstr.str());
}
if(estr.str().empty())
{
return;
}
//再生成导入模块
ostringstream ostr;
if (bNeedStream)
{
ostr << "import * as " << IDL_NAMESPACE_STR << "Stream from \"" << _sStreamPath << "\";" << endl;
}
if (bNeedRpc)
{
ostr << "import * as " << IDL_NAMESPACE_STR << "Rpc from \"" << _sRpcPath << "\";" << endl;
}
for (map<string, ImportFile>::iterator it = _mapFiles.begin(); it != _mapFiles.end(); it++)
{
if (it->second.sModule.empty()) continue;
if (estr.str().find(it->second.sModule + ".") == string::npos) continue;
ostr << "import * as " << it->second.sModule << " from \"" << TC_File::excludeFileExt(it->second.sFile) << "\";" << endl;
}
ostringstream str;
str << printHeaderRemark("Server");
str << ostr.str() << endl;
str << estr.str() << endl;
string sFileName = TC_File::excludeFileExt(_sToPath + TC_File::extractFileName(pPtr->getFileName())) + ".d.ts";
TC_File::makeDirRecursive(_sToPath, 0755);
makeUTF8File(sFileName, str.str());
}

123
src/gen_server_imp.cpp Normal file
View File

@ -0,0 +1,123 @@
/**
* 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 "code_generator.h"
string CodeGenerator::generateJSServerImp(const NamespacePtr &nPtr, const InterfacePtr &pPtr, const OperationPtr &oPtr)
{
ostringstream str;
str << nPtr->getId() << "." << pPtr->getId() << "Imp.prototype." << oPtr->getId() << " = function (current";
vector<ParamDeclPtr> & vParamDecl = oPtr->getAllParamDeclPtr();
for (size_t i = 0; i < vParamDecl.size(); i++)
{
str << ", " << vParamDecl[i]->getTypeIdPtr()->getId();
}
str << ") {" << endl;
INC_TAB;
str << TAB << "//TODO::" << endl;
DEL_TAB;
str << "};" << endl << endl;
return str.str();
}
string CodeGenerator::generateJSServerImp(const NamespacePtr &nPtr, const InterfacePtr &pPtr)
{
ostringstream str;
vector<OperationPtr> & vOperation = pPtr->getAllOperationPtr();
for (size_t i = 0; i < vOperation.size(); i++)
{
str << generateJSServerImp(nPtr, pPtr, vOperation[i]);
}
return str.str();
}
string CodeGenerator::generateJSServerImp(const ContextPtr &cPtr, const NamespacePtr &nPtr)
{
ostringstream str;
vector<InterfacePtr> & is = nPtr->getAllInterfacePtr();
for (size_t i = 0; i < is.size(); i++)
{
str << generateJSServerImp(nPtr, is[i]) << endl;
}
return str.str();
}
void CodeGenerator::generateJSServerImp(const ContextPtr &cPtr)
{
string sFileName = TC_File::excludeFileExt(_sToPath + TC_File::extractFileName(cPtr->getFileName())) + "Imp.js";
if (TC_File::isFileExist(sFileName))
{
return ;
}
ostringstream str;
str << printHeaderRemark("Imp");
str << "\"use strict\";" << endl << endl;
vector<NamespacePtr> namespaces = cPtr->getNamespaces();
set<string> setNamespace;
for(size_t i = 0; i < namespaces.size(); i++)
{
if (setNamespace.count(namespaces[i]->getId()) != 0)
{
continue;
}
setNamespace.insert(namespaces[i]->getId());
str << "var " << namespaces[i]->getId() << " = require(\"./"
<< TC_File::excludeFileExt(TC_File::extractFileName(cPtr->getFileName())) << ".js\")."
<< namespaces[i]->getId() << ";" << endl;
str << "module.exports." << namespaces[i]->getId() << " = " << namespaces[i]->getId() << ";" << endl;
}
str << endl;
set<string> setInterface;
for(size_t i = 0; i < namespaces.size(); i++)
{
vector<InterfacePtr> & is = namespaces[i]->getAllInterfacePtr();
for (size_t ii = 0; ii < is.size(); ii++)
{
if (setInterface.count(namespaces[i]->getId() + "::" + is[ii]->getId()) != 0)
{
continue;
}
setInterface.insert(namespaces[i]->getId() + "::" + is[ii]->getId());
str << namespaces[i]->getId() << "." << is[ii]->getId() << "Imp.prototype.initialize = function () {" << endl;
INC_TAB;
str << TAB << "//TODO::" << endl;
DEL_TAB;
str << "};" << endl << endl;
}
}
for(size_t i = 0; i < namespaces.size(); i++)
{
str << generateJSServerImp(cPtr, namespaces[i]);
}
TC_File::makeDirRecursive(_sToPath, 0755);
makeUTF8File(sFileName, str.str());
}

175
src/idl_scan.cpp Normal file
View File

@ -0,0 +1,175 @@
/**
* 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 "code_generator.h"
string CodeGenerator::makeName()
{
int iHigh = uiNameIndex/26;
int iLow = uiNameIndex%26;
uiNameIndex++;
ostringstream s;
s << "_" << TC_Common::upper(IDL_NAMESPACE_STR) << "_MODULE_";
if (iHigh != 0)
{
s << string(1, (char)(iHigh + 65)) << string(1, (char)(iLow + 65));
}
else
{
s << string(1, (char)(iLow + 65));
}
s << "_";
return s.str();
}
bool CodeGenerator::isDependent(const string& sNamespace, const string& sName) const
{
return _depMembers.find(sNamespace + "::" + sName) != _depMembers.end();
}
string CodeGenerator::findName(const string& sNamespace, const string& sName)
{
#ifdef DUMP_FIND_NAME
cout << "FINDNAME BEGIN:" << sNamespace << "|" << sName << endl;
#endif
for (map<string, ImportFile>::iterator it = _mapFiles.begin(); it != _mapFiles.end(); it++)
{
#ifdef DUMP_FIND_NAME
for (map<string, ImportFileType>::iterator demo = it->second.mapVars.begin(); demo != it->second.mapVars.end(); demo++)
{
cout << "FINDNAME:" << it->second.sModule << "|" << demo->first << "|" << demo->second.sNamespace << "|" << demo->second.sName << endl;
}
#endif
map<string, ImportFileType>::iterator inIter = it->second.mapVars.find(sNamespace + "::" + sName);
if (inIter == it->second.mapVars.end())
{
continue;
}
#ifdef DUMP_FIND_NAME
cout << "DEPMEMBER:" << it->first << "|" << inIter->second.sNamespace << "::" << inIter->second.sName << endl;
#endif
_depMembers.insert(inIter->second.sNamespace + "::" + inIter->second.sName);
switch (inIter->second.iType)
{
case ImportFileType::EN_ENUM : // [[fallthrough]]
case ImportFileType::EN_STRUCT :
{
return it->second.sModule + (it->second.sModule.empty()?"":".") + inIter->second.sNamespace + "." + inIter->second.sName;
}
case ImportFileType::EN_ENUM_VALUE :
{
return it->second.sModule + (it->second.sModule.empty()?"":".") + inIter->second.sNamespace + "." + inIter->second.sTypeName + "." + inIter->second.sName;
}
default :
{
return it->second.sModule;
}
}
}
return "";
}
void CodeGenerator::scan(const string& sFile, bool bNotPrefix)
{
if (_mapFiles.find(sFile) != _mapFiles.end())
{
return ;
}
string sIdlFile = getRealFileInfo(sFile);
g_parse->parse(sIdlFile);
vector<ContextPtr> contexts = g_parse->getContexts();
for(size_t i = 0; i < contexts.size(); i++)
{
if (sIdlFile == contexts[i]->getFileName())
{
ImportFile item;
item.sFile = "./" + TC_File::excludeFileExt(TC_File::extractFileName(sFile)) + IDL_TYPE + ".js";
item.sModule = bNotPrefix?"":makeName();
vector<NamespacePtr> namespaces = contexts[i]->getNamespaces();
for (size_t ii = 0; ii < namespaces.size(); ii++)
{
string sNamespace = namespaces[ii]->getId();
vector<StructPtr> & ss = namespaces[ii]->getAllStructPtr();
for (size_t iii = 0; iii < ss.size(); iii++)
{
vector<string> vecNames = TC_Common::sepstr<string>(ss[iii]->getSid(), "::");
ImportFileType temp;
temp.iType = ImportFileType::EN_STRUCT;
temp.sNamespace = sNamespace;
temp.sName = vecNames[1];
item.mapVars.insert(make_pair(temp.sNamespace + "::" + temp.sName, temp));
}
vector<EnumPtr> & es = namespaces[ii]->getAllEnumPtr();
for (size_t iii = 0; iii < es.size(); iii++)
{
vector<string> vecNames = TC_Common::sepstr<string>(es[iii]->getSid(), "::");
ImportFileType temp;
temp.iType = ImportFileType::EN_ENUM;
temp.sNamespace = sNamespace;
temp.sName = vecNames[1];
item.mapVars.insert(make_pair(temp.sNamespace + "::" + temp.sName, temp));
vector<TypeIdPtr> & eMember = es[iii]->getAllMemberPtr();
for (size_t iiii = 0; iiii < eMember.size(); iiii++)
{
ImportFileType temp;
temp.iType = ImportFileType::EN_ENUM_VALUE;
temp.sNamespace = sNamespace;
temp.sTypeName = vecNames[1];
temp.sName = eMember[iiii]->getId();
#ifdef DUMP_FIND_NAME
cout << "GET::::" << temp.sNamespace << "|" << temp.sName << endl;
#endif
item.mapVars.insert(make_pair(temp.sNamespace + "::" + temp.sName, temp));
}
}
}
_mapFiles.insert(make_pair(sFile, item));
vector<string> vecFiles = contexts[i]->getIncludes();
for (size_t ii = 0; ii < vecFiles.size(); ii++)
{
string sFileName = TC_File::extractFilePath(vecFiles[ii]) + TC_File::excludeFileExt(TC_File::extractFileName(vecFiles[ii])) + "." + TC_Common::lower(IDL_TYPE);
scan(sFileName, false);
}
}
}
}

440
src/idl_util.cpp Normal file
View File

@ -0,0 +1,440 @@
/**
* 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 "code_generator.h"
string CodeGenerator::toFunctionName(const TypeIdPtr& pPtr, const string& sAction)
{
BuiltinPtr bPtr = BuiltinPtr::dynamicCast(pPtr->getTypePtr());
if (bPtr)
{
switch (bPtr->kind())
{
case Builtin::KindBool : return sAction + "Boolean";
case Builtin::KindString : return sAction + "String";
case Builtin::KindByte : return sAction + "Int8";
case Builtin::KindShort : return sAction + (bPtr->isUnsigned() ? "UInt8" : "Int16");
case Builtin::KindInt : return sAction + (bPtr->isUnsigned() ? "UInt16" : "Int32");
case Builtin::KindLong : return sAction + (bPtr->isUnsigned() ? "UInt32" : "Int64");
case Builtin::KindFloat : return sAction + "Float";
case Builtin::KindDouble : return sAction + "Double";
default : assert(false);
}
}
VectorPtr vPtr = VectorPtr::dynamicCast(pPtr->getTypePtr());
if (vPtr)
{
BuiltinPtr vbPtr = BuiltinPtr::dynamicCast(vPtr->getTypePtr());
if (vbPtr && vbPtr->kind() == Builtin::KindByte)
{
return sAction + "Bytes";
}
return sAction + "List";
}
StructPtr sPtr = StructPtr::dynamicCast(pPtr->getTypePtr());
if (sPtr)
{
return sAction + "Struct";
}
EnumPtr ePtr = EnumPtr::dynamicCast(pPtr->getTypePtr());
if (ePtr)
{
return sAction + "Int32";
}
MapPtr mPtr = MapPtr::dynamicCast(pPtr->getTypePtr());
if (mPtr)
{
return sAction + "Map";
}
return "";
}
bool CodeGenerator::isRawOrString(const TypePtr& pPtr) const
{
BuiltinPtr bPtr = BuiltinPtr::dynamicCast(pPtr);
if (bPtr)
{
if (_bUseStringRepresent && bPtr->kind() == Builtin::KindLong)
{
return true;
}
if (_bStringBinaryEncoding && bPtr->kind() == Builtin::KindString)
{
return true;
}
}
return false;
}
string CodeGenerator::getClassName(const TypePtr& pPtr)
{
BuiltinPtr bPtr = BuiltinPtr::dynamicCast(pPtr);
if (bPtr)
{
switch (bPtr->kind())
{
case Builtin::KindBool : return "bool";
case Builtin::KindString : return "string";
case Builtin::KindByte : return "char";
case Builtin::KindShort : return "short";
case Builtin::KindInt : return "int32";
case Builtin::KindLong : return "int64";
case Builtin::KindFloat : return "float";
case Builtin::KindDouble : return "double";
default : assert(false);
}
}
VectorPtr vPtr = VectorPtr::dynamicCast(pPtr);
if (vPtr)
{
return "list(" + getDataType(vPtr->getTypePtr()) + ")";
}
StructPtr sPtr = StructPtr::dynamicCast(pPtr);
if (sPtr)
{
vector<string> vecNames = TC_Common::sepstr<string>(sPtr->getSid(), "::");
assert(vecNames.size() == 2);
return findName(vecNames[0], vecNames[1]);
}
MapPtr mPtr = MapPtr::dynamicCast(pPtr);
if (mPtr)
{
return "map(" + getClassName(mPtr->getLeftTypePtr()) + ", " + getClassName(mPtr->getRightTypePtr()) + ")";
}
EnumPtr ePtr = EnumPtr::dynamicCast(pPtr);
if (ePtr)
{
return "int32";
}
return "void";
}
string CodeGenerator::getDataType(const TypePtr& pPtr)
{
BuiltinPtr bPtr = BuiltinPtr::dynamicCast(pPtr);
if (bPtr)
{
switch (bPtr->kind())
{
case Builtin::KindBool : return IDL_NAMESPACE_STR + "Stream.Boolean";
case Builtin::KindString : return IDL_NAMESPACE_STR + "Stream.String";
case Builtin::KindByte : return IDL_NAMESPACE_STR + "Stream.Int8";
case Builtin::KindShort : return IDL_NAMESPACE_STR + "Stream.Int16";
case Builtin::KindInt : return IDL_NAMESPACE_STR + "Stream.Int32";
case Builtin::KindLong : return IDL_NAMESPACE_STR + "Stream.Int64";
case Builtin::KindFloat : return IDL_NAMESPACE_STR + "Stream.Float";
case Builtin::KindDouble : return IDL_NAMESPACE_STR + "Stream.Double";
default : assert(false);
}
}
VectorPtr vPtr = VectorPtr::dynamicCast(pPtr);
if (vPtr)
{
BuiltinPtr vbPtr = BuiltinPtr::dynamicCast(vPtr->getTypePtr());
if (vbPtr && vbPtr->kind() == Builtin::KindByte)
{
return IDL_NAMESPACE_STR + "Stream.BinBuffer";
}
return IDL_NAMESPACE_STR + "Stream.List(" + getDataType(vPtr->getTypePtr()) + (isRawOrString(vPtr->getTypePtr()) ? ", 1" : "") + ")";
}
StructPtr sPtr = StructPtr::dynamicCast(pPtr);
if (sPtr)
{
vector<string> vecNames = TC_Common::sepstr<string>(sPtr->getSid(), "::");
assert(vecNames.size() == 2);
return findName(vecNames[0], vecNames[1]);
}
MapPtr mPtr = MapPtr::dynamicCast(pPtr);
if (mPtr)
{
bool bLeft = isRawOrString(mPtr->getLeftTypePtr());
bool bRight = isRawOrString(mPtr->getRightTypePtr());
if (!bRight && !bLeft) {
return IDL_NAMESPACE_STR + "Stream.Map(" + getDataType(mPtr->getLeftTypePtr()) + ", " + getDataType(mPtr->getRightTypePtr()) + ")";
} else if (bRight && bLeft) {
return IDL_NAMESPACE_STR + "Stream.Map(" + getDataType(mPtr->getLeftTypePtr()) + ", " + getDataType(mPtr->getRightTypePtr()) + ", 1, 1)";
} else if (bRight) {
return IDL_NAMESPACE_STR + "Stream.Map(" + getDataType(mPtr->getLeftTypePtr()) + ", " + getDataType(mPtr->getRightTypePtr()) + ", 0, 1)";
} else if (bLeft) {
return IDL_NAMESPACE_STR + "Stream.Map(" + getDataType(mPtr->getLeftTypePtr()) + ", " + getDataType(mPtr->getRightTypePtr()) + ", 1)";
} else {
assert(false);
}
}
EnumPtr ePtr = EnumPtr::dynamicCast(pPtr);
if (ePtr)
{
vector<string> vecNames = TC_Common::sepstr<string>(ePtr->getSid(), "::");
assert(vecNames.size() == 2);
return findName(vecNames[0], vecNames[1]);
}
assert(false);
return "";
}
string CodeGenerator::getDtsType(const TypePtr &pPtr, const bool bStream)
{
BuiltinPtr bPtr = BuiltinPtr::dynamicCast(pPtr);
if (bPtr)
{
switch (bPtr->kind())
{
case Builtin::KindBool : return "boolean";
case Builtin::KindString : return _bStringBinaryEncoding ? (bStream ? (IDL_NAMESPACE_STR + "Stream.BinBuffer") : "Buffer") : "string";
case Builtin::KindByte : return "number";
case Builtin::KindShort : return "number";
case Builtin::KindInt : return "number";
case Builtin::KindLong : return _bUseStringRepresent ? "string" : "number";
case Builtin::KindFloat : return "number";
case Builtin::KindDouble : return "number";
default : assert(false);
}
}
VectorPtr vPtr = VectorPtr::dynamicCast(pPtr);
if (vPtr)
{
BuiltinPtr vbPtr = BuiltinPtr::dynamicCast(vPtr->getTypePtr());
if (vbPtr && vbPtr->kind() == Builtin::KindByte)
{
return bStream ? (IDL_NAMESPACE_STR + "Stream.BinBuffer") : "Buffer";
}
return (bStream ? (IDL_NAMESPACE_STR + "Stream.List") : "Array") + string("<") + getDtsType(vPtr->getTypePtr(), bStream) + string(">");
}
StructPtr sPtr = StructPtr::dynamicCast(pPtr);
if (sPtr)
{
vector<string> vecNames = TC_Common::sepstr<string>(sPtr->getSid(), "::");
assert(vecNames.size() == 2);
return findName(vecNames[0], vecNames[1]) + (bStream ? "" : "$OBJ");
}
MapPtr mPtr = MapPtr::dynamicCast(pPtr);
if (mPtr)
{
if (bStream)
{
return IDL_NAMESPACE_STR + "Stream.Map<" + getDtsType(mPtr->getLeftTypePtr(), bStream) + ", " + getDtsType(mPtr->getRightTypePtr(), bStream) + ">";
}
else
{
const string& sLeftType = getDtsType(mPtr->getLeftTypePtr(), bStream);
if (sLeftType == "number" || sLeftType == "string")
{
return "{[key: " + getDtsType(mPtr->getLeftTypePtr(), bStream) + "]: " + getDtsType(mPtr->getRightTypePtr(), bStream) + "}";
}
else if (isSimple(mPtr->getLeftTypePtr()) && sLeftType != "Buffer")
{
return "object";
}
else
{
return "";
}
}
}
EnumPtr ePtr = EnumPtr::dynamicCast(pPtr);
if (ePtr)
{
vector<string> vecNames = TC_Common::sepstr<string>(ePtr->getSid(), "::");
assert(vecNames.size() == 2);
return findName(vecNames[0], vecNames[1]);
}
assert(false);
return "";
}
bool CodeGenerator::isSimple(const TypePtr & pPtr) const
{
BuiltinPtr bPtr = BuiltinPtr::dynamicCast(pPtr);
if (bPtr)
{
return true;
}
EnumPtr ePtr = EnumPtr::dynamicCast(pPtr);
if (ePtr)
{
return true;
}
return false;
}
bool CodeGenerator::isBinBuffer(const TypePtr & pPtr) const
{
VectorPtr vPtr = VectorPtr::dynamicCast(pPtr);
if (vPtr)
{
BuiltinPtr vbPtr = BuiltinPtr::dynamicCast(vPtr->getTypePtr());
if (vbPtr && vbPtr->kind() == Builtin::KindByte)
{
return true;
}
}
return false;
}
string CodeGenerator::getDefault(const TypeIdPtr & pPtr, const string &sDefault, const string& sNamespace)
{
return getDefault(pPtr, sDefault, sNamespace, true);
}
string CodeGenerator::getDefault(const TypeIdPtr & pPtr, const string &sDefault, const string& sNamespace, const bool bGlobal)
{
BuiltinPtr bPtr = BuiltinPtr::dynamicCast(pPtr->getTypePtr());
if (bPtr)
{
switch (bPtr->kind())
{
case Builtin::KindBool :
return sDefault.empty() ? "false" : sDefault;
case Builtin::KindString :
{
if (_bStringBinaryEncoding)
{
return IDL_NAMESPACE_STR + "Stream.BinBuffer.from(\"" + TC_Common::replace(sDefault, "\"", "\\\"") + "\").toNodeBufferUnSafe()";
}
else
{
return "\"" + TC_Common::replace(sDefault, "\"", "\\\"") + "\"";
}
}
case Builtin::KindByte : // [[fallthrough]]
case Builtin::KindShort : // [[fallthrough]]
case Builtin::KindInt : // [[fallthrough]]
case Builtin::KindLong :
{
string sTemp = TC_Common::trim(sDefault);
if (sTemp.empty())
{
sTemp = "0";
}
else
{
if (TC_Common::tostr(TC_Common::strto<long>(sTemp)) != sTemp)
{
//有可能是枚举值，在枚举值中查找
vector<string> vecNames = TC_Common::sepstr<string>(sDefault, "::");
if (vecNames.size() == 2)
{
sTemp = findName(vecNames[0], vecNames[1]);
}
else
{
sTemp = findName(sNamespace, sDefault);
}
}
}
if (sTemp.empty())
{
sTemp = "0";
}
if (_bUseStringRepresent)
{
if (bPtr->kind() == Builtin::KindLong)
{
sTemp = "\"" + sTemp + "\"";
}
}
return sTemp;
}
case Builtin::KindFloat : // [[fallthrough]]
case Builtin::KindDouble :
return sDefault.empty()?"0.0":sDefault;
default :
assert(false);
}
}
EnumPtr ePtr = EnumPtr::dynamicCast(pPtr->getTypePtr());
if (ePtr)
{
if (sDefault.empty())
{
vector<TypeIdPtr>& eMember = ePtr->getAllMemberPtr();
if (eMember.size() > 0)
{
vector<string> vecNames = TC_Common::sepstr<string>(ePtr->getSid(), "::");
string sModule = findName(vecNames[0], eMember[0]->getId());
return sModule;
}
}
if (!TC_Common::isdigit(sDefault))
{
string s1 = sNamespace;
string s2 = sDefault;
string::size_type index = sDefault.find("::");
if (index != string::npos)
{
s1 = sDefault.substr(0, index);
s2 = sDefault.substr(index + 2);
}
string sModule = findName(s1, s2);
assert(!sModule.empty());
return sModule;
}
return sDefault;
}
if (bGlobal)
{
return "new " + getDataType(pPtr->getTypePtr());
}
return sDefault;
}

198
src/main.cpp Normal file
View File

@ -0,0 +1,198 @@
/**
* 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 "util/tc_option.h"
#include "code_generator.h"
void version()
{
cout << "v" << GENERATOR_VERSION << endl;
}
void usage()
{
cout << "Version : " << GENERATOR_VERSION << endl;
cout << "Usage : " << EXECUTE_FILENAME << " [OPTION] " << "*." << TC_Common::lower(IDL_TYPE) << " file" << endl;
cout << " --version print " << EXECUTE_FILENAME << " version" << endl;
cout << " --rpc-path=[DIRECTORY] specify the path of rpc module." << endl;
cout << " --stream-path=[DIRECTORY] specify the path of stream module." << endl;
cout << " --allow-reserved-namespace allow you to use reserved word as a namespace." << endl;
cout << " --dir=DIRECTORY generate source file to DIRECTORY." << endl;
cout << " --relative use current path." << endl;
cout << " --base=DIRECTORY where to search " << IDL_TYPE << " files." << endl;
cout << " --r generate source all " << IDL_TYPE << " files." << endl;
cout << " --r-minimal minimize the dependent members." << endl;
cout << " --r-reserved list of names(split by \",\") that should be keeped." << endl;
cout << " --client just for client side source file." << endl;
cout << " --server just for server side source file." << endl;
cout << " --dts generate d.ts file." << endl;
cout << " --use-string-represent use string represent long type." << endl;
cout << " --string-binary-encoding get string raw bytes <BinBuffer>." << endl;
cout << " --optimize=[0|s] support \"s\" to reduce code size, default is 0." << endl;
cout << endl;
cout << EXECUTE_FILENAME << " support type: boolean char short int long float double list map" << endl;
exit(0);
}
void check(vector<string> &vFiles)
{
for(size_t i = 0; i < vFiles.size(); i++)
{
string ext = TC_File::extractFileExt(vFiles[i]);
if(ext == TC_Common::lower(IDL_TYPE))
{
if(!TC_File::isFileExist(vFiles[i]))
{
cerr << "file '" << vFiles[i] << "' not exists" << endl;
usage();
exit(0);
}
}
else
{
cerr << "only support "<< TC_Common::lower(IDL_TYPE) << " file." << endl;
usage();
exit(0);
}
}
}
int main(int argc, char* argv[])
{
if(argc < 2)
{
usage();
}
try
{
TC_Option option;
option.decode(argc, argv);
vector<string> vFiles = option.getSingle();
check(vFiles);
if (option.hasParam("version"))
{
version();
return 0;
}
if(option.hasParam("help"))
{
usage();
return 0;
}
if(option.hasParam("base"))
{
if (::chdir(option.getValue("base").c_str()) != 0) {
return -1;
}
}
#define ALLOW_USE_RESERVED_NAMESPACE_V(name, keeped) \
g_parse->set##name(keeped);
#define ALLOW_USE_RESERVED_NAMESPACE_BASE(name, keeped) \
ALLOW_USE_RESERVED_NAMESPACE_V(name, keeped)
#define ALLOW_USE_RESERVED_NAMESPACE(keeped) \
ALLOW_USE_RESERVED_NAMESPACE_BASE(IDL_NAMESPACE, keeped)
ALLOW_USE_RESERVED_NAMESPACE(option.hasParam("allow-reserved-namespace"));
#undef ALLOW_USE_RESERVED_NAMESPACE
#undef ALLOW_USE_RESERVED_NAMESPACE_BASE
#undef ALLOW_USE_RESERVED_NAMESPACE_V
g_parse->setUseCurrentPath(option.hasParam("relative"));
CodeGenerator generator;
generator.setRpcPath(option.hasParam("rpc-path")?option.getValue("rpc-path"):RPC_MODULE_PATH);
generator.setStreamPath(option.hasParam("stream-path")?option.getValue("stream-path"):STREAM_MODULE_PATH);
generator.setEnableClient(option.hasParam("client"));
generator.setEnableServer(option.hasParam("server"));
generator.setTargetPath(option.hasParam("dir")?option.getValue("dir"):"./");
generator.setUseSpecialPath(option.hasParam("relative"));
generator.setUseStringRepresent(option.hasParam("use-string-represent"));
generator.setStringBinaryEncoding(option.hasParam("string-binary-encoding"));
generator.setEnableDTS(option.hasParam("dts"));
if (option.hasParam("optimize"))
{
string level = TC_Common::lower(option.getValue("optimize"));
if (level == "s")
{
generator.setOptimize(CodeGenerator::Os);
}
else
{
generator.setOptimize(CodeGenerator::O0);
}
}
bool _bRecursive = option.hasParam("r");
bool _bMinimalMembers = option.hasParam("r-minimal");
if (option.hasParam("r-reserved"))
{
set<string> vMembers;
vector<string> vReserved = TC_Common::sepstr<string>(option.getValue("r-reserved"), ",");
for (size_t i = 0; i < vReserved.size(); i++)
{
if (vReserved[i].empty())
{
continue;
}
if (TC_Common::sepstr<string>(vReserved[i], "::").size() == 2)
{
vMembers.insert(TC_Common::trim(vReserved[i]));
} else {
cerr << "reserved member must be match <Module>::<Member> pattern" << endl;
return 0;
}
}
if (vMembers.size() > 0)
{
_bMinimalMembers = true;
generator.setDependent(vMembers);
}
}
if (!_bRecursive && _bMinimalMembers)
{
cerr << "Missing --r flag" << endl;
return 0;
}
generator.setRecursive(_bRecursive);
generator.setMinimalMembers(_bMinimalMembers);
for (size_t i = 0; i < vFiles.size(); i++)
{
generator.createFile(vFiles[i], true);
}
}
catch(exception& e)
{
cerr<<e.what()<<endl;
}
return 0;
}

2261
third_partly/tarsgrammar/tars.lex.cpp Normal file
View File

File diff suppressed because it is too large Load Diff

3083
third_partly/tarsgrammar/tars.tab.cpp Normal file
View File

File diff suppressed because it is too large Load Diff

94
third_partly/tarsgrammar/tars.tab.hpp Normal file
View File

@ -0,0 +1,94 @@
/* A Bison parser, made by GNU Bison 3.0.4. */
/* Bison interface for Yacc-like parsers in C
Copyright (C) 1984, 1989-1990, 2000-2015 Free Software Foundation, Inc.
This program is free software: you can redistribute it and/or modify
it under the terms of the GNU General Public License as published by
the Free Software Foundation, either version 3 of the License, or
(at your option) any later version.
This program is distributed in the hope that it will be useful,
but WITHOUT ANY WARRANTY; without even the implied warranty of
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
GNU General Public License for more details.
You should have received a copy of the GNU General Public License
along with this program. If not, see <http://www.gnu.org/licenses/>. */
/* As a special exception, you may create a larger work that contains
part or all of the Bison parser skeleton and distribute that work
under terms of your choice, so long as that work isn't itself a
parser generator using the skeleton or a modified version thereof
as a parser skeleton. Alternatively, if you modify or redistribute
the parser skeleton itself, you may (at your option) remove this
special exception, which will cause the skeleton and the resulting
Bison output files to be licensed under the GNU General Public
License without this special exception.
This special exception was added by the Free Software Foundation in
version 2.2 of Bison. */
#ifndef YY_YY_TARS_TAB_HPP_INCLUDED
# define YY_YY_TARS_TAB_HPP_INCLUDED
/* Debug traces. */
#ifndef YYDEBUG
# define YYDEBUG 1
#endif
#if YYDEBUG
extern int yydebug;
#endif
/* Token type. */
#ifndef YYTOKENTYPE
# define YYTOKENTYPE
enum yytokentype
{
TARS_VOID = 258,
TARS_STRUCT = 259,
TARS_BOOL = 260,
TARS_BYTE = 261,
TARS_SHORT = 262,
TARS_INT = 263,
TARS_DOUBLE = 264,
TARS_FLOAT = 265,
TARS_LONG = 266,
TARS_STRING = 267,
TARS_VECTOR = 268,
TARS_MAP = 269,
TARS_NAMESPACE = 270,
TARS_INTERFACE = 271,
TARS_IDENTIFIER = 272,
TARS_OUT = 273,
TARS_OP = 274,
TARS_KEY = 275,
TARS_ROUTE_KEY = 276,
TARS_REQUIRE = 277,
TARS_OPTIONAL = 278,
TARS_CONST_INTEGER = 279,
TARS_CONST_FLOAT = 280,
TARS_FALSE = 281,
TARS_TRUE = 282,
TARS_STRING_LITERAL = 283,
TARS_SCOPE_DELIMITER = 284,
TARS_CONST = 285,
TARS_ENUM = 286,
TARS_UNSIGNED = 287,
BAD_CHAR = 288
};
#endif
/* Value type. */
#if ! defined YYSTYPE && ! defined YYSTYPE_IS_DECLARED
typedef int YYSTYPE;
# define YYSTYPE_IS_TRIVIAL 1
# define YYSTYPE_IS_DECLARED 1
#endif
extern YYSTYPE yylval;
int yyparse (void);
#endif /* !YY_YY_TARS_TAB_HPP_INCLUDED */

339
third_partly/tarsparse/element.cpp Normal file
View File

@ -0,0 +1,339 @@
/**
* 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 "element.h"
#include <iostream>
#include <cassert>
#include "parse.h"
#include "util/tc_file.h"
#include "util/tc_common.h"
/*************************************************************************************************/
const char* Builtin::builtinTable[] =
{
"void",
"bool",
"byte",
"short",
"int",
"long",
"float",
"double",
"string",
"vector",
"map"
};
Builtin::Builtin(Kind kind,bool isUnsigned = false) : _kind(kind),_isUnsigned(isUnsigned)
{
}
Builtin::Kind Builtin::kind() const
{
return _kind;
}
bool Builtin::isSimple() const
{
switch(_kind)
{
case Builtin::KindBool:
case Builtin::KindByte:
case Builtin::KindShort:
case Builtin::KindInt:
case Builtin::KindLong:
case Builtin::KindFloat:
case Builtin::KindDouble:
return true;
default:
return false;
}
return true;
}
string Builtin::def() const
{
switch(_kind)
{
case Builtin::KindBool:
return "true";
case Builtin::KindByte:
case Builtin::KindShort:
case Builtin::KindInt:
case Builtin::KindLong:
case Builtin::KindFloat:
case Builtin::KindDouble:
return "0";
case Builtin::KindString:
return "";
default:
assert(false);
}
return "";
}
/*************************************************************************************************/
Vector::Vector(const TypePtr& ptr) : _ptr(ptr)
{
}
/*************************************************************************************************/
Map::Map(const TypePtr& pleft, const TypePtr& pright): _pleft(pleft), _pright(pright)
{
}
/*************************************************************************************************/
TypeId::TypeId(const TypePtr& ptr, const string&id)
: _ptr(ptr)
, _id(id)
, _bRequire(true)
, _tag(0)
, _bHasDefault(false)
, _size(0)
, _array(false)
{
BuiltinPtr bPtr = BuiltinPtr::dynamicCast(_ptr);
if(bPtr)
{
_bHasDefault = true;
_default = bPtr->def();
}
}
void TypeId::setRequire(int tag)
{
_bRequire = true;
_tag = tag;
}
void TypeId::setDefault(const string &def)
{
BuiltinPtr bPtr = BuiltinPtr::dynamicCast(_ptr);
EnumPtr ePtr = EnumPtr::dynamicCast(_ptr);
assert(bPtr || ePtr);
_bHasDefault = true;
_default = def;
}
void TypeId::disableDefault()
{
BuiltinPtr bPtr = BuiltinPtr::dynamicCast(_ptr);
assert(bPtr);
_bHasDefault = false;
}
void TypeId::setOptional(int tag)
{
_bRequire = false;
_tag = tag;
}
/*************************************************************************************************/
NamespacePtr Container::createNamespace(const string &id)
{
NamespacePtr np = NamespacePtr::dynamicCast(this);
if(np)
{
g_parse->error("namespace can't be nested!");
}
_ns.push_back(new Namespace(id));
g_parse->currentContextPtr()->addNamespacePtr(_ns.back());
return _ns.back();
}
/*************************************************************************************************/
ParamDeclPtr Operation::createParamDecl(const TypeIdPtr &typeIdPtr, bool v, bool k)
{
_itag++;
if(_itag > 255)
{
g_parse->error("operation '" + _id + "' can't have more than 255 paramters!");
}
if(typeIdPtr->getId() == getId())
{
g_parse->error("paramters name '" + _id + "' conflicts with operation name!");
}
typeIdPtr->setRequire(_itag);
_ps.push_back(new ParamDecl(typeIdPtr, v, k));
return _ps.back();
}
/*************************************************************************************************/
OperationPtr Interface::createOperation(const string &id, const TypePtr &typePtr)
{
for(size_t i = 0; i < _ops.size(); i++)
{
if(_ops[i]->getId() == id)
{
g_parse->error("operation '" + id + "' exists in interface '" + _id + "'");
return NULL;
}
}
_ops.push_back(new Operation(id, typePtr));
return _ops.back();
}
/*************************************************************************************************/
InterfacePtr Namespace::createInterface(const string &id)
{
for(size_t i = 0; i < _is.size(); i++)
{
if(_is[i]->getId() == id)
{
g_parse->error("interface '" + id + "' exists in namespace '" + _id + "'");
return NULL;
}
}
if(id == _id)
{
g_parse->error("interface '" + id + "' must not be same with namespace '" + _id + "'");
return NULL;
}
_is.push_back(new Interface(id));
return _is.back();
}
StructPtr Namespace::createStruct(const string& id)
{
g_parse->checkConflict(_id + "::" + id);
_ss.push_back(new Struct(id, _id + "::" + id));
g_parse->addStructPtr(_ss.back());
return _ss.back();
}
EnumPtr Namespace::createEnum(const string &id)
{
g_parse->checkConflict(_id + "::" + id);
_es.push_back(new Enum(id, _id + "::" + id));
g_parse->addEnumPtr(_es.back());
return _es.back();
}
ConstPtr Namespace::createConst(TypeIdPtr &pPtr, ConstGrammarPtr &cPtr)
{
for(size_t i = 0; i < _cs.size(); i++)
{
if(_cs[i]->getTypeIdPtr()->getId() == pPtr->getId())
{
g_parse->error("const type '" + pPtr->getId() + "' exists in namespace '" + _id + "'");
return NULL;
}
}
g_parse->checkConstValue(pPtr, cPtr->t);
_cs.push_back(new Const(pPtr, cPtr));
return _cs.back();
}
/*************************************************************************************************/
void Struct::addTypeId(const TypeIdPtr &typeIdPtr)
{
StructPtr sp = StructPtr::dynamicCast(typeIdPtr->getTypePtr());
if(sp)
{
if(sp->getSid() == getSid())
{
g_parse->error("struct can't take self as member data");
}
}
for(size_t i = 0; i < _members.size(); i++)
{
if(_members[i]->getId() == typeIdPtr->getId())
{
g_parse->error("data member '" + typeIdPtr->getId() + "' duplicate definition");
}
if(_members[i]->getTag() == typeIdPtr->getTag())
{
g_parse->error("data member '" + typeIdPtr->getId() + "' has equal tag with '" + _members[i]->getId() + "'");
}
if(_members[i]->getTag() > typeIdPtr->getTag())
{
_members.insert(_members.begin() + i, typeIdPtr);
return;
}
}
_members.push_back(typeIdPtr);
}
void Struct::addKey(const string &member)
{
size_t i;
for(i = 0; i < _members.size(); i++)
{
if(_members[i]->getId() == member)
{
break;
}
}
if(i == _members.size())
{
g_parse->error("key member '" + member + "' is not struct member");
}
_key.push_back(member);
}
/***********************************************************************************/
void Enum::addMember(const TypeIdPtr &typeIdPtr)
{
for(size_t i = 0; i < _members.size(); i++)
{
if(_members[i]->getId() == typeIdPtr->getId())
{
g_parse->error("enum member '" + typeIdPtr->getId() + "' duplicate definition");
}
}
_members.push_back(typeIdPtr);
}
/***********************************************************************************/
void Context::addInclude(const string &incl)
{
if(incl == _filename)
{
g_parse->error("can't include self");
}
_includes.push_back(tars::TC_File::excludeFileExt(incl) + ".h");
}

861
third_partly/tarsparse/element.h Normal file
View File

@ -0,0 +1,861 @@
/**
* 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.
*/
#ifndef _ELEMENT_H
#define _ELEMENT_H
#include <iostream>
#include <string>
#include <vector>
#include <algorithm>
#include <iterator>
#include <map>
#include <stack>
#include <sstream>
#include "errno.h"
#include "util/tc_autoptr.h"
using namespace std;
void yyerror(char const *msg);
/**
* 语法元素基类
*/
class GrammarBase : virtual public tars::TC_HandleBase
{
public:
virtual ~GrammarBase(){};
};
typedef tars::TC_AutoPtr<GrammarBase> GrammarBasePtr;
/**
* 解析过程中的字符串
*/
class StringGrammar : public GrammarBase
{
public:
StringGrammar() { }
string v;
};
typedef tars::TC_AutoPtr<StringGrammar> StringGrammarPtr;
/**
* 解析过程中的整形数
*/
class IntergerGrammar : public GrammarBase
{
public:
IntergerGrammar():v(0) { }
long long v;
};
typedef tars::TC_AutoPtr<IntergerGrammar> IntergerGrammarPtr;
/**
* 解析过程中的浮点数
*/
class FloatGrammar : public GrammarBase
{
public:
FloatGrammar():v(0.0f) { }
double v;
};
typedef tars::TC_AutoPtr<FloatGrammar> FloatGrammarPtr;
/**
* 解析过程中的bool类型
*/
class BoolGrammar : public GrammarBase
{
public:
BoolGrammar():v(false) { }
bool v;
};
typedef tars::TC_AutoPtr<BoolGrammar> BoolGrammarPtr;
/**
* 解析过程中的常类型
*/
class ConstGrammar : public GrammarBase
{
public:
ConstGrammar():t(VALUE) { }
enum
{
VALUE,
STRING,
BOOL,
ENUM
};
int t;
string v;
};
typedef tars::TC_AutoPtr<ConstGrammar> ConstGrammarPtr;
///////////////////////////////////////////////
/**
* 类型基类
*/
class Type : virtual public GrammarBase
{
public:
/**
* 构造函数
* @param Type
*/
Type():_size(0),_array(false),_pointer(false){};
//是否简单类型
virtual bool isSimple() const { return false;}
//是否数组类型
virtual bool isArray() const { return _array;}
virtual void setArray(int size) {_array = true;_size = size;}
virtual int getSize() const {return _size;};
//是否指针类型
virtual bool isPointer() const { return _pointer;}
virtual void setPointer(bool b) {_pointer = b;}
public:
int _size;
bool _array;
bool _pointer;
};
typedef tars::TC_AutoPtr<Type> TypePtr;
/**
* 内建类型
*/
class Builtin : public Type
{
public:
enum Kind
{
KindVoid,
KindBool,
KindByte,
KindShort,
KindInt,
KindLong,
KindFloat,
KindDouble,
KindString,
KindVector,
KindMap
};
/**
* 构造函数
* @param kind
*/
Builtin(Kind kind,bool isUnsigned);
/**
* 类型
*
* @return Kind
*/
Kind kind() const;
/**
* 是否是简单类型
*
* @return bool
*/
bool isSimple() const;
/**
* 是否是unsined类型
*
* @return bool
*/
bool isUnsigned() const {return _isUnsigned;}
/**
* 设置是unsined类型
*
* @return bool
*/
void setUnsigned(bool isUnsigned = false) {_isUnsigned = isUnsigned;}
/**
* 缺省值
*
* @return string
*/
string def() const;
/**
* 字符串标示
*/
static const char* builtinTable[];
protected:
Kind _kind;
bool _isUnsigned;
};
typedef tars::TC_AutoPtr<Builtin> BuiltinPtr;
/**
* Vector类型
*/
class Vector : public Type
{
public:
/**
* 构造函数
* @param ptr
*/
Vector(const TypePtr& ptr);
/**
* 获取类型
*
* @return TypePtr&
*/
TypePtr& getTypePtr() {return _ptr;}
protected:
TypePtr _ptr;
};
typedef tars::TC_AutoPtr<Vector> VectorPtr;
/**
* Map类型
*/
class Map : public Type
{
public:
/**
* 构造函数
* @param pleft
* @param pright
*/
Map(const TypePtr& pleft, const TypePtr& pright);
/**
* 获取左类型
*
* @return TypePtr&
*/
TypePtr& getLeftTypePtr() {return _pleft;}
/**
* 获取右类型
*
* @return TypePtr&
*/
TypePtr& getRightTypePtr() {return _pright;}
protected:
TypePtr _pleft;
TypePtr _pright;
};
typedef tars::TC_AutoPtr<Map> MapPtr;
/**
* 变量定义
*/
class TypeId : public GrammarBase
{
public:
/**
* 构造函数
* @param ptr
* @param id
*/
TypeId(const TypePtr& ptr, const string& id);
/**
* 变量名称
*
* @return string
*/
string getId() const { return _id; }
/**
* 变量类型
*
* @return Type*
*/
TypePtr& getTypePtr() { return _ptr;}
/**
* 是否需要该参数
*
* @return bool
*/
bool isRequire() const { return _bRequire; }
/**
* 设置必选字段
* @param tag
*/
void setRequire(int tag);
/**
* 设置可选字段
* 只有基本类型才有缺省值
*/
void setDefault(const string &def);
/**
*
* 只有基本类型才有缺省值
*/
void disableDefault();
/**
* 设置可选字段
* @param tag
*/
void setOptional(int tag);
/**
* tag标识
*/
int getTag() const { return _tag; }
/**
* 缺省值
*
* @return string
*/
string def() const { return _default; }
/**
* 是否有缺省值
*
* @return bool
*/
bool hasDefault() const { return _bHasDefault; }
protected:
TypePtr _ptr;
string _id;
bool _bRequire;
int _tag;
bool _bHasDefault;
string _default;
public:
int _size;
bool _array;
};
typedef tars::TC_AutoPtr<TypeId> TypeIdPtr;
////////////////////////////////////////////////////
//
class Namespace;
typedef tars::TC_AutoPtr<Namespace> NamespacePtr;
/**
* 容器基类, 所有可以包含其他元素的元素都从该基类继承
*/
class Container : virtual public GrammarBase
{
public:
/**
* 构造函数
* @param id
*/
Container(const string &id) : _id(id)
{
}
/**
* 生成名字空间
* @param id
*
* @return NamespacePtr
*/
NamespacePtr createNamespace(const string &id);
/**
* 获取ID
*
* @return string
*/
string getId() const { return _id;}
/**
* 获取所有的名字空间
*
* @return vector<NamespacePtr>&
*/
vector<NamespacePtr> &getAllNamespacePtr() { return _ns; }
protected:
string _id;
vector<NamespacePtr> _ns;
};
typedef tars::TC_AutoPtr<Container> ContainerPtr;
////////////////////////////////////////////////////
//
class Const : public GrammarBase
{
public:
/**
*
* @param tPtr
* @param cPtr
*/
Const(TypeIdPtr &tPtr, ConstGrammarPtr &cPtr) : _tPtr(tPtr), _cPtr(cPtr)
{
}
/**
*
*
* @return TypePtr&
*/
TypeIdPtr &getTypeIdPtr() { return _tPtr; }
/**
*
*
* @return ConstGrammarPtr
*/
ConstGrammarPtr getConstGrammarPtr() { return _cPtr; }
protected:
TypeIdPtr _tPtr;
ConstGrammarPtr _cPtr;
};
typedef tars::TC_AutoPtr<Const> ConstPtr;
/////////////////////////////////////////////////////////
//
class Enum : virtual public Container, virtual public Type
{
public:
Enum(const string &id, const string &sid) : Container(id), _sid(sid)
{
}
/**
* 添加成员变量
* @param ptid
*/
void addMember(const TypeIdPtr &ptr);
/**
* 获取所有的成员变量
*
* @return vector<TypeIdPtr>&
*/
vector<TypeIdPtr>& getAllMemberPtr() {return _members;}
/**
* 获取结构的名称
*
* @return string
*/
string getSid() const { return _sid; }
/**
* 是否是简单类型
*
* @return bool
*/
virtual bool isSimple() const { return true; }
protected:
/**
* 每个变量名称
*/
vector<TypeIdPtr> _members;
/**
* 包含名字的空间的名称
*/
string _sid;
};
typedef tars::TC_AutoPtr<Enum> EnumPtr;
////////////////////////////////////////////////////
//
/**
* 结构
*/
class Struct : virtual public Container, virtual public Type
{
public:
/**
* 构造函数
* @param id
*/
Struct(const string& id, const string &sid) : Container(id), _sid(sid)
{
}
/**
* 添加成员变量
* @param ptid
*/
void addTypeId(const TypeIdPtr &ptr);
/**
* 获取所有的成员变量
*
* @return vector<TypeIdPtr>&
*/
vector<TypeIdPtr>& getAllMemberPtr() {return _members;}
/**
* 获取结构的名称
*
* @return string
*/
string getSid() const { return _sid; }
/**
* 增加小于memeber
* @param member
*/
void addKey(const string &member);
vector<string> getKey() { return _key; }
protected:
vector<TypeIdPtr> _members;
vector<string> _key;
string _sid;
};
typedef tars::TC_AutoPtr<Struct> StructPtr;
////////////////////////////////////////////////////
//
/**
* 参数描述
*/
class ParamDecl : public GrammarBase
{
public:
/**
* 构造
* @param typeIdPtr
* @param v
*/
ParamDecl(const TypeIdPtr &typeIdPtr, bool v, bool k)
: _typeIdPtr(typeIdPtr), _v(v), _k(k)
{
}
/**
* 变量声明
*
* @return TypeIdPtr&
*/
TypeIdPtr& getTypeIdPtr() { return _typeIdPtr; }
/**
* 是否是输出参数
*
* @return bool
*/
bool isOut() const { return _v; }
/**
* 是否是需要路由的字段
*
* @return bool
*/
bool isRouteKey() const { return _k; }
protected:
TypeIdPtr _typeIdPtr;
bool _v;
bool _k;
};
typedef tars::TC_AutoPtr<ParamDecl> ParamDeclPtr;
///////////////////////////////////////////////////////
//
/**
* 操作类
*/
class Operation : public Container
{
public:
/**
* 构造函数
* @param id
* @param typePtr
*/
Operation(const string &id, const TypePtr &typePtr) : Container(id), _itag(0)
{
_retPtr = new TypeId(typePtr, "_ret");
_retPtr->setRequire(_itag);
}
/**
* 生成一个参数
* @param typeIdPtr
* @param v
* @param k
*
* @return ParamDeclPtr
*/
ParamDeclPtr createParamDecl(const TypeIdPtr &typeIdPtr, bool v, bool k);
/**
* 获取返回值类型
*
* @return TypePtr&
*/
TypeIdPtr &getReturnPtr() { return _retPtr; }
/**
* 获取所有参数
*
* @return vector<ParamDeclPtr>&
*/
vector<ParamDeclPtr> &getAllParamDeclPtr() { return _ps; }
protected:
int _itag;
TypeIdPtr _retPtr;
vector<ParamDeclPtr> _ps;
};
typedef tars::TC_AutoPtr<Operation> OperationPtr;
///////////////////////////////////////////////////////
//
/**
* 接口描述
*/
class Interface : public Container
{
public:
/**
* 构造
* @param id
*/
Interface(const string &id) : Container(id)
{
}
/**
* 生成一个操作
* @param id
* @param typePtr
*
* @return OperationPtr
*/
OperationPtr createOperation(const string &id, const TypePtr &typePtr);
/**
* 获取所有操作
*
* @return vector<OperationPtr>&
*/
vector<OperationPtr> &getAllOperationPtr() { return _ops; }
protected:
vector<OperationPtr> _ops;
};
typedef tars::TC_AutoPtr<Interface> InterfacePtr;
/////////////////////////////////////////////////////////
//
/**
* 名字空间
*/
class Namespace : public Container
{
public:
/**
* 构造函数
* @param id
*/
Namespace(const string &id) : Container(id)
{
}
/**
* 生成接口
* @param id
*
* @return InterfacePtr
*/
InterfacePtr createInterface(const string &id);
/**
* 生成结构
* @param id
*
* @return StructPtr
*/
StructPtr createStruct(const string& id);
/**
* 生成枚举类型
* @param id
*
* @return EnumPtr
*/
EnumPtr createEnum(const string &id);
/**
*
* @param pPtr
* @param cPtr
*
* @return ConstPtr
*/
ConstPtr createConst(TypeIdPtr &pPtr, ConstGrammarPtr &cPtr);
/**
* 是否有接口
*
* @return bool
*/
bool hasInterface() const { return !_is.empty(); }
/**
* 获取所有的接口
*
* @return vector<InterfacePtr>&
*/
vector<InterfacePtr> &getAllInterfacePtr() { return _is; }
/**
* 获取所有的结构
*
* @return vector<StructPtr>&
*/
vector<StructPtr> &getAllStructPtr() { return _ss; }
/**
* 生成枚举类型
*
* @return vector<EnumPtr>&
*/
vector<EnumPtr> &getAllEnumPtr() { return _es; }
/**
* 常量类型
*
* @return vector<ConstPtr>&
*/
vector<ConstPtr> &getAllConstPtr() { return _cs; }
protected:
vector<InterfacePtr> _is;
vector<StructPtr> _ss;
vector<EnumPtr> _es;
vector<ConstPtr> _cs;
};
/////////////////////////////////////////////////////////////////
/**
* 上下文
*/
class Context : public tars::TC_HandleBase
{
public:
/**
* 构造函数
*/
Context(const string &file) : _currline(1), _filename(file)
{
}
/**
* 下一行
*/
void nextLine() { _currline++; }
/**
* 目前的行
*
* @return size_t
*/
size_t getCurrLine() { return _currline; }
/**
* 当前文件名
*
* @return string
*/
string getFileName() { return _filename; }
/**
* 添加include的文件
* @param incl
*/
void addInclude(const string &incl);
/**
* 添加属于这个文件的名字空间
* @param c
*/
void addNamespacePtr(const NamespacePtr &c)
{
_namespaces.push_back(c);
}
/**
* 获取includes的文件
*
* @return vector<string>
*/
vector<string> getIncludes() { return _includes; }
/**
* 获取名字空间
*
* @return vector<NamespacePtr>
*/
vector<NamespacePtr> getNamespaces() { return _namespaces; }
protected:
size_t _currline;
string _filename;
vector<string> _includes;
vector<NamespacePtr> _namespaces;
};
typedef tars::TC_AutoPtr<Context> ContextPtr;
#endif

603
third_partly/tarsparse/parse.cpp Normal file
View File

@ -0,0 +1,603 @@
/**
* 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 "parse.h"
#include "tars.tab.hpp"
#include <errno.h>
#include <fstream>
#include <string.h>
#include "util/tc_common.h"
#include "util/tc_file.h"
TarsParsePtr g_parse = new TarsParse();
void yyerror(char const *msg)
{
g_parse->error(msg);
}
TarsParse::TarsParse()
{
_bWithTars = false;
_bUseCurrentPath = false;
_itab = 0;
_bUseCurrentPathFirst = false;
initScanner();
}
void TarsParse::setTars(bool bWithTars)
{
_bWithTars = bWithTars;
}
void TarsParse::setHeader(const string &sHeader)
{
_sHeader = sHeader;
}
void TarsParse::setCurrentPriority(bool bFlag)
{
_bUseCurrentPathFirst = bFlag;
}
string TarsParse::getHeader()
{
if(_sHeader.empty())
return _sHeader;
return _sHeader + "/";
}
void TarsParse::clear()
{
while(!_contexts.empty()) _contexts.pop();
while(!_contains.empty()) _contains.pop();
_vcontexts.clear();
_structs.clear();
_enums.clear();
_namespaces.clear();
}
void TarsParse::parse(const string& sFileName)
{
if (_bUseCurrentPath)
{
std::string sTemp = sFileName;
bool isExist = tars::TC_File::isFileExist(sFileName);
if (!isExist)
{
sTemp = tars::TC_File::extractFileName(sFileName);
}
clear();
_contains.push(new Container(""));
if(!(yyin = fopen(sTemp.c_str(), "r")))
{
error("open file '" + sFileName + "(" + sTemp + ")" + "' error :" + string(strerror(errno)));
}
pushFile(sFileName);
yyparse();
return ;
}
if (_bUseCurrentPathFirst)
{
std::string sTemp = tars::TC_File::extractFileName(sFileName);
bool isExist = tars::TC_File::isFileExist(sTemp);
if (!isExist)
{
sTemp = sFileName;
}
clear();
_contains.push(new Container(""));
if(!(yyin = fopen(sTemp.c_str(), "r")))
{
error("open file '" + sFileName + "(" + sTemp + ")" + "' error :" + string(strerror(errno)));
}
pushFile(sFileName);
yyparse();
return ;
}
{
clear();
_contains.push(new Container(""));
if(!(yyin = fopen(sFileName.c_str(), "r")))
{
error("open file '" + sFileName + "' error :" + string(strerror(errno)));
}
pushFile(sFileName);
yyparse();
}
}
void TarsParse::pushFile(const string &file)
{
ContextPtr c = new Context(file);
_contexts.push(c);
_vcontexts.push_back(c);
}
ContextPtr TarsParse::popFile()
{
ContextPtr c = _contexts.top();
_contexts.pop();
return c;
}
bool TarsParse::getFilePath(const string &s, string &file)
{
if (_bUseCurrentPath)
{
if(s.length() < 2 || s[0] != '\"' || s[s.length()-1] != '\"')
{
error("#include need \"FILENAME\"");
}
file = s.substr(1, s.length() - 2);
std::string sTemp = file;
bool isExist = tars::TC_File::isFileExist(file);
if (!isExist)
{
file = tars::TC_File::extractFileName(file);
}
for(size_t i = 0; i < _vcontexts.size(); i++)
{
if(_vcontexts[i]->getFileName() == file)
{
return false;
}
}
return true;
}
if(_bUseCurrentPathFirst)
{
if(s.length() < 2 || s[0] != '\"' || s[s.length()-1] != '\"')
{
error("#include need \"FILENAME\"");
}
file = s.substr(1, s.length() - 2);
std::string sTemp = tars::TC_File::extractFileName(file);
bool isExist = tars::TC_File::isFileExist(sTemp);
if (isExist)
{
file = sTemp;
}
for(size_t i = 0; i < _vcontexts.size(); i++)
{
if(_vcontexts[i]->getFileName() == file)
{
return false;
}
}
return true;
}
if(s.length() < 2 || s[0] != '\"' || s[s.length()-1] != '\"')
{
error("#include need \"FILENAME\"");
}
file = s.substr(1, s.length() - 2);
if (!tars::TC_File::isFileExist(file))
{
for (size_t i = 0; i < _vIncludePath.size(); i++)
{
if (tars::TC_File::isFileExist(_vIncludePath[i] + "/" + file))
{
file = _vIncludePath[i] + "/" + file;
break;
}
}
}
for(size_t i = 0; i < _vcontexts.size(); i++)
{
if(_vcontexts[i]->getFileName() == file)
{
return false;
}
}
return true;
}
string TarsParse::getCurrFileName()
{
return _contexts.top()->getFileName();
}
void TarsParse::nextLine()
{
_contexts.top()->nextLine();
}
ContextPtr TarsParse::currentContextPtr()
{
return _contexts.top();
}
void TarsParse::error(const string &msg)
{
cerr << _contexts.top()->getFileName() << ": " << _contexts.top()->getCurrLine() << ": error: " << msg << endl;
exit(-1);
}
int TarsParse::checkKeyword(const string& s)
{
std::map<std::string, int>::const_iterator it = _keywordMap.find(s);
if(it != _keywordMap.end())
{
return it->second;
}
if(!_bWithTars)
{
string sPrefix = "tars";
//不能以tars开头
if((s.length() >= sPrefix.length()) && (s.substr(0, sPrefix.length()) == sPrefix))
{
error("identifier can't start with 'tars'");
}
}
return TARS_IDENTIFIER;
}
void TarsParse::initScanner()
{
_keywordMap["void"] = TARS_VOID;
_keywordMap["struct"] = TARS_STRUCT;
_keywordMap["bool"] = TARS_BOOL;
_keywordMap["byte"] = TARS_BYTE;
_keywordMap["short"] = TARS_SHORT;
_keywordMap["int"] = TARS_INT;
_keywordMap["double"] = TARS_DOUBLE;
_keywordMap["float"] = TARS_FLOAT;
_keywordMap["long"] = TARS_LONG;
_keywordMap["string"] = TARS_STRING;
_keywordMap["vector"] = TARS_VECTOR;
_keywordMap["map"] = TARS_MAP;
_keywordMap["key"] = TARS_KEY;
_keywordMap["routekey"] = TARS_ROUTE_KEY;
_keywordMap["module"] = TARS_NAMESPACE;
_keywordMap["interface"]= TARS_INTERFACE;
_keywordMap["out"] = TARS_OUT;
_keywordMap["require"] = TARS_REQUIRE;
_keywordMap["optional"] = TARS_OPTIONAL;
_keywordMap["false"] = TARS_FALSE;
_keywordMap["true"] = TARS_TRUE;
_keywordMap["enum"] = TARS_ENUM;
_keywordMap["const"] = TARS_CONST;
_keywordMap["unsigned"] = TARS_UNSIGNED;
}
string TarsParse::getTab()
{
ostringstream s;
for(int i = 0; i < _itab; i++)
{
s << " ";
}
return s.str();
}
BuiltinPtr TarsParse::createBuiltin(Builtin::Kind kind,bool isUnsigned)
{
return new Builtin(kind,isUnsigned);
}
VectorPtr TarsParse::createVector(const TypePtr &ptr)
{
return new Vector(ptr);
}
MapPtr TarsParse::createMap(const TypePtr &pleft, const TypePtr &pright)
{
return new Map(pleft, pright);
}
void TarsParse::addNamespacePtr(const NamespacePtr &nPtr)
{
_namespaces.push_back(nPtr);
}
NamespacePtr TarsParse::findNamespace(const string &id)
{
for(size_t i = 0; i < _namespaces.size(); i++)
{
if(_namespaces[i]->getId() == id)
{
return _namespaces[i];
}
}
return NULL;
}
NamespacePtr TarsParse::currentNamespace()
{
return _namespaces.back();
}
void TarsParse::addStructPtr(const StructPtr &sPtr)
{
_structs.push_back(sPtr);
}
void TarsParse::addEnumPtr(const EnumPtr &ePtr)
{
_enums.push_back(ePtr);
}
StructPtr TarsParse::findStruct(const string &sid)
{
string ssid = sid;
//在当前namespace中查找
NamespacePtr np = currentNamespace();
if(ssid.find("::") == string::npos)
{
ssid = np->getId() + "::" + ssid;
}
for(size_t i = 0; i < _structs.size(); i++)
{
if(_structs[i]->getSid() == ssid)
{
return _structs[i];
}
}
return NULL;
}
EnumPtr TarsParse::findEnum(const string &sid)
{
string ssid = sid;
//在当前namespace中查找
NamespacePtr np = currentNamespace();
if(ssid.find("::") == string::npos)
{
ssid = np->getId() + "::" + sid;
}
for(size_t i = 0; i < _enums.size(); i++)
{
if(_enums[i]->getSid() == ssid)
{
return _enums[i];
}
}
return NULL;
}
bool TarsParse::checkEnum(const string &idName)
{
for(size_t i = 0; i < _enums.size(); i++)
{
vector<TypeIdPtr> & list = _enums[i]->getAllMemberPtr();
for (size_t j = 0; j < list.size(); j++)
{
if (list[j]->getId() == idName)
{
return true;
}
}
}
return false;
}
void TarsParse::checkConflict(const string &sid)
{
//是否和枚举重名
if(findEnum(sid))
{
error("conflicts with enum '" + sid + "'");
}
//是否和结构重名
if(findStruct(sid))
{
error("conflicts with struct '" + sid + "'");
}
}
TypePtr TarsParse::findUserType(const string &sid)
{
StructPtr sPtr = findStruct(sid);
if(sPtr) return sPtr;
EnumPtr ePtr = findEnum(sid);
if(ePtr) return ePtr;
return NULL;
}
ContainerPtr TarsParse::currentContainer()
{
return _contains.top();
}
void TarsParse::pushContainer(const ContainerPtr &c)
{
_contains.push(c);
NamespacePtr np = NamespacePtr::dynamicCast(c);
if(np)
{
addNamespacePtr(np);
}
}
ContainerPtr TarsParse::popContainer()
{
ContainerPtr c = _contains.top();
_contains.pop();
return c;
}
void TarsParse::checkTag(int i)
{
if(i >= 256)
{
error("struct memeber tag can't beyond 256");
}
if(i < 0)
{
error("struct memeber tag can't less then 0");
}
}
void TarsParse::checkSize(int i)
{
if(i >= 1024*1024)
{
error("struct memeber size can't beyond 1M");
}
if(i < 1)
{
error("struct memeber size can't less than 1");
}
}
void TarsParse::checkArrayVaid(TypePtr &tPtr,int size)
{
checkSize(size);
//只有string/vector可以为array类型
//vector不可以嵌套array类型 例如不允许vector<string:8>:2;
VectorPtr vPtr = VectorPtr::dynamicCast(tPtr);
if(vPtr )
{
if(vPtr->getTypePtr()->isArray())
{
error("fixed array type can not be nested");
}
return;
}
BuiltinPtr bPtr = BuiltinPtr::dynamicCast(tPtr);
if(bPtr && bPtr->kind()== Builtin::KindString)
{
return;
}
error("only string or vector can use fix array");
}
void TarsParse::checkPointerVaid(TypePtr &tPtr)
{
//必须为vector<Byte>类型
VectorPtr vPtr = VectorPtr::dynamicCast(tPtr);
if(vPtr)
{
BuiltinPtr bPtr = BuiltinPtr::dynamicCast(vPtr->getTypePtr());
if( bPtr && bPtr->kind()== Builtin::KindByte)
return;
}
error("only 'byte *' can be pointer type");
}
void TarsParse::checkConstValue(TypeIdPtr &tPtr, int c)
{
//只有内建类型才能有缺省值
BuiltinPtr bPtr = BuiltinPtr::dynamicCast(tPtr->getTypePtr());
EnumPtr ePtr = EnumPtr::dynamicCast(tPtr->getTypePtr());
if(!bPtr && !ePtr)
{
error("only base/enum type can have default value");
}
if (ePtr)
{
if (c != ConstGrammar::VALUE && c != ConstGrammar::ENUM)
{
error("default value of enum only be int or enum_type");
}
return ;
}
int b = bPtr->kind();
if(c == ConstGrammar::VALUE)
{
if(b == Builtin::KindBool)
{
error("default value of bool can only be true or false");
}
if(b == Builtin::KindString)
{
error("default value of string can only be \"string\"");
}
}
else if(c == ConstGrammar::BOOL)
{
if(b != Builtin::KindBool)
{
error("only bool type can be true or false");
}
}
else if(c == ConstGrammar::STRING)
{
if(b != Builtin::KindString)
{
error("only string type can be \"string\"");
}
}
}
string TarsParse::printHeaderRemark()
{
ostringstream s;
s << "// **********************************************************************" << endl;
s << "// This file was generated by a TARS parser!" << endl;
s << "// TARS version " << TARS_VERSION << "." << endl;
s << "// **********************************************************************" << endl;
s << endl;
return s.str();
}

368
third_partly/tarsparse/parse.h Normal file
View File

@ -0,0 +1,368 @@
/**
* 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.
*/
#ifndef __TARS_PARSE_H_
#define __TARS_PARSE_H_
#include <iostream>
#include <string>
#include <vector>
#include <algorithm>
#include <iterator>
#include <map>
#include <stack>
#include <sstream>
#include <sys/types.h>
#include <sys/stat.h>
#include "element.h"
#include "util/tc_common.h"
#include "util/tc_autoptr.h"
using namespace std;
/**
* Tars文件解析类
*
*/
class TarsParse : public tars::TC_HandleBase
{
public:
/**
* 构造函数
*/
TarsParse();
/**
* 是否支持tars开头的标识
* @param bWithTars
*/
void setTars(bool bWithTars);
/**
* 头文件路径
* @param sHeader
*/
void setHeader(const string &sHeader);
/**
* 是否支持优先使用当前tars文件
* @param bWithTars
*/
void setCurrentPriority(bool bFlag);
/**
* 获取头文件路径
*/
string getHeader();
/**
* 解析某一个文件
* @param sFileName
*/
void parse(const string &sFileName);
/**
* 错误提示
* @param msg
*/
void error(const string &msg);
/**
* 检查关键字
* @param s
*
* @return int
*/
int checkKeyword(const string &s);
/**
* 下一行
*/
void nextLine();
/**
* 目前解析的文件名称
*
* @return string
*/
string getCurrFileName();
/**
* tab
*
* @return string
*/
string getTab();
/**
* 增加tab数
*/
void incTab() { _itab++; }
/**
* 减少tab数
*/
void delTab() { _itab--; }
/**
* 解析文件
* @param file
*/
void pushFile(const string &file);
/**
* 弹出解析文件
*/
ContextPtr popFile();
/**
* 获取所有的上下文
*
* @return std::vector<ContextPtr>
*/
std::vector<ContextPtr> getContexts() { return _vcontexts; }
/**
* 获取目前的容器
*
* @return ContainerPtr
*/
ContainerPtr currentContainer();
/**
* push容器
* @param c
*/
void pushContainer(const ContainerPtr &c);
/**
* 目前的上下文
*
* @return ContextPtr
*/
ContextPtr currentContextPtr();
/**
* 弹出容器
*
* @return ContainerPtr
*/
ContainerPtr popContainer();
/**
* 生成Builtin元素
* @param kind
*
* @return BuiltinPtr
*/
BuiltinPtr createBuiltin(Builtin::Kind kind,bool isUnsigned = false);
/**
* 生成Vector元素
* @param ptr
*
* @return VectorPtr
*/
VectorPtr createVector(const TypePtr &ptr);
/**
* 生成Map元素
* @param pleft
* @param pright
*
* @return MapPtr
*/
MapPtr createMap(const TypePtr &pleft, const TypePtr &pright);
/**
* 添加结构元素
* @param sPtr
*/
void addStructPtr(const StructPtr &sPtr);
/**
* 查找结构
* @param id
*
* @return StructPtr
*/
StructPtr findStruct(const string &sid);
/**
* 添加枚举元素
* @param ePtr
*/
void addEnumPtr(const EnumPtr &ePtr);
/**
* 查找结构
* @param id
*
* @return EnumPtr
*/
EnumPtr findEnum(const string &sid);
/**
* 检查冲突
* @param id
*/
void checkConflict(const string &sid);
/**
* 查找自定义类型
* @param sid
*
* @return TypePtr
*/
TypePtr findUserType(const string &sid);
/**
* 查找名字空间
* @param id
*
* @return NamespacePtr
*/
NamespacePtr findNamespace(const string &id);
/**
* 目前的名字空间
*
* @return NamespacePtr
*/
NamespacePtr currentNamespace();
/**
* 检查tag的合法性
* @param i
*/
void checkTag(int i);
/**
* 检查szie的合法性
* @param i
*/
void checkSize(int i);
/**
* 检查array的合法性
* @param i
*/
void checkArrayVaid(TypePtr &tPtr,int size);
/**
* 检查pointer的合法性
* @param i
*/
void checkPointerVaid(TypePtr &tPtr);
/**
* 检查常量类型和值是否一致
* @param c
* @param b
*/
void checkConstValue(TypeIdPtr &tPtr, int b);
/**
* 获取文件名
* @param s
*
* @return string
*/
bool getFilePath(const string &s, string &file);
void setKeyStruct(const StructPtr& key)
{
_key = key;
}
StructPtr getKeyStruct()
{
return _key;
}
/**
* 打印文件开头注释
*/
string printHeaderRemark();
/**
* 判断当前的标识符是否是枚举变量
*/
bool checkEnum(const string &idName);
/**
* 设置include的tars文件全部从当前文件搜寻
*/
void setUseCurrentPath(const bool & bEnable) { _bUseCurrentPath = bEnable; }
/**
* 查找tars文件时,设置include路径
*/
void addIncludePath(const string &include)
{
vector<string> v = tars::TC_Common::sepstr<string>(include, "; ,", false);
_vIncludePath.insert(_vIncludePath.end(), v.begin(), v.end());
}
protected:
/**
* 添加名字空间
* @param nPtr
*/
void addNamespacePtr(const NamespacePtr &nPtr);
/**
* 初始化
*/
void initScanner();
/**
* 清除
*/
void clear();
protected:
bool _bWithTars;
std::map<std::string, int> _keywordMap;
int _itab;
StructPtr _key;
std::stack<ContextPtr> _contexts;
std::stack<ContainerPtr> _contains;
std::vector<ContextPtr> _vcontexts;
std::vector<StructPtr> _structs;
std::vector<EnumPtr> _enums;
std::vector<NamespacePtr> _namespaces;
string _sHeader;
bool _bUseCurrentPath;
bool _bUseCurrentPathFirst;
std::vector<string> _vIncludePath;
};
extern int yyparse();
extern int yylex();
extern FILE *yyin, *yyout;
typedef tars::TC_AutoPtr<TarsParse> TarsParsePtr;
extern TarsParsePtr g_parse;
#endif

199
third_partly/util/include/util/tc_atomic.h Normal file
View File

@ -0,0 +1,199 @@
/**
* 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.
*/
#ifndef __TC_ATOMIC_H
#define __TC_ATOMIC_H
#include <stdint.h>
namespace tars
{
/////////////////////////////////////////////////
/**
* @file tc_atomic.h
* @brief 原子计数类.
*/
__BEGIN_DECLS
#define TARS_LOCK "lock ; "
typedef struct { volatile int counter; } tars_atomic_t;
#define tars_atomic_read(v) ((v)->counter)
#define tars_atomic_set(v,i) (((v)->counter) = (i))
__END_DECLS
/**
* @brief 原子操作类,对int做原子操作
*/
class TC_Atomic
{
public:
/**
* 原子类型
*/
typedef int atomic_type;
/**
* @brief 构造函数,初始化为0
*/
TC_Atomic(atomic_type at = 0)
{
set(at);
}
TC_Atomic& operator++()
{
inc();
return *this;
}
TC_Atomic& operator--()
{
dec();
return *this;
}
operator atomic_type() const
{
return get();
}
TC_Atomic& operator+=(atomic_type n)
{
add(n);
return *this;
}
TC_Atomic& operator-=(atomic_type n)
{
sub(n);
return *this;
}
TC_Atomic& operator=(atomic_type n)
{
set(n);
return *this;
}
/**
* @brief 获取值
*
* @return int
*/
atomic_type get() const { return _value.counter; }
/**
* @brief 添加
* @param i
*
* @return int
*/
atomic_type add(atomic_type i) { return add_and_return(i); }
/**
* @brief 减少
* @param i
*
* @return int
*/
atomic_type sub(atomic_type i) { return add_and_return(-i); }
/**
* @brief 自加1
*
* @return int
*/
atomic_type inc() { return add(1); }
/**
* @brief 自减1
*/
atomic_type dec() { return sub(1); }
/**
* @brief 自加1
*
* @return void
*/
void inc_fast()
{
__asm__ __volatile__(
TARS_LOCK "incl %0"
:"=m" (_value.counter)
:"m" (_value.counter));
}
/**
* @brief 自减1
* Atomically decrements @_value by 1 and returns true if the
* result is 0, or false for all other
*/
bool dec_and_test()
{
unsigned char c;
__asm__ __volatile__(
TARS_LOCK "decl %0; sete %1"
:"=m" (_value.counter), "=qm" (c)
:"m" (_value.counter) : "memory");
return c != 0;
}
/**
* @brief 设置值
*/
atomic_type set(atomic_type i)
{
_value.counter = i;
return i;
}
protected:
/**
* @brief 增加并返回值
*/
int add_and_return(int i)
{
/* Modern 486+ processor */
int __i = i;
__asm__ __volatile__(
TARS_LOCK "xaddl %0, %1;"
:"=r"(i)
:"m"(_value.counter), "0"(i));
return i + __i;
}
protected:
/**
* 值
*/
tars_atomic_t _value;
};
}
#endif

508
third_partly/util/include/util/tc_autoptr.h Normal file
View File

@ -0,0 +1,508 @@
/**
* 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.
*/
#ifndef __TC_AUTOPTR_H
#define __TC_AUTOPTR_H
#include "util/tc_atomic.h"
#include "util/tc_ex.h"
namespace tars
{
///////////////////////////////////////////////////////
/**
* @file tc_autoptr.h
* @brief 智能指针类(智能指针不能相互引用, 否则内存泄漏).
*/
//////////////////////////////////////////////////////
/**
* @brief 空指针异常
*/
struct TC_AutoPtrNull_Exception : public TC_Exception
{
TC_AutoPtrNull_Exception(const string &buffer) : TC_Exception(buffer){};
~TC_AutoPtrNull_Exception() throw(){};
};
/**
* @brief 智能指针基类.
*
* 所有需要智能指针支持的类都需要从该对象继承，
*
* 内部采用引用计数TC_Atomic实现，对象可以放在容器中；
*/
template<class T>
class TC_HandleBaseT
{
public:
/** 原子计数类型*/
typedef T atomic_type;
/**
* @brief 复制.
*
* @return TC_HandleBase&
*/
TC_HandleBaseT& operator=(const TC_HandleBaseT&)
{
return *this;
}
/**
* @brief 增加计数
*/
void incRef() { _atomic.inc_fast(); }
/**
* @brief 减少计数, 当计数==0时, 且需要删除数据时, 释放对象
*/
void decRef()
{
if(_atomic.dec_and_test() && !_bNoDelete)
{
_bNoDelete = true;
delete this;
}
}
/**
* @brief 获取计数.
*
* @return int 计数值
*/
int getRef() const { return _atomic.get(); }
/**
* @brief 设置不自动释放.
*
* @param b 是否自动删除,true or false
*/
void setNoDelete(bool b) { _bNoDelete = b; }
protected:
/**
* @brief 构造函数
*/
TC_HandleBaseT() : _atomic(0), _bNoDelete(false)
{
}
/**
* @brief 拷贝构造
*/
TC_HandleBaseT(const TC_HandleBaseT&) : _atomic(0), _bNoDelete(false)
{
}
/**
* @brief 析够
*/
virtual ~TC_HandleBaseT()
{
}
protected:
/**
* 计数
*/
atomic_type _atomic;
/**
* 是否自动删除
*/
bool _bNoDelete;
};
template<>
inline void TC_HandleBaseT<int>::incRef()
{
//__sync_fetch_and_add(&_atomic,1);
++_atomic;
}
template<>
inline void TC_HandleBaseT<int>::decRef()
{
//int c = __sync_fetch_and_sub(&_atomic, 1);
//if(c == 1 && !_bNoDelete)
if(--_atomic == 0 && !_bNoDelete)
{
_bNoDelete = true;
delete this;
}
}
template<>
inline int TC_HandleBaseT<int>::getRef() const
{
//return __sync_fetch_and_sub(const_cast<volatile int*>(&_atomic), 0);
return _atomic;
}
typedef TC_HandleBaseT<TC_Atomic> TC_HandleBase;
/**
* @brief 智能指针模板类.
*
* 可以放在容器中,且线程安全的智能指针.
*
* 通过它定义智能指针，该智能指针通过引用计数实现，
*
* 可以放在容器中传递.
*
* template<typename T> T必须继承于TC_HandleBase
*/
template<typename T>
class TC_AutoPtr
{
public:
/**
* 元素类型
*/
typedef T element_type;
/**
* @brief 用原生指针初始化, 计数+1.
*
* @param p
*/
TC_AutoPtr(T* p = 0)
{
_ptr = p;
if(_ptr)
{
_ptr->incRef();
}
}
/**
* @brief 用其他智能指针r的原生指针初始化, 计数+1.
*
* @param Y
* @param r
*/
template<typename Y>
TC_AutoPtr(const TC_AutoPtr<Y>& r)
{
_ptr = r._ptr;
if(_ptr)
{
_ptr->incRef();
}
}
/**
* @brief 拷贝构造, 计数+1.
*
* @param r
*/
TC_AutoPtr(const TC_AutoPtr& r)
{
_ptr = r._ptr;
if(_ptr)
{
_ptr->incRef();
}
}
/**
* @brief 析构
*/
~TC_AutoPtr()
{
if(_ptr)
{
_ptr->decRef();
}
}
/**
* @brief 赋值, 普通指针.
*
* @param p
* @return TC_AutoPtr&
*/
TC_AutoPtr& operator=(T* p)
{
if(_ptr != p)
{
if(p)
{
p->incRef();
}
T* ptr = _ptr;
_ptr = p;
if(ptr)
{
ptr->decRef();
}
}
return *this;
}
/**
* @brief 赋值, 其他类型智能指针.
*
* @param Y
* @param r
* @return TC_AutoPtr&
*/
template<typename Y>
TC_AutoPtr& operator=(const TC_AutoPtr<Y>& r)
{
if(_ptr != r._ptr)
{
if(r._ptr)
{
r._ptr->incRef();
}
T* ptr = _ptr;
_ptr = r._ptr;
if(ptr)
{
ptr->decRef();
}
}
return *this;
}
/**
* @brief 赋值, 该类型其他执政指针.
*
* @param r
* @return TC_AutoPtr&
*/
TC_AutoPtr& operator=(const TC_AutoPtr& r)
{
if(_ptr != r._ptr)
{
if(r._ptr)
{
r._ptr->incRef();
}
T* ptr = _ptr;
_ptr = r._ptr;
if(ptr)
{
ptr->decRef();
}
}
return *this;
}
/**
* @brief 将其他类型的智能指针换成当前类型的智能指针.
*
* @param Y
* @param r
* @return TC_AutoPtr
*/
template<class Y>
static TC_AutoPtr dynamicCast(const TC_AutoPtr<Y>& r)
{
return TC_AutoPtr(dynamic_cast<T*>(r._ptr));
}
/**
* @brief 将其他原生类型的指针转换成当前类型的智能指针.
*
* @param Y
* @param p
* @return TC_AutoPtr
*/
template<class Y>
static TC_AutoPtr dynamicCast(Y* p)
{
return TC_AutoPtr(dynamic_cast<T*>(p));
}
/**
* @brief 获取原生指针.
*
* @return T*
*/
T* get() const
{
return _ptr;
}
/**
* @brief 调用.
*
* @return T*
*/
T* operator->() const
{
if(!_ptr)
{
throwNullHandleException();
}
return _ptr;
}
/**
* @brief 引用.
*
* @return T&
*/
T& operator*() const
{
if(!_ptr)
{
throwNullHandleException();
}
return *_ptr;
}
/**
* @brief 是否有效.
*
* @return bool
*/
operator bool() const
{
return _ptr ? true : false;
}
/**
* @brief 交换指针.
*
* @param other
*/
void swap(TC_AutoPtr& other)
{
std::swap(_ptr, other._ptr);
}
protected:
/**
* @brief 抛出异常
*/
void throwNullHandleException() const;
public:
T* _ptr;
};
/**
* @brief 抛出异常.
*
* @param T
* @param file
* @param line
*/
template<typename T> inline void
TC_AutoPtr<T>::throwNullHandleException() const
{
throw TC_AutoPtrNull_Exception("autoptr null handle error");
}
/**
* @brief ==判断.
*
* @param T
* @param U
* @param lhs
* @param rhs
*
* @return bool
*/
template<typename T, typename U>
inline bool operator==(const TC_AutoPtr<T>& lhs, const TC_AutoPtr<U>& rhs)
{
T* l = lhs.get();
U* r = rhs.get();
if(l && r)
{
return *l == *r;
}
else
{
return !l && !r;
}
}
/**
* @brief 不等于判断.
*
* @param T
* @param U
* @param lhs
* @param rhs
*
* @return bool
*/
template<typename T, typename U>
inline bool operator!=(const TC_AutoPtr<T>& lhs, const TC_AutoPtr<U>& rhs)
{
T* l = lhs.get();
U* r = rhs.get();
if(l && r)
{
return *l != *r;
}
else
{
return l || r;
}
}
/**
* @brief 小于判断, 用于放在map等容器中.
*
* @param T
* @param U
* @param lhs
* @param rhs
*
* @return bool
*/
template<typename T, typename U>
inline bool operator<(const TC_AutoPtr<T>& lhs, const TC_AutoPtr<U>& rhs)
{
T* l = lhs.get();
U* r = rhs.get();
if(l && r)
{
return *l < *r;
}
else
{
return !l && r;
}
}
}
#endif

785
third_partly/util/include/util/tc_common.h Normal file
View File

@ -0,0 +1,785 @@
/**
* 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.
*/
#ifndef __TC_COMMON_H
#define __TC_COMMON_H
#ifndef __USE_XOPEN
#define __USE_XOPEN
#endif
#include <time.h>
#include <errno.h>
#include <unistd.h>
#include <stdint.h>
#include <stdlib.h>
#include <fcntl.h>
#include <stdarg.h>
#include <sys/stat.h>
#include <sys/types.h>
#include <cassert>
#include <cstdio>
#include <string>
#include <iostream>
#include <sstream>
#include <stdexcept>
#include <algorithm>
#include <map>
#include <stack>
#include <vector>
using namespace std;
namespace tars
{
/////////////////////////////////////////////////
/**
* @file tc_common.h
* @brief 帮助类,都是静态函数,提供一些常用的函数 .
*
*/
/////////////////////////////////////////////////
/**
* @brief 基础工具类，提供了一些非常基本的函数使用.
*
* 这些函数都是以静态函数提供。 包括以下几种函数:
*
* Trim类函数,大小写转换函数,分隔字符串函数（直接分隔字符串，
*
* 数字等）,时间相关函数,字符串转换函数,二进制字符串互转函数,
*
* 替换字符串函数,Ip匹配函数,判断一个数是否是素数等
*/
class TC_Common
{
public:
/**
* @brief 去掉头部以及尾部的字符或字符串.
*
* @param sStr 输入字符串
* @param s 需要去掉的字符
* @param bChar 如果为true, 则去掉s中每个字符; 如果为false, 则去掉s字符串
* @return string 返回去掉后的字符串
*/
static string trim(const string &sStr, const string &s = " \r\n\t", bool bChar = true);
/**
* @brief 去掉左边的字符或字符串.
*
* @param sStr 输入字符串
* @param s 需要去掉的字符
* @param bChar 如果为true, 则去掉s中每个字符; 如果为false, 则去掉s字符串
* @return string 返回去掉后的字符串
*/
static string trimleft(const string &sStr, const string &s = " \r\n\t", bool bChar = true);
/**
* @brief 去掉右边的字符或字符串.
*
* @param sStr 输入字符串
* @param s 需要去掉的字符
* @param bChar 如果为true, 则去掉s中每个字符; 如果为false, 则去掉s字符串
* @return string 返回去掉后的字符串
*/
static string trimright(const string &sStr, const string &s = " \r\n\t", bool bChar = true);
/**
* @brief 字符串转换成小写.
*
* @param sString 字符串
* @return string 转换后的字符串
*/
static string lower(const string &sString);
/**
* @brief 字符串转换成大写.
*
* @param sString 字符串
* @return string 转换后的大写的字符串
*/
static string upper(const string &sString);
/**
* @brief 字符串是否都是数字的.
*
* @param sString 字符串
* @return bool 是否是数字
*/
static bool isdigit(const string &sInput);
/**
* @brief 字符串转换成时间结构.
*
* @param sString 字符串时间格式
* @param sFormat 格式
* @param stTm 时间结构
* @return 0: 成功, -1:失败
*/
static int str2tm(const string &sString, const string &sFormat, struct tm &stTm);
/**
* @brief GMT格式的时间转化为时间结构
*
* eg.Sat, 06 Feb 2010 09:29:29 GMT, %a, %d %b %Y %H:%M:%S GMT
*
* 可以用mktime换成time_t, 但是注意时区 可以用mktime(&stTm)
*
* - timezone换成本地的秒(time(NULL)值相同) timezone是系统的 ,
*
* 需要extern long timezone;
*
* @param sString GMT格式的时间
* @param stTm 转换后的时间结构
* @return 0: 成功, -1:失败
*/
static int strgmt2tm(const string &sString, struct tm &stTm);
/**
* @brief 时间转换成字符串.
*
* @param stTm 时间结构
* @param sFormat 需要转换的目标格式，默认为紧凑格式
* @return string 转换后的时间字符串
*/
static string tm2str(const struct tm &stTm, const string &sFormat = "%Y%m%d%H%M%S");
/**
* @brief 时间转换成字符串.
*
* @param t 时间结构
* @param sFormat 需要转换的目标格式，默认为紧凑格式
* @return string 转换后的时间字符串
*/
static string tm2str(const time_t &t, const string &sFormat = "%Y%m%d%H%M%S");
/**
* @brief 当前时间转换成紧凑格式字符串
* @param sFormat 格式，默认为紧凑格式
* @return string 转换后的时间字符串
*/
static string now2str(const string &sFormat = "%Y%m%d%H%M%S");
/**
* @brief 时间转换成GMT字符串，GMT格式:Fri, 12 Jan 2001 18:18:18 GMT
* @param stTm 时间结构
* @return string GMT格式的时间字符串
*/
static string tm2GMTstr(const struct tm &stTm);
/**
* @brief 时间转换成GMT字符串，GMT格式:Fri, 12 Jan 2001 18:18:18 GMT
* @param stTm 时间结构
* @return string GMT格式的时间字符串
*/
static string tm2GMTstr(const time_t &t);
/**
* @brief 当前时间转换成GMT字符串，GMT格式:Fri, 12 Jan 2001 18:18:18 GMT
* @return string GMT格式的时间字符串
*/
static string now2GMTstr();
/**
* @brief 当前的日期(年月日)转换成字符串(%Y%m%d).
*
* @return string (%Y%m%d)格式的时间字符串
*/
static string nowdate2str();
/**
* @brief 当前的时间(时分秒)转换成字符串(%H%M%S).
*
* @return string (%H%M%S)格式的时间字符串
*/
static string nowtime2str();
/**
* @brief 获取当前时间的的毫秒数.
*
* @return int64_t 当前时间的的毫秒数
*/
static int64_t now2ms();
/**
* @brief 取出当前时间的微秒.
*
* @return int64_t 取出当前时间的微秒
*/
static int64_t now2us();
/**
* @brief 字符串转化成T型，如果T是数值类型, 如果str为空,则T为0.
*
* @param sStr 要转换的字符串
* @return T T型类型
*/
template<typename T>
static T strto(const string &sStr);
/**
* @brief 字符串转化成T型.
*
* @param sStr 要转换的字符串
* @param sDefault 缺省值
* @return T 转换后的T类型
*/
template<typename T>
static T strto(const string &sStr, const string &sDefault);
/**
* @brief 解析字符串,用分隔符号分隔,保存在vector里
*
* 例子: |a|b||c|
*
* 如果withEmpty=true时, 采用|分隔为:"","a", "b", "", "c", ""
*
* 如果withEmpty=false时, 采用|分隔为:"a", "b", "c"
*
* 如果T类型为int等数值类型, 则分隔的字符串为"", 则强制转化为0
*
* @param sStr 输入字符串
* @param sSep 分隔字符串(每个字符都算为分隔符)
* @param withEmpty true代表空的也算一个元素, false时空的过滤
* @return 解析后的字符vector
*/
template<typename T>
static vector<T> sepstr(const string &sStr, const string &sSep, bool withEmpty = false);
/**
* @brief T型转换成字符串，只要T能够使用ostream对象用<<重载,即可以被该函数支持
* @param t 要转换的数据
* @return 转换后的字符串
*/
template<typename T>
static string tostr(const T &t);
/**
* @brief vector转换成string.
*
* @param t 要转换的vector型的数据
* @return 转换后的字符串
*/
template<typename T>
static string tostr(const vector<T> &t);
/**
* @brief 把map输出为字符串.
*
* @param map<K, V, D, A> 要转换的map对象
* @return string 输出的字符串
*/
template<typename K, typename V, typename D, typename A>
static string tostr(const map<K, V, D, A> &t);
/**
* @brief map输出为字符串.
*
* @param multimap<K, V, D, A> map对象
* @return 输出的字符串
*/
template<typename K, typename V, typename D, typename A>
static string tostr(const multimap<K, V, D, A> &t);
/**
* @brief pair 转化为字符串，保证map等关系容器可以直接用tostr来输出
* @param pair<F, S> pair对象
* @return 输出的字符串
*/
template<typename F, typename S>
static string tostr(const pair<F, S> &itPair);
/**
* @brief container 转换成字符串.
*
* @param iFirst 迭代器
* @param iLast 迭代器
* @param sSep 两个元素之间的分隔符
* @return 转换后的字符串
*/
template <typename InputIter>
static string tostr(InputIter iFirst, InputIter iLast, const string &sSep = "|");
/**
* @brief 二进制数据转换成字符串.
*
* @param buf 二进制buffer
* @param len buffer长度
* @param sSep 分隔符
* @param lines 多少个字节换一行, 默认0表示不换行
* @return 转换后的字符串
*/
static string bin2str(const void *buf, size_t len, const string &sSep = "", size_t lines = 0);
/**
* @brief 二进制数据转换成字符串.
*
* @param sBinData 二进制数据
* @param sSep 分隔符
* @param lines 多少个字节换一行, 默认0表示不换行
* @return 转换后的字符串
*/
static string bin2str(const string &sBinData, const string &sSep = "", size_t lines = 0);
/**
* @brief 字符串转换成二进制.
*
* @param psAsciiData 字符串
* @param sBinData 二进制数据
* @param iBinSize 需要转换的字符串长度
* @return 转换长度，小于等于0则表示失败
*/
static int str2bin(const char *psAsciiData, unsigned char *sBinData, int iBinSize);
/**
* @brief 字符串转换成二进制.
*
* @param sBinData 要转换的字符串
* @param sSep 分隔符
* @param lines 多少个字节换一行, 默认0表示不换行
* @return 转换后的二进制数据
*/
static string str2bin(const string &sBinData, const string &sSep = "", size_t lines = 0);
/**
* @brief 替换字符串.
*
* @param sString 输入字符串
* @param sSrc 原字符串
* @param sDest 目的字符串
* @return string 替换后的字符串
*/
static string replace(const string &sString, const string &sSrc, const string &sDest);
/**
* @brief 批量替换字符串.
*
* @param sString 输入字符串
* @param mSrcDest map<原字符串,目的字符串>
* @return string 替换后的字符串
*/
static string replace(const string &sString, const map<string,string>& mSrcDest);
/**
* @brief 匹配以.分隔的字符串，pat中*则代表通配符，代表非空的任何字符串
* s为空, 返回false ，pat为空, 返回true
* @param s 普通字符串
* @param pat 带*则被匹配的字符串，用来匹配ip地址
* @return 是否匹配成功
*/
static bool matchPeriod(const string& s, const string& pat);
/**
* @brief 匹配以.分隔的字符串.
*
* @param s 普通字符串
* @param pat vector中的每个元素都是带*则被匹配的字符串，用来匹配ip地址
* @return 是否匹配成功
*/
static bool matchPeriod(const string& s, const vector<string>& pat);
/**
* @brief 判断一个数是否为素数.
*
* @param n 需要被判断的数据
* @return true代表是素数，false表示非素数
*/
static bool isPrimeNumber(size_t n);
/**
* @brief daemon
*/
static void daemon();
/**
* @brief 忽略管道异常
*/
static void ignorePipe();
/**
* @brief 将一个string类型转成一个字节 .
*
* @param sWhat 要转换的字符串
* @return char 转换后的字节
*/
static char x2c(const string &sWhat);
/**
* @brief 大小字符串换成字节数，支持K, M, G两种 例如: 1K, 3M, 4G, 4.5M, 2.3G
* @param s 要转换的字符串
* @param iDefaultSize 格式错误时, 缺省的大小
* @return 字节数
*/
static size_t toSize(const string &s, size_t iDefaultSize);
/**
* @brief 生成基于16进制字符的随机串
* @param p 存储随机字符串
* @param len 字符串大小
*/
static void getRandomHexChars(char* p, unsigned int len);
};
namespace p
{
template<typename D>
struct strto1
{
D operator()(const string &sStr)
{
string s = "0";
if(!sStr.empty())
{
s = sStr;
}
istringstream sBuffer(s);
D t;
sBuffer >> t;
return t;
}
};
template<>
struct strto1<char>
{
char operator()(const string &sStr)
{
if(!sStr.empty())
{
return sStr[0];
}
return 0;
}
};
template<>
struct strto1<short>
{
short operator()(const string &sStr)
{
if(!sStr.empty())
{
return atoi(sStr.c_str());
}
return 0;
}
};
template<>
struct strto1<unsigned short>
{
unsigned short operator()(const string &sStr)
{
if(!sStr.empty())
{
return strtoul(sStr.c_str(), NULL, 10);
}
return 0;
}
};
template<>
struct strto1<int>
{
int operator()(const string &sStr)
{
if(!sStr.empty())
{
return atoi(sStr.c_str());
}
return 0;
}
};
template<>
struct strto1<unsigned int>
{
unsigned int operator()(const string &sStr)
{
if(!sStr.empty())
{
return strtoul(sStr.c_str(), NULL, 10);
}
return 0;
}
};
template<>
struct strto1<long>
{
long operator()(const string &sStr)
{
if(!sStr.empty())
{
return atol(sStr.c_str());
}
return 0;
}
};
template<>
struct strto1<long long>
{
long long operator()(const string &sStr)
{
if(!sStr.empty())
{
return atoll(sStr.c_str());
}
return 0;
}
};
template<>
struct strto1<unsigned long>
{
unsigned long operator()(const string &sStr)
{
if(!sStr.empty())
{
return strtoul(sStr.c_str(), NULL, 10);
}
return 0;
}
};
template<>
struct strto1<float>
{
float operator()(const string &sStr)
{
if(!sStr.empty())
{
return atof(sStr.c_str());
}
return 0;
}
};
template<>
struct strto1<double>
{
double operator()(const string &sStr)
{
if(!sStr.empty())
{
return atof(sStr.c_str());
}
return 0;
}
};
template<typename D>
struct strto2
{
D operator()(const string &sStr)
{
istringstream sBuffer(sStr);
D t;
sBuffer >> t;
return t;
}
};
template<>
struct strto2<string>
{
string operator()(const string &sStr)
{
return sStr;
}
};
}
template<typename T>
T TC_Common::strto(const string &sStr)
{
using strto_type = typename std::conditional<std::is_arithmetic<T>::value, p::strto1<T>, p::strto2<T>>::type;
return strto_type()(sStr);
}
template<typename T>
T TC_Common::strto(const string &sStr, const string &sDefault)
{
string s;
if(!sStr.empty())
{
s = sStr;
}
else
{
s = sDefault;
}
return strto<T>(s);
}
template<typename T>
vector<T> TC_Common::sepstr(const string &sStr, const string &sSep, bool withEmpty)
{
vector<T> vt;
string::size_type pos = 0;
string::size_type pos1 = 0;
while(true)
{
string s;
pos1 = sStr.find_first_of(sSep, pos);
if(pos1 == string::npos)
{
if(pos + 1 <= sStr.length())
{
s = sStr.substr(pos);
}
}
else if(pos1 == pos)
{
s = "";
}
else
{
s = sStr.substr(pos, pos1 - pos);
pos = pos1;
}
if(withEmpty)
{
vt.push_back(strto<T>(s));
}
else
{
if(!s.empty())
{
T tmp = strto<T>(s);
vt.push_back(tmp);
}
}
if(pos1 == string::npos)
{
break;
}
pos++;
}
return vt;
}
template<typename T>
string TC_Common::tostr(const T &t)
{
ostringstream sBuffer;
sBuffer << t;
return sBuffer.str();
}
template<typename T>
string TC_Common::tostr(const vector<T> &t)
{
string s;
for(size_t i = 0; i < t.size(); i++)
{
s += tostr(t[i]);
s += " ";
}
return s;
}
template<typename K, typename V, typename D, typename A>
string TC_Common::tostr(const map<K, V, D, A> &t)
{
string sBuffer;
typename map<K, V, D, A>::const_iterator it = t.begin();
while(it != t.end())
{
sBuffer += " [";
sBuffer += tostr(it->first);
sBuffer += "]=[";
sBuffer += tostr(it->second);
sBuffer += "] ";
++it;
}
return sBuffer;
}
template<typename K, typename V, typename D, typename A>
string TC_Common::tostr(const multimap<K, V, D, A> &t)
{
string sBuffer;
typename multimap<K, V, D, A>::const_iterator it = t.begin();
while(it != t.end())
{
sBuffer += " [";
sBuffer += tostr(it->first);
sBuffer += "]=[";
sBuffer += tostr(it->second);
sBuffer += "] ";
++it;
}
return sBuffer;
}
template<typename F, typename S>
string TC_Common::tostr(const pair<F, S> &itPair)
{
string sBuffer;
sBuffer += "[";
sBuffer += tostr(itPair.first);
sBuffer += "]=[";
sBuffer += tostr(itPair.second);
sBuffer += "]";
return sBuffer;
}
template <typename InputIter>
string TC_Common::tostr(InputIter iFirst, InputIter iLast, const string &sSep)
{
string sBuffer;
InputIter it = iFirst;
while(it != iLast)
{
sBuffer += tostr(*it);
++it;
if(it != iLast)
{
sBuffer += sSep;
}
else
{
break;
}
}
return sBuffer;
}
}
#endif

143
third_partly/util/include/util/tc_encoder.h Normal file
View File

@ -0,0 +1,143 @@
/**
* 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.
*/
#ifndef __TC_ENCODER_H_
#define __TC_ENCODER_H_
#include <vector>
#include "util/tc_ex.h"
namespace tars
{
/////////////////////////////////////////////////
/**
* @file tc_encoder.h
* @brief 转码类
*
*
*/
/////////////////////////////////////////////////
/**
* @brief 转码异常类
*/
struct TC_Encoder_Exception : public TC_Exception
{
TC_Encoder_Exception(const string &buffer) : TC_Exception(buffer){};
TC_Encoder_Exception(const string &buffer, int err) : TC_Exception(buffer, err){};
~TC_Encoder_Exception() throw(){};
};
/**
* @brief 该类提供基本常用编码的转换.
*
* Gbk到utf8之间的相互转码函数，通过静态函数提供.
*
* 1：GBK ==> UTF8的转换
*
* 2：UTF8 ==> GBK的转换
*/
class TC_Encoder
{
public:
/**
* @brief gbk 转换到 utf8.
*
* @param sOut 输出buffer
* @param iMaxOutLen 输出buffer最大的长度/sOut的长度
* @param sIn 输入buffer
* @param iInLen 输入buffer长度
* @throws TC_Encoder_Exception
* @return
*/
static void gbk2utf8(char *sOut, int &iMaxOutLen, const char *sIn, int iInLen);
/**
* @brief gbk 转换到 utf8.
*
* @param sIn 输入buffer*
* @throws TC_Encoder_Exception
* @return 转换后的utf8编码
*/
static string gbk2utf8(const string &sIn);
/**
* @brief gbk 转换到 utf8.
*
* @param sIn 输入buffer
* @param vtStr 输出gbk的vector
* @throws TC_Encoder_Exception
* @return
*/
static void gbk2utf8(const string &sIn, vector<string> &vtStr);
/**
* @brief utf8 转换到 gbk.
*
* @param sOut 输出buffer
* @param iMaxOutLen 输出buffer最大的长度/sOut的长度
* @param sIn 输入buffer
* @param iInLen 输入buffer长度
* @throws TC_Encoder_Exception
* @return
*/
static void utf82gbk(char *sOut, int &iMaxOutLen, const char *sIn, int iInLen);
/**
* @brief utf8 转换到 gbk.
*
* @param sIn 输入buffer
* @throws TC_Encoder_Exception
* @return 转换后的gbk编码
*/
static string utf82gbk(const string &sIn);
/**
* @brief 将string的\n替换掉,转义字符串中的某个字符
*
* 缺省:\n 转换为 \r\0; \r转换为\,
*
* 主要用于将string记录在一行，通常用于写bin-log的地方;
* @param str 待转换字符串
* @param f 需要转义的字符
* @param t 转义后的字符
* @param u 借用的转义符
* @return str 转换后的字符串
*/
static string transTo(const string& str, char f = '\n', char t = '\r', char u = '\0');
/**
* @brief 从替换的数据恢复源数据,将 transTo 的字符串还原，
*
* 缺省:\r\0 还原为\n，\r\r还原为,
*
* 主要用于将string记录在一行，通常用于写bin-log的地方
* @param str 待还原的字符串(必须是transTo后得到的字符串)
* @param f 被转义的字符
* @param t 转义后的字符
* @param u 借用的转义符
* @return str 还原后的字符串
*/
static string transFrom(const string& str, char f = '\n', char t = '\r', char u = '\0');
};
}
#endif

94
third_partly/util/include/util/tc_ex.h Normal file
View File

@ -0,0 +1,94 @@
/**
* 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.
*/
#ifndef __TC_EX_H
#define __TC_EX_H
#include <stdexcept>
using namespace std;
namespace tars
{
/////////////////////////////////////////////////
/**
* @file tc_ex.h
* @brief 异常类
*/
/////////////////////////////////////////////////
/**
* @brief 异常类.
*/
class TC_Exception : public exception
{
public:
/**
* @brief 构造函数，提供了一个可以传入errno的构造函数，
*
* 异常抛出时直接获取的错误信息
*
* @param buffer 异常的告警信息
*/
explicit TC_Exception(const string &buffer);
/**
* @brief 构造函数,提供了一个可以传入errno的构造函数，
*
* 异常抛出时直接获取的错误信息
*
* @param buffer 异常的告警信息
* @param err 错误码, 可用strerror获取错误信息
*/
TC_Exception(const string &buffer, int err);
/**
* @brief 析够数函
*/
virtual ~TC_Exception() throw();
/**
* @brief 错误信息.
*
* @return const char*
*/
virtual const char* what() const throw();
/**
* @brief 获取错误码
*
* @return 成功获取返回0
*/
int getErrCode() { return _code; }
private:
void getBacktrace();
private:
/**
* 异常的相关信息
*/
string _buffer;
/**
* 错误码
*/
int _code;
};
}
#endif

289
third_partly/util/include/util/tc_file.h Normal file
View File

@ -0,0 +1,289 @@
/**
* 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.
*/
#ifndef __TC_FILE_H_
#define __TC_FILE_H_
#include <iostream>
#include <fstream>
#include <unistd.h>
#include <sys/stat.h>
#include <sys/types.h>
#include <dirent.h>
#include <fnmatch.h>
#include "util/tc_ex.h"
#include "util/tc_common.h"
namespace tars
{
/////////////////////////////////////////////////
/**
* @file tc_file.h
* @brief 文件处理类.
*
*/
/////////////////////////////////////////////////
/**
* @brief 文件异常类.
*
*/
struct TC_File_Exception : public TC_Exception
{
TC_File_Exception(const string &buffer) : TC_Exception(buffer){};
TC_File_Exception(const string &buffer, int err) : TC_Exception(buffer, err){};
~TC_File_Exception() throw(){};
};
/**
* @brief 常用文件名处理函数.
*
*/
class TC_File
{
public:
/**
* @brief 获取文件大小, 如果文件不存在, 则返回-1.
*
* @param sFullFileName 文件全路径(所在目录和文件名)
* @return ofstream::pos_type类型文件大小
*/
static ifstream::pos_type getFileSize(const string &sFullFileName);
/**
* @brief 判断是否为绝对路径, 忽略空格以'/'开头.
*
* @param sFullFileName 文件全路径(所在目录和文件名)
* @return ture是绝对路径，false代表非绝对路径
*/
static bool isAbsolute(const string &sFullFileName);
/**
* @brief 判断给定路径的文件是否存在.
* 如果文件是符号连接,则以符号连接判断而不是以符号连接指向的文件判断
* @param sFullFileName 文件全路径
* @param iFileType 文件类型, 缺省S_IFREG
* @return true代表存在，fals代表不存在
*/
static bool isFileExist(const string &sFullFileName, mode_t iFileType = S_IFREG);
/**
* @brief 判断给定路径的文件是否存在.
* 注意: 如果文件是符号连接,则以符号连接指向的文件判断
* @param sFullFileName 文件全路径
* @param iFileType 文件类型, 缺省S_IFREG
* @return true-存在，fals-不存在
*/
static bool isFileExistEx(const string &sFullFileName, mode_t iFileType = S_IFREG);
/**
* @brief 规则化目录名称, 把一些不用的去掉, 例如./等.
*
* @param path 目录名称
* @return 规范后的目录名称
*/
static string simplifyDirectory(const string& path);
/**
* @brief 创建目录, 如果目录已经存在, 则也返回成功.
*
* @param sFullPath 要创建的目录名称
* @param iFlag 权限, 默认 S_IRUSR | S_IWUSR | S_IXUSR | S_IRGRP| S_IXGRP | S_IROTH | S_IXOTH
* @return bool true-创建成功 ，false-创建失败
*/
static bool makeDir(const string &sDirectoryPath, mode_t iFlag = S_IRUSR | S_IWUSR | S_IXUSR | S_IRGRP | S_IXGRP | S_IROTH | S_IXOTH);
/**
*@brief 循环创建目录, 如果目录已经存在, 则也返回成功.
*
* @param sFullPath 要创建的目录名称
* @param iFlag 权限, 默认 S_IRUSR | S_IWUSR | S_IXUSR | S_IRGRP | S_IXGRP | S_IROTH | S_IXOTH
* @return true-创建成功，false-创建失败
*/
static bool makeDirRecursive(const string &sDirectoryPath, mode_t iFlag = S_IRUSR | S_IWUSR | S_IXUSR | S_IRGRP | S_IXGRP | S_IROTH | S_IXOTH);
/**
* @brief 设置文件是否可执行.
*
* @param sFullFileName 文件全路径
* @param canExecutable true表示可执行, false代表不可之行
* @return 成功返回0, 其他失败
*/
static int setExecutable(const string &sFullFileName, bool canExecutable);
/**
* @brief 判断文件是否可执行.
*
* @param sFullFileName 文件全路径
* @return true-可执行, false-不可执行
*/
static bool canExecutable(const string &sFullFileName);
/**
* @brief 删除一个文件或目录.
*
* @param sFullFileName 文件或者目录的全路径
* @param bRecursive 如果是目录是否递归删除
* @return 0-成功，失败可以通过errno查看失败的原因
*/
static int removeFile(const string &sFullFileName, bool bRecursive);
/**
* @brief 读取文件到string
* 文件存在则返回文件数据，不存在或者读取文件错误的时候, 返回为空
* @param sFullFileName 文件名称
* @return 文件数据
*/
static string load2str(const string &sFullFileName);
/**
* @brief 读取文件到vector<char>
* 文件存在则返回文件数据，不存在或者读取文件错误的时候, 返回为空
* @param sFullFileName 文件名称
*/
static void load2str(const string &sFullFileName, vector<char> &buffer);
/**
* @brief 写文件.
*
* @param sFullFileName 文件名称
* @param sFileData 文件内容
* @return
*/
static void save2file(const string &sFullFileName, const string &sFileData);
/**
* @brief 写文件.
*
* @param sFullFileName 文件名
* @param sFileData 数据指针
* @param length 写入长度
* @return 0-成功,-1-失败
*/
static int save2file(const string &sFullFileName, const char *sFileData, size_t length);
/**
* @brief 获取前当可执行文件路径.
*
* @return string 可执行文件的路径全名称
*/
static string getExePath();
/**
* @brief 提取文件名称
*从一个完全文件名中去掉路径，例如:/usr/local/temp.gif获取temp.gif
*@param sFullFileName 文件的完全名称
*@return string 提取后的文件名称
*/
static string extractFileName(const string &sFullFileName);
/**
* @brief 从一个完全文件名中提取文件的路径.
*
* 例如1: "/usr/local/temp.gif" 获取"/usr/local/"
* 例如2: "temp.gif" 获取 "./"
* @param sFullFileName 文件的完全名称
* @return 提取后的文件路径
*/
static string extractFilePath(const string &sFullFileName);
/**
* @brief 提取文件扩展名.
*
* 例如1: "/usr/local/temp.gif" 获取"gif"
* 例如2: "temp.gif" 获取"gif"
*@param sFullFileName 文件名称
*@return 文件扩展名
*/
static string extractFileExt(const string &sFullFileName);
/**
* @brief 提取文件名称,去掉扩展名.
* 例如1: "/usr/local/temp.gif" 获取"/usr/local/temp"
* 例如2: "temp.gif" 获取"temp"
* @param sFullFileName 文件名称
* @return 去掉扩展名的文件名称
*/
static string excludeFileExt(const string &sFullFileName);
/**
* @brief 替换文件扩展名
*
* 改变文件类型，如果无扩展名,则加上扩展名 =?1:
* 例如1："/usr/temp.gif" 替 换 "jpg" 得到"/usr/temp.jpg"
* 例如2: "/usr/local/temp" 替 换 "jpg" 得到"/usr/local/temp.jpg"
* @param sFullFileName 文件名称
* @param sExt 扩展名
* @return 替换扩展名后的文件名
*/
static string replaceFileExt(const string &sFullFileName, const string &sExt);
/**
* @brief 从一个url中获取完全文件名.
*
* 获取除http://外,第一个'/'后面的所有字符
* 例如1:http://www.qq.com/tmp/temp.gif 获取tmp/temp.gif
* 例如2:www.qq.com/tmp/temp.gif 获取tmp/temp.gif
* 例如3:/tmp/temp.gif 获取tmp/temp.gif
* @param sUrl url字符串
* @return 文件名称
*/
static string extractUrlFilePath(const string &sUrl);
/**
* @brief 遍历文件时确定是否选择.
*
* @return 1-选择, 0-不选择
*/
typedef int (*FILE_SELECT)(const dirent *);
/**
* @brief 扫描一个目录.
*
* @param sFilePath 需要扫描的路径
* @param vtMatchFiles 返回的文件名矢量表
* @param f 匹配函数,为NULL表示所有文件都获取
* @param iMaxSize 最大文件个数,iMaxSize <=0时,返回所有匹配文件
* @return 文件个数
*/
static size_t scanDir(const string &sFilePath, vector<string> &vtMatchFiles, FILE_SELECT f = NULL, int iMaxSize = 0);
/**
* @brief 遍历目录, 获取目录下面的所有文件和子目录.
*
* @param path 需要遍历的路径
* @param files 目标路径下面所有文件
* @param bRecursive 是否递归子目录
*
**/
static void listDirectory(const string &path, vector<string> &files, bool bRecursive);
/**
* @brief 复制文件或目录.
* 将文件或者目录从sExistFile复制到sNewFile
* @param sExistFile 复制的文件或者目录源路径
* @param sNewFile 复制的文件或者目录目标路径
* @param bRemove 是否先删除sNewFile再copy ，防止Textfile busy导致复制失败
* @return
*/
static void copyFile(const string &sExistFile, const string &sNewFile,bool bRemove = false);
};
}
#endif // TC_FILE_H

111
third_partly/util/include/util/tc_option.h Normal file
View File

@ -0,0 +1,111 @@
/**
* 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.
*/
#ifndef __TC_OPTION_H
#define __TC_OPTION_H
#include <map>
#include <vector>
#include <string>
#include <sstream>
using namespace std;
namespace tars
{
/////////////////////////////////////////////////
/**
* @file tc_option.h
* @brief 命令行参数解析类.
*
*/
/////////////////////////////////////////////////
/**
* @brief 命令解析类，通常用于解析命令行参数
*
* 支持以下形式的参数: ./main.exe --name=value --with abc def
*/
class TC_Option
{
public:
/**
* @brief 构造函数
*/
TC_Option(){};
/**
* @brief 解码.
*
* @param argc 参数个数
* @param argv 参数数组
*
*/
void decode(int argc, char *argv[]);
/**
* @brief 是否存在某个--标识的参数.
*
* @param sName 要判断的标识
* @return bool 存在返回true，否则返回false
*/
bool hasParam(const string &sName);
/**
* @brief 获取某个--表示的参数，如果参数不存在或者参数值为空 ,
* 都返回""
* @param sName 标识
* @return string 标识的参数值
*/
string getValue(const string &sName);
/**
* @brief 获取所有--标识的参数.
*
* @return map<string,string> map类型的标识和参数值的对应关系
*/
map<string, string>& getMulti();
/**
* @brief 获取所有普通的参数, 例子中的abc,
* def，参数按照顺序在vector中
* @return vector<string> 顺序存放参数的vector
*/
vector<string>& getSingle();
protected:
/**
* @brief 解析字符串，取出标识和其对应的参数值，
* 对型如--name=value 的字符串进行解析，取出name和vaue
* @param s 要解析的字符串
*/
void parse(const string &s);
protected:
/**
*存放标识和其对应参数的对应关系，例如：对于--name=value，存放name和value
*/
map<string, string> _mParam;
/**
*存放普通参数的vetor
*/
vector<string> _vSingle;
};
}
#endif

757
third_partly/util/src/tc_common.cpp Normal file
View File

@ -0,0 +1,757 @@
/**
* 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 "util/tc_common.h"
#include <signal.h>
#include <sys/time.h>
#include <string.h>
#include <cmath>
namespace tars
{
template <>
string TC_Common::tostr<bool>(const bool &t)
{
char buf[2];
buf[0] = t ? '1' : '0';
buf[1] = '\0';
return string(buf);
}
template <>
string TC_Common::tostr<char>(const char &t)
{
char buf[2];
snprintf(buf, 2, "%c", t);
return string(buf);
}
template <>
string TC_Common::tostr<unsigned char>(const unsigned char &t)
{
char buf[2];
snprintf(buf, 2, "%c", t);
return string(buf);
}
template <>
string TC_Common::tostr<short>(const short &t)
{
char buf[16];
snprintf(buf, 16, "%d", t);
return string(buf);
}
template <>
string TC_Common::tostr<unsigned short>(const unsigned short &t)
{
char buf[16];
snprintf(buf, 16, "%u", t);
return string(buf);
}
template <>
string TC_Common::tostr<int>(const int &t)
{
char buf[16];
snprintf(buf, 16, "%d", t);
return string(buf);
}
template <>
string TC_Common::tostr<unsigned int>(const unsigned int &t)
{
char buf[16];
snprintf(buf, 16, "%u", t);
return string(buf);
}
template <>
string TC_Common::tostr<long>(const long &t)
{
char buf[32];
snprintf(buf, 32, "%ld", t);
return string(buf);
}
template <>
string TC_Common::tostr<long long>(const long long &t)
{
char buf[32];
snprintf(buf, 32, "%lld", t);
return string(buf);
}
template <>
string TC_Common::tostr<unsigned long>(const unsigned long &t)
{
char buf[32];
snprintf(buf, 32, "%lu", t);
return string(buf);
}
template <>
string TC_Common::tostr<float>(const float &t)
{
char buf[32];
snprintf(buf, 32, "%.5f", t);
string s(buf);
//去掉无效0, eg. 1.0300 -> 1.03;1.00 -> 1
bool bFlag = false;
int pos = int(s.size() - 1);
for(; pos > 0; --pos)
{
if(s[pos] == '0')
{
bFlag = true;
if(s[pos-1] == '.')
{
//-2为了去掉"."号
pos -= 2;
break;
}
}
else
{
break;
}
}
if(bFlag)
s = s.substr(0, pos+1);
return s;
}
template <>
string TC_Common::tostr<double>(const double &t)
{
char buf[32];
snprintf(buf, 32, "%.5f", t);
string s(buf);
//去掉无效0, eg. 1.0300 -> 1.03;1.00 -> 1
bool bFlag = false;
int pos = int(s.size() - 1);
for(; pos > 0; --pos)
{
if(s[pos] == '0')
{
bFlag = true;
if(s[pos-1] == '.')
{
//-2为了去掉"."号
pos -= 2;
break;
}
}
else
{
break;
}
}
if(bFlag)
s = s.substr(0, pos+1);
return s;
}
template <>
string TC_Common::tostr<long double>(const long double &t)
{
char buf[32];
snprintf(buf, 32, "%Lf", t);
string s(buf);
//去掉无效0, eg. 1.0300 -> 1.03;1.00 -> 1
bool bFlag = false;
int pos = int(s.size() - 1);
for(; pos > 0; --pos)
{
if(s[pos] == '0')
{
bFlag = true;
if(s[pos-1] == '.')
{
//-2为了去掉"."号
pos -= 2;
break;
}
}
else
{
break;
}
}
if(bFlag)
s = s.substr(0, pos+1);
return s;
}
template <>
string TC_Common::tostr<std::string>(const std::string &t)
{
return t;
}
string TC_Common::trim(const string &sStr, const string &s, bool bChar)
{
if(sStr.empty())
{
return sStr;
}
/**
* 将完全与s相同的字符串去掉
*/
if(!bChar)
{
return trimright(trimleft(sStr, s, false), s, false);
}
return trimright(trimleft(sStr, s, true), s, true);
}
string TC_Common::trimleft(const string &sStr, const string &s, bool bChar)
{
if(sStr.empty())
{
return sStr;
}
/**
* 去掉sStr左边的字符串s
*/
if(!bChar)
{
if(sStr.length() < s.length())
{
return sStr;
}
if(sStr.compare(0, s.length(), s) == 0)
{
return sStr.substr(s.length());
}
return sStr;
}
/**
* 去掉sStr左边的 字符串s中的字符
*/
string::size_type pos = 0;
while(pos < sStr.length())
{
if(s.find_first_of(sStr[pos]) == string::npos)
{
break;
}
pos++;
}
if(pos == 0) return sStr;
return sStr.substr(pos);
}
string TC_Common::trimright(const string &sStr, const string &s, bool bChar)
{
if(sStr.empty())
{
return sStr;
}
/**
* 去掉sStr右边的字符串s
*/
if(!bChar)
{
if(sStr.length() < s.length())
{
return sStr;
}
if(sStr.compare(sStr.length() - s.length(), s.length(), s) == 0)
{
return sStr.substr(0, sStr.length() - s.length());
}
return sStr;
}
/**
* 去掉sStr右边的 字符串s中的字符
*/
string::size_type pos = sStr.length();
while(pos != 0)
{
if(s.find_first_of(sStr[pos-1]) == string::npos)
{
break;
}
pos--;
}
if(pos == sStr.length()) return sStr;
return sStr.substr(0, pos);
}
string TC_Common::lower(const string &s)
{
string sString = s;
for (string::iterator iter = sString.begin(); iter != sString.end(); ++iter)
{
*iter = tolower(*iter);
}
return sString;
}
string TC_Common::upper(const string &s)
{
string sString = s;
for (string::iterator iter = sString.begin(); iter != sString.end(); ++iter)
{
*iter = toupper(*iter);
}
return sString;
}
bool TC_Common::isdigit(const string &sInput)
{
string::const_iterator iter = sInput.begin();
if(sInput.empty())
{
return false;
}
while(iter != sInput.end())
{
if (!::isdigit(*iter))
{
return false;
}
++iter;
}
return true;
}
int TC_Common::str2tm(const string &sString, const string &sFormat, struct tm &stTm)
{
char *p = strptime(sString.c_str(), sFormat.c_str(), &stTm);
return (p != NULL) ? 0 : -1;
}
int TC_Common::strgmt2tm(const string &sString, struct tm &stTm)
{
return str2tm(sString, "%a, %d %b %Y %H:%M:%S GMT", stTm);
}
string TC_Common::tm2str(const struct tm &stTm, const string &sFormat)
{
char sTimeString[255] = "\0";
strftime(sTimeString, sizeof(sTimeString), sFormat.c_str(), &stTm);
return string(sTimeString);
}
string TC_Common::tm2str(const time_t &t, const string &sFormat)
{
struct tm tt;
localtime_r(&t, &tt);
return tm2str(tt, sFormat);
}
string TC_Common::now2str(const string &sFormat)
{
time_t t = time(NULL);
return tm2str(t, sFormat.c_str());
}
string TC_Common::now2GMTstr()
{
time_t t = time(NULL);
return tm2GMTstr(t);
}
string TC_Common::tm2GMTstr(const time_t &t)
{
struct tm tt;
gmtime_r(&t, &tt);
return tm2str(tt, "%a, %d %b %Y %H:%M:%S GMT");
}
string TC_Common::tm2GMTstr(const struct tm &stTm)
{
return tm2str(stTm, "%a, %d %b %Y %H:%M:%S GMT");
}
string TC_Common::nowdate2str()
{
return now2str("%Y%m%d");
}
string TC_Common::nowtime2str()
{
return now2str("%H%M%S");
}
int64_t TC_Common::now2ms()
{
struct timeval tv;
gettimeofday(&tv, 0);
return tv.tv_sec * (int64_t)1000 + tv.tv_usec/1000;
}
int64_t TC_Common::now2us()
{
struct timeval tv;
gettimeofday(&tv, 0);
return tv.tv_sec * (int64_t)1000000 + tv.tv_usec;
}
//参照phorix的优化
static char c_b2s[256][4]={"00","01","02","03","04","05","06","07","08","09","0a","0b","0c","0d","0e","0f","10","11","12","13","14","15","16","17","18","19","1a","1b","1c","1d","1e","1f","20","21","22","23","24","25","26","27","28","29","2a","2b","2c","2d","2e","2f","30","31","32","33","34","35","36","37","38","39","3a","3b","3c","3d","3e","3f","40","41","42","43","44","45","46","47","48","49","4a","4b","4c","4d","4e","4f","50","51","52","53","54","55","56","57","58","59","5a","5b","5c","5d","5e","5f","60","61","62","63","64","65","66","67","68","69","6a","6b","6c","6d","6e","6f","70","71","72","73","74","75","76","77","78","79","7a","7b","7c","7d","7e","7f","80","81","82","83","84","85","86","87","88","89","8a","8b","8c","8d","8e","8f","90","91","92","93","94","95","96","97","98","99","9a","9b","9c","9d","9e","9f","a0","a1","a2","a3","a4","a5","a6","a7","a8","a9","aa","ab","ac","ad","ae","af","b0","b1","b2","b3","b4","b5","b6","b7","b8","b9","ba","bb","bc","bd","be","bf","c0","c1","c2","c3","c4","c5","c6","c7","c8","c9","ca","cb","cc","cd","ce","cf","d0","d1","d2","d3","d4","d5","d6","d7","d8","d9","da","db","dc","dd","de","df","e0","e1","e2","e3","e4","e5","e6","e7","e8","e9","ea","eb","ec","ed","ee","ef","f0","f1","f2","f3","f4","f5","f6","f7","f8","f9","fa","fb","fc","fd","fe","ff"};
string TC_Common::bin2str(const void *buf, size_t len, const string &sSep, size_t lines)
{
if(buf == NULL || len <=0 )
{
return "";
}
string sOut;
const unsigned char *p = (const unsigned char *) buf;
for (size_t i = 0; i < len; ++i, ++p)
{
sOut += c_b2s[*p][0];
sOut += c_b2s[*p][1];
sOut += sSep;
//换行
if ((lines != 0) && ((i+1) % lines == 0))
{
sOut += "\n";
}
}
return sOut;
}
string TC_Common::bin2str(const string &sBinData, const string &sSep, size_t lines)
{
return bin2str((const void *)sBinData.data(), sBinData.length(), sSep, lines);
}
int TC_Common::str2bin(const char *psAsciiData, unsigned char *sBinData, int iBinSize)
{
int iAsciiLength = strlen(psAsciiData);
int iRealLength = (iAsciiLength/2 > iBinSize)?iBinSize:(iAsciiLength/2);
for (int i = 0 ; i < iRealLength ; i++)
{
sBinData[i] = x2c(psAsciiData + i*2);
}
return iRealLength;
}
string TC_Common::str2bin(const string &sString, const string &sSep, size_t lines)
{
const char *psAsciiData = sString.c_str();
int iAsciiLength = sString.length();
string sBinData;
for (int i = 0 ; i < iAsciiLength ; i++)
{
sBinData += x2c(psAsciiData + i);
i++;
i += sSep.length(); //过滤掉分隔符
if (lines != 0 && sBinData.length()%lines == 0)
{
i++; //过滤掉回车
}
}
return sBinData;
}
char TC_Common::x2c(const string &sWhat)
{
register char digit;
if(sWhat.length() < 2)
{
return '\0';
}
digit = (sWhat[0] >= 'A' ? ((sWhat[0] & 0xdf) - 'A')+10 : (sWhat[0] - '0'));
digit *= 16;
digit += (sWhat[1] >= 'A' ? ((sWhat[1] & 0xdf) - 'A')+10 : (sWhat[1] - '0'));
return(digit);
}
string TC_Common::replace(const string &sString, const string &sSrc, const string &sDest)
{
if(sSrc.empty())
{
return sString;
}
string sBuf = sString;
string::size_type pos = 0;
while( (pos = sBuf.find(sSrc, pos)) != string::npos)
{
sBuf.replace(pos, sSrc.length(), sDest);
pos += sDest.length();
}
return sBuf;
}
string TC_Common::replace(const string &sString, const map<string,string>& mSrcDest)
{
if(sString.empty())
{
return sString;
}
string tmp = sString;
map<string,string>::const_iterator it = mSrcDest.begin();
while(it != mSrcDest.end())
{
string::size_type pos = 0;
while((pos = tmp.find(it->first, pos)) != string::npos)
{
tmp.replace(pos, it->first.length(), it->second);
pos += it->second.length();
}
++it;
}
return tmp;
}
bool TC_Common::matchPeriod(const string& s, const string& pat)
{
if(s.empty())
{
return false;
}
if(pat.empty())
{
return true;
}
if(pat.find('*') == string::npos)
{
return s == pat;
}
string::size_type sIndex = 0;
string::size_type patIndex = 0;
do
{
if(pat[patIndex] == '*')
{
if(s[sIndex] == '.')
{
return false;
}
while(sIndex < s.size() && s[sIndex] != '.')
{
++sIndex;
}
patIndex++;
}
else
{
if(pat[patIndex] != s[sIndex])
{
return false;
}
++sIndex;
++patIndex;
}
}
while(sIndex < s.size() && patIndex < pat.size());
return sIndex == s.size() && patIndex == pat.size();
}
bool TC_Common::matchPeriod(const string& s, const vector<string>& pat)
{
for(size_t i = 0; i < pat.size(); i++)
{
if(TC_Common::matchPeriod(s,pat[i]))
{
return true;
}
}
return false;
}
void TC_Common::daemon()
{
pid_t pid;
if ((pid = fork()) != 0)
{
exit(0);
}
setsid();
signal(SIGINT, SIG_IGN);
signal(SIGHUP, SIG_IGN);
signal(SIGQUIT, SIG_IGN);
signal(SIGPIPE, SIG_IGN);
signal(SIGTTOU, SIG_IGN);
signal(SIGTTIN, SIG_IGN);
signal(SIGCHLD, SIG_IGN);
signal(SIGTERM, SIG_IGN);
ignorePipe();
if((pid = fork()) != 0)
{
//父进程结束,变成daemon
exit(0);
}
umask(0);
// chdir("/");
}
void TC_Common::ignorePipe()
{
struct sigaction sig;
memset(&sig,0,sizeof(struct sigaction));
sig.sa_handler = SIG_IGN;
sig.sa_flags = 0;
sigemptyset(&sig.sa_mask);
sigaction(SIGPIPE,&sig,NULL);
}
bool TC_Common::isPrimeNumber(size_t n)
{
size_t nn = (size_t)sqrt((double)n);
for(size_t i = 2; i < nn;i++)
{
if(n % i == 0)
{
return false;
}
}
return true;
}
size_t TC_Common::toSize(const string &s, size_t iDefaultSize)
{
if (s.empty())
{
return iDefaultSize;
}
char c = s[s.length()-1];
if(c != 'K' && c != 'M' && c != 'G' && TC_Common::trim(s) == TC_Common::tostr(TC_Common::strto<size_t>(s)))
{
//没有后缀, 且转换是正确的
return (size_t)(TC_Common::strto<size_t>(s));
}
else if(c == 'K' || c == 'M' || c == 'G')
{
if (s.length() == 1)
{
return iDefaultSize;
}
float n = TC_Common::strto<float>(s.substr(0, s.length()-1));
if(TC_Common::trim(s.substr(0, s.length()-1)) != TC_Common::tostr(n))
{
return iDefaultSize;
}
if(c == 'K')
{
return (size_t)(n * 1024);
}
if(c == 'M')
{
return (size_t)(n * 1024*1024);
}
if(c == 'G')
{
return (size_t)(n * 1024 * 1024 * 1024);
}
}
return iDefaultSize;
}
// Generate the randome string, a SHA1-sized random number
void TC_Common::getRandomHexChars(char* p, unsigned int len)
{
const char* const chars = "0123456789abcdef";
FILE *fp = fopen("/dev/urandom","r");
if (!fp || fread(p,len,1,fp) == 0)
{
for (unsigned int j = 0; j < len; j++)
p[j] ^= rand();
}
if (fp) fclose(fp);
for (unsigned int j = 0; j < len; j++)
p[j] = chars[p[j] & 0x0F];
}
}

310
third_partly/util/src/tc_encoder.cpp Normal file
View File

@ -0,0 +1,310 @@
/**
* 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 <iconv.h>
#include <errno.h>
#include <string.h>
#include "util/tc_encoder.h"
namespace tars
{
void TC_Encoder::gbk2utf8(char *sOut, int &iMaxOutLen, const char *sIn, int iInLen)
{
char * pIn = (char*)sIn;
char * pEnd = pIn+iInLen;
char * pOut = sOut;
size_t iLeftLen;
size_t iGbkLen;
iconv_t cd;
if (iInLen > iMaxOutLen)
{
throw TC_Encoder_Exception("[TC_Encoder::gbk2utf8] iInLen > iMaxOutLen error : ", errno);
}
cd = iconv_open("UTF-8","GBK");
if (cd == (iconv_t)-1)
{
throw TC_Encoder_Exception("[TC_Encoder::gbk2utf8] iconv_open error : ", errno);
}
iLeftLen = iMaxOutLen;
while(pIn < pEnd)
{
if((unsigned char)(*pIn)==0x80)
{
//注意GBK的0x80转换为UTF-8时为E2 82 AC
*pOut = 0xe2; pOut++; iLeftLen--;
*pOut = 0x82; pOut++; iLeftLen--;
*pOut = 0xac; pOut++; iLeftLen--;
pIn++;
}
else if((unsigned char)(*pIn)<0x80)
{
//单字节(GBK: 0x00-0x7F)
*pOut = *pIn;
pIn++;pOut++;iLeftLen--;
}
else
{
//双字节
iGbkLen=2;
int iRet=iconv(cd, &pIn, (size_t *)&iGbkLen, (char **)&pOut, (size_t *)&iLeftLen);
if(iRet < 0)
{
*pOut = ' '; //转换不了替换为空格
pIn+=2; pOut++; iLeftLen--;
}
}
}
iconv_close(cd);
sOut[iMaxOutLen - iLeftLen] = '\0';
iMaxOutLen = iMaxOutLen - iLeftLen;
}
string TC_Encoder::gbk2utf8(const string &sIn)
{
iconv_t cd;
cd = iconv_open("UTF-8","GBK");
if (cd == (iconv_t)-1)
{
throw TC_Encoder_Exception("[TC_Encoder::gbk2utf8] iconv_open error", errno);
}
string sOut;
for(string::size_type pos = 0; pos < sIn.length(); ++pos)
{
if((unsigned char)sIn[pos] == 0x80)
{
//注意GBK的0x80转换为UTF-8时为E2 82 AC
sOut += 0xe2;
sOut += 0x82;
sOut += 0xac;
}
else if((unsigned char)sIn[pos] < 0x80)
{
//单字节(GBK: 0x00-0x7F)
sOut += sIn[pos];
}
else
{
//双字节
size_t sizeGbkLen = 2;
char pIn[128] = "\0";
strncpy(pIn, sIn.c_str() + pos, sizeGbkLen);
char *p = pIn;
size_t sizeLeftLen = 128;
char pOut[128] = "\0";
char *o = pOut;
int iRet = iconv(cd, &p, &sizeGbkLen, (char **)&o, &sizeLeftLen);
if(iRet < 0)
{
//转换不了, 暂时替换为空格
sOut += ' ';
}
else
{
sOut += pOut;
}
++pos;
}
}
iconv_close(cd);
return sOut;
}
void TC_Encoder::gbk2utf8(const string &sIn, vector<string> &vtStr)
{
iconv_t cd;
cd = iconv_open("UTF-8","GBK");
if (cd == (iconv_t)-1)
{
throw TC_Encoder_Exception("[TC_Encoder::gbk2utf8] iconv_open error", errno);
}
vtStr.clear();
for(string::size_type pos = 0; pos < sIn.length(); ++pos)
{
string sOut;
if((unsigned char)sIn[pos] == 0x80)
{
//注意GBK的0x80转换为UTF-8时为E2 82 AC
sOut += 0xe2;
sOut += 0x82;
sOut += 0xac;
}
else if((unsigned char)sIn[pos] < 0x80)
{
//单字节(GBK: 0x00-0x7F)
sOut += sIn[pos];
}
else
{
//双字节
size_t iGbkLen = 2;
char pIn[128] = "\0";
strncpy(pIn, sIn.c_str() + pos, iGbkLen);
char *p = pIn;
size_t iLeftLen = 128;
char pOut[128] = "\0";
char *o = pOut;
int iRet = iconv(cd, &p, (size_t *)&iGbkLen, (char **)&o, (size_t *)&iLeftLen);
if(iRet < 0)
{
//转换不了, 暂时替换为空格
sOut += ' ';
}
else
{
sOut += pOut;
}
++pos;
}
vtStr.push_back(sOut);
}
iconv_close(cd);
}
void TC_Encoder::utf82gbk(char *sOut, int &iMaxOutLen, const char *sIn, int iInLen)
{
iconv_t cd;
cd = iconv_open("GBK","UTF-8");
if (cd == (iconv_t)-1)
{
throw TC_Encoder_Exception("[TC_Encoder::utf82gbk] iconv_open error", errno);
}
char * pIn = (char*)sIn;
size_t sizeLeftLen = iMaxOutLen;
size_t sizeInLen = iInLen;
char* pOut = sOut;
size_t ret = iconv(cd, &pIn, &sizeInLen, (char **)&sOut, &sizeLeftLen);
if (ret == (size_t) - 1)
{
iMaxOutLen = 0;
iconv_close(cd);
throw TC_Encoder_Exception("[TC_Encoder::utf82gbk] iconv error", errno);
return;
}
iconv_close(cd);
pOut[iMaxOutLen - (int)sizeLeftLen] = '\0';
iMaxOutLen = iMaxOutLen - (int)sizeLeftLen;
}
string TC_Encoder::utf82gbk(const string &sIn)
{
if(sIn.length() == 0)
{
return "";
}
string sOut;
int iLen = sIn.length() * 2 + 1;
char *pOut = new char[iLen];
try
{
utf82gbk(pOut, iLen, sIn.c_str(), sIn.length());
}
catch (TC_Encoder_Exception& e)
{
delete[] pOut;
throw e;
}
sOut.assign(pOut, iLen);
delete[] pOut;
return sOut;
}
/**
* \n -> \r\0
* \r -> \r\r
*/
string TC_Encoder::transTo(const string& str, char f /*='\n'*/, char t /*= '\r'*/, char u /*= '\0'*/)
{
string ret = str;
for (size_t i = 0; i < ret.length(); ++i)
{
if (ret[i] == f)
{
ret[i] = t;
ret.insert(++i, 1, u);
}
else if (ret[i] == t)
{
ret.insert(++i, 1, t);
}
}
return ret;
}
/**
* \r\0 -> \n
* \r\r -> \r
*/
string TC_Encoder::transFrom(const string& str, char f /*= '\n'*/, char t /*= '\r'*/, char u /*= '\0'*/)
{
string ret = "";
for (string::const_iterator it = str.begin()
; it != str.end()
; ++it)
{
ret.append(1, *it);
if (*it == t)
{
if (*(++it) == u)
{
*ret.rbegin() = f;
}
}
}
return ret;
}
}

62
third_partly/util/src/tc_ex.cpp Normal file
View File

@ -0,0 +1,62 @@
/**
* 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 "util/tc_ex.h"
#include <execinfo.h>
#include <string.h>
#include <stdlib.h>
#include <cerrno>
namespace tars
{
TC_Exception::TC_Exception(const string &buffer)
:_buffer(buffer), _code(0)
{
// getBacktrace();
}
TC_Exception::TC_Exception(const string &buffer, int err)
{
_buffer = buffer + " :" + strerror(err);
_code = err;
// getBacktrace();
}
TC_Exception::~TC_Exception() throw()
{
}
const char* TC_Exception::what() const throw()
{
return _buffer.c_str();
}
void TC_Exception::getBacktrace()
{
void * array[64];
int nSize = backtrace(array, 64);
char ** symbols = backtrace_symbols(array, nSize);
for (int i = 0; i < nSize; i++)
{
_buffer += symbols[i];
_buffer += "\n";
}
free(symbols);
}
}

514
third_partly/util/src/tc_file.cpp Normal file
View File

@ -0,0 +1,514 @@
/**
* 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 "util/tc_file.h"
#include <string.h>
namespace tars
{
ifstream::pos_type TC_File::getFileSize(const string &sFullFileName)
{
ifstream ifs(sFullFileName.c_str());
ifs.seekg(0, ios_base::end);
return ifs.tellg();
}
bool TC_File::isAbsolute(const string &sFullFileName)
{
if(sFullFileName.empty())
{
return false;
}
unsigned i = 0;
while(isspace(sFullFileName[i]))
{
++i;
}
return sFullFileName[i] == '/';
}
bool TC_File::isFileExist(const string &sFullFileName, mode_t iFileType)
{
struct stat f_stat;
if (lstat(sFullFileName.c_str(), &f_stat) == -1)
{
return false;
}
if (!(f_stat.st_mode & iFileType))
{
return false;
}
return true;
}
bool TC_File::isFileExistEx(const string &sFullFileName, mode_t iFileType)
{
struct stat f_stat;
if (stat(sFullFileName.c_str(), &f_stat) == -1)
{
return false;
}
if (!(f_stat.st_mode & iFileType))
{
return false;
}
return true;
}
bool TC_File::makeDir(const string &sDirectoryPath, mode_t iFlag)
{
int iRetCode = mkdir(sDirectoryPath.c_str(), iFlag);
if(iRetCode < 0 && errno == EEXIST)
{
return isFileExistEx(sDirectoryPath, S_IFDIR);
}
return iRetCode == 0;
}
bool TC_File::makeDirRecursive(const string &sDirectoryPath, mode_t iFlag)
{
string simple = simplifyDirectory(sDirectoryPath);
string::size_type pos = 0;
for(; pos != string::npos; )
{
pos = simple.find("/", pos + 1);
string s;
if(pos == string::npos)
{
s = simple.substr(0, simple.size());
return makeDir(s.c_str(), iFlag);
}
else
{
s = simple.substr(0, pos);
if(!makeDir(s.c_str(), iFlag)) return false;
}
}
return true;
}
int TC_File::setExecutable(const string &sFullFileName, bool canExecutable)
{
struct stat f_stat;
if (stat(sFullFileName.c_str(), &f_stat) == -1)
{
return -1;
}
return chmod(sFullFileName.c_str(), canExecutable ? f_stat.st_mode | S_IXUSR : f_stat.st_mode & ~S_IXUSR);
}
bool TC_File::canExecutable(const string &sFullFileName)
{
struct stat f_stat;
if (stat(sFullFileName.c_str(), &f_stat) == -1)
{
return false;
}
return f_stat.st_mode & S_IXUSR;
}
int TC_File::removeFile(const string &sFullFileName, bool bRecursive)
{
string path = simplifyDirectory(sFullFileName);
if(isFileExist(path, S_IFDIR))
{
if(bRecursive)
{
vector<string> files;
listDirectory(path, files, false);
for(size_t i = 0; i < files.size(); i++)
{
removeFile(files[i], bRecursive);
}
if(path != "/")
{
if(::rmdir(path.c_str()) == -1)
{
return -1;
}
return 0;
}
}
else
{
if(::rmdir(path.c_str()) == -1)
{
return -1;
}
}
}
else
{
if(::remove(path.c_str()) == -1)
{
return -1;
}
}
return 0;
}
string TC_File::simplifyDirectory(const string& path)
{
string result = path;
string::size_type pos;
pos = 0;
while((pos = result.find("//", pos)) != string::npos)
{
result.erase(pos, 1);
}
pos = 0;
while((pos = result.find("/./", pos)) != string::npos)
{
result.erase(pos, 2);
}
while(result.substr(0, 4) == "/../")
{
result.erase(0, 3);
}
if(result == "/.")
{
return result.substr(0, result.size() - 1);
}
if(result.size() >= 2 && result.substr(result.size() - 2, 2) == "/.")
{
result.erase(result.size() - 2, 2);
}
if(result == "/")
{
return result;
}
if(result.size() >= 1 && result[result.size() - 1] == '/')
{
result.erase(result.size() - 1);
}
if(result == "/..")
{
result = "/";
}
return result;
}
string TC_File::load2str(const string &sFullFileName)
{
FILE* fp = fopen(sFullFileName.data(), "rb");
if (!fp)
return "";
fseek(fp, 0L, SEEK_END);
long size = ftell(fp);
fseek(fp, 0L, SEEK_SET);
string s(size, '\0');
fread((void*)s.data(), size, 1, fp);
fclose(fp);
return s;
}
void TC_File::load2str(const string &sFullFileName, vector<char> &buffer)
{
buffer.clear();
FILE* fp = fopen(sFullFileName.data(), "rb");
if (!fp)
return;
fseek(fp, 0L, SEEK_END);
long size = ftell(fp);
fseek(fp, 0L, SEEK_SET);
buffer.resize(size);
fread((void*)&buffer[0], size, 1, fp);
fclose(fp);
}
void TC_File::save2file(const string &sFullFileName, const string &sFileData)
{
ofstream ofs((sFullFileName).c_str());
ofs << sFileData;
ofs.close();
}
int TC_File::save2file(const string &sFullFileName, const char *sFileData, size_t length)
{
FILE *fp = fopen(sFullFileName.c_str(), "wb");
if(fp == NULL)
{
return -1;
}
size_t ret = fwrite((void*)sFileData, 1, length, fp);
fclose(fp);
if(ret == length)
{
return 0;
}
return -1;
}
string TC_File::getExePath()
{
string proc = "/proc/self/exe";
char buf[2048] = "\0";
int bufsize = sizeof(buf)/sizeof(char);
int count = readlink(proc.c_str(), buf,bufsize);
if ( count < 0 )
{
throw TC_File_Exception("[TC_File::getExePath] could not get exe path error", errno);
}
count = (count>=bufsize)?(bufsize-1):count;
buf[count] = '\0';
return buf;
}
string TC_File::extractFileName(const string &sFullFileName)
{
if(sFullFileName.length() <= 0)
{
return "";
}
string::size_type pos = sFullFileName.rfind('/');
if(pos == string::npos)
{
return sFullFileName;
}
return sFullFileName.substr(pos + 1);
}
string TC_File::extractFilePath(const string &sFullFileName)
{
if(sFullFileName.length() <= 0)
{
return "./";
}
string::size_type pos = 0;
for(pos = sFullFileName.length(); pos != 0 ; --pos)
{
if(sFullFileName[pos-1] == '/')
{
return sFullFileName.substr(0, pos);
}
}
return "./";
}
string TC_File::extractFileExt(const string &sFullFileName)
{
string::size_type pos;
if ((pos = sFullFileName.rfind('.')) == string::npos)
{
return string("");
}
return sFullFileName.substr(pos+1);
}
string TC_File::excludeFileExt(const string &sFullFileName)
{
string::size_type pos;
if ((pos = sFullFileName.rfind('.')) == string::npos)
{
return sFullFileName;
}
return sFullFileName.substr(0, pos);
}
string TC_File::replaceFileExt(const string &sFullFileName, const string &sExt)
{
return excludeFileExt(sFullFileName) + "." + sExt;
}
string TC_File::extractUrlFilePath(const string &sUrl)
{
string sLowerUrl = TC_Common::lower(sUrl);
string::size_type pos = sLowerUrl.find("http://");
if(pos == 0)
{
pos += strlen("http://");
}
else if(pos == string::npos)
{
pos = 0;
}
for( ; pos < sUrl.length(); ++pos)
{
if(sUrl[pos] == '/')
{
if(pos < sUrl.length() - 1)
{
pos++;
break;
}
else
{
return "";
}
}
}
if(pos == string::npos || pos == sUrl.length())
{
pos = 0;
}
return sUrl.substr(pos);
}
size_t TC_File::scanDir(const string &sFilePath, vector<string> &vtMatchFiles, FILE_SELECT f, int iMaxSize )
{
vtMatchFiles.clear();
struct dirent **namelist;
int n = scandir(sFilePath.c_str(), &namelist, f, alphasort);
if (n < 0)
{
return 0;
}
else
{
while(n-- )
{
if(iMaxSize > 0 && vtMatchFiles.size() >= (size_t)iMaxSize )
{
free(namelist[n]);
break;
}
else
{
vtMatchFiles.push_back(namelist[n]->d_name);
free(namelist[n]);
}
}
free(namelist);
}
return vtMatchFiles.size();
}
void TC_File::listDirectory(const string &path, vector<string> &files, bool bRecursive)
{
vector<string> tf;
scanDir(path, tf, 0, 0);
for(size_t i = 0; i < tf.size(); i++)
{
if(tf[i] == "." || tf[i] == "..")
continue;
string s = path + "/" + tf[i];
if(isFileExist(s, S_IFDIR))
{
files.push_back(simplifyDirectory(s));
if(bRecursive)
{
listDirectory(s, files, bRecursive);
}
}
else
{
files.push_back(simplifyDirectory(s));
}
}
}
void TC_File::copyFile(const string &sExistFile, const string &sNewFile,bool bRemove)
{
if(TC_File::isFileExist(sExistFile, S_IFDIR))
{
TC_File::makeDir(sNewFile);
vector<string> tf;
TC_File::scanDir(sExistFile,tf, 0, 0);
for(size_t i = 0; i <tf.size(); i++)
{
if(tf[i] == "." || tf[i] == "..")
continue;
string s = sExistFile + "/" + tf[i];
string d = sNewFile + "/" + tf[i];
copyFile(s, d,bRemove);
}
}
else
{
if(bRemove) std::remove(sNewFile.c_str());
std::ifstream fin(sExistFile.c_str());
if(!fin)
{
throw TC_File_Exception("[TC_File::copyFile] error: "+sExistFile, errno);
}
std::ofstream fout(sNewFile.c_str());
if(!fout )
{
throw TC_File_Exception("[TC_File::copyFile] error: "+sNewFile, errno);
}
struct stat f_stat;
if (stat(sExistFile.c_str(), &f_stat) == -1)
{
throw TC_File_Exception("[TC_File::copyFile] error: "+sExistFile, errno);
}
chmod(sNewFile.c_str(),f_stat.st_mode);
fout<<fin.rdbuf();
fin.close();
fout.close();
}
}
}

85
third_partly/util/src/tc_option.cpp Normal file
View File

@ -0,0 +1,85 @@
/**
* 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 "util/tc_option.h"
#include "util/tc_common.h"
namespace tars
{
void TC_Option::decode(int argc, char *argv[])
{
_mParam.clear();
vector<string> v;
for(int i = 1; i < argc; i++)
{
v.push_back(argv[i]);
}
for(size_t i = 0; i < v.size(); i++)
{
if(v[i].length() > 2 && v[i].substr(0,2) == "--")
{
parse(v[i]);
}
else
{
_vSingle.push_back(v[i]);
}
}
}
void TC_Option::parse(const string &s)
{
string::size_type pos = s.find('=');
if( pos != string::npos)
{
_mParam[s.substr(2, pos-2)] = s.substr(pos+1);
}
else
{
_mParam[s.substr(2, pos-2)] = "";
}
}
string TC_Option::getValue(const string &sName)
{
if(_mParam.find(sName) != _mParam.end())
{
return _mParam[sName];
}
return "";
}
bool TC_Option::hasParam(const string &sName)
{
return _mParam.find(sName) != _mParam.end();
}
vector<string>& TC_Option::getSingle()
{
return _vSingle;
}
map<string, string>& TC_Option::getMulti()
{
return _mParam;
}
}