Parker
4 years ago
parent
e98b06e7fe
commit
60fa1c3bfc
@ -0,0 +1,76 @@
|
|||||||
|
/**
|
||||||
|
* Copyright 2020 OPSLI 快速开发平台 https://www.opsli.com
|
||||||
|
* <p>
|
||||||
|
* Licensed under the Apache License, Version 2.0 (the "License"); you may not
|
||||||
|
* use this file except in compliance with the License. You may obtain a copy of
|
||||||
|
* the License at
|
||||||
|
* <p>
|
||||||
|
* http://www.apache.org/licenses/LICENSE-2.0
|
||||||
|
* <p>
|
||||||
|
* Unless required by applicable law or agreed to in writing, software
|
||||||
|
* distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
|
||||||
|
* WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
|
||||||
|
* License for the specific language governing permissions and limitations under
|
||||||
|
* the License.
|
||||||
|
*/
|
||||||
|
package org.opsli.core.cache.pushsub.handler;
|
||||||
|
|
||||||
|
import com.alibaba.fastjson.JSONObject;
|
||||||
|
import lombok.extern.slf4j.Slf4j;
|
||||||
|
import org.apache.commons.lang3.StringUtils;
|
||||||
|
import org.opsli.common.constants.CacheConstants;
|
||||||
|
import org.opsli.core.cache.local.CacheUtil;
|
||||||
|
import org.opsli.core.cache.pushsub.enums.MsgArgsType;
|
||||||
|
import org.opsli.core.cache.pushsub.enums.PushSubType;
|
||||||
|
import org.opsli.core.utils.MenuUtil;
|
||||||
|
import org.opsli.core.utils.OrgUtil;
|
||||||
|
import org.opsli.plugins.cache.EhCachePlugin;
|
||||||
|
import org.springframework.beans.factory.annotation.Autowired;
|
||||||
|
|
||||||
|
/**
|
||||||
|
* @BelongsProject: opsli-boot
|
||||||
|
* @BelongsPackage: org.opsli.core.cache.pushsub.handler
|
||||||
|
* @Author: Parker
|
||||||
|
* @CreateTime: 2020-09-15 16:24
|
||||||
|
* @Description: 用户组织消息处理
|
||||||
|
*/
|
||||||
|
@Slf4j
|
||||||
|
public class OrgHandler implements RedisPushSubHandler{
|
||||||
|
|
||||||
|
@Autowired
|
||||||
|
EhCachePlugin ehCachePlugin;
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public PushSubType getType() {
|
||||||
|
return PushSubType.ORG;
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public void handler(JSONObject msgJson) {
|
||||||
|
// 用户刷新
|
||||||
|
this.orgHandler(msgJson);
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 用户组织数据处理
|
||||||
|
* @param msgJson
|
||||||
|
*/
|
||||||
|
private void orgHandler(JSONObject msgJson){
|
||||||
|
JSONObject data = msgJson.getJSONObject(MsgArgsType.ORG_USER_DATA.toString());
|
||||||
|
// 数据为空则不执行
|
||||||
|
if(data == null) return;
|
||||||
|
|
||||||
|
// 获得用户ID
|
||||||
|
String userId = (String) msgJson.get(MsgArgsType.ORG_USER_ID.toString());
|
||||||
|
if(StringUtils.isEmpty(userId)){
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
// 先删除
|
||||||
|
ehCachePlugin.delete(CacheConstants.HOT_DATA, OrgUtil.PREFIX_CODE + userId);
|
||||||
|
// 清除空拦截
|
||||||
|
CacheUtil.delNilFlag(OrgUtil.PREFIX_CODE + userId);
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
}
|
||||||
@ -0,0 +1,61 @@
|
|||||||
|
/**
|
||||||
|
* Copyright 2020 OPSLI 快速开发平台 https://www.opsli.com
|
||||||
|
* <p>
|
||||||
|
* Licensed under the Apache License, Version 2.0 (the "License"); you may not
|
||||||
|
* use this file except in compliance with the License. You may obtain a copy of
|
||||||
|
* the License at
|
||||||
|
* <p>
|
||||||
|
* http://www.apache.org/licenses/LICENSE-2.0
|
||||||
|
* <p>
|
||||||
|
* Unless required by applicable law or agreed to in writing, software
|
||||||
|
* distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
|
||||||
|
* WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
|
||||||
|
* License for the specific language governing permissions and limitations under
|
||||||
|
* the License.
|
||||||
|
*/
|
||||||
|
package org.opsli.core.cache.pushsub.msgs;
|
||||||
|
|
||||||
|
import com.alibaba.fastjson.JSONObject;
|
||||||
|
import lombok.Data;
|
||||||
|
import lombok.experimental.Accessors;
|
||||||
|
import org.opsli.api.wrapper.system.menu.MenuModel;
|
||||||
|
import org.opsli.api.wrapper.system.user.UserOrgRefModel;
|
||||||
|
import org.opsli.core.cache.pushsub.enums.MsgArgsType;
|
||||||
|
import org.opsli.core.cache.pushsub.enums.PushSubType;
|
||||||
|
import org.opsli.core.cache.pushsub.receiver.RedisPushSubReceiver;
|
||||||
|
import org.opsli.plugins.redis.pushsub.entity.BaseSubMessage;
|
||||||
|
|
||||||
|
/**
|
||||||
|
* @BelongsProject: opsli-boot
|
||||||
|
* @BelongsPackage: org.opsli.core.cache.pushsub.msgs
|
||||||
|
* @Author: Parker
|
||||||
|
* @CreateTime: 2020-09-15 16:50
|
||||||
|
* @Description: 用户组织消息
|
||||||
|
*/
|
||||||
|
|
||||||
|
@Data
|
||||||
|
@Accessors(chain = true)
|
||||||
|
public final class OrgMsgFactory extends BaseSubMessage{
|
||||||
|
|
||||||
|
/** 通道 */
|
||||||
|
private static final String CHANNEL = RedisPushSubReceiver.BASE_CHANNEL+RedisPushSubReceiver.CHANNEL;
|
||||||
|
|
||||||
|
private OrgMsgFactory(){}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 构建消息 - 组织
|
||||||
|
*/
|
||||||
|
public static BaseSubMessage createOrgMsg(UserOrgRefModel orgRefModel){
|
||||||
|
BaseSubMessage baseSubMessage = new BaseSubMessage();
|
||||||
|
// 数据
|
||||||
|
JSONObject jsonObj = new JSONObject();
|
||||||
|
jsonObj.put(MsgArgsType.ORG_USER_ID.toString(), orgRefModel.getUserId());
|
||||||
|
jsonObj.put(MsgArgsType.ORG_USER_DATA.toString(), orgRefModel);
|
||||||
|
|
||||||
|
// 用户
|
||||||
|
baseSubMessage.build(CHANNEL,PushSubType.ORG.toString(),jsonObj);
|
||||||
|
return baseSubMessage;
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
}
|
||||||
@ -0,0 +1,183 @@
|
|||||||
|
/**
|
||||||
|
* Copyright 2020 OPSLI 快速开发平台 https://www.opsli.com
|
||||||
|
* <p>
|
||||||
|
* Licensed under the Apache License, Version 2.0 (the "License"); you may not
|
||||||
|
* use this file except in compliance with the License. You may obtain a copy of
|
||||||
|
* the License at
|
||||||
|
* <p>
|
||||||
|
* http://www.apache.org/licenses/LICENSE-2.0
|
||||||
|
* <p>
|
||||||
|
* Unless required by applicable law or agreed to in writing, software
|
||||||
|
* distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
|
||||||
|
* WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
|
||||||
|
* License for the specific language governing permissions and limitations under
|
||||||
|
* the License.
|
||||||
|
*/
|
||||||
|
package org.opsli.core.utils;
|
||||||
|
|
||||||
|
import lombok.extern.slf4j.Slf4j;
|
||||||
|
import org.apache.commons.lang3.StringUtils;
|
||||||
|
import org.opsli.api.base.result.ResultVo;
|
||||||
|
import org.opsli.api.web.system.menu.MenuApi;
|
||||||
|
import org.opsli.api.web.system.user.UserApi;
|
||||||
|
import org.opsli.api.wrapper.system.menu.MenuModel;
|
||||||
|
import org.opsli.api.wrapper.system.user.UserOrgRefModel;
|
||||||
|
import org.opsli.core.cache.local.CacheUtil;
|
||||||
|
import org.opsli.core.cache.pushsub.msgs.MenuMsgFactory;
|
||||||
|
import org.opsli.core.cache.pushsub.msgs.OrgMsgFactory;
|
||||||
|
import org.opsli.plugins.redis.RedisLockPlugins;
|
||||||
|
import org.opsli.plugins.redis.RedisPlugin;
|
||||||
|
import org.opsli.plugins.redis.lock.RedisLock;
|
||||||
|
import org.springframework.beans.factory.annotation.Autowired;
|
||||||
|
import org.springframework.boot.autoconfigure.AutoConfigureAfter;
|
||||||
|
import org.springframework.context.annotation.Lazy;
|
||||||
|
import org.springframework.core.annotation.Order;
|
||||||
|
import org.springframework.stereotype.Component;
|
||||||
|
|
||||||
|
import static org.opsli.common.constants.OrderConstants.UTIL_ORDER;
|
||||||
|
|
||||||
|
/**
|
||||||
|
* @BelongsProject: opsli-boot
|
||||||
|
* @BelongsPackage: org.opsli.core.utils
|
||||||
|
* @Author: Parker
|
||||||
|
* @CreateTime: 2020-09-19 20:03
|
||||||
|
* @Description: 组织机构工具类
|
||||||
|
*/
|
||||||
|
@Slf4j
|
||||||
|
@Order(UTIL_ORDER)
|
||||||
|
@Component
|
||||||
|
@AutoConfigureAfter({RedisPlugin.class , RedisLockPlugins.class, UserApi.class})
|
||||||
|
@Lazy(false)
|
||||||
|
public class OrgUtil {
|
||||||
|
|
||||||
|
/** 前缀 */
|
||||||
|
public static final String PREFIX_CODE = "org:userId:";
|
||||||
|
|
||||||
|
|
||||||
|
/** Redis插件 */
|
||||||
|
private static RedisPlugin redisPlugin;
|
||||||
|
|
||||||
|
/** Redis分布式锁 */
|
||||||
|
private static RedisLockPlugins redisLockPlugins;
|
||||||
|
|
||||||
|
/** 用户 Api */
|
||||||
|
private static UserApi userApi;
|
||||||
|
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 根据 userId 获得用户组织
|
||||||
|
* @param userId
|
||||||
|
* @return
|
||||||
|
*/
|
||||||
|
public static UserOrgRefModel getOrgByUserId(String userId){
|
||||||
|
// 先从缓存里拿
|
||||||
|
UserOrgRefModel orgRefModel = CacheUtil.get(PREFIX_CODE + userId, UserOrgRefModel.class);
|
||||||
|
if (orgRefModel != null){
|
||||||
|
return orgRefModel;
|
||||||
|
}
|
||||||
|
|
||||||
|
// 拿不到 --------
|
||||||
|
// 防止缓存穿透判断
|
||||||
|
boolean hasNilFlag = CacheUtil.hasNilFlag(PREFIX_CODE + userId);
|
||||||
|
if(hasNilFlag){
|
||||||
|
return null;
|
||||||
|
}
|
||||||
|
|
||||||
|
// 锁凭证 redisLock 贯穿全程
|
||||||
|
RedisLock redisLock = new RedisLock();
|
||||||
|
redisLock.setLockName(PREFIX_CODE + userId)
|
||||||
|
.setAcquireTimeOut(3000L)
|
||||||
|
.setLockTimeOut(5000L);
|
||||||
|
|
||||||
|
try {
|
||||||
|
// 这里增加分布式锁 防止缓存击穿
|
||||||
|
// ============ 尝试加锁
|
||||||
|
redisLock = redisLockPlugins.tryLock(redisLock);
|
||||||
|
if(redisLock == null){
|
||||||
|
return null;
|
||||||
|
}
|
||||||
|
|
||||||
|
// 如果获得锁 则 再次检查缓存里有没有， 如果有则直接退出， 没有的话才发起数据库请求
|
||||||
|
orgRefModel = CacheUtil.get(PREFIX_CODE + userId, UserOrgRefModel.class);
|
||||||
|
if (orgRefModel != null){
|
||||||
|
return orgRefModel;
|
||||||
|
}
|
||||||
|
|
||||||
|
// 查询数据库
|
||||||
|
ResultVo<UserOrgRefModel> resultVo = userApi.getOrgInfoByUserId(userId);
|
||||||
|
if(resultVo.isSuccess()){
|
||||||
|
orgRefModel = resultVo.getData();
|
||||||
|
// 存入缓存
|
||||||
|
CacheUtil.put(PREFIX_CODE + userId, orgRefModel);
|
||||||
|
}
|
||||||
|
}catch (Exception e){
|
||||||
|
log.error(e.getMessage(),e);
|
||||||
|
}finally {
|
||||||
|
// ============ 释放锁
|
||||||
|
redisLockPlugins.unLock(redisLock);
|
||||||
|
redisLock = null;
|
||||||
|
}
|
||||||
|
|
||||||
|
if(orgRefModel == null){
|
||||||
|
// 设置空变量 用于防止穿透判断
|
||||||
|
CacheUtil.putNilFlag(PREFIX_CODE + userId);
|
||||||
|
return null;
|
||||||
|
}
|
||||||
|
|
||||||
|
return orgRefModel;
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
// ============== 刷新缓存 ==============
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 刷新用户组织 - 删就完了
|
||||||
|
* @param userId
|
||||||
|
* @return
|
||||||
|
*/
|
||||||
|
public static void refreshMenu(String userId){
|
||||||
|
if(StringUtils.isEmpty(userId)){
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
UserOrgRefModel orgRefModel = CacheUtil.get(PREFIX_CODE + userId, UserOrgRefModel.class);
|
||||||
|
boolean hasNilFlag = CacheUtil.hasNilFlag(PREFIX_CODE + userId);
|
||||||
|
|
||||||
|
// 只要不为空 则执行刷新
|
||||||
|
if (hasNilFlag){
|
||||||
|
// 清除空拦截
|
||||||
|
CacheUtil.delNilFlag(PREFIX_CODE + userId);
|
||||||
|
}
|
||||||
|
|
||||||
|
if(orgRefModel != null){
|
||||||
|
// 先删除
|
||||||
|
CacheUtil.del(PREFIX_CODE + userId);
|
||||||
|
|
||||||
|
// 发送通知消息
|
||||||
|
redisPlugin.sendMessage(
|
||||||
|
OrgMsgFactory.createOrgMsg(orgRefModel)
|
||||||
|
);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
// =====================================
|
||||||
|
|
||||||
|
@Autowired
|
||||||
|
public void setRedisPlugin(RedisPlugin redisPlugin) {
|
||||||
|
OrgUtil.redisPlugin = redisPlugin;
|
||||||
|
}
|
||||||
|
|
||||||
|
@Autowired
|
||||||
|
public void setRedisLockPlugins(RedisLockPlugins redisLockPlugins) {
|
||||||
|
OrgUtil.redisLockPlugins = redisLockPlugins;
|
||||||
|
}
|
||||||
|
|
||||||
|
@Autowired
|
||||||
|
public void setUserApi(UserApi userApi) {
|
||||||
|
OrgUtil.userApi = userApi;
|
||||||
|
}
|
||||||
|
|
||||||
|
}
|
||||||
Loading…
Reference in new issue