115 lines
3.3 KiB
Go
115 lines
3.3 KiB
Go
package managecenterMgr
|
||
|
||
import (
|
||
"encoding/json"
|
||
"errors"
|
||
"fmt"
|
||
|
||
"Framework/managecenterModel"
|
||
. "Framework/managecenterModel"
|
||
"goutil/logUtil"
|
||
"goutil/webUtil"
|
||
)
|
||
|
||
var (
|
||
whiteListMap = make(map[int32]map[string]*WhiteList, 64)
|
||
whiteListHash string
|
||
)
|
||
|
||
// 重新加载白名单
|
||
func reloadWhiteList(isInit bool) error {
|
||
//logUtil.DebugLog("开始刷新白名单列表")
|
||
|
||
url := getManageCenterUrl("/API/UserWhiteList.ashx")
|
||
|
||
// 定义请求参数
|
||
postDict := make(map[string]string)
|
||
postDict["HashValue"] = whiteListHash
|
||
|
||
//请求url,请求头
|
||
header := webUtil.GetFormHeader()
|
||
transport := webUtil.NewTransport()
|
||
transport.DisableKeepAlives = true
|
||
transport = webUtil.GetTimeoutTransport(transport, 30)
|
||
|
||
statusCode, returnBytes, err := webUtil.PostMapData(url, postDict, header, transport)
|
||
//statusCode, returnBytes, err := webUtil.PostMapData(url, postDict, header, nil)
|
||
if err != nil {
|
||
logUtil.ErrorLog(fmt.Sprintf("获取白名单列表出错，url:%s,错误信息为:%s", url, err))
|
||
return err
|
||
}
|
||
if statusCode != 200 {
|
||
logUtil.ErrorLog(fmt.Sprintf("获取白名单列表出错，url:%s,错误码为:%d", url, statusCode))
|
||
return err
|
||
}
|
||
|
||
// 解析返回值
|
||
returnObj := new(ReturnObject)
|
||
if err = json.Unmarshal(returnBytes, &returnObj); err != nil {
|
||
logUtil.ErrorLog(fmt.Sprintf("获取白名单列表出错，反序列化返回值出错，错误信息为：%s, str:%s", err, string(returnBytes)))
|
||
return err
|
||
}
|
||
|
||
// 判断返回状态是否为成功
|
||
if returnObj.Code != 0 {
|
||
// 数据没有变化，所以没有获取到新的数据，不能算错误。
|
||
if returnObj.Code == 47 || returnObj.Message == "DataNotChanged" {
|
||
//如果本地集合为空，且数据又没变化时，重新初始化一下本地hash值
|
||
if len(whiteListMap) == 0 {
|
||
whiteListHash = ""
|
||
}
|
||
return nil
|
||
} else {
|
||
msg := fmt.Sprintf("获取白名单列表出错，返回状态：%d，信息为：%s", returnObj.Code, returnObj.Message)
|
||
logUtil.ErrorLog(msg)
|
||
return errors.New(msg)
|
||
}
|
||
}
|
||
|
||
// 解析Data
|
||
tmpWhiteList := make([]*WhiteList, 0, 128)
|
||
if data, ok := returnObj.Data.(string); !ok {
|
||
msg := "获取白名单列表出错，返回的数据不是string类型"
|
||
logUtil.ErrorLog(msg)
|
||
return errors.New(msg)
|
||
} else {
|
||
if err = json.Unmarshal([]byte(data), &tmpWhiteList); err != nil {
|
||
logUtil.ErrorLog(fmt.Sprintf("获取白名单列表出错，反序列化数据出错，错误信息为：%s", err))
|
||
return err
|
||
}
|
||
}
|
||
|
||
//logUtil.DebugLog(fmt.Sprintf("刷新白名单信息结束，白名单数量:%d", len(tmpWhiteList)))
|
||
|
||
tmpWhiteListMap := make(map[int32]map[string]*WhiteList, 64)
|
||
for _, item := range tmpWhiteList {
|
||
if _, exist := tmpWhiteListMap[item.PartnerId]; !exist {
|
||
tmpWhiteListMap[item.PartnerId] = make(map[string]*WhiteList, 8)
|
||
}
|
||
|
||
tmpWhiteListMap[item.PartnerId][item.UserId] = item
|
||
}
|
||
|
||
// 赋值给最终的whiteListMap
|
||
whiteListMap = tmpWhiteListMap
|
||
whiteListHash = returnObj.HashValue
|
||
|
||
//通知变更
|
||
mcDataChangeCallBack(managecenterModel.UserWhiteListData, isInit)
|
||
|
||
return nil
|
||
}
|
||
|
||
// 判断用户是否在白名单里面
|
||
// partnerId: 合作商ID
|
||
// userId: userId
|
||
func IsInWhiteList(partnerId int32, userId string) bool {
|
||
subWhiteListMap, exist := whiteListMap[partnerId]
|
||
if !exist {
|
||
return false
|
||
}
|
||
|
||
_, exist = subWhiteListMap[userId]
|
||
return exist
|
||
}
|