程序員的自我救贖---3.2：SSO及應用案例

using System;
using System.Collections.Generic;
using System.Dynamic;
using System.Linq;
using System.Net;
using System.Web;
using System.Web.Mvc;
using System.Xml;
using Winner.Framework.MVC.Attribute;
using Winner.Framework.MVC.GlobalContext;
using Winner.Framework.MVC.Models;
using Winner.Framework.MVC.Models.Account;
using Winner.Framework.Utils;

namespace Winner.Framework.MVC
{
    /// <summary>
    /// PC Web用戶登錄檢查
    /// </summary>
    public class AuthLoginAttribute : AuthorizationFilterAttribute
    {
        /// <summary>
        /// 實例化一個新的驗證對象
        /// </summary>
        /// <param name="ignore">是否忽略檢查</param>
        public AuthLoginAttribute(bool ignore = false)
            : base(ignore)
        {
        }

        /// <summary>
        /// 登錄驗證
        /// </summary>
        /// <param name="context">當前上下文</param>
        protected override bool OnAuthorizationing(AuthorizationContext context)
        {
            //Ajax請求但又未登陸時則返回信息
            if (!ApplicationContext.Current.IsLogined && base.ContextProvider.IsAjaxRequest)
            {
                OutputResult("未登陸或者會話已過時，請從新登陸！", 401);
                return false;
            }
            if (context.HttpContext.Session == null)
            {
                throw new Exception("服務器Session不可用！");
            }
            try
            {
                //調用提供者進行登錄
                ProviderManager.LoginProvider.Login();
            }
            catch (Exception ex)
            {
                Log.Error(ex);
                if (ex.InnerException != null)
                {
                    Log.Error(ex.InnerException);
                }
                OutputResult("登錄時出現系統繁忙，請稍後再試！", 401);
                return false;
            }


            //若是沒有登錄則返回
            if (!ApplicationContext.Current.IsLogined)
            {
                OutputResult("未登陸", 401);
                return false;
            }
            return true;
        }
    }
}

咱們看到一開始咱們有base(ignore)；這個我在前面的篇章中有講到過，這個能夠經過配置文件配置，目的是省去咱們每一個項目開發的時候都要去登陸。

在配置文件中默認一個登陸帳號，這樣調試時候能省不少時間。

咱們判斷的步驟是這樣的：

第一步：若是用戶是ajax請求，而且用戶信息不存在的話直接返回false。這裏是應對用戶登陸以後用戶長時間未操做形成用戶信息過時失效

由於咱們的Winner框架基本都是Ajax請求，因此當兩個條件都存在的時候就直接返回401錯誤。若是界面顯示401則從新刷新一下，

由於刷新就不是Ajax了，因此就會跳到登陸頁去登陸。

第二步：判斷Session是否可用，不可用就直接拋異常了，就是我上面說的禁用Session這種狀況。

第三步：在ProviderManager.LoginProvider.Login()，咱們纔是作了具體的操做，咱們看一下Login()代碼：

   public void Login()
        {
            //檢查是否有SSO站點POST過來的用戶退出數據
            string str = HttpContext.Current.Request.Url.Query;
            if (str.Contains("logout"))
            {
                //TODO：退出本地登錄
                Logout();
                return;
            }
            //檢查本地系統是否已登錄
            if (ApplicationContext.Current.IsLogined)
                return;

            //判斷是否有配置自動登錄
            if (GlobalConfig.IsAutoLogin)
            {
                //代理登錄配置文件所配置的用戶
                var autoResult = ApplicationContext.UserLogin(GlobalConfig.DefaultAutoLoginUserId, true);
                if (!autoResult.Success)
                {
                    throw new Exception(autoResult.Message);
                }
                HttpCookie cookie = new HttpCookie("ticket");
                cookie.Value = GlobalConfig.DefaultAutoLoginToken;
                HttpContext.Current.Response.AppendCookie(cookie);
                return;
            }

            //若是沒有Ticket直接跳轉到SSO進行檢查
            int userId;
            if (!ApplicationContext.GetNodeIdByTicket(out userId))
            {
                SSOLogin();
                return;
            }
            Log.Debug("user_id={0}", userId);
            //登錄到本地系統
            var result = ApplicationContext.UserLogin(userId, false);
            if (!result.Success)
            {
                throw new Exception(result.Message);
            }
        }

  private void SSOLogin()
        {
            string service = HttpContext.Current.Request.Url.AbsoluteUri;
            service = Regex.Replace(service, @"\?ticket[^&]*.", "");
            string url = string.Concat(GlobalConfig.SSO_LoginURL, "?service=", HttpContext.Current.Server.UrlEncode(service));
            HttpContext.Current.Response.Redirect(url);
        }