AutoInit.cs 2.1 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455
  1. using JLHHJSvr.LJException;
  2. using System;
  3. using System.Data.SqlClient;
  4. using System.Diagnostics;
  5. using System.Linq;
  6. using System.Text;
  7. namespace JLHHJSvr.Tools
  8. {
  9. public class AutoInit
  10. {
  11. public static void AutoInitS <T>(SqlCommand cmd, T instance){
  12. cmd.CommandText = @"SELECT COLUMN_NAME, IS_NULLABLE FROM INFORMATION_SCHEMA.COLUMNS WHERE TABLE_NAME = @tableName";
  13. cmd.Parameters.Clear();
  14. cmd.Parameters.AddWithValue("@tableName", instance.GetType().Name);
  15. using (var reader = cmd.ExecuteReader())
  16. {
  17. while (reader.Read())
  18. {
  19. var column = Convert.ToString(reader["COLUMN_NAME"]).Trim();
  20. var isNullable = Convert.ToString(reader["IS_NULLABLE"]).Trim();
  21. var property = instance.GetType().GetProperties().FirstOrDefault(prop => prop.Name.Equals(column, StringComparison.OrdinalIgnoreCase));
  22. if(property == null)
  23. {
  24. //throw new LJCommonException($"数据库字段:{column}在模型中找不到相关属性!");
  25. continue;
  26. }
  27. if (isNullable.Equals("NO") && property.GetValue(instance) == null)
  28. {
  29. var type = property.PropertyType;
  30. if (type.IsGenericType && type.GetGenericTypeDefinition() == typeof(Nullable<>))
  31. {
  32. type = type.GetGenericArguments()[0];
  33. }
  34. if (type.IsAssignableFrom(typeof(string)))
  35. {
  36. property.SetValue(instance, "");
  37. }
  38. else if (type.IsAssignableFrom(typeof(int))
  39. || type.IsAssignableFrom(typeof(decimal))
  40. || type.IsAssignableFrom(typeof(byte)))
  41. {
  42. property.SetValue(instance, Activator.CreateInstance(type));
  43. }
  44. }
  45. }
  46. }
  47. }
  48. }
  49. }