.NET MVC 3自定义控制器属性

时间:2011-05-13 17:27:49

标签: asp.net-mvc-3

这可能是天空中的馅饼,但我想知道是否可以使用自定义控制器属性完成以下操作。

对于我的大多数控制器,我将向控制器中的每个操作传递一个名为“r”的URL参数。 “r”与我数据库中种族表中的种族ID相关联。

我想要发生的是,无论何时调用控制器动作,它都会自动检查是否存在“r”,查询数据库以确保“r”属于登录用户并设置名为ViewBag.RaceId的viewbag变量等于“r”。

如果不满足任何条件,它会将它们重定向回登录页面。

我正在努力让我的代码尽可能干。

非常感谢任何指导。

1 个答案:

答案 0 :(得分:6)

您可以编写自定义Authorize属性:

public class MyAuthorizeAttribute : AuthorizeAttribute
{
    protected override bool AuthorizeCore(HttpContextBase httpContext)
    {
        var isAuthorized = base.AuthorizeCore(httpContext);
        if (isAuthorized)
        {
            var request = httpContext.Request;
            // Fetch "r" from the route data or request
            var r = request.RequestContext.RouteData.Values["r"] 
                ?? request["r"];
            var currentUser = httpContext.User.Identity.Name;
            if (!CheckIfRBelongsToTheCurrentLoggedInUser(currentUser, r))
            {
                return false;
            }
        }
        return isAuthorized;
    }
}

现在剩下的就是用这个自定义属性装饰你的控制器/动作:

[MyAuthorize]
public ActionResult Foo()
{
    //...
}

如果你想在ViewBag中添加一些东西,你可以暂时将它存储在httpContext.Items方法中的AuthorizeCore中,以便成功,然后覆盖OnAuthorization方法,检查上下文中是否存在此项目。如果它存在,您可以将其存储在filterContext.Controller.ViewBag

相关问题