Выполните хранимую процедуру SQL перед каждым запросом, сгенерированным EntityFramework
Мне нужно выполнять хранимую процедуру SQL каждый раз, прежде чем я запрашиваю свой ObjectContext. Чего я хочу добиться - это установить CONTEXT_INFO
к значению, которое будет позже использовано с большинством моих запросов.
Кто-нибудь делал это? Это возможно?
[РЕДАКТИРОВАТЬ]
В настоящее время я достигаю этого, открывая соединение и выполняя хранимую процедуру в моем конструкторе ObjectContext следующим образом:
public partial class MyEntitiesContext
{
public MyEntitiesContext(int contextInfo) : this()
{
if (Connection.State != ConnectionState.Open)
{
Connection.Open(); // open connection if not already open
}
var connection = ((EntityConnection)Connection).StoreConnection;
using (var cmd = connection.CreateCommand())
{
// run stored procedure to set ContextInfo to contextInfo
cmd.CommandType = CommandType.StoredProcedure;
cmd.CommandText = "[dbo].[SetContextInfo]";
cmd.Parameters.Add(new SqlParameter("@ci", _contextInfo));
cmd.ExecuteNonQuery();
}
// leave the connection open to reuse later
}
}
Тогда в моем интеграционном тесте:
[TestMethod]
public void TestMethod1()
{
using (var ctx = new MyEntitiesContext(1))
{
Assert.AreEqual(2, ctx.Roles.ToList().Count);
Assert.AreEqual(2, ctx.Users.ToList().Count);
}
}
Но для этого необходимо, чтобы я оставил соединение открытым - это подвержено ошибкам, так как мне всегда понадобится CONTEXT_INFO, и другой разработчик может легко это сделать:
[TestMethod]
public void TestMethod2()
{
using (var ctx = new MyEntitiesContext(1))
{
// do something here
// ... more here :)
ctx.Connection.Close(); // then out of the blue comes Close();
// do something here
Assert.AreEqual(2, ctx.Roles.ToList().Count);
Assert.AreEqual(2, ctx.Users.ToList().Count); // this fails since the where
// clause will be:
// WHERE ColumnX = CAST(CAST(CONTEXT_INFO() AS BINARY(4)) AS INT)
// and CONTEXT_INFO is empty - there are no users with ColumnX set to 0
// while there are 2 users with it set to 1 so this test should pass
}
}
Вышеприведенное означает, что я могу написать код, как в моем тесте, и все будет зеленым (ДАЙ!), Но тогда мой коллега использует код из TestMethod2 где-то в своей бизнес-логике, и все вышло - и никто не знает, где и почему, так как все тесты зеленые: /
[EDIT2]
Этот пост, конечно, не отвечает на мой вопрос, но на самом деле решает мою проблему. Возможно, переход на NHibernate лучше подойдет для моих целей:)
2 ответа
Наконец я нашел ответ. Я могу обернуть соединение с помощью набора инструментов EFProvider от EFProviderWrappers. Чтобы сделать это, я в основном должен извлечь из EFProviderWrapperConnection
и переопределить DbConnection.Open()
метод. Я уже пробовал это с провайдером трассировки, и он работает нормально. Как только я протестирую это с моим решением, я добавлю больше информации.
Мы использовали этот шаблон.
Но мы сделали так, чтобы хранимая процедура была первой операцией внутри каждого контекста БД.