123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519 |
- /**
- * 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 __TARS2CPP_H_
- #define __TARS2CPP_H_
- #include "parse.h"
- #include <cassert>
- #include <string>
- using namespace tars;
- /**
- * 根据tars生成c++文件
- * 包括结构的编解码以及生成Proxy和Servant
- */
- class Tars2Cpp
- {
- public:
- Tars2Cpp();
- /**
- * 生成
- * @param file
- */
- void createFile(const string &file);//, const vector<string> &vsCoder);
- /**
- * 设置生成文件的目录
- *
- */
- void setBaseDir(const std::string & sPath) { _baseDir = sPath; }
- /**
- * 设置是否需要打包默认值
- */
- void setCheckDefault(bool bCheck) { _checkDefault = bCheck; }
- /**
- * 根据命令选项设置是否需要生成sql支持
- */
- void setSqlSupport(bool bSqlSupport) { _bSqlSupport = bSqlSupport;}
- /**
- * 根据命令选项设置是否需要生成xml支持
- */
- void setXmlSupport(bool bXmlSupport, const vector<string>& vXmlIntf)
- {
- _bXmlSupport = bXmlSupport;
- _vXmlIntf = vXmlIntf;
- }
- /**
- * 根据命令选项设置是否需要生成json支持
- */
- void setJsonSupport(bool bJsonSupport) { _bJsonSupport = bJsonSupport; }
- /**
- * 根据命令选项设置是否需要生成json支持
- */
- void setJsonSupport(const vector<string>& vJsonIntf) { _vJsonIntf = vJsonIntf; }
- /**
- * 设置是否只生成struct
- */
- void setOnlyStruct(bool bOnlyStruct) { _onlyStruct = bOnlyStruct; }
- /**
- * setUnknownField for UnknownField support
- */
- // void setUnknownField(bool bUnkownField) { _unknownField = bUnkownField; }
- /**
- * added for master info
- */
- void setTarsMaster(bool bTarsMaster) { _tarsMaster = bTarsMaster; }
- //下面是编解码的源码生成
- protected:
- /**
- * 生成xml
- * @param pPtr
- *
- * @return string
- */
- string writeToXml(const TypeIdPtr &pPtr) const;
- /**
- * 生成xml
- * @param pPtr
- *
- * @return string
- */
- string readFromXml(const TypeIdPtr &pPtr, bool bIsRequire = true) const;
- /**
- * 生成sql
- * @param pPtr
- *
- * @return string
- */
- string writeToSql(const TypeIdPtr &pPtr) const;
- /**
- * 生成sql
- * @param pPtr
- *
- * @return string
- */
- string readFromSql(const TypeIdPtr &pPtr, bool bIsRequire = true) const;
- /**
- * 生成json
- * @param pPtr
- *
- * @return string
- */
- string writeToJson(const TypeIdPtr& pPtr) const;
- /**
- * 生成json
- * @param pPtr
- *
- * @return string
- */
- string readFromJson(const TypeIdPtr& pPtr, bool bIsRequire = true) const;
- /**
- * 生成某类型的解码源码
- * @param pPtr
- *
- * @return string
- */
- string writeTo(const TypeIdPtr &pPtr) const;
- /**
- * 生成某类型的编码源码
- * @param pPtr
- *
- * @return string
- */
- string readFrom(const TypeIdPtr &pPtr, bool bIsRequire = true) const;
- /**
- * 生成某类型的编码源码
- * @param pPtr
- *
- * @return string
- */
- // string promiseReadFrom(const TypeIdPtr &pPtr, bool bIsRequire = true) const;
- /**
- *
- * @param pPtr
- *
- * @return string
- */
- // string readUnknown(const TypeIdPtr &pPtr) const;
- /**
- *
- * @param pPtr
- *
- * @return string
- */
- // string writeUnknown()const;
- /**
- *
- * @param pPtr
- *
- * @return string
- */
- string display(const TypeIdPtr &pPtr) const;
- /**
- *
- * @param pPtr
- *
- * @return string
- */
- string displaySimple(const TypeIdPtr &pPtr, bool bSep) const;
- /**
- *
- * @param pPtr
- *
- * @return string
- */
- string generateCollection(const TypeIdPtr &pPtr, bool bSep) const;
- //下面是类型描述的源码生成
- protected:
- /**
- * 生成某类型的字符串描述源码
- * @param pPtr
- *
- * @return string
- */
- string tostr(const TypePtr &pPtr) const;
- /**
- * 生成内建类型的字符串源码
- * @param pPtr
- *
- * @return string
- */
- string tostrBuiltin(const BuiltinPtr &pPtr) const;
- /**
- * 生成vector的字符串描述
- * @param pPtr
- *
- * @return string
- */
- string tostrVector(const VectorPtr &pPtr) const;
- /**
- * 生成map的字符串描述
- * @param pPtr
- *
- * @return string
- */
- string tostrMap(const MapPtr &pPtr) const;
- /**
- * 生成某种结构的符串描述
- * @param pPtr
- *
- * @return string
- */
- string tostrStruct(const StructPtr &pPtr) const;
- /**
- * 生成某种枚举的符串描述
- * @param pPtr
- *
- * @return string
- */
- string tostrEnum(const EnumPtr &pPtr) const;
- /**
- * 获取定长数组的坐标
- * @param pPtr
- *
- * @return string
- */
- string toStrSuffix(const TypeIdPtr &pPtr) const;
- /**
- * 获取定长数组的坐标
- * @param pPtr
- *
- * @return int
- */
- int getSuffix(const TypeIdPtr &pPtr) const;
- /**
- * 生成unsigned转换为signed的符串描述
- * @param pPtr
- *
- * @return string
- */
- //string unsigned2signedStr(const BuiltinPtr &pPtr) const;
- /**
- * 生成类型变量的解码源码
- * @param pPtr
- *
- * @return string
- */
- string decode(const TypeIdPtr &pPtr) const;
- /**
- * 生成类型变量的编码源码
- * @param pPtr
- *
- * @return string
- */
- string encode(const TypeIdPtr &pPtr) const;
- //以下是h文件的具体生成
- protected:
- /**
- * 结构的md5
- * @param pPtr
- *
- * @return string
- */
- string MD5(const StructPtr &pPtr) const;
- /**
- * 生成结构的头文件内容
- * @param pPtr
- *
- * @return string
- */
- string generateH(const StructPtr &pPtr, const string& namespaceId) const;
- /**
- * 生成容器的头文件源码
- * @param pPtr
- *
- * @return string
- */
- string generateH(const ContainerPtr &pPtr) const;
- /**
- * 生成参数声明的头文件内容
- * @param pPtr
- *
- * @return string
- */
- string generateH(const ParamDeclPtr &pPtr) const;
- /**
- *
- * @param pPtr
- *
- * @return string
- */
- string generateOutH(const ParamDeclPtr &pPtr) const;
- /**
- *
- * @param pPtr
- * @param cn
- *
- * @return string
- */
- string generateAsyncResponseCpp(const OperationPtr &pPtr, const string &cn) const;
- /**
- * 生成参数声明的内容
- * @param pPtr
- *
- * @return string
- */
- string generateParamDecl(const ParamDeclPtr &pPtr) const;
- /**
- * 生成操作函数调用分发的源码
- * @param pPtr
- * @param cn
- *
- * @return string
- */
- string generateDispatchAsync(const OperationPtr &pPtr, const string &cn) const;
- /**
- * 生成操作函数调用分发的源码
- * @param pPtr
- * @param cn
- *
- * @return string
- */
- string generateDispatchCoroAsync(const OperationPtr &pPtr, const string &cn) const;
- /**
- * 生成servant操作函数调用的源码
- * @param pPtr
- * @param cn
- *
- * @return string
- */
- string generateServantDispatch(const OperationPtr& pPtr, const string& cn) const;
- /**
- * 生成操作的servant的头文件源码
- * @param pPtr
- * @param bVirtual
- *
- * @return string
- */
- string generateHAsync(const OperationPtr &pPtr) const;
- /**
- * 生成操作的servant的头文件源码
- * @param pPtr
- * @param bVirtual
- *
- * @return string
- */
- string generateHAsync(const OperationPtr &pPtr, const string& interfaceId) const;
- /**
- * 生成操作的servant的头文件源码
- * @param pPtr
- * @param bVirtual
- *
- * @return string
- */
- // string generateHPromiseAsync(const InterfacePtr &pInter, const OperationPtr &pPtr) const;
- /**
- * 生成操作函数调用分发的源码
- * @param pPtr
- * @param cn
- *
- * @return string
- */
- // string generateDispatchPromiseAsync(const OperationPtr &pPtr, const string &cn) const;
- /**
- * 生成操作的servant的头文件源码
- * @param pPtr
- * @param bVirtual
- *
- * @return string
- */
- string generateH(const OperationPtr &pPtr, bool bVirtual, const string& interfaceId) const;
- /**
- * 生成接口的头文件源码
- * @param pPtr
- *
- * @return string
- */
- string generateH(const InterfacePtr &pPtr, const NamespacePtr &nPtr) const;
- /**
- * 生成枚举的头文件源码
- * @param pPtr
- *
- * @return string
- */
- string generateH(const EnumPtr &pPtr) const;
- /**
- * 生成常量头文件源码
- * @param pPtr
- *
- * @return string
- */
- string generateH(const ConstPtr &pPtr) const;
- /**
- * 生成名字空间的头文件源码
- * @param pPtr
- *
- * @return string
- */
- string generateH(const NamespacePtr &pPtr) const;
- /**
- * 生成名字空间的头文件源码
- * @param pPtr
- *
- * @return string
- */
- void generateH(const ContextPtr &pPtr) const;
- /**
- *
- * 根据名字查找结构
- * @param pPtr
- * @param id
- *
- * @return StructPtr
- */
- StructPtr findStruct(const ContextPtr &pPtr,const string &id);
- // /**
- // *
- // * 生成接口编解码代码
- // * @param pPtr
- // * @param interface
- // */
- // void generateCoder(const ContextPtr &pPtr,const string &interface) const;
- // string generateCoder(const NamespacePtr &pPtr,const string & sInterface) const;
- // string generateCoder(const InterfacePtr &pPtr) const;
- // string generateCoder(const OperationPtr &pPtr) const;
- string generateInitValue(const TypeIdPtr &pPtr) const;
- // bool isPromiseDispatchInitValue(const TypeIdPtr &pPtr) const;
- private:
- std::string _baseDir;
- bool _checkDefault;
- bool _onlyStruct;
- bool _bSqlSupport;
- bool _bXmlSupport;
- bool _bJsonSupport;
- vector<string> _vJsonIntf;
- vector<string> _vXmlIntf;
- std::string _namespace ;
- // bool _unknownField;
- bool _tarsMaster;
- };
- #endif
|