public static string CreateServerInterfaceCode(FunctionAttr functionAttrInterface, string inerfaceName)
{
string strs = "func (slf *" + StringTo.ToUpper(inerfaceName) + "Interface) " + functionAttrInterface.FuncName + "Interface(context *rpcb.Context, req *rpcs.Request) error{\n";
strs += "\t" + functionAttrInterface.FuncArgMap.VarName + "FB := GetRootAs" + functionAttrInterface.FuncArgMap.TypeName + "FB(req.Pop(), 0)\n";
strs += "\t" + functionAttrInterface.FuncArgMap.VarName + " := " + functionAttrInterface.FuncArgMap.TypeName + "{}\n";
strs += "\t//input flatbuffer code for " + functionAttrInterface.FuncArgMap.VarName + "FB class\n";
strs += "" + ParseStruct.CreateDeserializeCodeForFlatbuffer((ParseStruct)Vars.GetStruct(functionAttrInterface.FuncArgMap.TypeName),
functionAttrInterface.FuncArgMap.VarName, functionAttrInterface.FuncArgMap.VarName + "FB") + "\n\n";
strs += "\tdeleageService := " + StringTo.ToUpper(inerfaceName) + "Service{_implcontext:context}\n";
strs += "\t" + functionAttrInterface.FuncReturn.VarName + " := deleageService." + functionAttrInterface.FuncName + "(" + functionAttrInterface.FuncArgMap.VarName + ")\n";
strs += "\tbuilder := slf._implsrv.MallocFlatBuilder()\n";
strs += "\tdefer slf._implsrv.FreeFlatBuilder(builder)\n";
strs += "\t//input flatbuffer code for " + functionAttrInterface.FuncReturn.TypeName + "\n";
strs += "\t" + ParseStruct.CreateSerializeCodeForFlatbuffer((ParseStruct)Vars.GetStruct(functionAttrInterface.FuncReturn.TypeName),
functionAttrInterface.FuncReturn.VarName);
strs += "\tbuilder.Finish(" + functionAttrInterface.FuncReturn.VarName + "Pos)\n";
strs += "\tresp := rpcs.NewReponse(req, builder.FinishedBytes())\n";
strs += "\tif err := slf._implsrv.Response(resp, rpcc.MRPC_PT_FLATBUFF); err != nil {\n";
strs += "\t\tcontext.Error(\"response fail:%v\", err)\n";
strs += "\t}\n";
strs += "\treturn nil\n}\n\n";
return(strs);
}
public static string CreateClientServiceCode(FunctionAttr functionAttrInterface, string inerfaceName)
{
string serviceName = StringTo.ToUpper(inerfaceName, 0) + "Service";
string interfaceName = StringTo.ToUpper(inerfaceName, 0) + "Interface";
string selfName = inerfaceName[0].ToString().ToLower();
FunctionArg v = functionAttrInterface.FuncArgMap;
string funcArgsStr = v.VarName + " " + v.TypeName;
string strs = "func (" + selfName + " *" + serviceName + ") " + StringTo.ToUpper(functionAttrInterface.FuncName, 0) + "(";
strs += funcArgsStr + ", timeout uint32" + ") (*" + functionAttrInterface.FuncReturn.TypeName + ", error) {\n";
strs += "\ts := " + interfaceName + "{}\n";
strs += "\treturn s." + StringTo.ToLower(functionAttrInterface.FuncName, 0) + "Interface(" + selfName + "._brokerImpl" +
", " + v.VarName + ", " + selfName + "._compressType, timeout)\n";
strs += "}\n\n";
return(strs);
}
public static string CreateClientInterfaceCode(FunctionAttr functionAttrInterface, string inerfaceName)
{
string strs = "";
string funcArgsStr = "";
string funcArgsStructStr = "";
string funcInterfaceName = StringTo.ToUpper(inerfaceName, 0) + "Interface";
FunctionArg v = functionAttrInterface.FuncArgMap;
funcArgsStr = v.VarName + " " + v.TypeName;
funcArgsStructStr = v.VarName;
strs += "func (" + inerfaceName[0].ToString().ToLower() + " *" + funcInterfaceName + ") " +
StringTo.ToLower(functionAttrInterface.FuncName, 0) + "Interface(broker" + " *rpclient.Broker, " +
funcArgsStr + ", compressType rpcc.MRPC_PACKAGE_COMPRESS, timeout uint32) (*" + functionAttrInterface.FuncReturn.TypeName + ", error) {\n";
strs += "\tconst funcID = " + functionAttrInterface.FuncHash + "\n";
strs += "\tbuilder := flatbuffers.NewBuilder(512)\n";
strs += "\t//input flatbuffer code for " + v.TypeName + "FB class\n";
strs += ParseStruct.CreateSerializeCodeForFlatbuffer((ParseStruct)Vars.GetStruct(v.TypeName),
v.VarName);
strs += "\tbuilder.Finish(" + v.VarName + "Pos)\n";
strs += "\treq := rpclient.NewRequest(funcID, compressType,timeout, builder.FinishedBytes())\n";
strs += "\tresp, err := broker.SyncCall(req, rpcc.MRPC_PT_FLATBUFF)\n";
strs += "\tif err != nil{\n";
strs += "\t\treturn nil, err\n";
strs += "\t}\n\n";
strs += "\tif resp.GetStatus() != rpcc.RS_OK {\n";
strs += "\t\treturn nil, errors.New(\"rpc sync call error, function:" + functionAttrInterface.FuncName + "\")\n";
strs += "\t}\n\n";
strs += "\t" + functionAttrInterface.FuncReturn.VarName + "FB := GetRootAs" +
functionAttrInterface.FuncReturn.TypeName + "FB(resp.Pop(), 0)\n";
strs += "\tvar " + functionAttrInterface.FuncReturn.VarName + " " + functionAttrInterface.FuncReturn.TypeName + "\n";
strs += "\t//input flatbuffer code for %" + functionAttrInterface.FuncReturn.TypeName + "FB class\n\n";
strs += ParseStruct.CreateDeserializeCodeForFlatbuffer((ParseStruct)Vars.GetStruct(functionAttrInterface.FuncReturn.TypeName),
functionAttrInterface.FuncReturn.VarName, functionAttrInterface.FuncReturn.VarName + "FB") + "\n";
strs += "\treturn &" + functionAttrInterface.FuncReturn.VarName + ", nil\n";
strs += "}\n";
return(strs);
}
public static string CreateServerServiceCode(FunctionAttr functionAttrInterface, string inerfaceName)
{
string strs = "";
string funcArgsStrs = "";
string returnStrs = StringTo.ToLower(functionAttrInterface.FuncReturn.TypeName) + "Ret";
string serviceName = StringTo.ToUpper(inerfaceName) + "Service";
FunctionArg v = functionAttrInterface.FuncArgMap;
funcArgsStrs += v.VarName + " " + v.TypeName;
strs += "func (" + StringTo.ToLower(functionAttrInterface.FuncReturn.TypeName)[0] + " *" + serviceName + ") " +
functionAttrInterface.FuncName + "(" + funcArgsStrs + ") " + functionAttrInterface.FuncReturn.TypeName + " {\n";
strs += "\tvar " + returnStrs + " " + functionAttrInterface.FuncReturn.TypeName + "\n";
strs += "\t//TODO: 编写函数代码\n\n";
strs += "\treturn " + returnStrs + "\n";
strs += "}\n";
return(strs);
}
public bool Parse(string filePath, string name, string bodys)
{
this.m_interfaceName = name;
this.m_filePath = filePath;
string[] memberAttrList = bodys.Split(";");
if (memberAttrList.Length < 1)
{
throw new System.Exception("parse service member attr is failed, " + bodys);
}
foreach (string attr in memberAttrList)
{
if (attr.Length > 1)
{
FunctionAttr funcAttr = new FunctionAttr(m_filePath + this.m_interfaceName, attr);
m_functions.Insert(m_funcIdx++, funcAttr);
//m_functionList[] = funcAttr;
}
}
return(true);
}
