|
2 | 2 |
|
3 | 3 | namespace StackExchange.Profiling.Data
|
4 | 4 | {
|
| 5 | + using System.Collections.Generic; |
5 | 6 | using System.Data.Common;
|
6 | 7 | using System.Data.Entity.Core.Common;
|
7 | 8 | using System.Data.Entity.Core.Common.CommandTrees;
|
8 | 9 | using System.Data.Entity.Core.Metadata.Edm;
|
| 10 | + using System.Data.Entity.Spatial; |
9 | 11 | using System.Diagnostics;
|
| 12 | + using System.Linq; |
10 | 13 | using System.Reflection;
|
11 | 14 | using StackExchange.Profiling;
|
12 | 15 |
|
@@ -160,5 +163,74 @@ private static DbConnection GetRealConnection(DbConnection connection)
|
160 | 163 |
|
161 | 164 | return connection;
|
162 | 165 | }
|
| 166 | + |
| 167 | + private static DbDataReader GetSpatialDataReader(DbDataReader fromReader) |
| 168 | + { |
| 169 | + var profiled = fromReader as ProfiledDbDataReader; |
| 170 | + if (profiled != null) |
| 171 | + { |
| 172 | + fromReader = profiled.WrappedReader; |
| 173 | + } |
| 174 | + return fromReader; |
| 175 | + } |
| 176 | + |
| 177 | + public override object GetService(Type type, object key) |
| 178 | + { |
| 179 | + return _tail.GetService(type, key); |
| 180 | + } |
| 181 | + |
| 182 | + public override IEnumerable<object> GetServices(Type type, object key) |
| 183 | + { |
| 184 | + return _tail.GetServices(type, key); |
| 185 | + } |
| 186 | + |
| 187 | + protected override DbSpatialDataReader GetDbSpatialDataReader(DbDataReader fromReader, string manifestToken) |
| 188 | + { |
| 189 | + var setDbParameterValueMethod = |
| 190 | + _tail.GetType().GetMethods(BindingFlags.Instance | BindingFlags.NonPublic).FirstOrDefault(f => f.Name.Equals("GetDbSpatialDataReader")); |
| 191 | + |
| 192 | + var reader = GetSpatialDataReader(fromReader); |
| 193 | + |
| 194 | + |
| 195 | + if (setDbParameterValueMethod == null) |
| 196 | + { |
| 197 | + return base.GetDbSpatialDataReader(reader, manifestToken); |
| 198 | + } |
| 199 | + |
| 200 | + var result = setDbParameterValueMethod.Invoke(_tail, new object[] { reader, manifestToken }); |
| 201 | + return result as DbSpatialDataReader; |
| 202 | + } |
| 203 | + |
| 204 | + protected override DbSpatialServices DbGetSpatialServices(string manifestToken) |
| 205 | + { |
| 206 | + var dbGetSpatialServices = |
| 207 | + _tail.GetType().GetMethods(BindingFlags.Instance | BindingFlags.NonPublic).FirstOrDefault(f => f.Name.Equals("DbGetSpatialServices")); |
| 208 | + |
| 209 | + return dbGetSpatialServices.Invoke(_tail, new[] { manifestToken }) as DbSpatialServices; |
| 210 | + } |
| 211 | + |
| 212 | + |
| 213 | + |
| 214 | + |
| 215 | + protected override void SetDbParameterValue(DbParameter parameter, TypeUsage parameterType, object value) |
| 216 | + { |
| 217 | + // if this is available in _tail, use it |
| 218 | + var setDbParameterValueMethod = _tail.GetType().GetMethods(BindingFlags.Instance | BindingFlags.NonPublic).FirstOrDefault(f => f.Name.Equals("SetDbParameterValue")); |
| 219 | + if (setDbParameterValueMethod != null) |
| 220 | + { |
| 221 | + setDbParameterValueMethod.Invoke(_tail, new[] { parameter, parameterType, value }); |
| 222 | + return; |
| 223 | + } |
| 224 | + |
| 225 | + // this should never need to be called, but just in case get the Provider Value |
| 226 | + if (value is DbGeography) |
| 227 | + { |
| 228 | + value = ((DbGeography)value).ProviderValue; |
| 229 | + } |
| 230 | + |
| 231 | + base.SetDbParameterValue(parameter, parameterType, value); |
| 232 | + } |
| 233 | + |
| 234 | + |
163 | 235 | }
|
164 | 236 | }
|
0 commit comments