PostgreSQLDbBind.cs 9.4 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158
  1. using System;
  2. using System.Collections.Generic;
  3. using System.Linq;
  4. namespace SqlSugar
  5. {
  6. public class PostgreSQLDbBind : DbBindProvider
  7. {
  8. public override string GetDbTypeName(string csharpTypeName)
  9. {
  10. if (csharpTypeName == UtilConstants.ByteArrayType.Name)
  11. return "bytea";
  12. if (csharpTypeName.ToLower() == "int32")
  13. csharpTypeName = "int";
  14. if (csharpTypeName.ToLower() == "int16")
  15. csharpTypeName = "short";
  16. if (csharpTypeName.ToLower() == "int64")
  17. csharpTypeName = "long";
  18. if (csharpTypeName.ToLower().IsIn("boolean", "bool"))
  19. csharpTypeName = "bool";
  20. if (csharpTypeName == "DateTimeOffset")
  21. csharpTypeName = "DateTime";
  22. var mappings = this.MappingTypes.Where(it => it.Value.ToString().Equals(csharpTypeName, StringComparison.CurrentCultureIgnoreCase)).ToList();
  23. if (mappings != null && mappings.Count > 0)
  24. return mappings.First().Key;
  25. else
  26. return "varchar";
  27. }
  28. public override string GetPropertyTypeName(string dbTypeName)
  29. {
  30. dbTypeName = dbTypeName.ToLower();
  31. var propertyTypes = MappingTypes.Where(it => it.Value.ToString().ToLower() == dbTypeName || it.Key.ToLower() == dbTypeName);
  32. if (propertyTypes == null)
  33. {
  34. return "other";
  35. }
  36. else if (dbTypeName == "xml" || dbTypeName == "string"|| dbTypeName == "jsonb"|| dbTypeName == "json")
  37. {
  38. return "string";
  39. }else if (dbTypeName == "bpchar")//数据库char datatype 查询出来的时候是 bpchar
  40. {
  41. return "char";
  42. }
  43. if (dbTypeName == "byte[]")
  44. {
  45. return "byte[]";
  46. }
  47. else if (propertyTypes == null || propertyTypes.Count() == 0)
  48. {
  49. if (dbTypeName.StartsWith("_"))
  50. {
  51. var dbTypeName2 = dbTypeName.TrimStart('_');
  52. return MappingTypes.Where(it => it.Value.ToString().ToLower() == dbTypeName2 || it.Key.ToLower() == dbTypeName2).Select(it => it.Value + "[]").First();
  53. }
  54. Check.ThrowNotSupportedException(string.Format(" \"{0}\" Type NotSupported, DbBindProvider.GetPropertyTypeName error.", dbTypeName));
  55. return null;
  56. }
  57. else if (propertyTypes.First().Value == CSharpDataType.byteArray)
  58. {
  59. return "byte[]";
  60. }
  61. else
  62. {
  63. return propertyTypes.First().Value.ToString();
  64. }
  65. }
  66. public override List<KeyValuePair<string, CSharpDataType>> MappingTypes
  67. {
  68. get
  69. {
  70. var extService = this.Context.CurrentConnectionConfig.ConfigureExternalServices;
  71. if (extService != null && extService.AppendDataReaderTypeMappings.HasValue())
  72. {
  73. return extService.AppendDataReaderTypeMappings.Union(MappingTypesConst).ToList();
  74. }
  75. else
  76. {
  77. return MappingTypesConst;
  78. }
  79. }
  80. }
  81. public static List<KeyValuePair<string, CSharpDataType>> MappingTypesConst = new List<KeyValuePair<string, CSharpDataType>>(){
  82. new KeyValuePair<string, CSharpDataType>("int2",CSharpDataType.@short),
  83. //new KeyValuePair<string, CSharpDataType>("int1",CSharpDataType.@byte),
  84. new KeyValuePair<string, CSharpDataType>("smallint",CSharpDataType.@short),
  85. new KeyValuePair<string, CSharpDataType>("smallint",CSharpDataType.@byte),
  86. new KeyValuePair<string, CSharpDataType>("int4",CSharpDataType.@int),
  87. new KeyValuePair<string, CSharpDataType>("serial",CSharpDataType.@int),
  88. new KeyValuePair<string, CSharpDataType>("integer",CSharpDataType.@int),
  89. new KeyValuePair<string, CSharpDataType>("int8",CSharpDataType.@long),
  90. new KeyValuePair<string, CSharpDataType>("long",CSharpDataType.@long),
  91. new KeyValuePair<string, CSharpDataType>("bigint",CSharpDataType.@long),
  92. new KeyValuePair<string, CSharpDataType>("float4",CSharpDataType.@float),
  93. new KeyValuePair<string, CSharpDataType>("float4",CSharpDataType.Single),
  94. new KeyValuePair<string, CSharpDataType>("real",CSharpDataType.@float),
  95. new KeyValuePair<string, CSharpDataType>("float8",CSharpDataType.@double),
  96. new KeyValuePair<string, CSharpDataType>("double precision",CSharpDataType.@int),
  97. new KeyValuePair<string, CSharpDataType>("numeric",CSharpDataType.@decimal),
  98. new KeyValuePair<string, CSharpDataType>("decimal",CSharpDataType.@decimal),
  99. new KeyValuePair<string, CSharpDataType>("path",CSharpDataType.@decimal),
  100. new KeyValuePair<string, CSharpDataType>("point",CSharpDataType.@decimal),
  101. new KeyValuePair<string, CSharpDataType>("polygon",CSharpDataType.@decimal),
  102. new KeyValuePair<string, CSharpDataType>("boolean",CSharpDataType.@bool),
  103. new KeyValuePair<string, CSharpDataType>("bool",CSharpDataType.@bool),
  104. new KeyValuePair<string, CSharpDataType>("box",CSharpDataType.@bool),
  105. new KeyValuePair<string, CSharpDataType>("bytea",CSharpDataType.byteArray),
  106. new KeyValuePair<string, CSharpDataType>("varchar",CSharpDataType.@string),
  107. new KeyValuePair<string, CSharpDataType>("clob",CSharpDataType.@string),
  108. new KeyValuePair<string, CSharpDataType>("nvarchar2",CSharpDataType.@string),
  109. new KeyValuePair<string, CSharpDataType>("varchar2",CSharpDataType.@string),
  110. new KeyValuePair<string, CSharpDataType>("character varying",CSharpDataType.@string),
  111. new KeyValuePair<string, CSharpDataType>("geometry",CSharpDataType.@string),
  112. new KeyValuePair<string, CSharpDataType>("name",CSharpDataType.@string),
  113. new KeyValuePair<string, CSharpDataType>("text",CSharpDataType.@string),
  114. new KeyValuePair<string, CSharpDataType>("char",CSharpDataType.@string),
  115. new KeyValuePair<string, CSharpDataType>("character",CSharpDataType.@string),
  116. new KeyValuePair<string, CSharpDataType>("cidr",CSharpDataType.@string),
  117. new KeyValuePair<string, CSharpDataType>("circle",CSharpDataType.@string),
  118. new KeyValuePair<string, CSharpDataType>("tsquery",CSharpDataType.@string),
  119. new KeyValuePair<string, CSharpDataType>("tsvector",CSharpDataType.@string),
  120. new KeyValuePair<string, CSharpDataType>("txid_snapshot",CSharpDataType.@string),
  121. new KeyValuePair<string, CSharpDataType>("uuid",CSharpDataType.Guid),
  122. new KeyValuePair<string, CSharpDataType>("xml",CSharpDataType.@string),
  123. new KeyValuePair<string, CSharpDataType>("json",CSharpDataType.@string),
  124. new KeyValuePair<string, CSharpDataType>("interval",CSharpDataType.@decimal),
  125. new KeyValuePair<string, CSharpDataType>("lseg",CSharpDataType.@decimal),
  126. new KeyValuePair<string, CSharpDataType>("macaddr",CSharpDataType.@decimal),
  127. new KeyValuePair<string, CSharpDataType>("money",CSharpDataType.@decimal),
  128. new KeyValuePair<string, CSharpDataType>("timestamp",CSharpDataType.DateTime),
  129. new KeyValuePair<string, CSharpDataType>("timestamp with time zone",CSharpDataType.DateTime),
  130. new KeyValuePair<string, CSharpDataType>("timestamptz",CSharpDataType.DateTime),
  131. new KeyValuePair<string, CSharpDataType>("timestamp without time zone",CSharpDataType.DateTime),
  132. new KeyValuePair<string, CSharpDataType>("date",CSharpDataType.DateTime),
  133. new KeyValuePair<string, CSharpDataType>("time",CSharpDataType.DateTime),
  134. new KeyValuePair<string, CSharpDataType>("time with time zone",CSharpDataType.DateTime),
  135. new KeyValuePair<string, CSharpDataType>("timetz",CSharpDataType.DateTime),
  136. new KeyValuePair<string, CSharpDataType>("time without time zone",CSharpDataType.DateTime),
  137. new KeyValuePair<string, CSharpDataType>("bit",CSharpDataType.byteArray),
  138. new KeyValuePair<string, CSharpDataType>("bit varying",CSharpDataType.byteArray),
  139. new KeyValuePair<string, CSharpDataType>("varbit",CSharpDataType.@byte),
  140. new KeyValuePair<string, CSharpDataType>("time",CSharpDataType.TimeSpan),
  141. new KeyValuePair<string, CSharpDataType>("public.geometry",CSharpDataType.@object),
  142. new KeyValuePair<string, CSharpDataType>("inet",CSharpDataType.@object)
  143. };
  144. public override List<string> StringThrow
  145. {
  146. get
  147. {
  148. return new List<string>() { "int32", "datetime", "decimal", "double", "byte" };
  149. }
  150. }
  151. }
  152. }