mirror of
https://github.com/RRQM/TouchSocket.git
synced 2025-12-20 10:26:43 +08:00
316 lines
13 KiB
C#
316 lines
13 KiB
C#
//------------------------------------------------------------------------------
|
||
// 此代码版权归作者本人若汝棋茗所有
|
||
// 源代码使用协议遵循本仓库的开源协议及附加协议，若本仓库没有设置，则按MIT开源协议授权
|
||
// CSDN博客：https://blog.csdn.net/qq_40374647
|
||
// 哔哩哔哩视频：https://space.bilibili.com/94253567
|
||
// Gitee源代码仓库：https://gitee.com/RRQM_Home
|
||
// Github源代码仓库：https://github.com/RRQM
|
||
// 交流QQ群：234762506
|
||
// 感谢您的下载和使用
|
||
//------------------------------------------------------------------------------
|
||
//------------------------------------------------------------------------------
|
||
using System;
|
||
using System.Collections.Generic;
|
||
using System.Linq;
|
||
using System.Reflection;
|
||
using System.Text;
|
||
|
||
namespace RRQMSocket.RPC.RRQMRPC
|
||
{
|
||
/// <summary>
|
||
/// 代码辅助类
|
||
/// </summary>
|
||
public class PropertyCodeMap
|
||
{
|
||
private static readonly string[] listType = { "List`1", "HashSet`1", "IList`1", "ISet`1", "ICollection`1", "IEnumerable`1" };
|
||
|
||
private static readonly string[] dicType = { "Dictionary`2", "IDictionary`2" };
|
||
|
||
private static readonly Type intType = typeof(int);
|
||
private static readonly Type byteType = typeof(byte);
|
||
private static readonly Type shortType = typeof(short);
|
||
private static readonly Type longType = typeof(long);
|
||
|
||
/// <summary>
|
||
/// 构造函数
|
||
/// </summary>
|
||
public PropertyCodeMap(Assembly assembly, string nameSpace)
|
||
{
|
||
this.Assembly = assembly;
|
||
codeString = new StringBuilder();
|
||
this.nameSpace = nameSpace;
|
||
}
|
||
|
||
/// <summary>
|
||
/// 所属程序集
|
||
/// </summary>
|
||
public Assembly Assembly { get; private set; }
|
||
|
||
/// <summary>
|
||
/// 获取属性代码
|
||
/// </summary>
|
||
public string GetPropertyCode()
|
||
{
|
||
codeString.Clear();
|
||
|
||
codeString.AppendLine("using System;");
|
||
codeString.AppendLine("using RRQMSocket.RPC;");
|
||
codeString.AppendLine("using RRQMCore.Exceptions;");
|
||
codeString.AppendLine("using System.Collections.Generic;");
|
||
codeString.AppendLine("using System.Diagnostics;");
|
||
codeString.AppendLine("using System.Text;");
|
||
codeString.AppendLine("using System.Threading.Tasks;");
|
||
|
||
codeString.AppendLine(string.Format("namespace {0}", nameSpace));
|
||
|
||
codeString.AppendLine("{");
|
||
foreach (var item in propertyDic.Values)
|
||
{
|
||
codeString.AppendLine(item);
|
||
}
|
||
codeString.AppendLine("}");
|
||
return codeString.ToString();
|
||
}
|
||
|
||
private StringBuilder codeString;
|
||
private string nameSpace;
|
||
private Dictionary<Type, string> propertyDic = new Dictionary<Type, string>();
|
||
private Dictionary<Type, string> genericTypeDic = new Dictionary<Type, string>();
|
||
|
||
internal void AddTypeString(Type type)
|
||
{
|
||
if (type.IsByRef)
|
||
{
|
||
string typeName = type.FullName.Replace("&", string.Empty);
|
||
type = Type.GetType(typeName);
|
||
if (type == null)
|
||
{
|
||
type = this.Assembly.GetType(typeName);
|
||
}
|
||
}
|
||
|
||
if (!type.IsPrimitive && type != typeof(string))
|
||
{
|
||
if (type.IsArray)
|
||
{
|
||
AddTypeString(type.GetElementType());
|
||
}
|
||
else if (type.IsGenericType)
|
||
{
|
||
Type[] types = type.GetGenericArguments();
|
||
foreach (Type itemType in types)
|
||
{
|
||
AddTypeString(itemType);
|
||
}
|
||
|
||
if (listType.Contains(type.Name))
|
||
{
|
||
string typeInnerString = this.GetTypeFullName(types[0]);
|
||
string typeString = $"{type.Name.Replace("`1", string.Empty)}<{typeInnerString}>";
|
||
if (!genericTypeDic.ContainsKey(type))
|
||
{
|
||
genericTypeDic.Add(type, typeString);
|
||
}
|
||
}
|
||
else if (dicType.Contains(type.Name))
|
||
{
|
||
string keyString = this.GetTypeFullName(types[0]);
|
||
string valueString = this.GetTypeFullName(types[1]);
|
||
string typeString = $"{type.Name.Replace("`2", string.Empty)}<{keyString},{valueString}>";
|
||
if (!genericTypeDic.ContainsKey(type))
|
||
{
|
||
genericTypeDic.Add(type, typeString);
|
||
}
|
||
}
|
||
}
|
||
else if (type.IsInterface || type.IsAbstract)
|
||
{
|
||
throw new RRQMRPCException("服务参数类型不允许接口或抽象类");
|
||
}
|
||
else if (type.IsEnum)
|
||
{
|
||
Type baseType = Enum.GetUnderlyingType(type);
|
||
StringBuilder stringBuilder = new StringBuilder();
|
||
if (baseType == byteType)
|
||
{
|
||
stringBuilder.AppendLine($"public enum {type.Name}:byte");
|
||
stringBuilder.AppendLine("{");
|
||
Array array = Enum.GetValues(type);
|
||
foreach (object item in array)
|
||
{
|
||
string enumString = item.ToString();
|
||
stringBuilder.AppendLine($"{enumString}={(byte)item},");
|
||
}
|
||
}
|
||
else if (baseType == shortType)
|
||
{
|
||
stringBuilder.AppendLine($"public enum {type.Name}:short");
|
||
stringBuilder.AppendLine("{");
|
||
Array array = Enum.GetValues(type);
|
||
foreach (object item in array)
|
||
{
|
||
string enumString = item.ToString();
|
||
stringBuilder.AppendLine($"{enumString}={(short)item},");
|
||
}
|
||
}
|
||
else if (baseType == intType)
|
||
{
|
||
stringBuilder.AppendLine($"public enum {type.Name}:int");
|
||
stringBuilder.AppendLine("{");
|
||
Array array = Enum.GetValues(type);
|
||
foreach (object item in array)
|
||
{
|
||
string enumString = item.ToString();
|
||
stringBuilder.AppendLine($"{enumString}={(int)item},");
|
||
}
|
||
}
|
||
else if (baseType == longType)
|
||
{
|
||
stringBuilder.AppendLine($"public enum {type.Name}:long");
|
||
stringBuilder.AppendLine("{");
|
||
Array array = Enum.GetValues(type);
|
||
foreach (object item in array)
|
||
{
|
||
string enumString = item.ToString();
|
||
stringBuilder.AppendLine($"{enumString}={(long)item},");
|
||
}
|
||
}
|
||
|
||
stringBuilder.AppendLine("}");
|
||
if (!propertyDic.ContainsKey(type))
|
||
{
|
||
propertyDic.Add(type, stringBuilder.ToString());
|
||
}
|
||
}
|
||
else if (type.IsClass)
|
||
{
|
||
if (type.Assembly == this.Assembly)
|
||
{
|
||
StringBuilder stringBuilder = new StringBuilder();
|
||
|
||
stringBuilder.AppendLine("");
|
||
stringBuilder.AppendLine($"public class {type.Name}");
|
||
if (type.BaseType != typeof(object))
|
||
{
|
||
AddTypeString(type.BaseType);
|
||
if (type.BaseType.IsGenericType)
|
||
{
|
||
Type[] types = type.BaseType.GetGenericArguments();
|
||
foreach (Type itemType in types)
|
||
{
|
||
AddTypeString(itemType);
|
||
}
|
||
if (listType.Contains(type.BaseType.Name))
|
||
{
|
||
string typeString = this.GetTypeFullName(types[0]);
|
||
stringBuilder.Append($":{type.BaseType.Name.Replace("`1", string.Empty)}<{typeString}>");
|
||
}
|
||
else if (dicType.Contains(type.BaseType.Name))
|
||
{
|
||
string keyString = this.GetTypeFullName(types[0]);
|
||
string valueString = this.GetTypeFullName(types[1]);
|
||
stringBuilder.Append($": {type.BaseType.Name.Replace("`2", string.Empty)}<{keyString},{valueString}>");
|
||
}
|
||
}
|
||
else if (type.BaseType.IsClass)
|
||
{
|
||
stringBuilder.AppendLine($": {this.GetTypeFullName(type.BaseType)}");
|
||
}
|
||
}
|
||
stringBuilder.AppendLine("{");
|
||
PropertyInfo[] propertyInfos = type.GetProperties(BindingFlags.Instance | BindingFlags.Public | BindingFlags.DeclaredOnly | BindingFlags.GetProperty | BindingFlags.SetProperty);
|
||
|
||
foreach (PropertyInfo itemProperty in propertyInfos)
|
||
{
|
||
AddTypeString(itemProperty.PropertyType);
|
||
if (propertyDic.ContainsKey(itemProperty.PropertyType))
|
||
{
|
||
stringBuilder.Append($"public {itemProperty.PropertyType.Name} {itemProperty.Name}");
|
||
}
|
||
else if (itemProperty.PropertyType.IsGenericType)
|
||
{
|
||
Type[] types = itemProperty.PropertyType.GetGenericArguments();
|
||
foreach (Type itemType in types)
|
||
{
|
||
AddTypeString(itemType);
|
||
}
|
||
|
||
if (listType.Contains(itemProperty.PropertyType.Name))
|
||
{
|
||
string typeString = this.GetTypeFullName(types[0]);
|
||
stringBuilder.Append($"public {itemProperty.PropertyType.Name.Replace("`1", string.Empty)}<{typeString}> {itemProperty.Name}");
|
||
}
|
||
else if (dicType.Contains(itemProperty.PropertyType.Name))
|
||
{
|
||
string keyString = this.GetTypeFullName(types[0]);
|
||
string valueString = this.GetTypeFullName(types[1]);
|
||
stringBuilder.Append($"public {itemProperty.PropertyType.Name.Replace("`2", string.Empty)}<{keyString},{valueString}> {itemProperty.Name}");
|
||
}
|
||
}
|
||
else
|
||
{
|
||
AddTypeString(itemProperty.PropertyType);
|
||
stringBuilder.Append($"public {itemProperty.PropertyType.FullName} {itemProperty.Name}");
|
||
}
|
||
|
||
stringBuilder.AppendLine("{get;set;}");
|
||
}
|
||
|
||
stringBuilder.AppendLine("}");
|
||
|
||
if (!propertyDic.ContainsKey(type))
|
||
{
|
||
propertyDic.Add(type, stringBuilder.ToString());
|
||
}
|
||
}
|
||
}
|
||
}
|
||
}
|
||
|
||
/// <summary>
|
||
/// 获取类型全名
|
||
/// </summary>
|
||
/// <param name="type"></param>
|
||
/// <returns></returns>
|
||
public string GetTypeFullName(Type type)
|
||
{
|
||
if (type.FullName == null)
|
||
{
|
||
return type.Name.Replace("&", string.Empty);
|
||
}
|
||
else if (type == typeof(void))
|
||
{
|
||
return null;
|
||
}
|
||
|
||
if (type.IsByRef)
|
||
{
|
||
string typeName = type.FullName.Replace("&", string.Empty);
|
||
type = Type.GetType(typeName);
|
||
if (type == null)
|
||
{
|
||
type = this.Assembly.GetType(typeName);
|
||
}
|
||
}
|
||
|
||
if (type.IsPrimitive || type == typeof(string))
|
||
{
|
||
return type.FullName;
|
||
}
|
||
else if (listType.Contains(type.Name) || dicType.Contains(type.Name))
|
||
{
|
||
return genericTypeDic[type];
|
||
}
|
||
else if (propertyDic.ContainsKey(type))
|
||
{
|
||
return this.nameSpace + "." + type.Name;
|
||
}
|
||
else
|
||
{
|
||
return type.FullName;
|
||
}
|
||
}
|
||
|
||
|
||
}
|
||
} |