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 CreateServerCode(ParseStruct parse)
{
string strs = "type " + parse.StructName + " struct {\n";
foreach (MemberAttr attr in parse.MemberAttrs)
{
if (attr.IsArray)
{
strs += "\t" + attr.VarName + "[] " + attr.TypeName + "\n";
}
else
{
strs += "\t" + attr.VarName + " " + attr.TypeName + "\n";
}
}
strs += "}\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);
}
public static string CreateDeserializeCodeForFlatbuffer(ParseStruct structInfo, string varName, string fbName)
{
string strs = "";
string spacesStr = "";
List <MemberAttr> memberAttrInfo = structInfo.MemberAttrs;
string structName = structInfo.StructName;
deserializeRecursive++;
for (int j = 0; j <= deserializeRecursive; j++)
{
spacesStr += "\t";
}
for (int i = 0; i < memberAttrInfo.Count; i++)
{
string iterName = memberAttrInfo[i].VarName;
string iterType = memberAttrInfo[i].TypeName;
//for (int j = 0; j <= deserializeRecursive; j++)
// speces += "\t";
if (memberAttrInfo[i].IsArray)
{
if (memberAttrInfo[i].IsClass)
{
string tmpName = iterType.ToLower() + "Tmp";
strs += spacesStr + "for i := 0;i < " + fbName + "." + StringTo.ToUpper(iterName) + "Length();i++ {\n";
strs += spacesStr + "\tvar " + tmpName + " " + iterType + "\n";
strs += spacesStr + "\tvar " + iterName + " " + iterType + "FB\n";
strs += spacesStr + "\tif ok := " + fbName + "." + StringTo.ToUpper(iterName) + "(&" + iterName + ",i);!ok {\n";
strs += spacesStr + "\t\treturn nil, errors.New(\"deserialize " + fbName + "=>" + StringTo.ToUpper(iterName) + " fail\")\n";
strs += spacesStr + "\t}\n\n";
strs += ParseStruct.CreateDeserializeCodeForFlatbuffer((ParseStruct)Vars.GetStruct(iterType), tmpName, iterName);
strs += spacesStr + "\t" + varName + "." + iterName + " = append(" + varName + "." + iterName + "," + tmpName + ")\n";
}
else
{
strs += spacesStr + "for i := 0;i < " + fbName + "." + StringTo.ToUpper(iterName) + "Length();i++ {\n";
strs += spacesStr + "\t" + varName + "." + iterName + " = append(" + varName + "." + iterName + ",";
if (memberAttrInfo[i].IsString)
{
strs += spacesStr + "string(" + fbName + "." + StringTo.ToUpper(iterName) + "(i)))\n";
}
else
{
strs += spacesStr + fbName + "." + StringTo.ToUpper(iterName) + "(i))\n";
}
}
strs += spacesStr + "}\n\n";
}
else if (memberAttrInfo[i].IsClass)
{
strs += ParseStruct.CreateDeserializeCodeForFlatbuffer((ParseStruct)Vars.GetStruct(iterType), varName + "." + iterName,
fbName + "." + StringTo.ToUpper(iterName) + "()");
}
else
{
if (memberAttrInfo[i].IsString)
{
strs += spacesStr + varName + "." + iterName + " = string(" + fbName + "." + StringTo.ToUpper(iterName) + "())\n";
}
else
{
strs += spacesStr + varName + "." + iterName + " = " + fbName + "." + StringTo.ToUpper(iterName) + "()\n";
}
}
}
deserializeRecursive--;
return(strs);
}
public static string CreateServerCode(ParseStruct parse)
{
return("");
}
public static string CreateFlatbufferCode(ParseStruct structInterface)
{
m_flatbufferVariable["bool"] = "bool";
m_flatbufferVariable["int8"] = "byte";
m_flatbufferVariable["uint8"] = "ubyte";
m_flatbufferVariable["int16"] = "short";
m_flatbufferVariable["uint16"] = "ushort";
m_flatbufferVariable["int"] = "int";
m_flatbufferVariable["uint"] = "uint";
m_flatbufferVariable["int32"] = "long";
m_flatbufferVariable["uint32"] = "ulong";
m_flatbufferVariable["float32"] = "float";
m_flatbufferVariable["float64"] = "double";
m_flatbufferVariable["string"] = "string";
string strs = "table " + structInterface.StructName + "FB {\n";
for (int i = 0; i < structInterface.MemberAttrs.Count; i++)
{
MemberAttr v = structInterface.MemberAttrs[i];
string typeName = getFlatbufferVariable(v.TypeName);
if (typeName == null)
{
ParseStruct pstruct = (ParseStruct)Vars.GetStruct(v.TypeName);
if (pstruct != null)
{
typeName = pstruct.GetName() + "FB";
}
}
if (typeName != null)
{
if (v.IsArray)
{
if (v.IsClass)
{
strs += "\t" + v.VarName + ":[" + v.TypeName + "FB];\n";
}
else
{
strs += "\t" + v.VarName + ":[" + v.TypeName + "];\n";
}
}
else
{
if (v.IsClass)
{
strs += "\t" + v.VarName + ":" + v.TypeName + "FB;\n";
}
else
{
strs += "\t" + v.VarName + ":" + v.TypeName + ";\n";
}
}
}
else
{
throw new System.Exception("create flatbuffer file is faild, message name: " +
structInterface.StructName + ", type:" + v.TypeName + " is not exits!");
}
}
strs += "}\n\n";
return(strs);
}
bool parse(string data)
{
data = Regex.Replace(data, @"\/\/[^\n]*", "");
data = Regex.Replace(data, @"[\n\r\t]", "");
data = Regex.Replace(data, @"\s{2,}", "");
string[] classes = Regex.Split(data, @"[@\}]");
if (classes.Length == 0)
{
throw new System.Exception("parse classes is failed, no class struct!!");
}
//
foreach (string c in classes)
{
string[] symbolFlag = c.Split('{');
if (symbolFlag.Length != 2)
{
continue;
}
string[] symbolAttr = symbolFlag[0].Split(":");
if (symbolAttr.Length != 2)
{
throw new Exception("parse symbol attr is failed, symbol missing :, " + symbolFlag[0]);
}
IBParse idlParse;
switch (symbolAttr[0])
{
case Symbol.Struct:
idlParse = new ParseStruct();
break;
case Symbol.Interface:
idlParse = new ParseInterface();
break;
case Symbol.Namespace:
idlParse = new ParseNamespace();
break;
default:
throw new Exception("parse symbol attr is error, symbol: " + symbolAttr[0]);
}
if (idlParse.Parse(m_fileName, symbolAttr[1].Trim(), symbolFlag[1].Trim()))
{
switch (symbolAttr[0])
{
case Symbol.Struct:
Vars.RegisterStruct(idlParse.GetName(), idlParse);
break;
case Symbol.Interface:
Vars.RegisterInterface(idlParse.GetName(), idlParse);
break;
case Symbol.Namespace:
Vars.RegisterNamespace(idlParse);
break;
}
}
}
//TODDO:解释代码修
createCode();
return(true);
}
