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 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 CreateSerializeCodeForFlatbuffer(ParseStruct structInfo, string varName)
{
string classStr = "";
string stringStr = "";
string argsStr = "";
string spacesStr = "";
List <MemberAttr> memberAttrInfo = structInfo.MemberAttrs;
string structName = structInfo.StructName;
serializeRecursive++;
for (int j = 0; j <= serializeRecursive; j++)
{
spacesStr += "\t";
}
for (int i = 0; i < memberAttrInfo.Count; i++)
{
string iterName = memberAttrInfo[i].VarName;
string iterType = memberAttrInfo[i].TypeName;
if (memberAttrInfo[i].IsArray)
{
if (memberAttrInfo[i].IsClass)
{
classStr += "\n";
classStr += spacesStr + "//class " + iterType + " vector serialize\n";
classStr += spacesStr + iterName + "PosArray := make([]flatbuffers.UOffsetT, len(" + varName + "." + iterName + "))\n";
classStr += spacesStr + "for i," + iterName + " := range " + varName + "." + iterName + " {\n";
classStr += spacesStr + ParseStruct.CreateSerializeCodeForFlatbuffer((ParseStruct)Vars.GetStruct(iterType), iterName);
classStr += spacesStr + "\t" + iterName + "PosArray[i] = " + iterName + "Pos\n";
}
else if (memberAttrInfo[i].IsString)
{
classStr += "\n";
classStr += spacesStr + "//string " + iterName + " vector serialize\n";
classStr += spacesStr + iterName + "PosArray := make([]flatbuffers.UOffsetT, len(" + varName + "." + iterName + "))\n";
classStr += spacesStr + "for i," + iterName + " := range " + varName + "." + iterName + " {\n";
classStr += spacesStr + "\t" + iterName + "PosArray[i] = builder.CreateString(" + iterName + ")\n";
}
else
{
string endOffset = varName + StringTo.ToUpper(iterName) + "EndOffset";
classStr += "\n";
classStr += spacesStr + "//" + iterType + " " + iterName + " vector serialize\n";
classStr += spacesStr + structName + "FBStart" + StringTo.ToUpper(iterName) + "Vector(builder, len(" + varName + "." + iterName + "))\n";
classStr += spacesStr + "for _," + iterName + " := range " + varName + "." + iterName + " {\n";
classStr += spacesStr + "\tbuilder.Prepend" + StringTo.ToUpper(iterType) + "(" + iterName + ")\n";
classStr += spacesStr + "}\n";
classStr += spacesStr + endOffset + " := builder.EndVector(len(" + varName + "." + iterName + "))\n\n";
argsStr += spacesStr + structName + "FBAdd" + StringTo.ToUpper(iterName) + "(builder," + endOffset + ")\n";
}
if (memberAttrInfo[i].IsClass || memberAttrInfo[i].IsString)
{
classStr += spacesStr + "}\n";
string endOffset = varName + StringTo.ToUpper(iterName) + "EndOffset";
classStr += spacesStr + structName + "FBStart" + StringTo.ToUpper(iterName) + "Vector(builder, len(" + iterName + "PosArray))\n";
classStr += spacesStr + "for _," + iterName + "Offset := range " + iterName + "PosArray {\n";
classStr += "\t\tbuilder.PrependUOffsetT(" + iterName + "Offset)\n";
classStr += "\t}\n";
classStr += spacesStr + endOffset + " := builder.EndVector(len(" + iterName + "PosArray))\n\n";
argsStr += spacesStr + structName + "FBAdd" + StringTo.ToUpper(iterName) + "(builder," +
varName + StringTo.ToUpper(iterName) + "EndOffset)\n";
}
}
else if (memberAttrInfo[i].IsClass)
{
classStr += "\n";
classStr += "//class " + iterType + " " + iterName + " serialize\n";
classStr += ParseStruct.CreateSerializeCodeForFlatbuffer((ParseStruct)Vars.GetStruct(iterType), iterName);
argsStr += spacesStr + structName + "FBAdd" + StringTo.ToUpper(iterName) + "(builder," + iterName + "Pos" + ")\n";
}
else if (memberAttrInfo[i].IsString)
{
string stringPos = varName + StringTo.ToUpper(iterName) + "StringPos";
stringStr += "\n";
stringStr += spacesStr + stringPos + " := builder.CreateString(" + varName + "." + iterName + ")\n";
argsStr += spacesStr + structName + "FBAdd" + StringTo.ToUpper(iterName) + "(builder," + stringPos + ")\n";
}
else
{
argsStr += spacesStr + structName + "FBAdd" + StringTo.ToUpper(iterName) + "(builder," + varName + "." + iterName + ")\n";
}
}
classStr += spacesStr + "//class " + structName + " serialize start\n";
classStr += stringStr;
classStr += spacesStr + structName + "FBStart(builder)\n";
classStr += argsStr;
classStr += spacesStr + varName + "Pos := " + structName + "FBEnd(builder)\n";
classStr += spacesStr + "//class " + structName + " serialize end\n\n";
serializeRecursive--;
return(classStr);
}
