395 lines
12 KiB
C#
395 lines
12 KiB
C#
/*******************************************************************************
|
||
* Copyright © 2016 HT.Cloud.Framework 版权所有
|
||
* Author: HT.Cloud
|
||
* Description: WaterCloud快速开发平台
|
||
* Website：
|
||
*********************************************************************************/
|
||
|
||
using System;
|
||
using System.Collections.Generic;
|
||
using System.Threading.Tasks;
|
||
|
||
namespace HT.Cloud.Code
|
||
{
|
||
public class OperatorProvider
|
||
{
|
||
//是否允许一个账户在多处登录
|
||
private static bool LoginMultiple = GlobalContext.SystemConfig.LoginMultiple;
|
||
|
||
//缓存过期时间
|
||
private static int LoginExpire = GlobalContext.SystemConfig.LoginExpire;
|
||
|
||
private static string projectPrefix = GlobalContext.SystemConfig.ProjectPrefix;
|
||
|
||
public static OperatorProvider Provider
|
||
{
|
||
get { return new OperatorProvider(); }
|
||
}
|
||
|
||
//watercloud_operator_pc_ PC端登录
|
||
//watercloud_operator_info_ 登录次数
|
||
//
|
||
/// <summary>
|
||
/// 缓存操作类
|
||
/// </summary>
|
||
private string cacheKeyOperator = projectPrefix + "_operator_";// +登录者token
|
||
|
||
private string cacheKeyToken = projectPrefix + "_token_";// +登录者token
|
||
private string cacheKeyError = projectPrefix + "_error_";// + Mark
|
||
|
||
/// <summary>
|
||
/// 秘钥
|
||
/// </summary>
|
||
private string LoginUserToken = projectPrefix + "_Token";
|
||
|
||
/// <summary>
|
||
/// 标记登录的浏览器
|
||
/// </summary>
|
||
private string LoginUserMarkKey = projectPrefix + "_Mark";
|
||
|
||
public string GetProvider(string key)
|
||
{
|
||
var token = GetToken();
|
||
if (!string.IsNullOrEmpty(token))
|
||
return token;
|
||
token = WebHelper.GetCookie(key).ToString();
|
||
if (!string.IsNullOrEmpty(token))
|
||
return token;
|
||
return WebHelper.GetSession(key).ToString();
|
||
}
|
||
|
||
public void SetProvider(string key, string value)
|
||
{
|
||
WebHelper.WriteCookie(key, value);
|
||
WebHelper.WriteSession(key, value);
|
||
}
|
||
|
||
public string GetToken()
|
||
{
|
||
try
|
||
{
|
||
if (GlobalContext.HttpContext == null)
|
||
{
|
||
return null;
|
||
}
|
||
//查请求头
|
||
string token = GlobalContext.HttpContext.Request.Headers[GlobalContext.SystemConfig.TokenName].ParseToString();
|
||
if (!String.IsNullOrEmpty(token)) return token;
|
||
|
||
//查参数
|
||
token = GlobalContext.HttpContext.Request.Query[GlobalContext.SystemConfig.TokenName];
|
||
if (!String.IsNullOrEmpty(token)) return token;
|
||
|
||
//查cookies
|
||
string cookie = GlobalContext.HttpContext.Request.Cookies[GlobalContext.SystemConfig.TokenName];
|
||
return cookie == null ? string.Empty : cookie;
|
||
}
|
||
catch (Exception)
|
||
{
|
||
return null;
|
||
}
|
||
}
|
||
|
||
public void RemoveProvider(string key)
|
||
{
|
||
WebHelper.RemoveCookie(key);
|
||
WebHelper.RemoveSession(key);
|
||
}
|
||
|
||
public OperatorModel GetCurrent()
|
||
{
|
||
OperatorModel operatorModel = new OperatorModel();
|
||
try
|
||
{
|
||
string loginMark = GetProvider(LoginUserMarkKey);
|
||
operatorModel = CacheHelper.Get<OperatorModel>(cacheKeyOperator + loginMark);
|
||
}
|
||
catch
|
||
{
|
||
operatorModel = null;
|
||
}
|
||
return operatorModel;
|
||
}
|
||
|
||
/// <summary>
|
||
/// 获取浏览器设配号
|
||
/// </summary>
|
||
/// <returns></returns>
|
||
public string GetMark()
|
||
{
|
||
string cookieMark = GetProvider(LoginUserMarkKey);
|
||
if (string.IsNullOrEmpty(cookieMark))
|
||
{
|
||
cookieMark = Guid.NewGuid().ToString();
|
||
SetProvider(LoginUserMarkKey, cookieMark);
|
||
}
|
||
return cookieMark;
|
||
}
|
||
|
||
/// <summary>
|
||
/// 登录者信息添加到缓存中
|
||
/// </summary>
|
||
/// <param name="userEntity">用户</param>
|
||
/// <param name="loginMark">设备标识uid</param>
|
||
/// <param name="facilityMark">设备类型</param>
|
||
/// <param name="cookie">是否保存cookie，默认是</param>
|
||
/// <returns></returns>
|
||
public async Task<string> AddLoginUser(OperatorModel operatorModel, string loginMark, string facilityMark)
|
||
{
|
||
string token = Guid.NewGuid().ToString();
|
||
try
|
||
{
|
||
// 填写登录信息
|
||
operatorModel.LoginToken = token;
|
||
//登录信息更新
|
||
if (string.IsNullOrEmpty(loginMark))
|
||
{
|
||
string cookieMark = GetProvider(LoginUserMarkKey);
|
||
if (string.IsNullOrEmpty(cookieMark))
|
||
{
|
||
operatorModel.loginMark = Guid.NewGuid().ToString();
|
||
SetProvider(LoginUserMarkKey, operatorModel.loginMark);
|
||
}
|
||
else
|
||
{
|
||
operatorModel.loginMark = cookieMark;
|
||
}
|
||
SetProvider(LoginUserToken, token);
|
||
}
|
||
else
|
||
{
|
||
operatorModel.loginMark = loginMark;
|
||
RemoveProvider(LoginUserMarkKey);
|
||
}
|
||
//redis 登录token列表更新
|
||
Dictionary<string, string> tokenMarkList = await CacheHelper.GetAsync<Dictionary<string, string>>(cacheKeyToken + operatorModel.UserId);
|
||
if (tokenMarkList == null)// 此账号第一次登录
|
||
{
|
||
tokenMarkList = new Dictionary<string, string>();
|
||
tokenMarkList.Add(operatorModel.loginMark, token);
|
||
}
|
||
else
|
||
{
|
||
if (tokenMarkList.ContainsKey(operatorModel.loginMark))
|
||
{
|
||
tokenMarkList[operatorModel.loginMark] = token;
|
||
}
|
||
else
|
||
{
|
||
tokenMarkList.Add(operatorModel.loginMark, token);
|
||
}
|
||
}
|
||
|
||
await CacheHelper.SetAsync(cacheKeyToken + operatorModel.UserId, tokenMarkList);
|
||
await CacheHelper.SetAsync(cacheKeyOperator + operatorModel.loginMark, operatorModel, LoginExpire);
|
||
await CacheHelper.RemoveAsync(cacheKeyOperator + facilityMark + operatorModel.UserId);
|
||
await CacheHelper.SetAsync(cacheKeyOperator + facilityMark + operatorModel.UserId, token, LoginExpire);
|
||
return token;
|
||
}
|
||
catch (Exception)
|
||
{
|
||
throw;
|
||
}
|
||
}
|
||
|
||
/// <summary>
|
||
/// 清空当前登录信息
|
||
/// </summary>
|
||
/// <param name="apitoken">apitoken</param>
|
||
/// <param name="facilityMark">设备类型</param>
|
||
public async Task EmptyCurrent(string facilityMark)
|
||
{
|
||
try
|
||
{
|
||
string token = GetProvider(LoginUserToken);
|
||
string loginMark = GetProvider(LoginUserMarkKey);
|
||
await EmptyCurrent(token, facilityMark, loginMark);
|
||
RemoveProvider(LoginUserMarkKey.Trim());
|
||
RemoveProvider(LoginUserToken.Trim());
|
||
}
|
||
catch (Exception)
|
||
{
|
||
}
|
||
}
|
||
|
||
/// <summary>
|
||
/// 清空当前登录信息
|
||
/// </summary>
|
||
/// <param name="token">登录票据</param>
|
||
/// <param name="facilityMark">登录设备</param>
|
||
/// <param name="loginMark">登录设备标识</param>
|
||
public async Task EmptyCurrent(string token, string facilityMark, string loginMark)
|
||
{
|
||
try
|
||
{
|
||
OperatorModel operatorInfo = await CacheHelper.GetAsync<OperatorModel>(cacheKeyOperator + loginMark);
|
||
if (operatorInfo != null)
|
||
{
|
||
Dictionary<string, string> tokenMarkList = await CacheHelper.GetAsync<Dictionary<string, string>>(cacheKeyToken + operatorInfo.UserId);
|
||
tokenMarkList.Remove(loginMark);
|
||
await CacheHelper.RemoveAsync(cacheKeyOperator + loginMark);
|
||
if (operatorInfo.LoginToken == token || facilityMark == "api_")
|
||
{
|
||
await CacheHelper.RemoveAsync(cacheKeyOperator + facilityMark + operatorInfo.UserId);
|
||
}
|
||
await CacheHelper.SetAsync(cacheKeyToken + operatorInfo.UserId, tokenMarkList);
|
||
await CacheHelper.RemoveAsync(facilityMark + GlobalContext.SystemConfig.TokenName + "_" + operatorInfo.UserId + "_" + operatorInfo.LoginTime);
|
||
}
|
||
}
|
||
catch (Exception)
|
||
{
|
||
}
|
||
}
|
||
|
||
/// <summary>
|
||
/// 判断登录状态
|
||
/// </summary>
|
||
/// <param name="facilityMark">登录设备</param>
|
||
/// <param name="apitoken">apitoken</param>
|
||
/// <returns>-1未登录,1登录成功,0登录过期,-2账号被顶</returns>
|
||
public async Task<OperatorResult> IsOnLine(string facilityMark)
|
||
{
|
||
try
|
||
{
|
||
string token = GetProvider(LoginUserToken);
|
||
string loginMark = GetProvider(LoginUserMarkKey);
|
||
return await IsOnLine(token, facilityMark, loginMark);
|
||
}
|
||
catch (Exception)
|
||
{
|
||
return new OperatorResult { stateCode = -1 };
|
||
}
|
||
}
|
||
|
||
/// <summary>
|
||
/// 判断登录状态
|
||
/// </summary>
|
||
/// <param name="token">登录票据</param>
|
||
/// <param name="facilityMark">登录设备</param>
|
||
/// <param name="loginMark">登录设备标识</param>
|
||
/// <returns>-1未登录,1登录成功,0登录过期,-2账号被顶</returns>
|
||
public async Task<OperatorResult> IsOnLine(string token, string facilityMark, string loginMark)
|
||
{
|
||
OperatorResult operatorResult = new OperatorResult();
|
||
operatorResult.stateCode = -1; // -1未登录,1登录成功,0登录过期
|
||
try
|
||
{
|
||
if (string.IsNullOrEmpty(token) || string.IsNullOrEmpty(loginMark))
|
||
{
|
||
return operatorResult;
|
||
}
|
||
OperatorModel operatorInfo = await CacheHelper.GetAsync<OperatorModel>(cacheKeyOperator + loginMark);
|
||
if (operatorInfo != null)
|
||
{
|
||
Dictionary<string, string> tokenMarkList = await CacheHelper.GetAsync<Dictionary<string, string>>(cacheKeyToken + operatorInfo.UserId);
|
||
if ((token == operatorInfo.LoginToken || facilityMark == "api_") && tokenMarkList.ContainsKey(operatorInfo.loginMark) && tokenMarkList[operatorInfo.loginMark] == operatorInfo.LoginToken)
|
||
{
|
||
////账号被顶(排除admin)
|
||
if (!LoginMultiple && !operatorInfo.IsSuperAdmin && operatorInfo.LoginToken != await CacheHelper.GetAsync<string>(cacheKeyOperator + facilityMark + operatorInfo.UserId))
|
||
{
|
||
operatorResult.stateCode = -2;
|
||
tokenMarkList = await CacheHelper.GetAsync<Dictionary<string, string>>(cacheKeyToken + operatorInfo.UserId);
|
||
tokenMarkList.Remove(loginMark);
|
||
await CacheHelper.SetAsync(cacheKeyToken + operatorInfo.UserId, tokenMarkList);
|
||
await CacheHelper.RemoveAsync(cacheKeyOperator + loginMark);
|
||
}
|
||
else
|
||
{
|
||
operatorResult.userInfo = operatorInfo;
|
||
operatorResult.stateCode = 1;
|
||
await CacheHelper.ExpireAsync(cacheKeyOperator + loginMark, LoginExpire);
|
||
await CacheHelper.ExpireAsync(cacheKeyOperator + facilityMark + operatorInfo.UserId, LoginExpire);
|
||
await CacheHelper.ExpireAsync(facilityMark + GlobalContext.SystemConfig.TokenName + "_" + operatorInfo.UserId + "_" + operatorInfo.LoginTime, LoginExpire);
|
||
}
|
||
}
|
||
}
|
||
return operatorResult;
|
||
}
|
||
catch (Exception)
|
||
{
|
||
return operatorResult;
|
||
}
|
||
}
|
||
|
||
#region 登录错误次数记录
|
||
|
||
/// <summary>
|
||
/// 获取当前登录错误次数
|
||
/// </summary>
|
||
/// <returns></returns>
|
||
public async Task<int> GetCurrentErrorNum()
|
||
{
|
||
int res = 0;
|
||
try
|
||
{
|
||
string cookieMark = GetProvider(LoginUserMarkKey);
|
||
if (string.IsNullOrEmpty(cookieMark))
|
||
{
|
||
cookieMark = Guid.NewGuid().ToString();
|
||
SetProvider(LoginUserMarkKey, cookieMark);
|
||
}
|
||
string num = await CacheHelper.GetAsync<string>(cacheKeyError + cookieMark);
|
||
if (!string.IsNullOrEmpty(num))
|
||
{
|
||
res = Convert.ToInt32(num);
|
||
}
|
||
}
|
||
catch (Exception)
|
||
{
|
||
}
|
||
return res;
|
||
}
|
||
|
||
/// <summary>
|
||
/// 增加错误次数
|
||
/// </summary>
|
||
/// <returns></returns>
|
||
public async Task<int> AddCurrentErrorNum()
|
||
{
|
||
int res = 0;
|
||
try
|
||
{
|
||
string cookieMark = GetProvider(LoginUserMarkKey);
|
||
if (string.IsNullOrEmpty(cookieMark))
|
||
{
|
||
cookieMark = Guid.NewGuid().ToString();
|
||
SetProvider(LoginUserMarkKey, cookieMark);
|
||
}
|
||
string num = await CacheHelper.GetAsync<string>(cacheKeyError + cookieMark);
|
||
if (!string.IsNullOrEmpty(num))
|
||
{
|
||
res = Convert.ToInt32(num);
|
||
}
|
||
res++;
|
||
num = res + "";
|
||
await CacheHelper.SetAsync(cacheKeyError + cookieMark, num, 24);
|
||
}
|
||
catch (Exception)
|
||
{
|
||
}
|
||
return res;
|
||
}
|
||
|
||
/// <summary>
|
||
/// 清除当前登录错误次数
|
||
/// </summary>
|
||
public async Task ClearCurrentErrorNum()
|
||
{
|
||
try
|
||
{
|
||
string cookieMark = GetProvider(LoginUserMarkKey);
|
||
if (string.IsNullOrEmpty(cookieMark))
|
||
{
|
||
cookieMark = Guid.NewGuid().ToString();
|
||
SetProvider(LoginUserMarkKey, cookieMark);
|
||
}
|
||
await CacheHelper.RemoveAsync(cacheKeyError + cookieMark);
|
||
}
|
||
catch (Exception)
|
||
{
|
||
}
|
||
}
|
||
|
||
#endregion 登录错误次数记录
|
||
}
|
||
} |