|
|
package SsoUtil
|
|
|
|
|
|
import (
|
|
|
"dsBigData/Utils/ConfigUtil"
|
|
|
"dsBigData/Utils/RedisUtil"
|
|
|
"encoding/base64"
|
|
|
"github.com/gin-gonic/gin"
|
|
|
"github.com/gomodule/redigo/redis"
|
|
|
"github.com/sirupsen/logrus"
|
|
|
"strconv"
|
|
|
"strings"
|
|
|
)
|
|
|
|
|
|
/**
|
|
|
功能：检查是不是合法用户
|
|
|
作者：黄海
|
|
|
时间：2020-02-22
|
|
|
*/
|
|
|
func IsAuthority(c *gin.Context) (bool, string) {
|
|
|
//检查Header头中是不是存在ds_sso_sessionid,就是说，如果客户端不支持Cookie(比如手机)，可以在Header中构建ds_sso_sessionid
|
|
|
sessionId := ""
|
|
|
//从 Header中获取会话ID
|
|
|
sessionId = c.Request.Header.Get(ConfigUtil.SessionId)
|
|
|
//如果没有找到，就到 cookie里继续寻找
|
|
|
if len(sessionId) == 0 {
|
|
|
cookie, e := c.Request.Cookie(ConfigUtil.SessionId)
|
|
|
if e == nil { //如果成功获取到的话
|
|
|
sessionId = cookie.Value
|
|
|
}
|
|
|
}
|
|
|
//如果两个位置都没有找到
|
|
|
if len(sessionId) == 0 {
|
|
|
return false, ""
|
|
|
} else { //有这个值，判断一下是不是正确
|
|
|
_map := RedisUtil.HGETALL(ConfigUtil.SessionId + "_" + sessionId)
|
|
|
if len(_map) == 0 {
|
|
|
//清除非法cookie，资料参考：https://www.tizi365.com/archives/273.html
|
|
|
c.SetCookie(ConfigUtil.SessionId, "", -1, "/", "", false, true)
|
|
|
return false, sessionId
|
|
|
} else {
|
|
|
return true, sessionId
|
|
|
}
|
|
|
}
|
|
|
}
|
|
|
|
|
|
/**
|
|
|
功能：判断当前登录人员是不是处于登录状态
|
|
|
作者：黄海
|
|
|
时间：2020-02-25
|
|
|
*/
|
|
|
func IsLoginStatus(username string) (bool, string) {
|
|
|
var value = ""
|
|
|
var r = true
|
|
|
value, err := redis.String(RedisUtil.GET(ConfigUtil.SearchSessionIdPrefix + username))
|
|
|
if err != nil {
|
|
|
r = false
|
|
|
}
|
|
|
return r, value
|
|
|
}
|
|
|
|
|
|
/**
|
|
|
功能：写入统一认证的信息
|
|
|
作者：黄海
|
|
|
时间：2020-02-22
|
|
|
*/
|
|
|
func WriteSsoInfo(c *gin.Context, device string, identityId int64, personId int64, username string, personName string, dsSsoSessionid string) {
|
|
|
//下面应该是双向的redis存储，即：（1）通过登录名能找到对应的会话ID，（2）通过会话ID，可以解析出是哪种设备，哪个身份，哪个人员ID
|
|
|
|
|
|
//一、写入登录名与会话的对应关系,由四部分组成，前缀+设备类型+身份+人员编号,通过这个键值可以查找到对应的会话
|
|
|
SearchSsoKey := ConfigUtil.SearchSessionIdPrefix + "_" + device + "_" + strconv.FormatInt(identityId, 10) + "_" + strconv.FormatInt(personId, 10)
|
|
|
//写入redis
|
|
|
RedisUtil.SET(SearchSsoKey, ConfigUtil.SessionId+"_"+dsSsoSessionid)
|
|
|
//加上过期时间
|
|
|
RedisUtil.EXPIRE(SearchSsoKey)
|
|
|
|
|
|
//二、写入SESSION_ID
|
|
|
_map := make(map[string]interface{})
|
|
|
_map["device"] = device
|
|
|
_map["identity_id"] = identityId
|
|
|
_map["person_id"] = personId
|
|
|
_map["username"] = username
|
|
|
_map["person_name"] = personName
|
|
|
RedisUtil.HMSET(ConfigUtil.SessionId+"_"+dsSsoSessionid, _map)
|
|
|
//加上过期时间
|
|
|
RedisUtil.EXPIRE(ConfigUtil.SessionId + "_" + dsSsoSessionid)
|
|
|
|
|
|
//三、写入主域的COOKIE值
|
|
|
c.SetCookie(ConfigUtil.SessionId, dsSsoSessionid, 0, "/", "", false, true)
|
|
|
}
|
|
|
|
|
|
/**
|
|
|
功能: 清理统一认证的登录信息，包括清理cookie+redis缓存
|
|
|
作者：黄海
|
|
|
时间：2020-02-22
|
|
|
*/
|
|
|
func ClearSsoInfo(sessionId string, c *gin.Context) string {
|
|
|
//一、删除人员与 sessionId的映射关系
|
|
|
DeletePersonSessionReleation(sessionId)
|
|
|
//二、删除redis中的会话id
|
|
|
DeleteSessionId(sessionId)
|
|
|
//三、清除cookie
|
|
|
deleteSsoCookie(c)
|
|
|
return sessionId
|
|
|
}
|
|
|
|
|
|
/**
|
|
|
功能：删除人员与会话的映射关系
|
|
|
作者：黄海
|
|
|
时间：2020-02-25
|
|
|
*/
|
|
|
func DeletePersonSessionReleation(sessionId string) {
|
|
|
//一、删除人员与 sessionId的映射关系
|
|
|
var identityId = -1
|
|
|
var personId = -1
|
|
|
var deviceId = -1
|
|
|
deviceId, identityId, personId = AnalyzeSessionId(sessionId) //通过会话ID获取到人员的信息
|
|
|
//删除人员与会话的对应关系
|
|
|
SearchSsoKey := ConfigUtil.SearchSessionIdPrefix + "_" + strconv.Itoa(deviceId) + "_" + strconv.Itoa(identityId) + "_" + strconv.Itoa(personId)
|
|
|
RedisUtil.DEL(SearchSsoKey)
|
|
|
}
|
|
|
|
|
|
/**
|
|
|
功能：解析SessionId的内容，换算出设备，身份和人员ID
|
|
|
作者：黄海
|
|
|
时间：2020-02-25
|
|
|
*/
|
|
|
func AnalyzeSessionId(sessionId string) (int, int, int) {
|
|
|
var identityId = -1
|
|
|
var personId = -1
|
|
|
var deviceId = -1
|
|
|
|
|
|
r, err := base64.RawURLEncoding.DecodeString(sessionId)
|
|
|
if err != nil {
|
|
|
logrus.Errorln(err)
|
|
|
} else {
|
|
|
s := string(r)
|
|
|
array := strings.Split(s, "_")
|
|
|
if len(array) > 1 {
|
|
|
deviceId, _ = strconv.Atoi(array[0])
|
|
|
identityId, _ = strconv.Atoi(array[1])
|
|
|
personId, _ = strconv.Atoi(array[2])
|
|
|
}
|
|
|
}
|
|
|
return deviceId, identityId, personId
|
|
|
}
|
|
|
|
|
|
/**
|
|
|
功能：删除redis中的会话id
|
|
|
作者：黄海
|
|
|
时间：2020-02-25
|
|
|
*/
|
|
|
func DeleteSessionId(sessionId string) {
|
|
|
//删除redis中的记录
|
|
|
RedisUtil.DEL(ConfigUtil.SessionId + "_" + sessionId)
|
|
|
}
|
|
|
|
|
|
/**
|
|
|
功能：读取统一认证的cookie值
|
|
|
作者：黄海
|
|
|
时间：2020-02-22
|
|
|
*/
|
|
|
func ReadSsoCookie(c *gin.Context) string {
|
|
|
cookie, e := c.Request.Cookie(ConfigUtil.SessionId)
|
|
|
if e == nil {
|
|
|
return cookie.Value
|
|
|
} else {
|
|
|
return ""
|
|
|
}
|
|
|
}
|
|
|
|
|
|
/**
|
|
|
功能：删除统一认证的cookie信息
|
|
|
作者：黄海
|
|
|
时间：2020-02-22
|
|
|
*/
|
|
|
func deleteSsoCookie(c *gin.Context) {
|
|
|
c.SetCookie(ConfigUtil.SessionId, "", -1, "/", "", false, true)
|
|
|
}
|
