Выполните хранимую процедуру 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() метод. Я уже пробовал это с провайдером трассировки, и он работает нормально. Как только я протестирую это с моим решением, я добавлю больше информации.

Мы использовали этот шаблон.

Но мы сделали так, чтобы хранимая процедура была первой операцией внутри каждого контекста БД.

Другие вопросы по тегам