DataTableBuilder.cs 2.5 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273
  1. using System;
  2. using System.Data;
  3. using System.Reflection;
  4. using System.Reflection.Emit;
  5. namespace Masuit.Tools.Database
  6. {
  7. internal sealed class DataTableBuilder<T>
  8. {
  9. private static readonly MethodInfo GetValueMethod = typeof(DataRow).GetMethod("get_Item", new[]
  10. {
  11. typeof(int)
  12. });
  13. private static readonly MethodInfo IsDbNullMethod = typeof(DataRow).GetMethod("IsNull", new[]
  14. {
  15. typeof(int)
  16. });
  17. private delegate T Load(DataRow dataRecord);
  18. private Load _handler;
  19. private DataTableBuilder()
  20. {
  21. }
  22. public T Build(DataRow dataRecord)
  23. {
  24. return _handler(dataRecord);
  25. }
  26. public static DataTableBuilder<T> CreateBuilder(DataRow dataRecord)
  27. {
  28. var methodCreateEntity = new DynamicMethod("DynamicCreateEntity", typeof(T), new[]
  29. {
  30. typeof(DataRow)
  31. }, typeof(T), true);
  32. var generator = methodCreateEntity.GetILGenerator();
  33. var result = generator.DeclareLocal(typeof(T));
  34. generator.Emit(OpCodes.Newobj, typeof(T).GetConstructor(Type.EmptyTypes));
  35. generator.Emit(OpCodes.Stloc, result);
  36. for (int i = 0; i < dataRecord.ItemArray.Length; i++)
  37. {
  38. var propertyInfo = typeof(T).GetProperty(dataRecord.Table.Columns[i].ColumnName);
  39. var endIfLabel = generator.DefineLabel();
  40. if (propertyInfo == null || propertyInfo.GetSetMethod() == null)
  41. {
  42. continue;
  43. }
  44. generator.Emit(OpCodes.Ldarg_0);
  45. generator.Emit(OpCodes.Ldc_I4, i);
  46. generator.Emit(OpCodes.Callvirt, IsDbNullMethod);
  47. generator.Emit(OpCodes.Brtrue, endIfLabel);
  48. generator.Emit(OpCodes.Ldloc, result);
  49. generator.Emit(OpCodes.Ldarg_0);
  50. generator.Emit(OpCodes.Ldc_I4, i);
  51. generator.Emit(OpCodes.Callvirt, GetValueMethod);
  52. generator.Emit(OpCodes.Unbox_Any, propertyInfo.PropertyType);
  53. generator.Emit(OpCodes.Callvirt, propertyInfo.GetSetMethod());
  54. generator.MarkLabel(endIfLabel);
  55. }
  56. generator.Emit(OpCodes.Ldloc, result);
  57. generator.Emit(OpCodes.Ret);
  58. return new DataTableBuilder<T>
  59. {
  60. _handler = (Load)methodCreateEntity.CreateDelegate(typeof(Load))
  61. };
  62. }
  63. }
  64. }