这可能是天空中的馅饼,但我想知道是否可以使用自定义控制器属性完成以下操作。
对于我的大多数控制器,我将向控制器中的每个操作传递一个名为“r”的URL参数。 “r”与我数据库中种族表中的种族ID相关联。
我想要发生的是,无论何时调用控制器动作,它都会自动检查是否存在“r”,查询数据库以确保“r”属于登录用户并设置名为ViewBag.RaceId的viewbag变量等于“r”。
如果不满足任何条件,它会将它们重定向回登录页面。
我正在努力让我的代码尽可能干。
非常感谢任何指导。
答案 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
。