mirrors/TarsCloud_TarsCpp
1
0
Fork 0
mirror of https://gitee.com/TarsCloud/TarsCpp.git synced 2025-01-05 17:42:24 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity
aa97998533
TarsCloud_TarsCpp/tools/tars2cpp/tars2cpp.cpp
linfengchen(陈林峰) ad8bae850d 1\fix tc_json bug;2\support tars2xml
2020-04-02 18:57:36 +08:00

3263 lines
104 KiB
C++
Executable File
Raw Blame History

This file contains ambiguous Unicode characters

This file contains Unicode characters that might be confused with other characters. If you think that this is intentional, you can safely ignore this warning. Use the Escape button to reveal them.

/**
* 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 "tars2cpp.h"
#include "util/tc_md5.h"
#include "util/tc_file.h"
#include "util/tc_common.h"
#include <string>
#define TAB g_parse->getTab()
#define INC_TAB g_parse->incTab()
#define DEL_TAB g_parse->delTab()
//////////////////////////////////////////////////////////////////////////////////
//
Tars2Cpp::Tars2Cpp()
: _checkDefault(false)
, _onlyStruct(false)
, _bSqlSupport(false)
, _bXmlSupport(false)
, _bJsonSupport(true)
, _namespace("tars")
// , _unknownField(false)
, _tarsMaster(false)
{
}
string Tars2Cpp::writeToXml(const TypeIdPtr &pPtr) const
{
ostringstream s;
if(EnumPtr::dynamicCast(pPtr->getTypePtr()))
{
s << TAB << "p->value[\"" << pPtr->getId() << "\"] = " + _namespace+ "::XmlOutput::writeXml((" + _namespace+ "::Int32)" << pPtr->getId() << ", _cdata_format);" << endl;
}
else if(pPtr->getTypePtr()->isArray())
{
s << TAB << "p->value[\"" << pPtr->getId() << "\"] = " + _namespace+ "::XmlOutput::writeXml((const "
<< tostr(pPtr->getTypePtr()) << " *)" << pPtr->getId() << "Len" << ");" << endl;
}
else if(pPtr->getTypePtr()->isPointer())
{
s << TAB << "p->value[\"" << pPtr->getId() << "\"] = " + _namespace+ "::XmlOutput::writeXml((const "
<< tostr(pPtr->getTypePtr()) << " )" << pPtr->getId() << "Len" << ");" << endl;
}
else
{
MapPtr mPtr = MapPtr::dynamicCast(pPtr->getTypePtr());
VectorPtr vPtr = VectorPtr::dynamicCast(pPtr->getTypePtr());
if (!_checkDefault || pPtr->isRequire() || (!pPtr->hasDefault() && !mPtr && !vPtr))
{
BuiltinPtr bPtr = BuiltinPtr::dynamicCast(pPtr->getTypePtr());
if (pPtr->getTypePtr()->isSimple() || (bPtr && bPtr->kind() == Builtin::KindString))
{
s << TAB << "p->value[\"" << pPtr->getId() << "\"] = " + _namespace + "::XmlOutput::writeXml(" << pPtr->getId() << ", _cdata_format);" << endl;
}
else
{
s << TAB << "p->value[\"" << pPtr->getId() << "\"] = " + _namespace + "::XmlOutput::writeXml(" << pPtr->getId() << ");" << endl;
}
}
else
{
string sDefault = pPtr->def();
BuiltinPtr bPtr = BuiltinPtr::dynamicCast(pPtr->getTypePtr());
if (bPtr && bPtr->kind() == Builtin::KindString)
{
sDefault = "\"" + TC_Common::replace(pPtr->def(), "\"", "\\\"") + "\"";
}
if (mPtr || vPtr)
{
s << TAB << "if (" << pPtr->getId() << ".size() > 0)" << endl;
}
else
{
s << TAB << "if (" << pPtr->getId() << " != " << sDefault << ")" << endl;
}
s << TAB << "{" << endl;
INC_TAB;
s << TAB << "p->value[\"" << pPtr->getId() << "\"] = " + _namespace+ "::XmlOutput::writeXml(" << pPtr->getId() << ");" << endl;
DEL_TAB;
s << TAB << "}" << endl;
}
}
return s.str();
}
string Tars2Cpp::readFromXml(const TypeIdPtr &pPtr, bool bIsRequire) const
{
ostringstream s;
if(EnumPtr::dynamicCast(pPtr->getTypePtr()))
{
s << TAB << "tars::XmlInput::readXml((tars::Int32&)" << pPtr->getId() <<", pObj->value[\"" << pPtr->getId() << "\"]";
}
else if(pPtr->getTypePtr()->isArray())
{
s << TAB << "tars::XmlInput::readXml(" << pPtr->getId() << "Len" <<", pObj->value[\"" << pPtr->getId() << "\"]" << getSuffix(pPtr);
}
else if(pPtr->getTypePtr()->isPointer())
{
// "not support";
}
else
{
s << TAB << "tars::XmlInput::readXml(" << pPtr->getId() << ",pObj->value[\"" << pPtr->getId() << "\"]";
}
s << ", " << ((pPtr->isRequire() && bIsRequire)?"true":"false") << ");" << endl;
return s.str();
}
string Tars2Cpp::writeToSql(const TypeIdPtr &pPtr) const
{
ostringstream s;
if(EnumPtr::dynamicCast(pPtr->getTypePtr()))
{
s << TAB << "_mycols[\"" << pPtr->getId() << "\"] = make_pair(tars::TC_Mysql::DB_INT, tars::TC_Common::tostr(" << pPtr->getId() << "));" << endl;
}
BuiltinPtr bPtr = BuiltinPtr::dynamicCast(pPtr->getTypePtr());
if (bPtr)
{
switch(bPtr->kind())
{
case Builtin::KindBool:
case Builtin::KindByte:
case Builtin::KindShort:
case Builtin::KindInt:
case Builtin::KindLong:
s << TAB << "_mycols[\"" << pPtr->getId() << "\"] = make_pair(tars::TC_Mysql::DB_INT, tars::TC_Common::tostr(" << pPtr->getId() << "));" << endl;
break;
case Builtin::KindFloat:
case Builtin::KindDouble:
s << TAB << "_mycols[\"" << pPtr->getId() << "\"] = make_pair(tars::TC_Mysql::DB_STR, tars::TC_Common::tostr(" << pPtr->getId() << "));" << endl;
break;
case Builtin::KindString:
s << TAB << "_mycols[\"" << pPtr->getId() << "\"] = make_pair(tars::TC_Mysql::DB_STR, tars::TC_Common::trim(" << pPtr->getId() << "));" << endl;
break;
default:
break;
}
}
else if (!pPtr->getTypePtr()->isSimple())
{
s << TAB << "_mycols[\"" << pPtr->getId() << "\"] = make_pair(tars::TC_Mysql::DB_STR, tars::TC_Json::writeValue(tars::JsonOutput::writeJson(" << pPtr->getId() << ")));" << endl;
}
return s.str();
}
string Tars2Cpp::readFromSql(const TypeIdPtr &pPtr, bool bIsRequire) const
{
ostringstream s;
EnumPtr ePtr = EnumPtr::dynamicCast(pPtr->getTypePtr());
if(ePtr)
{
s << TAB << pPtr->getId() << " = (" << ePtr->getSid() <<")TC_Common::strto<tars::Int32>(_mysrd[\"" << pPtr->getId() << "\"]);" << endl;
}
BuiltinPtr bPtr = BuiltinPtr::dynamicCast(pPtr->getTypePtr());
if (bPtr)
{
switch(bPtr->kind())
{
case Builtin::KindBool:
s << TAB << pPtr->getId() << " = TC_Common::strto<tars::Bool>(_mysrd[\"" << pPtr->getId() << "\"]);" << endl;
break;
case Builtin::KindByte:
s << TAB << pPtr->getId() << " = TC_Common::strto<tars::" << (bPtr->isUnsigned() ? "UInt8" : "Char") << ">(_mysrd[\"" << pPtr->getId() << "\"]);" << endl;
break;
case Builtin::KindShort:
s << TAB << pPtr->getId() << " = TC_Common::strto<tars::" << (bPtr->isUnsigned() ? "UInt16" : "Short") << ">(_mysrd[\"" << pPtr->getId() << "\"]);" << endl;
break;
case Builtin::KindInt:
s << TAB << pPtr->getId() << " = TC_Common::strto<tars::" << (bPtr->isUnsigned() ? "UInt32" : "Int32") << ">(_mysrd[\"" << pPtr->getId() << "\"]);" << endl;
break;
case Builtin::KindLong:
s << TAB << pPtr->getId() << " = TC_Common::strto<tars::" << (bPtr->isUnsigned() ? "UInt64" : "Int64") << ">(_mysrd[\"" << pPtr->getId() << "\"]);" << endl;
break;
case Builtin::KindFloat:
s << TAB << pPtr->getId() << " = TC_Common::strto<tars::Float>(_mysrd[\"" << pPtr->getId() << "\"]);" << endl;
break;
case Builtin::KindDouble:
s << TAB << pPtr->getId() << " = TC_Common::strto<tars::Double>(_mysrd[\"" << pPtr->getId() << "\"]);" << endl;
break;
case Builtin::KindString:
s << TAB << pPtr->getId() << " = TC_Common::trim(_mysrd[\"" << pPtr->getId() << "\"]);" << endl;
break;
default:
break;
}
}
else if (!pPtr->getTypePtr()->isSimple() )
{
s << TAB << "tars::JsonInput::readJson(" << pPtr->getId() << ", tars::TC_Json::getValue(_mysrd[\"" << pPtr->getId() << "\"]), false);" << endl;
}
return s.str();
}
string Tars2Cpp::writeToJson(const TypeIdPtr& pPtr) const
{
ostringstream s;
if (EnumPtr::dynamicCast(pPtr->getTypePtr()))
{
s << TAB << "p->value[\"" << pPtr->getId() << "\"] = " + _namespace + "::JsonOutput::writeJson((" + _namespace + "::Int32)"
<< pPtr->getId() << ");" << endl;
}
else if (pPtr->getTypePtr()->isArray())
{
s << TAB << "p->value[\"" << pPtr->getId() << "\"] = " + _namespace + "::JsonOutput::writeJson((const "
<< tostr(pPtr->getTypePtr()) << " *)" << pPtr->getId() << "Len" << ");" << endl;
}
else if (pPtr->getTypePtr()->isPointer())
{
s << TAB << "p->value[\"" << pPtr->getId() << "\"] = " + _namespace + "::JsonOutput::writeJson((const "
<< tostr(pPtr->getTypePtr()) << " )" << pPtr->getId() << "Len" << ");" << endl;
}
else
{
MapPtr mPtr = MapPtr::dynamicCast(pPtr->getTypePtr());
VectorPtr vPtr = VectorPtr::dynamicCast(pPtr->getTypePtr());
// 对于json， 不检测默认值
if (true || !_checkDefault || pPtr->isRequire() || (!pPtr->hasDefault() && !mPtr && !vPtr))
{
s << TAB << "p->value[\"" << pPtr->getId() << "\"] = " + _namespace + "::JsonOutput::writeJson("
<< pPtr->getId() << ");" << endl;
}
else
{
string sDefault = pPtr->def();
BuiltinPtr bPtr = BuiltinPtr::dynamicCast(pPtr->getTypePtr());
if (bPtr && bPtr->kind() == Builtin::KindString)
{
sDefault = "\"" + tars::TC_Common::replace(pPtr->def(), "\"", "\\\"") + "\"";
}
if (mPtr || vPtr)
{
s << TAB << "if (" << pPtr->getId() << ".size() > 0)" << endl;
}
else if (bPtr && (bPtr->kind() == Builtin::KindFloat || bPtr->kind() == Builtin::KindDouble))
{
s << TAB << "if (!tars::TC_Common::equal(" << pPtr->getId() << "," << sDefault << "))" << endl;
}
else
{
s << TAB << "if (" << pPtr->getId() << " != " << sDefault << ")" << endl;
}
s << TAB << "{" << endl;
INC_TAB;
s << TAB << "p->value[\"" << pPtr->getId() << "\"] = " + _namespace + "::JsonOutput::writeJson("
<< pPtr->getId() << ");" << endl;
DEL_TAB;
s << TAB << "}" << endl;
}
}
return s.str();
}
string Tars2Cpp::readFromJson(const TypeIdPtr& pPtr, bool bIsRequire) const
{
ostringstream s;
// if (EnumPtr::dynamicCast(pPtr->getTypePtr()))
// {
// s << TAB << _namespace + "::JsonInput::readJson((" + _namespace + "::Int32&)" << pPtr->getId() << ",pObj->value[\"" << pPtr->getId() << "\"]";
// }
// else
if (pPtr->getTypePtr()->isArray())
{
s << TAB << _namespace + "::JsonInput::readJson(" << pPtr->getId() << "Len" << ",pObj->value[\"" << pPtr->getId() << "\"]" << getSuffix(pPtr);
}
else if (pPtr->getTypePtr()->isPointer())
{
#if 0
s << TAB << pPtr->getId() <<" = ("<<tostr(pPtr->getTypePtr())<<")_is.cur();"<<endl;
s << TAB << "_is.read("<< pPtr->getId()<<", _is.left(), "<< pPtr->getId() << "Len";
#endif
s << TAB << "not support";
}
else
{
s << TAB << _namespace + "::JsonInput::readJson(" << pPtr->getId() << ",pObj->value[\"" << pPtr->getId() << "\"]";
}
s << ", " << ((pPtr->isRequire() && bIsRequire) ? "true" : "false") << ");" << endl;
#if 0
if(pPtr->getTypePtr()->isPointer())
s << TAB <<"_is.mapBufferSkip("<<pPtr->getId() << "Len);"<<endl;
#endif
return s.str();
}
string Tars2Cpp::writeTo(const TypeIdPtr& pPtr) const
{
ostringstream s;
if (EnumPtr::dynamicCast(pPtr->getTypePtr()))
{
s << TAB << "_os.write((" + _namespace + "::Int32)" << pPtr->getId() << ", " << pPtr->getTag() << ");" << endl;
}
else if (pPtr->getTypePtr()->isArray())
{
s << TAB << "_os.write((const " << tostr(pPtr->getTypePtr()) << " *)" << pPtr->getId() << ", " << pPtr->getId() << "Len" << ", " << pPtr->getTag() << ");" << endl;
}
else if (pPtr->getTypePtr()->isPointer())
{
s << TAB << "_os.write((const " << tostr(pPtr->getTypePtr()) << ")" << pPtr->getId() << ", " << pPtr->getId() << "Len" << ", " << pPtr->getTag() << ");" << endl;
}
else
{
MapPtr mPtr = MapPtr::dynamicCast(pPtr->getTypePtr());
VectorPtr vPtr = VectorPtr::dynamicCast(pPtr->getTypePtr());
if (!_checkDefault || pPtr->isRequire() || (!pPtr->hasDefault() && !mPtr && !vPtr))
{
s << TAB << "_os.write(" << pPtr->getId() << ", " << pPtr->getTag() << ");" << endl;
}
else
{
string sDefault = pPtr->def();
BuiltinPtr bPtr = BuiltinPtr::dynamicCast(pPtr->getTypePtr());
if (bPtr && bPtr->kind() == Builtin::KindString)
{
sDefault = "\"" + tars::TC_Common::replace(pPtr->def(), "\"", "\\\"") + "\"";
}
if (mPtr || vPtr)
{
s << TAB << "if (" << pPtr->getId() << ".size() > 0)" << endl;
}
else
{
s << TAB << "if (" << pPtr->getId() << " != " << sDefault << ")" << endl;
}
s << TAB << "{" << endl;
INC_TAB;
s << TAB << "_os.write(" << pPtr->getId() << ", " << pPtr->getTag() << ");" << endl;
DEL_TAB;
s << TAB << "}" << endl;
}
}
return s.str();
}
string Tars2Cpp::readFrom(const TypeIdPtr& pPtr, bool bIsRequire) const
{
ostringstream s;
if (EnumPtr::dynamicCast(pPtr->getTypePtr()))
{
//枚举强制类型转换在O2编译选项情况下会告警
string tmp = _namespace + "::Int32 eTemp" + TC_Common::tostr(pPtr->getTag()) + generateInitValue(pPtr);
s << TAB << tmp << endl;
s << TAB << "_is.read(eTemp" << TC_Common::tostr(pPtr->getTag());
}
else if (pPtr->getTypePtr()->isArray())
{
s << TAB << "_is.read(" << pPtr->getId() << ", " << getSuffix(pPtr) << ", " << pPtr->getId() << "Len";
}
else if (pPtr->getTypePtr()->isPointer())
{
s << TAB << pPtr->getId() << " = (" << tostr(pPtr->getTypePtr()) << ")_is.cur();" << endl;
s << TAB << "_is.read(" << pPtr->getId() << ", _is.left(), " << pPtr->getId() << "Len";
}
else
{
s << TAB << "_is.read(" << pPtr->getId();
}
s << ", " << pPtr->getTag() << ", " << ((pPtr->isRequire() && bIsRequire) ? "true" : "false") << ");" << endl;
if (EnumPtr::dynamicCast(pPtr->getTypePtr()))
{
s << TAB << pPtr->getId() << " = (" << tostr(pPtr->getTypePtr()) << ")eTemp" << TC_Common::tostr(pPtr->getTag()) << ";" << endl;
}
if(pPtr->getTypePtr()->isPointer())
s << TAB <<"_is.mapBufferSkip("<<pPtr->getId() << "Len);"<<endl;
return s.str();
}
// string Tars2Cpp::readUnknown(const TypeIdPtr& pPtr) const
// {
// ostringstream s;
// s << TAB << "_is.readUnknown(sUnknownField, " << pPtr->getTag() << ");" << endl;
// return s.str();
// }
// string Tars2Cpp::writeUnknown() const
// {
// ostringstream s;
// s << TAB << "_os.writeUnknown(sUnknownField);" << endl;
// return s.str();
// }
string Tars2Cpp::display(const TypeIdPtr& pPtr) const
{
ostringstream s;
if (EnumPtr::dynamicCast(pPtr->getTypePtr()))
{
s << TAB << "_ds.display((" + _namespace + "::Int32)" << pPtr->getId() << ",\"" << pPtr->getId() << "\");" << endl;;
}
else if (pPtr->getTypePtr()->isArray() || pPtr->getTypePtr()->isPointer())
{
s << TAB << "_ds.display(" << pPtr->getId() << ", " << pPtr->getId() << "Len" << ",\"" << pPtr->getId() << "\");" << endl;
}
else
{
s << TAB << "_ds.display(" << pPtr->getId() << ",\"" << pPtr->getId() << "\");" << endl;;
}
return s.str();
}
string Tars2Cpp::displaySimple(const TypeIdPtr& pPtr, bool bSep) const
{
ostringstream s;
if (EnumPtr::dynamicCast(pPtr->getTypePtr()))
{
s << TAB << "_ds.displaySimple((" + _namespace + "::Int32)" << pPtr->getId() << ", "
<< (bSep ? "true" : "false") << ");" << endl;
}
else if (pPtr->getTypePtr()->isArray())
{
s << TAB << "_ds.displaySimple(" << pPtr->getId() << ", " << pPtr->getId() << "Len" << ","
<< (bSep ? "true" : "false") << ");" << endl;
}
else if (pPtr->getTypePtr()->isPointer())
{
s << TAB << "_ds.displaySimple(";
s << (bSep ? "" : ("(const " + tostr(pPtr->getTypePtr()) + ")"));
s << pPtr->getId() << ", " << pPtr->getId() << "Len" << "," << (bSep ? "true" : "false") << ");" << endl;
}
else
{
s << TAB << "_ds.displaySimple(" << pPtr->getId() << ", "
<< (bSep ? "true" : "false") << ");" << endl;
}
return s.str();
}
string Tars2Cpp::generateCollection(const TypeIdPtr& pPtr, bool bSep) const
{
ostringstream s;
if (pPtr->getTypePtr()->isArray())
{
s << TAB << "_jj.generateCollection(" << "\"" << pPtr->getId() << "\"" << ", " << pPtr->getId() << ", " << pPtr->getId() << "Len"
<< ");" << "\n" << (bSep ? TAB + "_jj.append(\",\", false);" : "") << endl;
}
else
{
s << TAB << "_jj.generateCollection(" << "\"" << pPtr->getId() << "\"" << ", " << pPtr->getId()
<< ");" << "\n" << (bSep ? TAB + "_jj.append(\",\", false);" : "") << endl;
}
return s.str();
}
/*******************************获取定长数组坐标********************************/
int Tars2Cpp::getSuffix(const TypeIdPtr& pPtr) const
{
BuiltinPtr bPtr = BuiltinPtr::dynamicCast(pPtr->getTypePtr());
if (bPtr && bPtr->kind() == Builtin::KindString && bPtr->isArray())
{
return bPtr->getSize();
}
VectorPtr vPtr = VectorPtr::dynamicCast(pPtr->getTypePtr());
if (vPtr && vPtr->isArray())
{
return vPtr->getSize();
}
return -1;
}
/*******************************定长数组坐标********************************/
string Tars2Cpp::toStrSuffix(const TypeIdPtr& pPtr) const
{
ostringstream s;
int i = getSuffix(pPtr);
if (i >= 0)
{
s << "[" << i << "]";
}
return s.str();
}
/*******************************BuiltinPtr********************************/
string Tars2Cpp::tostr(const TypePtr& pPtr) const
{
BuiltinPtr bPtr = BuiltinPtr::dynamicCast(pPtr);
if (bPtr) return tostrBuiltin(bPtr);
VectorPtr vPtr = VectorPtr::dynamicCast(pPtr);
if (vPtr) return tostrVector(vPtr);
MapPtr mPtr = MapPtr::dynamicCast(pPtr);
if (mPtr) return tostrMap(mPtr);
StructPtr sPtr = StructPtr::dynamicCast(pPtr);
if (sPtr) return tostrStruct(sPtr);
EnumPtr ePtr = EnumPtr::dynamicCast(pPtr);
if (ePtr) return tostrEnum(ePtr);
if (!pPtr) return "void";
assert(false);
return "";
}
string Tars2Cpp::tostrBuiltin(const BuiltinPtr& pPtr) const
{
string s;
switch (pPtr->kind())
{
case Builtin::KindBool:
s = _namespace + "::Bool";
break;
case Builtin::KindByte:
s = _namespace + "::Char";
break;
case Builtin::KindShort:
//为了兼容java无unsigned, 编结码时把tars问件中 unsigned char 对应到short
//c++中需要还原回来
s = (pPtr->isUnsigned() ? _namespace + "::UInt8" : _namespace + "::Short");
break;
case Builtin::KindInt:
s = (pPtr->isUnsigned() ? _namespace + "::UInt16" : _namespace + "::Int32");
break;
case Builtin::KindLong:
s = (pPtr->isUnsigned() ? _namespace + "::" + "UInt32" : _namespace + "::Int64");
break;
case Builtin::KindFloat:
s = _namespace + "::Float";
break;
case Builtin::KindDouble:
s = _namespace + "::Double";
break;
case Builtin::KindString:
if(pPtr->isArray())
s = _namespace+ "::Char"; //char a [8];
else
s = "std::string";//string a;
break;
case Builtin::KindVector:
s = "std::vector";
break;
case Builtin::KindMap:
s = "std::map";
break;
default:
assert(false);
break;
}
return s;
}
/*******************************VectorPtr********************************/
string Tars2Cpp::tostrVector(const VectorPtr& pPtr) const
{
//数组类型
if (pPtr->isArray())
{
return tostr(pPtr->getTypePtr());
}
//指针类型
if (pPtr->isPointer())
{
return tostr(pPtr->getTypePtr()) + " *";
}
string s = Builtin::builtinTable[Builtin::KindVector] + string("<") + tostr(pPtr->getTypePtr());
if (MapPtr::dynamicCast(pPtr->getTypePtr()) || VectorPtr::dynamicCast(pPtr->getTypePtr()))
{
s += " >";
}
else
{
s += ">";
}
return s;
}
/*******************************MapPtr********************************/
string Tars2Cpp::tostrMap(const MapPtr& pPtr) const
{
string s = Builtin::builtinTable[Builtin::KindMap] + string("<") + tostr(pPtr->getLeftTypePtr()) + ", " + tostr(pPtr->getRightTypePtr());
if (MapPtr::dynamicCast(pPtr->getRightTypePtr()) || VectorPtr::dynamicCast(pPtr->getRightTypePtr()))
{
s += " >";
}
else
{
s += ">";
}
return s;
}
/*******************************StructPtr********************************/
string Tars2Cpp::tostrStruct(const StructPtr& pPtr) const
{
return pPtr->getSid();
}
string Tars2Cpp::MD5(const StructPtr& pPtr) const
{
string s;
vector<TypeIdPtr>& member = pPtr->getAllMemberPtr();
for (size_t j = 0; j < member.size(); j++)
{
s += "_" + tostr(member[j]->getTypePtr());
}
return "\"" + tars::TC_MD5::md5str(s) + "\"";
}
/////////////////////////////////////////////////////////////////////
string Tars2Cpp::tostrEnum(const EnumPtr& pPtr) const
{
return pPtr->getSid();
}
///////////////////////////////////////////////////////////////////////
string Tars2Cpp::generateH(const StructPtr& pPtr, const string& namespaceId) const
{
ostringstream s;
s << TAB << "struct " << pPtr->getId() << " : public " + _namespace + "::TarsStructBase" << endl;
s << TAB << "{" << endl;
s << TAB << "public:" << endl;
INC_TAB;
s << TAB << "static string className()" << endl;
s << TAB << "{" << endl;
INC_TAB;
s << TAB << "return " << "\"" << namespaceId << "." << pPtr->getId() << "\"" << ";" << endl;
DEL_TAB;
s << TAB << "}" << endl;
s << TAB << "static string MD5()" << endl;
s << TAB << "{" << endl;
INC_TAB;
s << TAB << "return " << MD5(pPtr) << ";" << endl;
DEL_TAB;
s << TAB << "}" << endl;
////////////////////////////////////////////////////////////
//定义缺省构造函数
s << TAB << pPtr->getId() << "()" << endl;
s << TAB << "{" << endl;
INC_TAB;
s << TAB << "resetDefautlt();" << endl;
vector<TypeIdPtr>& member = pPtr->getAllMemberPtr();
/*
bool b = false;
//定义初始化列表
for (size_t j = 0; j < member.size(); j++)
{
if (member[j]->getTypePtr()->isArray())
{
if (!b) s << TAB << ":";
else s << ",";
s << member[j]->getId() << "Len(0)";
b = true;
continue;
}
if (member[j]->getTypePtr()->isPointer())
{
if (!b) s << TAB << ":";
else s << ",";
s << member[j]->getId() << "Len(0)," << member[j]->getId() << "(NULL)";
b = true;
continue;
}
if (member[j]->hasDefault())
{
if (!b) s << TAB << ":";
else s << ",";
BuiltinPtr bPtr = BuiltinPtr::dynamicCast(member[j]->getTypePtr());
//string值要转义
if (bPtr && bPtr->kind() == Builtin::KindString)
{
string tmp = tars::TC_Common::replace(member[j]->def(), "\"", "\\\"");
s << member[j]->getId() << "(\"" << tmp << "\")";
}
else
{
s << member[j]->getId() << "(" << member[j]->def() << ")";
}
b = true;
}
else
{ //没有提供初始值才会走到这里,提供枚举类型初始化值
EnumPtr ePtr = EnumPtr::dynamicCast(member[j]->getTypePtr());
if (ePtr)
{
vector<TypeIdPtr>& eMember = ePtr->getAllMemberPtr();
if (eMember.size() > 0)
{
if (!b) s << TAB << ":";
else s << ",";
string sid = ePtr->getSid();
s << member[j]->getId() << "(" << sid.substr(0, sid.find_first_of("::")) << "::" << eMember[0]->getId() << ")";
b = true;
}
}
}
}
// if (b)
// {
// if (_unknownField)
// {
// s << ",sUnknownField(\"\")";
// }
// s << endl;
// }
s << TAB << "{" << endl;
INC_TAB;
for (size_t j = 0; j < member.size(); j++)
{
VectorPtr vPtr = VectorPtr::dynamicCast(member[j]->getTypePtr());
if (vPtr)
{
BuiltinPtr bPtr = BuiltinPtr::dynamicCast(vPtr->getTypePtr());
if (!bPtr || bPtr->kind() == Builtin::KindString) //非内建类型或者string 类型不能memset
{
continue;
}
}
if (!member[j]->getTypePtr()->isArray())
{
continue;
}
s << TAB << "memset(" << member[j]->getId() << ", 0, " << "sizeof(" << member[j]->getId() << "));" << endl;
}
*/
if (_bXmlSupport)
{
s << TAB << "_cdata_format = false;" << endl;
}
DEL_TAB;
s << TAB << "}" << endl;
//resetDefault()函数
s << TAB << "void resetDefautlt()" << endl;
s << TAB << "{" << endl;
INC_TAB;
member = pPtr->getAllMemberPtr();
for (size_t j = 0; j < member.size(); j++)
{
if (member[j]->getTypePtr()->isArray())
{
s << TAB << member[j]->getId() << "Len = 0;" << endl;
VectorPtr vPtr = VectorPtr::dynamicCast(member[j]->getTypePtr());
if (vPtr)
{
BuiltinPtr bPtr = BuiltinPtr::dynamicCast(vPtr->getTypePtr());
if (bPtr && bPtr->kind() != Builtin::KindString) //非内建类型或者string 类型不能memset
{
s << TAB << "memset(" << member[j]->getId() << ", 0, " << "sizeof(" << member[j]->getId() << "));" << endl;
}
}
continue;
}
if (member[j]->getTypePtr()->isPointer())
{
s << TAB << member[j]->getId() << "Len = 0;" << endl;
s << TAB << member[j]->getId() << " = NULL;" << endl;
continue;
}
if (member[j]->hasDefault())
{
BuiltinPtr bPtr = BuiltinPtr::dynamicCast(member[j]->getTypePtr());
//string值要转义
if (bPtr && bPtr->kind() == Builtin::KindString)
{
string tmp = tars::TC_Common::replace(member[j]->def(), "\"", "\\\"");
s << TAB << member[j]->getId() << " = \"" << tmp << "\";" << endl;
}
else
{
s << TAB << member[j]->getId() << " = " << member[j]->def() << ";" << endl;
}
}
}
DEL_TAB;
s << TAB << "}" << endl;
s << TAB << "template<typename WriterT>" << endl;
s << TAB << "void writeTo(" + _namespace + "::TarsOutputStream<WriterT>& _os) const" << endl;
s << TAB << "{" << endl;
INC_TAB;
for (size_t j = 0; j < member.size(); j++)
{
s << writeTo(member[j]);
}
// if (_unknownField)
// {
// s << writeUnknown();
// }
DEL_TAB;
s << TAB << "}" << endl;
///////////////////////////////////////////////////////////
s << TAB << "template<typename ReaderT>" << endl;
s << TAB << "void readFrom(" + _namespace + "::TarsInputStream<ReaderT>& _is)" << endl;
s << TAB << "{" << endl;
INC_TAB;
s << TAB << "resetDefautlt();" << endl;
for (size_t j = 0; j < member.size(); j++)
{
s << readFrom(member[j]);
}
// if (_unknownField)
// {
// s << readUnknown(member[member.size() - 1]);
// }
DEL_TAB;
s << TAB << "}" << endl;
if (_bJsonSupport)
{
s << TAB << "tars::JsonValueObjPtr writeToJson() const" << endl;
s << TAB << "{" << endl;
INC_TAB;
s << TAB << "tars::JsonValueObjPtr p = new tars::JsonValueObj();" << endl;
for (size_t j = 0; j < member.size(); j++)
{
s << writeToJson(member[j]);
}
s << TAB << "return p;" << endl;
DEL_TAB;
s << TAB << "}" << endl;
s << TAB << "string writeToJsonString() const" << endl;
s << TAB << "{" << endl;
INC_TAB;
s << TAB << "return tars::TC_Json::writeValue(writeToJson());" << endl;
DEL_TAB;
s << TAB << "}" << endl;
s << TAB << "void readFromJson(const tars::JsonValuePtr & p, bool isRequire = true)" << endl;
s << TAB << "{" << endl;
INC_TAB;
s << TAB << "resetDefautlt();" << endl;
s << TAB << "if(NULL == p.get() || p->getType() != tars::eJsonTypeObj)" << endl;
s << TAB << "{" << endl;
INC_TAB;
s << TAB << "char s[128];" << endl;
s << TAB << "snprintf(s, sizeof(s), \"read 'struct' type mismatch, get type: %d.\", (p.get() ? p->getType() : 0));" << endl;
s << TAB << "throw tars::TC_Json_Exception(s);" << endl;
DEL_TAB;
s << TAB << "}" << endl;
s << TAB << "tars::JsonValueObjPtr pObj=tars::JsonValueObjPtr::dynamicCast(p);" << endl;
for (size_t j = 0; j < member.size(); j++)
{
s << readFromJson(member[j]);
}
DEL_TAB;
s << TAB << "}" << endl;
s << TAB << "void readFromJsonString(const string & str)" << endl;
s << TAB << "{" << endl;
INC_TAB;
s << TAB << "readFromJson(tars::TC_Json::getValue(str));" << endl;
DEL_TAB;
s << TAB << "}" << endl;
}
if (_bXmlSupport)
{
s << TAB << "void setXmlFormat(bool cdata = false)" << endl;
s << TAB << "{" << endl;
INC_TAB;
s << TAB << "_cdata_format = cdata;" <<endl;
DEL_TAB;
s << TAB << "}" << endl;
s << TAB << "tars::XmlValueObjPtr writeToXml() const" << endl;
s << TAB << "{" << endl;
INC_TAB;
s << TAB << "tars::XmlValueObjPtr p = new tars::XmlValueObj();" << endl;
for(size_t j = 0; j < member.size(); j++)
{
s << writeToXml(member[j]);
}
s << TAB << "return p;" <<endl;
DEL_TAB;
s << TAB << "}" << endl;
s << TAB << "string writeToXmlString() const" << endl;
s << TAB << "{" << endl;
INC_TAB;
s << TAB << "return tars::TC_Xml::writeValue(writeToXml());" <<endl;
DEL_TAB;
s << TAB << "}" << endl;
s << TAB << "void readFromXml(const tars::XmlValuePtr & p, bool isRequire = true)" << endl;
s << TAB << "{" << endl;
INC_TAB;
s << TAB << "resetDefautlt();" << endl;
s << TAB << "if(NULL == p.get() || p->getType() != eXmlTypeObj)" << endl;
s << TAB << "{" << endl;
INC_TAB;
s << TAB << "char s[128];" << endl;
s << TAB << "snprintf(s, sizeof(s), \"read 'struct' type mismatch, get type: %d.\", p->getType());" << endl;
s << TAB << "throw TC_Xml_Exception(s);" << endl;
DEL_TAB;
s << TAB << "}" << endl;
s << TAB << "tars::XmlValueObjPtr pObj= tars::XmlValueObjPtr::dynamicCast(p);" << endl;
for(size_t j = 0; j < member.size(); j++)
{
s << readFromXml(member[j]);
}
DEL_TAB;
s << TAB << "}" << endl;
s << TAB << "void readFromXmlString(const string & str)" << endl;
s << TAB << "{" << endl;
INC_TAB;
s << TAB << "readFromXml(tars::TC_Xml::getValue(str));" <<endl;
DEL_TAB;
s << TAB << "}" << endl;
}
if (_bSqlSupport)
{
s << TAB << "tars::TC_Mysql::RECORD_DATA& toSql(tars::TC_Mysql::RECORD_DATA& _mycols) const" << endl;
s << TAB << "{" << endl;
INC_TAB;
for(size_t j = 0; j < member.size(); j++)
{
s << writeToSql(member[j]);
}
s << TAB << "return _mycols;" << endl;
DEL_TAB;
s << TAB << "}" << endl;
s << TAB << "void fromSql(tars::TC_Mysql::MysqlRecord& _mysrd)" << endl;
s << TAB << "{" << endl;
INC_TAB;
s << TAB << "resetDefautlt();" << endl;
for(size_t j = 0; j < member.size(); j++)
{
s << readFromSql(member[j]);
}
DEL_TAB;
s << TAB << "}" << endl;
}
s << TAB << "ostream& display(ostream& _os, int _level=0) const" << endl;
s << TAB << "{" << endl;
INC_TAB;
s << TAB << _namespace + "::TarsDisplayer _ds(_os, _level);" << endl;
for (size_t j = 0; j < member.size(); j++)
{
s << display(member[j]);
}
s << TAB << "return _os;" << endl;
DEL_TAB;
s << TAB << "}" << endl;
s << TAB << "ostream& displaySimple(ostream& _os, int _level=0) const" << endl;
s << TAB << "{" << endl;
INC_TAB;
s << TAB << _namespace + "::TarsDisplayer _ds(_os, _level);" << endl;
for (size_t j = 0; j < member.size(); j++)
{
s << displaySimple(member[j], (j != member.size() - 1 ? true : false));
}
s << TAB << "return _os;" << endl;
DEL_TAB;
s << TAB << "}" << endl;
DEL_TAB;
s << TAB << "public:" << endl;
INC_TAB;
//定义成员变量
for (size_t j = 0; j < member.size(); j++)
{
if (member[j]->getTypePtr()->isArray() || member[j]->getTypePtr()->isPointer()) //数组类型、指针类型需要定义长度
{
s << TAB << _namespace + "::" << "UInt32 " << member[j]->getId() << "Len" << ";" << endl;
}
s << TAB << tostr(member[j]->getTypePtr()) << " " << member[j]->getId() << toStrSuffix(member[j]) << ";" << endl;
}
if (_bXmlSupport)
{
DEL_TAB;
s << TAB << "private:" << endl;
INC_TAB;
s << TAB << "bool _cdata_format;" << endl;
}
// if (_unknownField)
// {
// s << TAB << "std::string sUnknownField;" << endl;
// }
DEL_TAB;
s << TAB << "};" << endl;
//定义==操作
s << TAB << "inline bool operator==(const " << pPtr->getId() << "&l, const " << pPtr->getId() << "&r)" << endl;
s << TAB << "{" << endl;
INC_TAB;
s << TAB << "return ";
for (size_t j = 0; j < member.size(); j++)
{
if (member[j]->getTypePtr()->isArray() || member[j]->getTypePtr()->isPointer()) //数组类型、指针类型
{
s << "!memcmp(l." << member[j]->getId() << ",r." << member[j]->getId() << ",l." << member[j]->getId() << "Len)";
}
else
{
s << "l." << member[j]->getId() << " == r." << member[j]->getId();
}
if (j != member.size() - 1)
{
s << " && ";
}
}
s << ";" << endl;
DEL_TAB;
s << TAB << "}" << endl;
//定义!=
s << TAB << "inline bool operator!=(const " << pPtr->getId() << "&l, const " << pPtr->getId() << "&r)" << endl;
s << TAB << "{" << endl;
INC_TAB;
s << TAB << "return !(l == r);" << endl;
DEL_TAB;
s << TAB << "}" << endl;
//定义 << >>
if (_bJsonSupport)
{
//重载 <<
s << TAB << "inline ostream& operator<<(ostream & os,const " << pPtr->getId() << "&r)" << endl;
s << TAB << "{" << endl;
INC_TAB;
s << TAB << "os << r.writeToJsonString();" << endl;
s << TAB << "return os;" << endl;
DEL_TAB;
s << TAB << "}" << endl;
//重载 >>
s << TAB << "inline istream& operator>>(istream& is," << pPtr->getId() << "&l)" << endl;
s << TAB << "{" << endl;
INC_TAB;
s << TAB << "std::istreambuf_iterator<char> eos;" << endl;
s << TAB << "std::string s(std::istreambuf_iterator<char>(is), eos);" << endl;
s << TAB << "l.readFromJsonString(s);" << endl;
s << TAB << "return is;" << endl;
DEL_TAB;
s << TAB << "}" << endl;
}
vector<string> key = pPtr->getKey();
//定义<
if (key.size() > 0)
{
s << TAB << "inline bool operator<(const " << pPtr->getId() << "&l, const " << pPtr->getId() << "&r)" << endl;
s << TAB << "{" << endl;
INC_TAB;
for (size_t i = 0; i < key.size(); i++)
{
s << TAB << "if(l." << key[i] << " != r." << key[i] << ") ";
for (size_t z = 0; z < member.size(); z++)
{
if (key[i] == member[z]->getId() && (member[z]->getTypePtr()->isArray() || member[z]->getTypePtr()->isPointer())) //数组类型、指针类型
{
s << "memcmp(l." << key[i] << ",r." << key[i] << ",l." << key[i] << "Len)< 0";
}
}
s << " return (l." << key[i] << " < r." << key[i] << ");" << endl;
}
s << TAB << "return false;" << endl;
DEL_TAB;
s << TAB << "}" << endl;
//定义<=
s << TAB << "inline bool operator<=(const " << pPtr->getId() << "&l, const " << pPtr->getId() << "&r)" << endl;
s << TAB << "{" << endl;
INC_TAB;
s << TAB << "return !(r < l);" << endl;
DEL_TAB;
s << TAB << "}" << endl;
//定义>
s << TAB << "inline bool operator>(const " << pPtr->getId() << "&l, const " << pPtr->getId() << "&r)" << endl;
s << TAB << "{" << endl;
INC_TAB;
s << TAB << "return r < l;" << endl;
DEL_TAB;
s << TAB << "}" << endl;
//定义>=
s << TAB << "inline bool operator>=(const " << pPtr->getId() << "&l, const " << pPtr->getId() << "&r)" << endl;
s << TAB << "{" << endl;
INC_TAB;
s << TAB << "return !(l < r);" << endl;
DEL_TAB;
s << TAB << "}" << endl;
}
return s.str();
}
/*******************************ContainerPtr********************************/
string Tars2Cpp::generateH(const ContainerPtr& pPtr) const
{
ostringstream s;
for (size_t i = 0; i < pPtr->getAllNamespacePtr().size(); i++)
{
s << generateH(pPtr->getAllNamespacePtr()[i]) << endl;
s << endl;
}
return s.str();
}
/******************************ParamDeclPtr***************************************/
string Tars2Cpp::generateH(const ParamDeclPtr& pPtr) const
{
ostringstream s;
//输出参数, 或简单类型
if (pPtr->isOut() || pPtr->getTypeIdPtr()->getTypePtr()->isSimple())
{
s << tostr(pPtr->getTypeIdPtr()->getTypePtr());
}
else
{
//结构, map, vector, string
s << "const " << tostr(pPtr->getTypeIdPtr()->getTypePtr()) << " &";
}
if (pPtr->isOut())
{
s << " &";
}
else
{
s << " ";
}
s << pPtr->getTypeIdPtr()->getId();
return s.str();
}
string Tars2Cpp::generateOutH(const ParamDeclPtr& pPtr) const
{
if (!pPtr->isOut()) return "";
ostringstream s;
//输出参数, 或简单类型
if (pPtr->getTypeIdPtr()->getTypePtr()->isSimple())
{
s << tostr(pPtr->getTypeIdPtr()->getTypePtr()) << " ";
}
else
{
//结构, map, vector, string
s << "const " << tostr(pPtr->getTypeIdPtr()->getTypePtr()) << " &";
}
s << pPtr->getTypeIdPtr()->getId();
return s.str();
}
string Tars2Cpp::generateParamDecl(const ParamDeclPtr& pPtr) const
{
ostringstream s;
if (pPtr->isOut() || pPtr->getTypeIdPtr()->getTypePtr()->isSimple())
{
s << tostr(pPtr->getTypeIdPtr()->getTypePtr());
if (pPtr->isOut()) s << " &";
else s << " ";
}
else
{
//输入参数
s << "const " << tostr(pPtr->getTypeIdPtr()->getTypePtr()) << " &";
}
s << pPtr->getTypeIdPtr()->getId();
return s.str();
}
string Tars2Cpp::generateDispatchAsync(const OperationPtr& pPtr, const string& cn) const
{
ostringstream s;
s << TAB << "if (msg->response->iRet != tars::TARSSERVERSUCCESS)" << endl
<< TAB << "{" << endl;
INC_TAB;
s << TAB << "callback_" << pPtr->getId() << "_exception(msg->response->iRet);" << endl;
s << endl;
s << TAB << "return msg->response->iRet;" << endl;
DEL_TAB;
s << TAB << "}" << endl;
s << TAB << _namespace + "::TarsInputStream<" + _namespace + "::BufferReader> _is;" << endl;
s << endl;
vector<ParamDeclPtr>& vParamDecl = pPtr->getAllParamDeclPtr();
s << TAB << "_is.setBuffer(msg->response->sBuffer);" << endl;
//对输出参数编码
if (pPtr->getReturnPtr()->getTypePtr())
{
s << TAB << tostr(pPtr->getReturnPtr()->getTypePtr()) << " " << pPtr->getReturnPtr()->getId() << generateInitValue(pPtr->getReturnPtr()) << ";" << endl;
s << readFrom(pPtr->getReturnPtr()) << endl;
}
for (size_t i = 0; i < vParamDecl.size(); i++)
{
if (vParamDecl[i]->isOut())
{
s << TAB << tostr(vParamDecl[i]->getTypeIdPtr()->getTypePtr()) << " "
<< vParamDecl[i]->getTypeIdPtr()->getId() << generateInitValue(vParamDecl[i]->getTypeIdPtr()) << ";" << endl;
s << readFrom(vParamDecl[i]->getTypeIdPtr());
}
}
//处理线程私有数据
s << TAB << "CallbackThreadData * pCbtd = CallbackThreadData::getData();" << endl;
s << TAB << "assert(pCbtd != NULL);" << endl;
s << endl;
s << TAB << "pCbtd->setResponseContext(msg->response->context);" << endl;
s << endl;
//异步回调都无返回值
s << TAB << "callback_" << pPtr->getId() << "(";
string sParams;
if (pPtr->getReturnPtr()->getTypePtr())
{
sParams = pPtr->getReturnPtr()->getId() + ", ";
}
for (size_t i = 0; i < vParamDecl.size(); i++)
{
if (vParamDecl[i]->isOut())
{
sParams += vParamDecl[i]->getTypeIdPtr()->getId() + ", ";
}
}
s << tars::TC_Common::trimright(sParams, ", ", false) << ");" << endl;
s << endl;
s << TAB << "pCbtd->delResponseContext();" << endl;
s << endl;
s << TAB << "return tars::TARSSERVERSUCCESS;" << endl;
return s.str();
}
///////////////////////////////////////////////////////////////////////////////////////////////
string Tars2Cpp::generateDispatchCoroAsync(const OperationPtr& pPtr, const string& cn) const
{
ostringstream s;
s << TAB << "if (msg->response->iRet != tars::TARSSERVERSUCCESS)" << endl
<< TAB << "{" << endl;
INC_TAB;
s << TAB << "callback_" << pPtr->getId() << "_exception(msg->response->iRet);" << endl;
s << endl;
s << TAB << "return msg->response->iRet;" << endl;
DEL_TAB;
s << TAB << "}" << endl;
s << TAB << _namespace + "::TarsInputStream<" + _namespace + "::BufferReader> _is;" << endl;
s << endl;
vector<ParamDeclPtr>& vParamDecl = pPtr->getAllParamDeclPtr();
s << TAB << "_is.setBuffer(msg->response->sBuffer);" << endl;
if(pPtr->getReturnPtr()->getTypePtr() || vParamDecl.size() >0)
{
s << TAB << "try" << endl;
s << TAB << "{" << endl;
INC_TAB;
}
//对输出参数编码
if (pPtr->getReturnPtr()->getTypePtr())
{
s << TAB << tostr(pPtr->getReturnPtr()->getTypePtr()) << " " << pPtr->getReturnPtr()->getId() << generateInitValue(pPtr->getReturnPtr()) << ";" << endl;
s << readFrom(pPtr->getReturnPtr()) << endl;
}
for (size_t i = 0; i < vParamDecl.size(); i++)
{
if (vParamDecl[i]->isOut())
{
s << TAB << tostr(vParamDecl[i]->getTypeIdPtr()->getTypePtr()) << " "
<< vParamDecl[i]->getTypeIdPtr()->getId() << generateInitValue(vParamDecl[i]->getTypeIdPtr()) << ";" << endl;
s << readFrom(vParamDecl[i]->getTypeIdPtr());
}
}
s << TAB << "setResponseContext(msg->response->context);" << endl;
s << endl;
//异步回调都无返回值
s << TAB << "callback_" << pPtr->getId() << "(";
string sParams;
if (pPtr->getReturnPtr()->getTypePtr())
{
sParams = pPtr->getReturnPtr()->getId() + ", ";
}
for (size_t i = 0; i < vParamDecl.size(); i++)
{
if (vParamDecl[i]->isOut())
{
sParams += vParamDecl[i]->getTypeIdPtr()->getId() + ", ";
}
}
s << tars::TC_Common::trimright(sParams, ", ", false) << ");" << endl;
s << endl;
if(pPtr->getReturnPtr()->getTypePtr() || vParamDecl.size() >0)
{
DEL_TAB;
s << TAB << "}" << endl;
s << TAB << "catch(std::exception &ex)" << endl;
s << TAB << "{" << endl;
INC_TAB;
s << TAB << "callback_" << pPtr->getId() << "_exception(tars::TARSCLIENTDECODEERR);" << endl;
s << endl;
s << TAB << "return tars::TARSCLIENTDECODEERR;" << endl;
DEL_TAB;
s << TAB << "}" << endl;
s << TAB << "catch(...)" << endl;
s << TAB << "{" << endl;
INC_TAB;
s << TAB << "callback_" << pPtr->getId() << "_exception(tars::TARSCLIENTDECODEERR);" << endl;
s << endl;
s << TAB << "return tars::TARSCLIENTDECODEERR;" << endl;
DEL_TAB;
s << TAB << "}" << endl;
s << endl;
}
s << TAB << "return tars::TARSSERVERSUCCESS;" << endl;
return s.str();
}
///////////////////////////////////////////////////////////////////////////////////////////////////////
string Tars2Cpp::generateHAsync(const OperationPtr& pPtr) const
{
ostringstream s;
vector<ParamDeclPtr>& vParamDecl = pPtr->getAllParamDeclPtr();
s << TAB << "virtual void " << "callback_" << pPtr->getId() << "(";
string sParams;
if (pPtr->getReturnPtr()->getTypePtr())
{
if (pPtr->getReturnPtr()->getTypePtr()->isSimple())
{
sParams = tostr(pPtr->getReturnPtr()->getTypePtr()) + " ret, ";
}
else
{
//结构, map, vector, string
sParams = "const " + tostr(pPtr->getReturnPtr()->getTypePtr()) + "& ret, ";
}
}
for (size_t i = 0; i < vParamDecl.size(); i++)
{
ParamDeclPtr& pPtr = vParamDecl[i];
if (pPtr->isOut())
{
//输出参数, 或简单类型
if (pPtr->getTypeIdPtr()->getTypePtr()->isSimple())
{
sParams += tostr(pPtr->getTypeIdPtr()->getTypePtr());
}
else
{
//结构, map, vector, string
sParams += " const " + tostr(pPtr->getTypeIdPtr()->getTypePtr()) + "&";
}
sParams += " " + pPtr->getTypeIdPtr()->getId() + ", ";
}
}
s << tars::TC_Common::trimright(sParams, ", ", false) << ")" << endl;
s << TAB << "{ throw std::runtime_error(\"callback_" << pPtr->getId() << "() override incorrect.\"); }" << endl;
s << TAB << "virtual void " << "callback_" << pPtr->getId() << "_exception(" + _namespace + "::Int32 ret)" << endl;
s << TAB << "{ throw std::runtime_error(\"callback_" << pPtr->getId() << "_exception() override incorrect.\"); }";
s << endl;
return s.str();
}
string Tars2Cpp::generateInitValue(const TypeIdPtr& pPtr) const
{
BuiltinPtr bPtr = BuiltinPtr::dynamicCast(pPtr->getTypePtr());
string init = "";
if (bPtr && Builtin::KindBool == bPtr->kind())
{
init = " = false";
}
//枚举强制类型转换在O2编译选项情况下会告警
EnumPtr ePtr = EnumPtr::dynamicCast(pPtr->getTypePtr());
if (ePtr)
{
if (pPtr->hasDefault())
{
string sid = ePtr->getSid();
init = " = " + pPtr->def() + ";";
}
else
{
vector<TypeIdPtr>& eMember = ePtr->getAllMemberPtr();
if (eMember.size() > 0)
{
string sid = ePtr->getSid();
init = " = " + sid.substr(0, sid.find_first_of("::")) + "::" + eMember[0]->getId() + ";";
}
}
}
return init;
}
//////////////////////////////////////////////////////////////////////////////////////////////////////
string Tars2Cpp::generateServantDispatch(const OperationPtr& pPtr, const string& cn) const
{
ostringstream s;
s << TAB << _namespace + "::TarsInputStream<" + _namespace + "::BufferReader> _is;" << endl;
s << TAB << "_is.setBuffer(_current->getRequestBuffer());" << endl;
vector<ParamDeclPtr>& vParamDecl = pPtr->getAllParamDeclPtr();
string routekey;
for(size_t i = 0; i < vParamDecl.size(); i++)
{
s << TAB << tostr(vParamDecl[i]->getTypeIdPtr()->getTypePtr()) << " "<< vParamDecl[i]->getTypeIdPtr()->getId()
<< generateInitValue(vParamDecl[i]->getTypeIdPtr()) << ";" << endl;
if (routekey.empty() && vParamDecl[i]->isRouteKey())
{
routekey = vParamDecl[i]->getTypeIdPtr()->getId();
}
}
s << TAB << "if (_current->getRequestVersion() == TUPVERSION)" << endl;
s << TAB << "{" << endl;
INC_TAB;
s << TAB << "UniAttribute<" + _namespace + "::BufferWriterVector, " + _namespace + "::BufferReader> tarsAttr;" << endl;
s << TAB << "tarsAttr.setVersion(_current->getRequestVersion());" << endl;
s << TAB << "tarsAttr.decode(_current->getRequestBuffer());" << endl;
for(size_t i = 0; i < vParamDecl.size(); i++)
{
string sParamName = vParamDecl[i]->getTypeIdPtr()->getId();
string sEnum2Int = (EnumPtr::dynamicCast(vParamDecl[i]->getTypeIdPtr()->getTypePtr())) ? "(" + _namespace + "::Int32)" : "";
if (!vParamDecl[i]->isOut())
{
//枚举类型转成int
if (EnumPtr::dynamicCast(vParamDecl[i]->getTypeIdPtr()->getTypePtr()))
{
s << TAB << sParamName << " = (" << tostr(vParamDecl[i]->getTypeIdPtr()->getTypePtr())
<< ") tarsAttr.get<" + _namespace + "::Int32>(\"" << sParamName << "\");" << endl;
}
else
{
s << TAB << "tarsAttr.get(\"" << sParamName << "\", " << sParamName << ");" << endl;
}
}
else
{
//枚举类型转成int
if (EnumPtr::dynamicCast(vParamDecl[i]->getTypeIdPtr()->getTypePtr()))
{
s << TAB << sParamName << " = (" << tostr(vParamDecl[i]->getTypeIdPtr()->getTypePtr())
<< ") tarsAttr.getByDefault<" + _namespace + "::Int32>(\"" << sParamName << "\", " << sEnum2Int << sParamName << ");" << endl;
}
else
{
s << TAB << "tarsAttr.getByDefault(\"" << sParamName << "\", " << sEnum2Int << sParamName << ", "
<< sEnum2Int << sParamName << ");" << endl;
}
}
}
DEL_TAB;
s << TAB << "}" << endl;
// 支持JSON协议分发
if (_bJsonSupport && tars::TC_Common::matchPeriod(pPtr->getId(), _vJsonIntf))
{
s << TAB << "else if (_current->getRequestVersion() == JSONVERSION)" << endl;
s << TAB << "{" << endl;
INC_TAB;
s << TAB << _namespace << "::JsonValueObjPtr _jsonPtr = " << _namespace << "::JsonValueObjPtr::dynamicCast(" << _namespace << "::TC_Json::getValue(_current->getRequestBuffer()));" << endl;
for(size_t i = 0; i < vParamDecl.size(); i++)
{
string sParamName = vParamDecl[i]->getTypeIdPtr()->getId();
string sEnum2Int = (EnumPtr::dynamicCast(vParamDecl[i]->getTypeIdPtr()->getTypePtr())) ? "(" + _namespace + "::Int32)" : "";
if (!vParamDecl[i]->isOut())
{
// tars::JsonInput::readJson(uin, _jsonPtr->value["uin"], true); 枚举类型转成int
s << TAB << _namespace << "::JsonInput::readJson(" << sParamName << ", _jsonPtr->value[\"" << sParamName << "\"], true);" << endl;
}
else
{
s << TAB << _namespace << "::JsonInput::readJson(" << sParamName << ", _jsonPtr->value[\"" << sParamName << "\"], false);" << endl;
}
}
DEL_TAB;
s << TAB << "}" << endl;
}
// 支持XML协议分发
if (_bXmlSupport && tars::TC_Common::matchPeriod(pPtr->getId(), _vXmlIntf))
{
s << TAB << "else if (_current->getRequestVersion() == XMLVERSION)" << endl;
s << TAB << "{" << endl;
INC_TAB;
s << TAB << "tars::XmlValueObjPtr _xmlPtr = tars::XmlValueObjPtr::dynamicCast(tars::TC_Xml::getValue(_current->getRequestBuffer()));" << endl;
for(size_t i = 0; i < vParamDecl.size(); i++)
{
string sParamName = vParamDecl[i]->getTypeIdPtr()->getId();
if(!vParamDecl[i]->isOut())
{
//枚举类型转成int
s << TAB << "tars::XmlInput::readXml(" << sParamName << ", _xmlPtr->value[\"" << sParamName << "\"], true);" << endl;
}
else
{
s << TAB << "tars::XmlInput::readXml(" << sParamName << ", _xmlPtr->value[\"" << sParamName << "\"], false);" << endl;
}
}
DEL_TAB;
s << TAB << "}" << endl;
}
s << TAB << "else" << endl;
s << TAB << "{" << endl;
INC_TAB;
//普通tars请求
for (size_t i = 0; i < vParamDecl.size(); i++)
{
s << readFrom(vParamDecl[i]->getTypeIdPtr(), (!vParamDecl[i]->isOut()));
}
DEL_TAB;
s << TAB << "}" << endl;
if(pPtr->getReturnPtr()->getTypePtr())
{
s << TAB << tostr(pPtr->getReturnPtr()->getTypePtr()) << " " << pPtr->getReturnPtr()->getId() << " = " << pPtr->getId() << "(";
}
else
{
s << TAB << pPtr->getId() << "(";
}
for(size_t i = 0; i < vParamDecl.size(); i++)
{
s << vParamDecl[i]->getTypeIdPtr()->getId();
if(i != vParamDecl.size() - 1)
s << ",";
else
s << ", _current);" << endl;
}
if (vParamDecl.size() == 0)
{
s << "_current);" << endl;
}
s << TAB << "if(_current->isResponse())" << endl;
s << TAB << "{" << endl;
INC_TAB;
s << TAB << "if (_current->getRequestVersion() == TUPVERSION )" << endl;
s << TAB << "{" << endl;
INC_TAB;
s << TAB << "UniAttribute<" + _namespace + "::BufferWriterVector, " + _namespace + "::BufferReader> tarsAttr;" << endl;
s << TAB << "tarsAttr.setVersion(_current->getRequestVersion());" << endl;
if(pPtr->getReturnPtr()->getTypePtr())
{
string sEnum2Int = (EnumPtr::dynamicCast(pPtr->getReturnPtr()->getTypePtr())) ? "(" + _namespace + "::Int32)" : "";
s << TAB << "tarsAttr.put(\"\", " << sEnum2Int << "_ret);" << endl;
}
for (size_t i = 0; i < vParamDecl.size(); i++)
{
string sParamName = vParamDecl[i]->getTypeIdPtr()->getId();
string sEnum2Int = (EnumPtr::dynamicCast(vParamDecl[i]->getTypeIdPtr()->getTypePtr())) ? "(" + _namespace + "::Int32)" : "";
if (vParamDecl[i]->isOut())
{
s << TAB << "tarsAttr.put(\"" << sParamName << "\", " << sEnum2Int << sParamName << ");" << endl;
}
}
s << TAB << "tarsAttr.encode(_sResponseBuffer);" << endl;
DEL_TAB;
s << TAB << "}" << endl;
// 支持JSON协议分发
if (_bJsonSupport && tars::TC_Common::matchPeriod(pPtr->getId(), _vJsonIntf))
{
s << TAB << "else if (_current->getRequestVersion() == JSONVERSION)" << endl;
s << TAB << "{" << endl;
INC_TAB;
s << TAB << _namespace << "::JsonValueObjPtr _p = new " << _namespace << "::JsonValueObj();" << endl;
for(size_t i = 0; i < vParamDecl.size(); i++)
{
string sParamName = vParamDecl[i]->getTypeIdPtr()->getId();
if (vParamDecl[i]->isOut())
{
s << TAB << "_p->value[\"" << sParamName << "\"] = "<< _namespace << "::JsonOutput::writeJson(" << sParamName << ");" << endl;
}
}
if (pPtr->getReturnPtr()->getTypePtr())
{
BuiltinPtr retPtr = BuiltinPtr::dynamicCast(pPtr->getReturnPtr()->getTypePtr());
if (retPtr->kind() >= Builtin::KindBool && retPtr->kind() <= Builtin::KindLong)
{
s << TAB << "_p->value[\"ret\"] = "<< _namespace << "::JsonOutput::writeJson(" << pPtr->getReturnPtr()->getId() << ");" << endl;
}
}
s << TAB << _namespace << "::TC_Json::writeValue(_p, _sResponseBuffer);" << endl;
DEL_TAB;
s << TAB << "}" << endl;
}
// 支持XML协议分发
if (_bXmlSupport && tars::TC_Common::matchPeriod(pPtr->getId(), _vXmlIntf))
{
s << TAB << "else if (_current->getRequestVersion() == XMLVERSION)" << endl;
s << TAB << "{" << endl;
INC_TAB;
s << TAB << "tars::XmlValueObjPtr _p = new tars::XmlValueObj();" << endl;
for(size_t i = 0; i < vParamDecl.size(); i++)
{
string sParamName = vParamDecl[i]->getTypeIdPtr()->getId();
if (vParamDecl[i]->isOut())
{
s << TAB << "_p->value[\"" << sParamName << "\"] = tars::XmlOutput::writeXml(" << sParamName << ");" << endl;
}
}
if (pPtr->getReturnPtr()->getTypePtr())
{
BuiltinPtr retPtr = BuiltinPtr::dynamicCast(pPtr->getReturnPtr()->getTypePtr());
if (retPtr->kind() >= Builtin::KindBool && retPtr->kind() <= Builtin::KindLong)
{
s << TAB << "_p->value[\"ret\"] = tars::XmlOutput::writeXml(" << pPtr->getReturnPtr()->getId() << ");" << endl;
}
}
s << TAB << "tars::TC_Xml::writeValue(_p, _sResponseBuffer);" << endl;
DEL_TAB;
s << TAB << "}" << endl;
}
s << TAB << "else" << endl;
//普通tars调用输出参数
s << TAB << "{" << endl;
INC_TAB;
s << TAB << _namespace + "::TarsOutputStream<" + _namespace + "::BufferWriterVector> _os;" << endl;
if (pPtr->getReturnPtr()->getTypePtr())
{
s << writeTo(pPtr->getReturnPtr());
}
//解码输出参数
for (size_t i = 0; i < vParamDecl.size(); i++)
{
if (vParamDecl[i]->isOut())
{
s << writeTo(vParamDecl[i]->getTypeIdPtr());
}
}
s << TAB << "_os.swap(_sResponseBuffer);" << endl;
DEL_TAB;
s << TAB << "}" << endl;
DEL_TAB;
s << TAB << "}" << endl;
s << TAB << "return tars::TARSSERVERSUCCESS;" << endl;
return s.str();
}
/*
string Tars2Cpp::promiseReadFrom(const TypeIdPtr &pPtr, bool bIsRequire) const
{
ostringstream s;
if(EnumPtr::dynamicCast(pPtr->getTypePtr()))
{
//枚举强制类型转换在O2编译选项情况下会告警
string tmp = _namespace + "::Int32 eTemp" + TC_Common::tostr(pPtr->getTag()) + generateInitValue(pPtr);
s << TAB << tmp <<endl;
s << TAB << "_is.read(eTemp"<<TC_Common::tostr(pPtr->getTag());
}
else if(pPtr->getTypePtr()->isArray())
{
s << TAB << "_is.read(ptr->"<< pPtr->getId()<<", "<<getSuffix(pPtr)<<", "<< pPtr->getId() << "Len";
}
else if(pPtr->getTypePtr()->isPointer())
{
s << TAB << pPtr->getId() <<" = ("<<tostr(pPtr->getTypePtr())<<")_is.cur();"<<endl;
s << TAB << "_is.read(ptr->"<< pPtr->getId()<<", _is.left(), "<< pPtr->getId() << "Len";
}
else
{
s << TAB << "_is.read(ptr->"<< pPtr->getId();
}
s << ", " << pPtr->getTag() << ", " << ((pPtr->isRequire() && bIsRequire)?"true":"false") << ");" << endl;
if(EnumPtr::dynamicCast(pPtr->getTypePtr()))
{
s << TAB << "ptr->" << pPtr->getId() << " = (" <<tostr(pPtr->getTypePtr()) <<")eTemp"<<TC_Common::tostr(pPtr->getTag())<<";"<<endl;
}
if(pPtr->getTypePtr()->isPointer())
s << TAB <<"_is.mapBufferSkip("<<pPtr->getId() << "Len);"<<endl;
return s.str();
}
bool Tars2Cpp::isPromiseDispatchInitValue(const TypeIdPtr &pPtr) const
{
BuiltinPtr bPtr = BuiltinPtr::dynamicCast(pPtr->getTypePtr());
string init = "";
if(bPtr && Builtin::KindBool == bPtr->kind())
{
return true;
}
//枚举强制类型转换在O2编译选项情况下会告警
EnumPtr ePtr = EnumPtr::dynamicCast(pPtr->getTypePtr());
if(ePtr)
{
if(pPtr->hasDefault())
{
return true;
}
else
{
vector<TypeIdPtr>& eMember = ePtr->getAllMemberPtr();
if(eMember.size() > 0)
{
return true;
}
}
}
return false;
}
*/
string Tars2Cpp::generateHAsync(const OperationPtr& pPtr, const string& cn) const
{
ostringstream s;
//生成函数声明
s << TAB << "void async_" << pPtr->getId() << "(";
s << cn << "PrxCallbackPtr callback,";
vector<ParamDeclPtr>& vParamDecl = pPtr->getAllParamDeclPtr();
string routekey = "";
for (size_t i = 0; i < vParamDecl.size(); i++)
{
if (!vParamDecl[i]->isOut() )
{
s << generateParamDecl(vParamDecl[i]) << ",";
}
if (routekey.empty() && vParamDecl[i]->isRouteKey())
{
routekey = vParamDecl[i]->getTypeIdPtr()->getId();
}
}
s << "const map<string, string>& context = TARS_CONTEXT())";
s << endl;
s << TAB << "{" << endl;
INC_TAB;
if (_tarsMaster)
{
s << TAB << "this->tars_setMasterFlag(true);" << endl;
}
s << TAB << _namespace + "::TarsOutputStream<" + _namespace + "::BufferWriterVector> _os;" << endl;
for (size_t i = 0; i < vParamDecl.size(); i++)
{
if (vParamDecl[i]->isOut())
{
continue;
}
s << writeTo(vParamDecl[i]->getTypeIdPtr());
}
s << TAB << "std::map<string, string> _mStatus;" << endl;
if (!routekey.empty())
{
ostringstream os;
os << routekey;
s << TAB << "_mStatus.insert(std::make_pair(ServantProxy::STATUS_GRID_KEY, " << os.str() << "));" << endl;
}
s << TAB << "tars_invoke_async(tars::TARSNORMAL,\"" << pPtr->getId() << "\", _os, context, _mStatus, callback);" << endl;
DEL_TAB;
s << TAB << "}" << endl;
s << TAB << endl;
/*
//promise异步的函数声明
string sStruct = pPtr->getId();
s << TAB << "tars::Future< " << cn <<"PrxCallbackPromise::Promise" << sStruct << "Ptr > promise_async_" << pPtr->getId() << "(";
for(size_t i = 0; i < vParamDecl.size(); i++)
{
if (!vParamDecl[i]->isOut())
{
s << generateParamDecl(vParamDecl[i]) << ",";
}
}
s << "const map<string, string>& context)" << endl;
s << TAB << "{" << endl;
INC_TAB;
if (_tarsMaster)
{
s << TAB << "this->tars_setMasterFlag(true);" << endl;
}
s << TAB << "tars::Promise< " << cn <<"PrxCallbackPromise::Promise" << sStruct << "Ptr > promise;" << endl;
s << TAB << cn << "PrxCallbackPromisePtr callback = new " << cn << "PrxCallbackPromise(promise);" << endl;
s << endl;
s << TAB << _namespace + "::TarsOutputStream<" + _namespace + "::BufferWriterVector> _os;" << endl;
for(size_t i = 0; i < vParamDecl.size(); i++)
{
if(vParamDecl[i]->isOut())
{
continue;
}
s << writeTo(vParamDecl[i]->getTypeIdPtr());
}
s << TAB << "std::map<string, string> _mStatus;" << endl;
if (!routekey.empty())
{
ostringstream os;
os << routekey;
s << TAB << "_mStatus.insert(std::make_pair(ServantProxy::STATUS_GRID_KEY, " << os.str() << "));" << endl;
}
s << TAB << "tars_invoke_async(tars::TARSNORMAL,\"" << pPtr->getId() << "\", _os, context, _mStatus, callback);" << endl;
s << endl;
s << TAB << "return promise.getFuture();" << endl;
DEL_TAB;
s << TAB << "}" << endl;
s << endl;
*/
//协程并行异步的函数声明
s << TAB << "void coro_" << pPtr->getId() << "(";
s << cn << "CoroPrxCallbackPtr callback,";
for (size_t i = 0; i < vParamDecl.size(); i++)
{
if (!vParamDecl[i]->isOut())
{
s << generateParamDecl(vParamDecl[i]) << ",";
}
}
s << "const map<string, string>& context = TARS_CONTEXT())";
s << endl;
s << TAB << "{" << endl;
INC_TAB;
if (_tarsMaster)
{
s << TAB << "this->tars_setMasterFlag(true);" << endl;
}
s << TAB << _namespace + "::TarsOutputStream<" + _namespace + "::BufferWriterVector> _os;" << endl;
for (size_t i = 0; i < vParamDecl.size(); i++)
{
if (vParamDecl[i]->isOut())
{
continue;
}
s << writeTo(vParamDecl[i]->getTypeIdPtr());
}
s << TAB << "std::map<string, string> _mStatus;" << endl;
if (!routekey.empty())
{
ostringstream os;
os << routekey;
s << TAB << "_mStatus.insert(std::make_pair(ServantProxy::STATUS_GRID_KEY, " << os.str() << "));" << endl;
}
s << TAB << "tars_invoke_async(tars::TARSNORMAL,\"" << pPtr->getId() << "\", _os, context, _mStatus, callback, true);" << endl;
DEL_TAB;
s << TAB << "}" << endl;
return s.str();
}
/////////////////////////////////////////////////////////////////////////////////////////////////
string Tars2Cpp::generateH(const OperationPtr& pPtr, bool bVirtual, const string& interfaceId) const
{
ostringstream s;
vector<ParamDeclPtr>& vParamDecl = pPtr->getAllParamDeclPtr();
s << TAB;
if (bVirtual) s << "virtual ";
s << tostr(pPtr->getReturnPtr()->getTypePtr()) << " " << pPtr->getId() << "(";
string routekey = "";
for (size_t i = 0; i < vParamDecl.size(); i++)
{
s << generateH(vParamDecl[i]) << ",";
if (routekey.empty() && vParamDecl[i]->isRouteKey())
{
routekey = vParamDecl[i]->getTypeIdPtr()->getId();
}
}
if (bVirtual)
{
s << "tars::TarsCurrentPtr current) = 0;";
}
else
{
s << "const map<string, string> &context = TARS_CONTEXT(),map<string, string> * pResponseContext = NULL)";
s << endl;
s << TAB << "{" << endl;
INC_TAB;
if (_tarsMaster)
{
s << TAB << "this->tars_setMasterFlag(true);" << endl;
}
s << TAB << _namespace + "::TarsOutputStream<" + _namespace + "::BufferWriterVector> _os;" << endl;
for (size_t i = 0; i < vParamDecl.size(); i++)
{
//if(vParamDecl[i]->isOut()) continue;
s << writeTo(vParamDecl[i]->getTypeIdPtr());
}
// s << TAB << "" + _namespace + "::ResponsePacket rep;" << endl;
s << TAB << "std::map<string, string> _mStatus;" << endl;
if (!routekey.empty())
{
ostringstream os;
os << routekey;
s << TAB << "_mStatus.insert(std::make_pair(ServantProxy::STATUS_GRID_KEY, " << os.str() << "));" << endl;
}
// s << TAB << "tars_invoke(tars::TARSNORMAL,\"" << pPtr->getId() << "\", _os.getByteBuffer(), context, _mStatus, rep);" << endl;
s << TAB << "shared_ptr<" + _namespace + "::ResponsePacket> rep = tars_invoke(tars::TARSNORMAL,\"" << pPtr->getId() << "\", _os, context, _mStatus);" << endl;
s << TAB << "if(pResponseContext)" << endl;
s << TAB << "{" << endl;
INC_TAB;
s << TAB << "pResponseContext->swap(rep->context);" << endl;
// s << TAB << "*pResponseContext = rep.context;" << endl;
DEL_TAB;
s << TAB << "}" << endl;
s << endl;
if (vParamDecl.size() > 0 || pPtr->getReturnPtr()->getTypePtr())
{
s << TAB << _namespace + "::TarsInputStream<" + _namespace + "::BufferReader> _is;" << endl;
s << TAB << "_is.setBuffer(rep->sBuffer);" << endl;
if (pPtr->getReturnPtr()->getTypePtr())
{
s << TAB << tostr(pPtr->getReturnPtr()->getTypePtr()) << " " << pPtr->getReturnPtr()->getId() << generateInitValue(pPtr->getReturnPtr()) << ";" << endl;
s << readFrom(pPtr->getReturnPtr());
}
for (size_t i = 0; i < vParamDecl.size(); i++)
{
if (vParamDecl[i]->isOut())
{
s << readFrom(vParamDecl[i]->getTypeIdPtr());
}
}
if (pPtr->getReturnPtr()->getTypePtr())
{
s << TAB << "return " << pPtr->getReturnPtr()->getId() << ";" << endl;
}
}
DEL_TAB;
s << TAB << "}" << endl;
}
s << endl;
if (bVirtual)
{
//异步回调
s << TAB << "static void async_response_" << pPtr->getId() << "(tars::TarsCurrentPtr current";
if (pPtr->getReturnPtr()->getTypePtr())
{
s << ", ";
if (pPtr->getReturnPtr()->getTypePtr()->isSimple())
{
s << tostr(pPtr->getReturnPtr()->getTypePtr()) << " ";
}
else
{
//结构, map, vector, string
s << "const " << tostr(pPtr->getReturnPtr()->getTypePtr()) << " &";
}
s << pPtr->getReturnPtr()->getId();
}
for (size_t i = 0; i < vParamDecl.size(); i++)
{
if(!vParamDecl[i]->isOut())
continue;
s << ", ";
s << generateOutH(vParamDecl[i]);
}
s << ")" << endl;
s << TAB << "{" << endl;
INC_TAB;
s << TAB << "if (current->getRequestVersion() == TUPVERSION )" << endl;
s << TAB << "{" << endl;
INC_TAB;
s << TAB << "UniAttribute<" + _namespace + "::BufferWriterVector, " + _namespace + "::BufferReader> tarsAttr;" << endl;
s << TAB << "tarsAttr.setVersion(current->getRequestVersion());" << endl;
if(pPtr->getReturnPtr()->getTypePtr())
{
string sEnum2Int = (EnumPtr::dynamicCast(pPtr->getReturnPtr()->getTypePtr())) ? "(" + _namespace + "::Int32)" : "";
s << TAB << "tarsAttr.put(\"\", " << sEnum2Int << "_ret);" << endl;
}
for(size_t i = 0; i < vParamDecl.size(); i++)
{
string sParamName = vParamDecl[i]->getTypeIdPtr()->getId();
string sEnum2Int = (EnumPtr::dynamicCast(vParamDecl[i]->getTypeIdPtr()->getTypePtr())) ? "(" + _namespace + "::Int32)" : "";
if(vParamDecl[i]->isOut())
{
s << TAB << "tarsAttr.put(\"" << sParamName << "\", " << sEnum2Int << sParamName << ");" << endl;
}
}
s << endl;
s << TAB << "vector<char> sTupResponseBuffer;" << endl;
s << TAB << "tarsAttr.encode(sTupResponseBuffer);"<< endl;
s << TAB << "current->sendResponse(tars::TARSSERVERSUCCESS, sTupResponseBuffer);" << endl;
DEL_TAB;
s << TAB << "}" << endl;
s << TAB << "else" << endl;
s << TAB << "{" << endl;
INC_TAB;
s << TAB << _namespace + "::TarsOutputStream<" + _namespace + "::BufferWriterVector> _os;" << endl;
if(pPtr->getReturnPtr()->getTypePtr())
{
s << writeTo(pPtr->getReturnPtr()) << endl;
}
for(size_t i = 0; i < vParamDecl.size(); i++)
{
if(!vParamDecl[i]->isOut())
continue;
s << writeTo(vParamDecl[i]->getTypeIdPtr()) << endl;
}
//s << TAB << "current->sendResponse(tars::TARSSERVERSUCCESS, string(_os.getBuffer(), _os.getLength()));" << endl;
s << TAB << "current->sendResponse(tars::TARSSERVERSUCCESS, _os.getByteBuffer());" << endl;
DEL_TAB;
s << TAB << "}" << endl;
DEL_TAB;
s << TAB << "}" << endl;
}
return s.str();
}
/**
* 对接口名排序
*/
struct SortOperation {
bool operator ()(const OperationPtr& o1, const OperationPtr& o2)
{
return o1->getId() < o2->getId();
}
};
/*
string Tars2Cpp::generateHPromiseAsync(const InterfacePtr &pInter, const OperationPtr &pPtr) const
{
ostringstream s;
string sStruct = pPtr->getId();
vector<ParamDeclPtr>& vParamDecl = pPtr->getAllParamDeclPtr();
////////
DEL_TAB;
s << TAB << "public:" << endl;
INC_TAB;
s << TAB << "struct Promise" << sStruct << ": virtual public TC_HandleBase" << endl;
s << TAB << "{" << endl;
s << TAB << "public:" << endl;
INC_TAB;
if (pPtr->getReturnPtr()->getTypePtr())
{
s << TAB << tostr(pPtr->getReturnPtr()->getTypePtr()) << " _ret;" << endl;
}
for(size_t i = 0; i < vParamDecl.size(); i++)
{
ParamDeclPtr& pPtr = vParamDecl[i];
if (pPtr->isOut())
{
s << TAB << tostr(pPtr->getTypeIdPtr()->getTypePtr()) << " " << pPtr->getTypeIdPtr()->getId() << ";" << endl;
}
}
s << TAB << "map<std::string, std::string> _mRspContext;" << endl;
DEL_TAB;
s << TAB << "};" << endl;
s << TAB << endl;
s << TAB << "typedef tars::TC_AutoPtr< " << pInter->getId() << "PrxCallbackPromise::Promise" << sStruct << " > Promise" << sStruct << "Ptr;" << endl;
s << endl;
s << TAB << pInter->getId() << "PrxCallbackPromise(const tars::Promise< " << pInter->getId() << "PrxCallbackPromise::Promise" << sStruct << "Ptr > &promise)" << endl;
s << TAB << ": _promise_" << sStruct << "(promise)" << endl;
s << TAB << "{}" << endl;
s << TAB << endl;
s << TAB << "virtual void " << "callback_" << pPtr->getId() << "(const " << pInter->getId() << "PrxCallbackPromise::Promise" << sStruct << "Ptr &ptr)" << endl;
s << TAB << "{" << endl;
INC_TAB;
s << TAB << "_promise_" << sStruct << ".setValue(ptr);" << endl;
DEL_TAB;
s << TAB << "}" << endl;
s << TAB << "virtual void " << "callback_" << pPtr->getId() << "_exception(" + _namespace + "::Int32 ret)" << endl;
s << TAB << "{" << endl;
INC_TAB;
s << TAB << "std::string str(\"\");" << endl;
s << TAB << "str += \"Function:" << pPtr->getId() << "_exception|Ret:\";" << endl;
s << TAB << "str += TC_Common::tostr(ret);" << endl;
s << TAB << "_promise_" << sStruct << ".setException(tars::copyException(str, ret));" << endl;
DEL_TAB;
s << TAB << "}" << endl;
s << endl;
DEL_TAB;
s << TAB << "protected:" << endl;
INC_TAB;
s << TAB << "tars::Promise< " << pInter->getId() << "PrxCallbackPromise::Promise" << sStruct << "Ptr > _promise_" << sStruct << ";" << endl;
return s.str();
}
string Tars2Cpp::generateDispatchPromiseAsync(const OperationPtr &pPtr, const string &cn) const
{
ostringstream s;
s << TAB << "if (msg->response->iRet != tars::TARSSERVERSUCCESS)" << endl
<< TAB << "{" << endl;
INC_TAB;
s << TAB << "callback_" << pPtr->getId() << "_exception(msg->response->iRet);" << endl;
s << endl;
s << TAB << "return msg->response->iRet;" << endl;
DEL_TAB;
s << TAB << "}" << endl;
s << TAB << _namespace + "::TarsInputStream<" + _namespace + "::BufferReader> _is;" << endl;
s << endl;
vector<ParamDeclPtr>& vParamDecl = pPtr->getAllParamDeclPtr();
s << TAB << "_is.setBuffer(msg->response->sBuffer);" << endl;
s << endl;
string sStruct = pPtr->getId();
s << TAB << cn << "PrxCallbackPromise::Promise" << sStruct << "Ptr ptr = new "<< cn << "PrxCallbackPromise::Promise" << sStruct << "();" << endl;
s << endl;
if(pPtr->getReturnPtr()->getTypePtr() || vParamDecl.size() >0)
{
s << TAB << "try" << endl;
s << TAB << "{" << endl;
INC_TAB;
}
//对输出参数编码
if(pPtr->getReturnPtr()->getTypePtr())
{
if(isPromiseDispatchInitValue(pPtr->getReturnPtr()))
s << TAB << "ptr->_ret " << generateInitValue(pPtr->getReturnPtr())<< ";"<< endl;
s << promiseReadFrom(pPtr->getReturnPtr()) << endl;
}
for(size_t i = 0; i < vParamDecl.size(); i++)
{
if(vParamDecl[i]->isOut())
{
if(isPromiseDispatchInitValue(vParamDecl[i]->getTypeIdPtr()))
s << TAB << "ptr->" << vParamDecl[i]->getTypeIdPtr()->getId() << generateInitValue(vParamDecl[i]->getTypeIdPtr())<< ";" << endl;
s << promiseReadFrom(vParamDecl[i]->getTypeIdPtr());
}
}
if(pPtr->getReturnPtr()->getTypePtr() || vParamDecl.size() >0)
{
DEL_TAB;
s << TAB << "}" << endl;
s << TAB << "catch(std::exception &ex)" << endl;
s << TAB << "{" << endl;
INC_TAB;
s << TAB << "callback_" << pPtr->getId() << "_exception(tars::TARSCLIENTDECODEERR);" << endl;
s << endl;
s << TAB << "return tars::TARSCLIENTDECODEERR;" << endl;
DEL_TAB;
s << TAB << "}" << endl;
s << TAB << "catch(...)" << endl;
s << TAB << "{" << endl;
INC_TAB;
s << TAB << "callback_" << pPtr->getId() << "_exception(tars::TARSCLIENTDECODEERR);" << endl;
s << endl;
s << TAB << "return tars::TARSCLIENTDECODEERR;" << endl;
DEL_TAB;
s << TAB << "}" << endl;
s << endl;
}
s << TAB << "ptr->_mRspContext = msg->response->context;" << endl;
s << endl;
s << TAB << "callback_" << pPtr->getId() << "(ptr);" << endl;
s << endl;
s << TAB << "return tars::TARSSERVERSUCCESS;" << endl;
return s.str();
}
*/
/******************************InterfacePtr***************************************/
string Tars2Cpp::generateH(const InterfacePtr &pPtr, const NamespacePtr &nPtr) const
{
ostringstream s;
vector<OperationPtr>& vOperation = pPtr->getAllOperationPtr();
std::sort(vOperation.begin(), vOperation.end(), SortOperation());
//生成异步回调Proxy
s << TAB << "/* callback of async proxy for client */" << endl;
s << TAB << "class " << pPtr->getId() << "PrxCallback: public tars::ServantProxyCallback" << endl;
s << TAB << "{" << endl;
s << TAB << "public:" << endl;
INC_TAB;
s << TAB << "virtual ~" << pPtr->getId() << "PrxCallback(){}" << endl;
for (size_t i = 0; i < vOperation.size(); i++)
{
s << generateHAsync(vOperation[i]) << endl;
}
DEL_TAB;
s << TAB << "public:" << endl;
INC_TAB;
s << TAB << "virtual const map<std::string, std::string> & getResponseContext() const" << endl;
s << TAB << "{" << endl;
INC_TAB;
s << TAB << "CallbackThreadData * pCbtd = CallbackThreadData::getData();" << endl;
s << TAB << "assert(pCbtd != NULL);" << endl;
s << endl;
s << TAB << "if(!pCbtd->getContextValid())" << endl;
s << TAB << "{" << endl;
INC_TAB;
s << TAB << "throw TC_Exception(\"cann't get response context\");" << endl;
DEL_TAB;
s << TAB << "}" << endl;
s << TAB << "return pCbtd->getResponseContext();" << endl;
DEL_TAB;
s << TAB << "}" << endl;
s << endl;
DEL_TAB;
s << TAB << "public:" << endl;
INC_TAB;
s << TAB << "virtual int onDispatch(tars::ReqMessagePtr msg)" << endl;
//生成异步回调接口
s << TAB << "{" << endl;
INC_TAB;
string dname = "__" + pPtr->getId() + "_all";
string dispatch = "static ::std::string " + dname;
s << TAB << dispatch << "[]=" << endl;
s << TAB << "{" << endl;
INC_TAB;
for (size_t i = 0; i < vOperation.size(); i++)
{
s << TAB << "\"" << vOperation[i]->getId() << "\"";
if (i != vOperation.size() - 1)
{
s << ",";
}
s << endl;
}
DEL_TAB;
s << TAB << "};" << endl;
s << TAB << "pair<string*, string*> r = equal_range(" << dname << ", " << dname << "+" << vOperation.size() << ", string(msg->request.sFuncName));" << endl;
s << TAB << "if(r.first == r.second) return tars::TARSSERVERNOFUNCERR;" << endl;
s << TAB << "switch(r.first - " << dname << ")" << endl;
s << TAB << "{" << endl;
INC_TAB;
for (size_t i = 0; i < vOperation.size(); i++)
{
s << TAB << "case " << i << ":" << endl;
s << TAB << "{" << endl;
INC_TAB;
s << generateDispatchAsync(vOperation[i], pPtr->getId()) << endl;
DEL_TAB;
s << TAB << "}" << endl;
}
DEL_TAB;
s << TAB << "}" << endl;
s << TAB << "return tars::TARSSERVERNOFUNCERR;" << endl;
DEL_TAB;
s << TAB << "}" << endl;
s << endl;
DEL_TAB;
s << TAB << "};" << endl;
s << TAB << "typedef tars::TC_AutoPtr<" << pPtr->getId() << "PrxCallback> " << pPtr->getId() << "PrxCallbackPtr;" << endl;
s << endl;
/*
//生成promise异步回调Proxy
s << TAB << "//callback of promise async proxy for client" << endl;
s << TAB << "class " << pPtr->getId() << "PrxCallbackPromise: public tars::ServantProxyCallback" << endl;
s << TAB << "{" << endl;
s << TAB << "public:" << endl;
INC_TAB;
s << TAB << "virtual ~" << pPtr->getId() << "PrxCallbackPromise(){}" << endl;
for(size_t i = 0; i < vOperation.size(); i++)
{
s << generateHPromiseAsync(pPtr, vOperation[i]) << endl;
}
DEL_TAB;
s << TAB << "public:" << endl;
INC_TAB;
s << TAB << "virtual int onDispatch(tars::ReqMessagePtr msg)" << endl;
s << TAB << "{" << endl;
INC_TAB;
dname = "__" + pPtr->getId() + "_all";
dispatch = "static ::std::string " + dname;
s << TAB << dispatch << "[]=" << endl;
s << TAB << "{" << endl;
INC_TAB;
for (size_t i = 0; i < vOperation.size(); i++)
{
s << TAB << "\"" << vOperation[i]->getId() << "\"";
if (i != vOperation.size() - 1)
{
s << ",";
}
s << endl;
}
DEL_TAB;
s << TAB << "};" << endl;
s << endl;
s << TAB << "pair<string*, string*> r = equal_range(" << dname << ", " << dname << "+" << vOperation.size() << ", string(msg->request.sFuncName));" << endl;
s << TAB << "if(r.first == r.second) return tars::TARSSERVERNOFUNCERR;" << endl;
s << TAB << "switch(r.first - " << dname << ")" << endl;
s << TAB << "{" << endl;
INC_TAB;
for(size_t i = 0; i < vOperation.size(); i++)
{
s << TAB << "case " << i << ":" << endl;
s << TAB << "{" << endl;
INC_TAB;
s << generateDispatchPromiseAsync(vOperation[i], pPtr->getId()) << endl;
DEL_TAB;
s << TAB << "}" << endl;
}
DEL_TAB;
s << TAB << "}" << endl;
s << TAB << "return tars::TARSSERVERNOFUNCERR;" << endl;
DEL_TAB;
s << TAB << "}" << endl;
s << endl;
DEL_TAB;
s << TAB << "};" << endl;
s << TAB << "typedef tars::TC_AutoPtr<" << pPtr->getId() << "PrxCallbackPromise> " << pPtr->getId() << "PrxCallbackPromisePtr;" << endl;
s << endl;
*/
//生成协程异步回调类，用于并发请求
s << TAB << "/* callback of coroutine async proxy for client */" << endl;
s << TAB << "class " << pPtr->getId() << "CoroPrxCallback: public " << pPtr->getId() << "PrxCallback" << endl;
s << TAB << "{" << endl;
s << TAB << "public:" << endl;
INC_TAB;
s << TAB << "virtual ~" << pPtr->getId() << "CoroPrxCallback(){}" << endl;
DEL_TAB;
s << TAB << "public:" << endl;
INC_TAB;
s << TAB << "virtual const map<std::string, std::string> & getResponseContext() const { return _mRspContext; }" << endl;
s << endl;
s << TAB << "virtual void setResponseContext(const map<std::string, std::string> &mContext) { _mRspContext = mContext; }" << endl;
s << endl;
DEL_TAB;
s << TAB << "public:" << endl;
INC_TAB;
//生成协程异步回调接口
s << TAB << "int onDispatch(tars::ReqMessagePtr msg)" << endl;
s << TAB << "{" << endl;
INC_TAB;
dname = "__" + pPtr->getId() + "_all";
dispatch = "static ::std::string " + dname;
s << TAB << dispatch << "[]=" << endl;
s << TAB << "{" << endl;
INC_TAB;
for (size_t i = 0; i < vOperation.size(); i++)
{
s << TAB << "\"" << vOperation[i]->getId() << "\"";
if (i != vOperation.size() - 1)
{
s << ",";
}
s << endl;
}
DEL_TAB;
s << TAB << "};" << endl;
s << endl;
s << TAB << "pair<string*, string*> r = equal_range(" << dname << ", " << dname << "+" << vOperation.size() << ", string(msg->request.sFuncName));" << endl;
s << TAB << "if(r.first == r.second) return tars::TARSSERVERNOFUNCERR;" << endl;
s << TAB << "switch(r.first - " << dname << ")" << endl;
s << TAB << "{" << endl;
INC_TAB;
for (size_t i = 0; i < vOperation.size(); i++)
{
s << TAB << "case " << i << ":" << endl;
s << TAB << "{" << endl;
INC_TAB;
s << generateDispatchCoroAsync(vOperation[i], pPtr->getId()) << endl;
DEL_TAB;
s << TAB << "}" << endl;
}
DEL_TAB;
s << TAB << "}" << endl;
s << TAB << "return tars::TARSSERVERNOFUNCERR;" << endl;
DEL_TAB;
s << TAB << "}" << endl;
s << endl;
////////////////////////////////////////////////////////////////////
DEL_TAB;
s << TAB << "protected:" << endl;
INC_TAB;
s << TAB << "map<std::string, std::string> _mRspContext;" << endl;
//s << TAB << "tars::ParallelSharedBasePtr _pPtr;" << endl;
DEL_TAB;
s << TAB << "};" << endl;
s << TAB << "typedef tars::TC_AutoPtr<" << pPtr->getId() << "CoroPrxCallback> " << pPtr->getId() << "CoroPrxCallbackPtr;" << endl;
s << endl;
//生成客户端代理
s << TAB << "/* proxy for client */" << endl;
s << TAB << "class " << pPtr->getId() << "Proxy : public tars::ServantProxy" << endl;
s << TAB << "{" << endl;
s << TAB << "public:" << endl;
INC_TAB;
s << TAB << "typedef map<string, string> TARS_CONTEXT;" << endl;
for (size_t i = 0; i < vOperation.size(); i++)
{
s << generateH(vOperation[i], false, pPtr->getId()); // << endl;
s << generateHAsync(vOperation[i], pPtr->getId()) << endl;
}
s << TAB << pPtr->getId() << "Proxy* tars_hash(int64_t key)" << endl;
s << TAB << "{" << endl;
INC_TAB;
s << TAB << "return (" << pPtr->getId() + "Proxy*)ServantProxy::tars_hash(key);" << endl;
DEL_TAB;
s << TAB << "}" << endl;
s << endl;
s << TAB << pPtr->getId() << "Proxy* tars_consistent_hash(int64_t key)" << endl;
s << TAB << "{" << endl;
INC_TAB;
s << TAB << "return (" << pPtr->getId() + "Proxy*)ServantProxy::tars_consistent_hash(key);" << endl;
DEL_TAB;
s << TAB << "}" << endl;
s << endl;
s << TAB << pPtr->getId() << "Proxy* tars_set_timeout(int msecond)" << endl;
s << TAB << "{" << endl;
INC_TAB;
s << TAB << "return (" << pPtr->getId() + "Proxy*)ServantProxy::tars_set_timeout(msecond);" << endl;
DEL_TAB;
s << TAB << "}" << endl;
s << endl;
s << TAB << "static const char* tars_prxname() { return \"" << pPtr->getId() << "Proxy\"; }" << endl;
DEL_TAB;
s << TAB << "};" << endl;
s << TAB << "typedef tars::TC_AutoPtr<" << pPtr->getId() << "Proxy> " << pPtr->getId() << "Prx;" << endl;
s << endl;
//生成服务端Servant
s << TAB << "/* servant for server */" << endl;
s << TAB << "class " << pPtr->getId() << " : public tars::Servant" << endl;
s << TAB << "{" << endl;
s << TAB << "public:" << endl;
INC_TAB;
s << TAB << "virtual ~" << pPtr->getId() << "(){}" << endl;
for (size_t i = 0; i < vOperation.size(); i++)
{
s << generateH(vOperation[i], true, pPtr->getId()) << endl;
}
DEL_TAB;
s << TAB << "public:" << endl;
INC_TAB;
s << TAB << "int onDispatch(tars::TarsCurrentPtr _current, vector<char> &_sResponseBuffer)" << endl;
s << TAB << "{" << endl;
INC_TAB;
dname = "__" + nPtr->getId() + "__" + pPtr->getId() + "_all";
dispatch = "static ::std::string " + dname;
s << TAB << dispatch << "[]=" << endl;
s << TAB << "{" << endl;
INC_TAB;
for (size_t i = 0; i < vOperation.size(); i++)
{
s << TAB << "\"" << vOperation[i]->getId() << "\"";
if (i != vOperation.size() - 1)
{
s << ",";
}
s << endl;
}
DEL_TAB;
s << TAB << "};" << endl;
s << endl;
s << TAB << "pair<string*, string*> r = equal_range(" << dname << ", " << dname << "+" << vOperation.size() << ", _current->getFuncName());" << endl;
s << TAB << "if(r.first == r.second) return tars::TARSSERVERNOFUNCERR;" << endl;
s << TAB << "switch(r.first - " << dname << ")" << endl;
s << TAB << "{" << endl;
INC_TAB;
for (size_t i = 0; i < vOperation.size(); i++)
{
s << TAB << "case " << i << ":" << endl;
s << TAB << "{" << endl;
INC_TAB;
s << generateServantDispatch(vOperation[i], pPtr->getId()) << endl;
DEL_TAB;
s << TAB << "}" << endl;
}
DEL_TAB;
s << TAB << "}" << endl;
s << TAB << "return tars::TARSSERVERNOFUNCERR;" << endl;
DEL_TAB;
s << TAB << "}" << endl;
DEL_TAB;
s << TAB << "};" << endl;
return s.str();
}
//struct SortOperation
//{
// bool operator()(const OperationPtr &o1, const OperationPtr &o2)
// {
// return o1->getId() < o2->getId();
// }
//};
/******************************EnumPtr***************************************/
string Tars2Cpp::generateH(const EnumPtr& pPtr) const
{
ostringstream s;
s << TAB << "enum " << pPtr->getId() << endl;
s << TAB << "{" << endl;
INC_TAB;
vector<TypeIdPtr>& member = pPtr->getAllMemberPtr();
for (size_t i = 0; i < member.size(); i++)
{
s << TAB << member[i]->getId();
if (member[i]->hasDefault())
{
s << " = " << member[i]->def();
}
s << "," << endl;
}
DEL_TAB;
s << TAB << "};" << endl;
//生成枚举转字符串函数
s << TAB << "inline string etos" << "(const " << pPtr->getId() << " & e)" << endl;
s << TAB << "{" << endl;
INC_TAB;
s << TAB << "switch(e)" << endl;
s << TAB << "{" << endl;
INC_TAB;
for (size_t i = 0; i < member.size(); i++)
{
s << TAB << "case " << member[i]->getId() << ": return "
<< "\"" << member[i]->getId() << "\";" << endl;
}
s << TAB << "default: return \"\";" << endl;
DEL_TAB;
s << TAB << "}" << endl;
//s << TAB << "return \"\";" << endl;
DEL_TAB;
s << TAB << "}" << endl;
//生成字符串转枚举函数
s << TAB << "inline int stoe" << "(const string & s, " << pPtr->getId() << " & e)" << endl;
s << TAB << "{" << endl;
INC_TAB;
for (size_t i = 0; i < member.size(); i++)
{
s << TAB << "if(s == \"" << member[i]->getId() << "\") { e=" << member[i]->getId() << "; return 0;}" << endl;
}
s << endl;
s << TAB << "return -1;" << endl;
DEL_TAB;
s << TAB << "}" << endl;
return s.str();
}
/******************************ConstPtr***************************************/
string Tars2Cpp::generateH(const ConstPtr& pPtr) const
{
ostringstream s;
if (pPtr->getConstGrammarPtr()->t == ConstGrammar::STRING)
{
string tmp = tars::TC_Common::replace(pPtr->getConstGrammarPtr()->v, "\"", "\\\"");
s << TAB << "const " << tostr(pPtr->getTypeIdPtr()->getTypePtr()) << " " << pPtr->getTypeIdPtr()->getId() << " = \"" << tmp << "\";" << endl;
}
else
{
s << TAB << "const " << tostr(pPtr->getTypeIdPtr()->getTypePtr()) << " " << pPtr->getTypeIdPtr()->getId() << " = " << pPtr->getConstGrammarPtr()->v
<< ((tostr(pPtr->getTypeIdPtr()->getTypePtr()) == _namespace + "::Int64") ? "LL;" : ";") << endl;
}
return s.str();
}
/******************************NamespacePtr***************************************/
string Tars2Cpp::generateH(const NamespacePtr& pPtr) const
{
ostringstream s;
vector<InterfacePtr>& is = pPtr->getAllInterfacePtr();
vector<StructPtr>& ss = pPtr->getAllStructPtr();
vector<EnumPtr>& es = pPtr->getAllEnumPtr();
vector<ConstPtr>& cs = pPtr->getAllConstPtr();
s << endl;
s << TAB << "namespace " << pPtr->getId() << endl;
s << TAB << "{" << endl;
INC_TAB;
for (size_t i = 0; i < cs.size(); i++)
{
s << generateH(cs[i]) << endl;
}
for (size_t i = 0; i < es.size(); i++)
{
s << generateH(es[i]) << endl;
}
for (size_t i = 0; i < ss.size(); i++)
{
s << generateH(ss[i], pPtr->getId()) << endl;
}
s << endl;
for (size_t i = 0; i < is.size() && _onlyStruct == false; i++)
{
s << generateH(is[i], pPtr) << endl;
s << endl;
}
DEL_TAB;
s << "}";
s << endl << endl;
return s.str();
}
void Tars2Cpp::generateH(const ContextPtr &pPtr) const
{
string n = g_parse->getFileName(pPtr->getFileName());
string fileH = g_parse->getAbsoluteFileName(_baseDir, g_parse->replaceFileName(n, "h"));
string fileCpp = g_parse->getAbsoluteFileName(_baseDir, g_parse->replaceFileName(n, "cpp"));
//
// string n = tars::TC_File::excludeFileExt(tars::TC_File::extractFileName(pPtr->getFileName()));
//
// string fileH = _baseDir + FILE_SEP + n + ".h";
string define = tars::TC_Common::upper("__" + n + "_h_");
ostringstream s;
s << g_parse->printHeaderRemark();
s << "#ifndef " << define << endl;
s << "#define " << define << endl;
s << endl;
s << "#include <map>" << endl;
s << "#include <string>" << endl;
s << "#include <vector>" << endl;
s << "#include \"tup/Tars.h\"" << endl;
if (_bJsonSupport) s << "#include \"tup/TarsJson.h\"" << endl;
if (_bSqlSupport) s << "#include \"util/tc_mysql.h\"" << endl;
if (_bXmlSupport) s << "#include \"tup/TarsXml.h\"" << endl;
s << "using namespace std;" << endl;
vector<string> include = pPtr->getIncludes();
for (size_t i = 0; i < include.size(); i++)
{
s << "#include \"" << g_parse->getHeader() << tars::TC_File::extractFileName(include[i]) << "\"" << endl;
}
vector<NamespacePtr> namespaces = pPtr->getNamespaces();
//名字空间有接口
for (size_t i = 0; i < namespaces.size() && _onlyStruct == false; i++)
{
if (namespaces[i]->hasInterface())
{
s << "#include \"servant/ServantProxy.h\"" << endl;
s << "#include \"servant/Servant.h\"" << endl;
// s << "#include \"promise/promise.h\"" << endl;
break;
}
}
s << endl;
for (size_t i = 0; i < namespaces.size(); i++)
{
s << generateH(namespaces[i]) << endl;
}
s << endl;
s << "#endif" << endl;
tars::TC_File::makeDirRecursive(_baseDir);
tars::TC_File::save2file(fileH, s.str());
}
void Tars2Cpp::createFile(const string& file)//, const vector<string>& vsCoder)
{
std::vector<ContextPtr> contexts = g_parse->getContexts();
for (size_t i = 0; i < contexts.size(); i++)
{
if (file == contexts[i]->getFileName())
{
// if (vsCoder.size() == 0)
// {
generateH(contexts[i]);
// }
// else
// {
// for (size_t j = 0; j < vsCoder.size(); j++)
// {
// generateCoder(contexts[i], vsCoder[j]);
// }
// }
}
}
}
StructPtr Tars2Cpp::findStruct(const ContextPtr& pPtr, const string& id)
{
string sid = id;
//在当前namespace中查找
vector<NamespacePtr> namespaces = pPtr->getNamespaces();
for (size_t i = 0; i < namespaces.size(); i++)
{
NamespacePtr np = namespaces[i];
vector<StructPtr> structs = np->getAllStructPtr();
for (size_t i = 0; i < structs.size(); i++)
{
if (structs[i]->getSid() == sid)
{
return structs[i];
}
}
}
return NULL;
}
// ////////////////////////////////
// //for coder generating
// ////////////////////////////////
// string Tars2Cpp::generateCoder(const NamespacePtr& pPtr, const string& sInterface) const
// {
// ostringstream s;
// vector<InterfacePtr>& is = pPtr->getAllInterfacePtr();
// vector<StructPtr>& ss = pPtr->getAllStructPtr();
// vector<EnumPtr>& es = pPtr->getAllEnumPtr();
// vector<ConstPtr>& cs = pPtr->getAllConstPtr();
// s << endl;
// s << TAB << "namespace " << pPtr->getId() << endl;
// s << TAB << "{" << endl;
// INC_TAB;
// for (size_t i = 0; i < cs.size(); i++)
// {
// s << generateH(cs[i]) << endl;
// }
// for (size_t i = 0; i < es.size(); i++)
// {
// s << generateH(es[i]) << endl;
// }
// for (size_t i = 0; i < ss.size(); i++)
// {
// s << generateH(ss[i], pPtr->getId()) << endl;
// }
// s << endl;
// for (size_t i = 0; i < is.size(); i++)
// {
// if (pPtr->getId() + "::" + is[i]->getId() == sInterface)
// {
// s << generateCoder(is[i]) << endl;
// s << endl;
// }
// }
// DEL_TAB;
// s << "}";
// s << endl << endl;
// return s.str();
// }
// string Tars2Cpp::generateCoder(const InterfacePtr& pPtr) const
// {
// ostringstream s;
// vector<OperationPtr>& vOperation = pPtr->getAllOperationPtr();
// //生成编解码类
// s << TAB << "// encode and decode for client" << endl;
// s << TAB << "class " << pPtr->getId() << "Coder" << endl;
// s << TAB << "{" << endl;
// s << TAB << "public:" << endl << endl;
// INC_TAB;
// s << TAB << "typedef map<string, string> TARS_CONTEXT;" << endl << endl;
// s << TAB << "enum enumResult" << endl;
// s << TAB << "{" << endl;
// INC_TAB;
// s << TAB << "eTarsServerSuccess = 0," << endl;
// s << TAB << "eTarsPacketLess = 1," << endl;
// s << TAB << "eTarsPacketErr = 2," << endl;
// s << TAB << "eTarsServerDecodeErr = -1," << endl;
// s << TAB << "eTarsServerEncodeErr = -2," << endl;
// s << TAB << "eTarsServerNoFuncErr = -3," << endl;
// s << TAB << "eTarsServerNoServantErr = -4," << endl;
// s << TAB << "eTarsServerQueueTimeout = -6," << endl;
// s << TAB << "eTarsAsyncCallTimeout = -7," << endl;
// s << TAB << "eTarsProxyConnectErr = -8," << endl;
// s << TAB << "eTarsServerUnknownErr = -99," << endl;
// DEL_TAB;
// s << TAB << "};" << endl << endl;
// for (size_t i = 0; i < vOperation.size(); i++)
// {
// s << generateCoder(vOperation[i]) << endl;
// }
// DEL_TAB;
// s << TAB << "protected:" << endl << endl;
// INC_TAB;
// s << TAB << "static " + _namespace + "::Int32 fetchPacket(const string & in, string & out)" << endl;
// s << TAB << "{" << endl;
// INC_TAB;
// s << TAB << "if(in.length() < sizeof(" + _namespace + "::Int32)) return eTarsPacketLess;" << endl;
// s << TAB << "" + _namespace + "::Int32 iHeaderLen;" << endl;
// s << TAB << "memcpy(&iHeaderLen, in.c_str(), sizeof(" + _namespace + "::Int32));" << endl;
// s << TAB << "iHeaderLen = ntohl(iHeaderLen);" << endl;
// s << TAB << "if(iHeaderLen < (" + _namespace + "::Int32)sizeof(" + _namespace + "::Int32) || iHeaderLen > 100000000) return eTarsPacketErr;" << endl;
// s << TAB << "if((" + _namespace + "::Int32)in.length() < iHeaderLen) return eTarsPacketLess;" << endl;
// s << TAB << "out = in.substr(sizeof(" + _namespace + "::Int32), iHeaderLen - sizeof(" + _namespace + "::Int32)); " << endl;
// s << TAB << "return 0;" << endl;
// DEL_TAB;
// s << TAB << "}" << endl;
// s << endl;
// s << TAB << "static string encodeBasePacket(const string & sServantName, const string & sFuncName, const vector<char> & buffer, "
// << "const map<string, string>& context = TARS_CONTEXT())" << endl;
// s << TAB << "{" << endl;
// INC_TAB;
// s << TAB << _namespace + "::TarsOutputStream<" + _namespace + "::BufferWriterVector> os;" << endl;
// s << TAB << "os.write(1, 1);" << endl;
// s << TAB << "os.write(0, 2);" << endl;
// s << TAB << "os.write(0, 3);" << endl;
// s << TAB << "os.write(0, 4);" << endl;
// s << TAB << "os.write(sServantName, 5);" << endl;
// s << TAB << "os.write(sFuncName, 6);" << endl;
// s << TAB << "os.write(buffer, 7);" << endl;
// s << TAB << "os.write(60, 8);" << endl;
// s << TAB << "os.write(context, 9);" << endl;
// s << TAB << "os.write(map<string, string>(), 10);" << endl;
// s << TAB << _namespace + "::Int32 iHeaderLen;" << endl;
// s << TAB << "iHeaderLen = htonl(sizeof(" + _namespace + "::Int32) + os.getLength());" << endl;
// s << TAB << "string s;" << endl;
// s << TAB << "s.append((const char*)&iHeaderLen, sizeof(" + _namespace + "::Int32));" << endl;
// s << TAB << "s.append(os.getBuffer(), os.getLength());" << endl;
// s << TAB << "return s;" << endl;
// DEL_TAB;
// s << TAB << "}" << endl;
// s << endl;
// s << TAB << "static " + _namespace + "::Int32 decodeBasePacket(const string & in, " + _namespace + "::Int32 & iServerRet, vector<char> & buffer)" << endl;
// s << TAB << "{" << endl;
// INC_TAB;
// s << TAB << _namespace + "::TarsInputStream<" + _namespace + "::BufferReader> is;" << endl;
// s << TAB << "is.setBuffer(in.c_str(), in.length());" << endl;
// s << TAB << "is.read(iServerRet, 5, true);" << endl;
// s << TAB << "is.read(buffer, 6, true);" << endl;
// s << TAB << "return 0;" << endl;
// DEL_TAB;
// s << TAB << "}" << endl;
// s << endl;
// DEL_TAB;
// s << TAB << "};" << endl;
// return s.str();
// }
// string Tars2Cpp::generateCoder(const OperationPtr& pPtr) const
// {
// ostringstream s;
// vector<ParamDeclPtr>& vParamDecl = pPtr->getAllParamDeclPtr();
// //编码函数
// s << TAB << "//encode & decode function for '" << pPtr->getId() << "()'" << endl << endl;
// s << TAB << "static string encode_" << pPtr->getId() << "(const string & sServantName, ";
// for (size_t i = 0; i < vParamDecl.size(); i++)
// {
// if (!vParamDecl[i]->isOut())
// {
// s << generateH(vParamDecl[i]) << ",";
// }
// }
// s << endl;
// s << TAB << " const map<string, string>& context = TARS_CONTEXT())" << endl;
// s << TAB << "{" << endl;
// INC_TAB;
// s << TAB << "try" << endl;
// s << TAB << "{" << endl;
// INC_TAB;
// s << TAB << _namespace + "::TarsOutputStream<" + _namespace + "::BufferWriterVector> _os;" << endl;
// for (size_t i = 0; i < vParamDecl.size(); i++)
// {
// if (vParamDecl[i]->isOut()) continue;
// s << writeTo(vParamDecl[i]->getTypeIdPtr());
// }
// s << TAB << "return encodeBasePacket(sServantName, \"" << pPtr->getId() << "\", _os.getByteBuffer(), context);" << endl;
// DEL_TAB;
// s << TAB << "}" << endl;
// s << TAB << "catch (" + _namespace + "::TarsException & ex)" << endl;
// s << TAB << "{" << endl;
// INC_TAB;
// s << TAB << "return \"\";" << endl;
// DEL_TAB;
// s << TAB << "}" << endl;
// DEL_TAB;
// s << TAB << "}" << endl;
// s << endl;
// //解码函数
// s << TAB << "static " + _namespace + "::Int32 decode_" << pPtr->getId() << "(const string & in ";
// if (pPtr->getReturnPtr()->getTypePtr())
// {
// s << ", " << tostr(pPtr->getReturnPtr()->getTypePtr()) << " & _ret ";
// }
// for (size_t i = 0; i < vParamDecl.size(); i++)
// {
// if(!vParamDecl[i]->isOut())
// continue;
// s << ", " << generateH(vParamDecl[i]);
// }
// s << ")" << endl;
// s << TAB << "{" << endl;
// INC_TAB;
// s << TAB << "try" << endl;
// s << TAB << "{" << endl;
// INC_TAB;
// s << TAB << "string out;" << endl;
// s << TAB << _namespace + "::Int32 iRet = 0;" << endl;
// s << TAB << "if((iRet = fetchPacket(in, out)) != 0) return iRet;" << endl;
// s << TAB << _namespace + "::TarsInputStream<" + _namespace + "::BufferReader> _is;" << endl;
// s << TAB << _namespace + "::Int32 iServerRet=0;" << endl;
// s << TAB << "vector<char> buffer;" << endl;
// s << TAB << "decodeBasePacket(out, iServerRet, buffer);" << endl;
// s << TAB << "if(iServerRet != 0) return iServerRet;" << endl;
// s << TAB << "_is.setBuffer(buffer);" << endl;
// if (pPtr->getReturnPtr()->getTypePtr())
// {
// s << readFrom(pPtr->getReturnPtr());
// }
// for (size_t i = 0; i < vParamDecl.size(); i++)
// {
// if (vParamDecl[i]->isOut())
// {
// s << readFrom(vParamDecl[i]->getTypeIdPtr());
// }
// }
// s << TAB << "return 0;" << endl;
// DEL_TAB;
// s << TAB << "}" << endl;
// s << TAB << "catch (" + _namespace + "::TarsException & ex)" << endl;
// s << TAB << "{" << endl;
// INC_TAB;
// s << TAB << "return eTarsPacketErr;" << endl;
// DEL_TAB;
// s << TAB << "}" << endl;
// DEL_TAB;
// s << TAB << "}" << endl;
// s << endl;
// return s.str();
// }
// void Tars2Cpp::generateCoder(const ContextPtr& pPtr, const string& sInterface) const
// {
// cout << "Interface:" << sInterface << endl;
// string n = tars::TC_File::excludeFileExt(tars::TC_File::extractFileName(pPtr->getFileName())) + "Coder";
// string fileH = _baseDir + FILE_SEP + n + ".h";
// string define = tars::TC_Common::upper("__" + n + "_h_");
// ostringstream s;
// s << g_parse->printHeaderRemark();
// s << "#ifndef " << define << endl;
// s << "#define " << define << endl;
// s << endl;
// s << "#include <map>" << endl;
// s << "#include <string>" << endl;
// s << "#include <vector>" << endl;
// s << "#include \"tup/Tars.h\"" << endl;
// s << "using namespace std;" << endl;
// vector<string> include = pPtr->getIncludes();
// for (size_t i = 0; i < include.size(); i++)
// {
// s << "#include \"" << g_parse->getHeader()
// << tars::TC_Common::replace(tars::TC_File::extractFileName(include[i]), ".h", "Coder.h") << "\"" << endl;
// }
// vector<NamespacePtr> namespaces = pPtr->getNamespaces();
// s << endl;
// for (size_t i = 0; i < namespaces.size(); i++)
// {
// s << generateCoder(namespaces[i], sInterface) << endl;
// }
// s << endl;
// s << "#endif" << endl;
//
Reference in New Issue View Git Blame Copy Permalink