AlphaaeGameStudio/TEngine
1
0
Fork 0
mirror of https://github.com/Alex-Rachel/TEngine.git synced 2025-08-14 16:51:28 +00:00
Code Issues Packages Projects Releases Wiki Activity

DotNet支持自定义协议目录

Browse Source 
DotNet支持自定义协议目录
This commit is contained in:
ALEXTANG
2023-08-24 17:47:21 +08:00
parent d55f6e29d9
commit f19b889deb
1 changed files with 200 additions and 168 deletions
Download Patch File Download Diff File Expand all files Collapse all files

368
Assets/GameScripts/DotNet/Core/Exporter/ProtoBuf/ProtoBufExporter.cs
View File

@@ -1,6 +1,7 @@
#if TENGINE_NET #if TENGINE_NET
using System.Text; using System.Text;
using TEngine.Core.Network; using TEngine.Core.Network;
#pragma warning disable CS8604 #pragma warning disable CS8604
#pragma warning disable CS8602 #pragma warning disable CS8602
#pragma warning disable CS8600 #pragma warning disable CS8600
@@ -32,22 +33,37 @@ public sealed class ProtoBufExporter
private uint _aRouteResponse; private uint _aRouteResponse;
private string _serverTemplate; private string _serverTemplate;
private string _clientTemplate; private string _clientTemplate;
private readonly List<OpcodeInfo> _opcodes = new(); private readonly List<OpcodeInfo> _opcodes = new();
public ProtoBufExporter() public ProtoBufExporter()
{ {
Console.OutputEncoding = Encoding.UTF8; Console.OutputEncoding = Encoding.UTF8;
if (!Directory.Exists(Define.ProtoBufServerDirectory)) if (!Directory.Exists(Define.ProtoBufServerDirectory))
{ {
Directory.CreateDirectory(Define.ProtoBufServerDirectory); Directory.CreateDirectory(Define.ProtoBufServerDirectory);
} }
if (!Directory.Exists(Define.ProtoBufClientDirectory)) if (!Directory.Exists(Define.ProtoBufClientDirectory))
{ {
Directory.CreateDirectory(Define.ProtoBufClientDirectory); Directory.CreateDirectory(Define.ProtoBufClientDirectory);
} }
if (!Directory.Exists($"{Define.ProtoBufDirectory}Inner"))
{
Directory.CreateDirectory($"{Define.ProtoBufDirectory}Inner");
}
if (!Directory.Exists($"{Define.ProtoBufDirectory}Outer"))
{
Directory.CreateDirectory($"{Define.ProtoBufDirectory}Outer");
}
if (!Directory.Exists($"{Define.ProtoBufDirectory}Bson"))
{
Directory.CreateDirectory($"{Define.ProtoBufDirectory}Bson");
}
var tasks = new Task[2]; var tasks = new Task[2];
tasks[0] = Task.Run(RouteType); tasks[0] = Task.Run(RouteType);
@@ -63,6 +79,7 @@ public sealed class ProtoBufExporter
private async Task Start(ProtoBufOpCodeType opCodeType) private async Task Start(ProtoBufOpCodeType opCodeType)
{ {
List<string> files = new List<string>();
var protoFile = ""; var protoFile = "";
var opCodeName = ""; var opCodeName = "";
var parameter = ""; var parameter = "";
@@ -73,8 +90,8 @@ public sealed class ProtoBufExporter
string customRouteType = null; string customRouteType = null;
_opcodes.Clear(); _opcodes.Clear();
var file = new StringBuilder(); var file = new StringBuilder();
var saveDirectory = new Dictionary<string,string>(); var saveDirectory = new Dictionary<string, string>();
switch (opCodeType) switch (opCodeType)
{ {
case ProtoBufOpCodeType.Outer: case ProtoBufOpCodeType.Outer:
@@ -89,6 +106,8 @@ public sealed class ProtoBufExporter
protoFile = $"{Define.ProtoBufDirectory}OuterMessage.proto"; protoFile = $"{Define.ProtoBufDirectory}OuterMessage.proto";
saveDirectory.Add(Define.ProtoBufServerDirectory, _serverTemplate); saveDirectory.Add(Define.ProtoBufServerDirectory, _serverTemplate);
saveDirectory.Add(Define.ProtoBufClientDirectory, _clientTemplate); saveDirectory.Add(Define.ProtoBufClientDirectory, _clientTemplate);
files.Add(protoFile);
files.AddRange(Directory.GetFiles($"{Define.ProtoBufDirectory}Outer").ToList());
break; break;
} }
case ProtoBufOpCodeType.Inner: case ProtoBufOpCodeType.Inner:
@@ -103,6 +122,8 @@ public sealed class ProtoBufExporter
opCodeName = "InnerOpcode"; opCodeName = "InnerOpcode";
protoFile = $"{Define.ProtoBufDirectory}InnerMessage.proto"; protoFile = $"{Define.ProtoBufDirectory}InnerMessage.proto";
saveDirectory.Add(Define.ProtoBufServerDirectory, _serverTemplate); saveDirectory.Add(Define.ProtoBufServerDirectory, _serverTemplate);
files.Add(protoFile);
files.AddRange(Directory.GetFiles($"{Define.ProtoBufDirectory}Inner").ToList());
break; break;
} }
case ProtoBufOpCodeType.InnerBson: case ProtoBufOpCodeType.InnerBson:
@@ -117,214 +138,223 @@ public sealed class ProtoBufExporter
opCodeName = "InnerBsonOpcode"; opCodeName = "InnerBsonOpcode";
protoFile = $"{Define.ProtoBufDirectory}InnerBsonMessage.proto"; protoFile = $"{Define.ProtoBufDirectory}InnerBsonMessage.proto";
saveDirectory.Add(Define.ProtoBufServerDirectory, _serverTemplate); saveDirectory.Add(Define.ProtoBufServerDirectory, _serverTemplate);
files.Add(protoFile);
files.AddRange(Directory.GetFiles($"{Define.ProtoBufDirectory}Bson").ToList());
break; break;
} }
} }
var protoFileText = await File.ReadAllTextAsync(protoFile);
foreach (var line in protoFileText.Split('\n'))
foreach (var filePath in files)
{ {
var currentLine = line.Trim(); var protoFileText = await File.ReadAllTextAsync(filePath);
if (string.IsNullOrWhiteSpace(currentLine)) foreach (var line in protoFileText.Split('\n'))
{ {
continue; var currentLine = line.Trim();
}
if (currentLine.StartsWith("///"))
{
file.AppendFormat(" /// <summary>\r\n" + " /// {0}\r\n" + " /// </summary>\r\n", currentLine.TrimStart(new[] {'/', '/', '/'}));
continue;
}
if (currentLine.StartsWith("message")) if (string.IsNullOrWhiteSpace(currentLine))
{
isMsgHead = true;
opcodeInfo = new OpcodeInfo();
file.AppendLine("\t[ProtoContract]");
className = currentLine.Split(Define.SplitChars, StringSplitOptions.RemoveEmptyEntries)[1];
var splits = currentLine.Split(new[] {"//"}, StringSplitOptions.RemoveEmptyEntries);
if (splits.Length > 1)
{ {
var parameterArray = currentLine.Split(new[] {"//"}, StringSplitOptions.RemoveEmptyEntries)[1].Trim().Split(','); continue;
parameter = parameterArray[0].Trim(); }
switch (parameterArray.Length) if (currentLine.StartsWith("///"))
{
file.AppendFormat(" /// <summary>\r\n" + " /// {0}\r\n" + " /// </summary>\r\n", currentLine.TrimStart(new[] { '/', '/', '/' }));
continue;
}
if (currentLine.StartsWith("message"))
{
isMsgHead = true;
opcodeInfo = new OpcodeInfo();
file.AppendLine("\t[ProtoContract]");
className = currentLine.Split(Define.SplitChars, StringSplitOptions.RemoveEmptyEntries)[1];
var splits = currentLine.Split(new[] { "//" }, StringSplitOptions.RemoveEmptyEntries);
if (splits.Length > 1)
{ {
case 2: var parameterArray = currentLine.Split(new[] { "//" }, StringSplitOptions.RemoveEmptyEntries)[1].Trim().Split(',');
responseTypeStr = parameterArray[1].Trim(); parameter = parameterArray[0].Trim();
break;
case 3: switch (parameterArray.Length)
{ {
customRouteType = parameterArray[1].Trim(); case 2:
responseTypeStr = parameterArray[1].Trim();
if (parameterArray.Length == 3) break;
case 3:
{ {
responseTypeStr = parameterArray[2].Trim(); customRouteType = parameterArray[1].Trim();
if (parameterArray.Length == 3)
{
responseTypeStr = parameterArray[2].Trim();
}
break;
} }
break;
} }
} }
}
else
{
parameter = "";
}
file.Append(string.IsNullOrWhiteSpace(parameter)
? $"\tpublic partial class {className} : AProto"
: $"\tpublic partial class {className} : AProto, {parameter}");
opcodeInfo.Name = className;
continue;
}
if (!isMsgHead)
{
continue;
}
switch (currentLine)
{
case "{":
{
file.AppendLine("\n\t{");
if (string.IsNullOrWhiteSpace(parameter) || parameter == "IMessage")
{
opcodeInfo.Code += ++_aMessage;
file.AppendLine($"\t\tpublic uint OpCode() {{ return {opCodeName}.{className}; }}");
}
else else
{ {
if (responseTypeStr != null) parameter = "";
}
file.Append(string.IsNullOrWhiteSpace(parameter)
? $"\tpublic partial class {className} : AProto"
: $"\tpublic partial class {className} : AProto, {parameter}");
opcodeInfo.Name = className;
continue;
}
if (!isMsgHead)
{
continue;
}
switch (currentLine)
{
case "{":
{
file.AppendLine("\n\t{");
if (string.IsNullOrWhiteSpace(parameter) || parameter == "IMessage")
{ {
file.AppendLine("\t\t[ProtoIgnore]"); opcodeInfo.Code += ++_aMessage;
file.AppendLine($"\t\tpublic {responseTypeStr} ResponseType {{ get; set; }}"); file.AppendLine($"\t\tpublic uint OpCode() {{ return {opCodeName}.{className}; }}");
responseTypeStr = null;
} }
else else
{ {
if (parameter.Contains("RouteRequest")) if (responseTypeStr != null)
{ {
Exporter.LogError($"{opcodeInfo.Name} 没指定ResponseType"); file.AppendLine("\t\t[ProtoIgnore]");
file.AppendLine($"\t\tpublic {responseTypeStr} ResponseType {{ get; set; }}");
responseTypeStr = null;
} }
} else
file.AppendLine($"\t\tpublic uint OpCode() {{ return {opCodeName}.{className}; }}");
if (customRouteType != null)
{
file.AppendLine($"\t\tpublic long RouteTypeOpCode() {{ return (long)RouteType.{customRouteType}; }}");
customRouteType = null;
}
else if (parameter is "IAddressableRouteRequest" or "IAddressableRouteMessage")
{
file.AppendLine($"\t\tpublic long RouteTypeOpCode() {{ return CoreRouteType.Addressable; }}");
}
else if (parameter.EndsWith("BsonRouteMessage") || parameter.EndsWith("BsonRouteRequest"))
{
file.AppendLine($"\t\tpublic long RouteTypeOpCode() {{ return CoreRouteType.BsonRoute; }}");
}
else if (parameter is "IRouteMessage" or "IRouteRequest")
{
file.AppendLine($"\t\tpublic long RouteTypeOpCode() {{ return CoreRouteType.Route; }}");
}
switch (parameter)
{
case "IRequest":
case "IBsonRequest":
{ {
opcodeInfo.Code += ++_aRequest; if (parameter.Contains("RouteRequest"))
break;
}
case "IResponse":
case "IBsonResponse":
{
opcodeInfo.Code += ++_aResponse;
file.AppendLine("\t\t[ProtoMember(91, IsRequired = true)]");
file.AppendLine("\t\tpublic uint ErrorCode { get; set; }");
break;
}
default:
{
if (parameter.EndsWith("RouteMessage") || parameter == "IRouteMessage")
{ {
opcodeInfo.Code += ++_aRouteMessage; Exporter.LogError($"{opcodeInfo.Name} 没指定ResponseType");
} }
else if (parameter.EndsWith("RouteRequest") || parameter == "IRouteRequest") }
file.AppendLine($"\t\tpublic uint OpCode() {{ return {opCodeName}.{className}; }}");
if (customRouteType != null)
{
file.AppendLine($"\t\tpublic long RouteTypeOpCode() {{ return (long)RouteType.{customRouteType}; }}");
customRouteType = null;
}
else if (parameter is "IAddressableRouteRequest" or "IAddressableRouteMessage")
{
file.AppendLine($"\t\tpublic long RouteTypeOpCode() {{ return CoreRouteType.Addressable; }}");
}
else if (parameter.EndsWith("BsonRouteMessage") || parameter.EndsWith("BsonRouteRequest"))
{
file.AppendLine($"\t\tpublic long RouteTypeOpCode() {{ return CoreRouteType.BsonRoute; }}");
}
else if (parameter is "IRouteMessage" or "IRouteRequest")
{
file.AppendLine($"\t\tpublic long RouteTypeOpCode() {{ return CoreRouteType.Route; }}");
}
switch (parameter)
{
case "IRequest":
case "IBsonRequest":
{ {
opcodeInfo.Code += ++_aRouteRequest; opcodeInfo.Code += ++_aRequest;
break;
} }
else if (parameter.EndsWith("RouteResponse") || parameter == "IRouteResponse") case "IResponse":
case "IBsonResponse":
{ {
opcodeInfo.Code += ++_aRouteResponse; opcodeInfo.Code += ++_aResponse;
file.AppendLine("\t\t[ProtoMember(91, IsRequired = true)]"); file.AppendLine("\t\t[ProtoMember(91, IsRequired = true)]");
file.AppendLine("\t\tpublic uint ErrorCode { get; set; }"); file.AppendLine("\t\tpublic uint ErrorCode { get; set; }");
break;
} }
default:
{
if (parameter.EndsWith("RouteMessage") || parameter == "IRouteMessage")
{
opcodeInfo.Code += ++_aRouteMessage;
}
else if (parameter.EndsWith("RouteRequest") || parameter == "IRouteRequest")
{
opcodeInfo.Code += ++_aRouteRequest;
}
else if (parameter.EndsWith("RouteResponse") || parameter == "IRouteResponse")
{
opcodeInfo.Code += ++_aRouteResponse;
file.AppendLine("\t\t[ProtoMember(91, IsRequired = true)]");
file.AppendLine("\t\tpublic uint ErrorCode { get; set; }");
}
break; break;
}
} }
} }
_opcodes.Add(opcodeInfo);
continue;
}
case "}":
{
isMsgHead = false;
file.AppendLine("\t}");
continue;
}
case "":
{
continue;
} }
_opcodes.Add(opcodeInfo);
continue;
} }
case "}":
if (currentLine.StartsWith("//"))
{ {
isMsgHead = false; file.AppendFormat("\t\t///<summary>\r\n" + "\t\t/// {0}\r\n" + "\t\t///</summary>\r\n", currentLine.TrimStart('/', '/'));
file.AppendLine("\t}");
continue; continue;
} }
case "":
if (currentLine.StartsWith("repeated"))
{ {
continue; Repeated(file, currentLine);
}
else
{
Members(file, currentLine);
} }
} }
if (currentLine.StartsWith("//")) var csName = $"{Path.GetFileNameWithoutExtension(filePath)}.cs";
foreach (var (directory, template) in saveDirectory)
{ {
file.AppendFormat("\t\t///<summary>\r\n" + "\t\t/// {0}\r\n" + "\t\t///</summary>\r\n", currentLine.TrimStart('/', '/')); var csFile = Path.Combine(directory, csName);
continue; var content = template.Replace("(Content)", file.ToString());
await File.WriteAllTextAsync(csFile, content);
} }
if (currentLine.StartsWith("repeated")) file.Clear();
{
Repeated(file, currentLine);
}
else
{
Members(file, currentLine);
}
}
var csName = $"{Path.GetFileNameWithoutExtension(protoFile)}.cs";
foreach (var (directory, template) in saveDirectory)
{
var csFile = Path.Combine(directory, csName);
var content = template.Replace("(Content)", file.ToString());
await File.WriteAllTextAsync(csFile, content);
} }
file.Clear(); #region GenerateOpCode
file.AppendLine("namespace TEngine"); file.AppendLine("namespace TEngine");
file.AppendLine("{"); file.AppendLine("{");
file.AppendLine($"\tpublic static partial class {opCodeName}"); file.AppendLine($"\tpublic static partial class {opCodeName}");
file.AppendLine("\t{"); file.AppendLine("\t{");
foreach (var opcode in _opcodes) foreach (var opcode in _opcodes)
{ {
file.AppendLine($"\t\t public const int {opcode.Name} = {opcode.Code};"); file.AppendLine($"\t\t public const int {opcode.Name} = {opcode.Code};");
} }
_opcodes.Clear(); _opcodes.Clear();
file.AppendLine("\t}"); file.AppendLine("\t}");
file.AppendLine("}"); file.AppendLine("}");
@@ -333,6 +363,8 @@ public sealed class ProtoBufExporter
var csFile = Path.Combine(directory, $"{opCodeName}.cs"); var csFile = Path.Combine(directory, $"{opCodeName}.cs");
await File.WriteAllTextAsync(csFile, file.ToString()); await File.WriteAllTextAsync(csFile, file.ToString());
} }
#endregion
} }
private async Task RouteType() private async Task RouteType()
@@ -352,11 +384,11 @@ public sealed class ProtoBufExporter
{ {
continue; continue;
} }
var splits = currentLine.Split(new[] {"//"}, StringSplitOptions.RemoveEmptyEntries); var splits = currentLine.Split(new[] { "//" }, StringSplitOptions.RemoveEmptyEntries);
var routeTypeStr = splits[0].Split("=", StringSplitOptions.RemoveEmptyEntries); var routeTypeStr = splits[0].Split("=", StringSplitOptions.RemoveEmptyEntries);
routeTypeFileSb.Append($"\t\t{routeTypeStr[0].Trim()} = {routeTypeStr[1].Trim()},"); routeTypeFileSb.Append($"\t\t{routeTypeStr[0].Trim()} = {routeTypeStr[1].Trim()},");
if (splits.Length > 1) if (splits.Length > 1)
{ {
routeTypeFileSb.Append($" // {splits[1].Trim()}\n"); routeTypeFileSb.Append($" // {splits[1].Trim()}\n");
@@ -372,7 +404,7 @@ public sealed class ProtoBufExporter
await File.WriteAllTextAsync($"{Define.ProtoBufServerDirectory}RouteType.cs", file); await File.WriteAllTextAsync($"{Define.ProtoBufServerDirectory}RouteType.cs", file);
await File.WriteAllTextAsync($"{Define.ProtoBufClientDirectory}RouteType.cs", file); await File.WriteAllTextAsync($"{Define.ProtoBufClientDirectory}RouteType.cs", file);
} }
private void Repeated(StringBuilder file, string newline) private void Repeated(StringBuilder file, string newline)
{ {
try try
@@ -393,7 +425,7 @@ public sealed class ProtoBufExporter
Exporter.LogError($"{newline}\n {e}"); Exporter.LogError($"{newline}\n {e}");
} }
} }
private void Members(StringBuilder file, string currentLine) private void Members(StringBuilder file, string currentLine)
{ {
try try
@@ -406,7 +438,7 @@ public sealed class ProtoBufExporter
var memberIndex = int.Parse(property[3]); var memberIndex = int.Parse(property[3]);
var typeCs = ConvertType(type); var typeCs = ConvertType(type);
string defaultValue = GetDefault(typeCs); string defaultValue = GetDefault(typeCs);
file.AppendLine($"\t\t[ProtoMember({memberIndex})]"); file.AppendLine($"\t\t[ProtoMember({memberIndex})]");
file.AppendLine($"\t\tpublic {typeCs} {name} {{ get; set; }}"); file.AppendLine($"\t\tpublic {typeCs} {name} {{ get; set; }}");
} }
@@ -415,7 +447,7 @@ public sealed class ProtoBufExporter
Exporter.LogError($"{currentLine}\n {e}"); Exporter.LogError($"{currentLine}\n {e}");
} }
} }
private string ConvertType(string type) private string ConvertType(string type)
{ {
return type switch return type switch
@@ -430,11 +462,11 @@ public sealed class ProtoBufExporter
_ => type _ => type
}; };
} }
private string GetDefault(string type) private string GetDefault(string type)
{ {
type = type.Trim(); type = type.Trim();
switch (type) switch (type)
{ {
case "byte": case "byte":
@@ -450,7 +482,7 @@ public sealed class ProtoBufExporter
return "null"; return "null";
} }
} }
private void LoadTemplate() private void LoadTemplate()
{ {
string[] lines = File.ReadAllLines(Define.ProtoBufTemplatePath, Encoding.UTF8); string[] lines = File.ReadAllLines(Define.ProtoBufTemplatePath, Encoding.UTF8);
@@ -468,12 +500,12 @@ public sealed class ProtoBufExporter
flag = 1; flag = 1;
continue; continue;
} }
else if(trim.StartsWith("#else")) else if (trim.StartsWith("#else"))
{ {
flag = 2; flag = 2;
continue; continue;
} }
else if(trim.StartsWith($"#endif")) else if (trim.StartsWith($"#endif"))
{ {
flag = 0; flag = 0;
continue; continue;