BaseController 缓存调整

opsli-base-support/opsli-core/src/main/java/org/opsli/core/base/controller/BaseRestController.java

@@ -45,11 +45,10 @@ import org.opsli.core.cache.local.CacheUtil;
 import org.opsli.core.msg.CoreMsg;
 import org.opsli.core.msg.TokenMsg;
 import org.opsli.core.security.shiro.realm.JwtRealm;
+import org.opsli.core.utils.DistributedLockUtil;
 import org.opsli.core.utils.ExcelUtil;
 import org.opsli.core.utils.UserUtil;
 import org.opsli.plugins.excel.exception.ExcelPluginException;
-import org.opsli.plugins.redis.RedisLockPlugins;
-import org.opsli.plugins.redis.lock.RedisLock;
 import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.web.bind.annotation.ModelAttribute;
 import org.springframework.web.bind.annotation.RequestParam;
@@ -98,10 +97,6 @@ public abstract class BaseRestController <T extends BaseEntity, E extends ApiWra
     @Autowired(required = false)
     protected S IService;
 
-    /** Redis分布式锁 */
-    @Autowired
-    private RedisLockPlugins redisLockPlugins;
-
     /** Excel工具类 */
     @Autowired
     private ExcelUtil excelUtil;
@@ -115,41 +110,37 @@ public abstract class BaseRestController <T extends BaseEntity, E extends ApiWra
      */
     @ModelAttribute
     public E get(@RequestParam(required=false) String id) {
-        E model = null;
+        E model;
         if (StringUtils.isNotBlank(id)){
             // 如果开启缓存 先从缓存读
             if(hotDataFlag){
+                // 缓存Key
+                String cacheKey = CacheConstants.HOT_DATA_PREFIX +"::"+ id;
+
                 model = WrapperUtil.transformInstance(
-                        CacheUtil.getTimed(entityClazz, CacheConstants.HOT_DATA_PREFIX +"::"+ id)
+                        CacheUtil.getTimed(entityClazz, cacheKey)
                         , modelClazz);
                 if(model != null){
                     return model;
                 }
-            }
+
                 // 防止缓存穿透判断
-            boolean hasNilFlag = false;
-            // 如果开启缓存 防止缓存穿透判断
-            if(hotDataFlag){
-                hasNilFlag = CacheUtil.hasNilFlag("get::" + id);
-            }
-            if(!hasNilFlag){
-                // 锁凭证 redisLock 贯穿全程
-                RedisLock redisLock = new RedisLock();
-                redisLock.setLockName("getLock::"+id)
-                        .setAcquireTimeOut(3000L)
-                        .setLockTimeOut(10000L);
-                // 这里增加分布式锁 防止缓存击穿
-                if(hotDataFlag){
+                boolean hasNilFlag = CacheUtil.hasNilFlag(cacheKey);
+                if(hasNilFlag){
+                    return null;
+                }
+
                 try {
-                        // 加锁
-                        redisLock = redisLockPlugins.tryLock(redisLock);
-                        if(redisLock == null){
+                    // 分布式加锁
+                    if(!DistributedLockUtil.lock(cacheKey)){
+                        // 无法申领分布式锁
+                        log.error(CoreMsg.REDIS_EXCEPTION_LOCK.getMessage());
                         throw new ServiceException(CoreMsg.CACHE_PUNCTURE_EXCEPTION);
                     }
 
                     // 如果获得锁 则 再次检查缓存里有没有， 如果有则直接退出， 没有的话才发起数据库请求
                     model = WrapperUtil.transformInstance(
-                                CacheUtil.getTimed(entityClazz, CacheConstants.HOT_DATA_PREFIX +"::"+ id)
+                            CacheUtil.getTimed(entityClazz, cacheKey)
                             , modelClazz);
                     if(model != null){
                         return model;
@@ -157,32 +148,37 @@ public abstract class BaseRestController <T extends BaseEntity, E extends ApiWra
 
                     // 如果缓存没读到 则去数据库读
                     model = WrapperUtil.transformInstance(IService.get(id), modelClazz);
+
+                    // 获得数据后处理
+                    if(model != null){
+                        // 这里会 同步更新到本地Ehcache 和 Redis缓存
+                        // 如果其他服务器缓存也丢失了 则 回去Redis拉取
+                        CacheUtil.put(cacheKey, model);
+                    }
                 }catch (Exception e){
                     log.error(e.getMessage(), e);
                 }finally {
                     // 释放锁
-                        redisLockPlugins.unLock(redisLock);
-                    }
-                }else{
-                    // 如果缓存没读到 则去数据库读
-                    model = WrapperUtil.transformInstance(IService.get(id),modelClazz);
+                    DistributedLockUtil.unlock(cacheKey);
                 }
 
-                // 获得数据后处理
-                if(model != null){
-                    // 如果开启缓存 将数据库查询对象 存如缓存
-                    if(hotDataFlag){
-                        // 这里会 同步更新到本地Ehcache 和 Redis缓存
-                        // 如果其他服务器缓存也丢失了 则 回去Redis拉取
-                        CacheUtil.put(id, model);
+                if(model == null){
+                    // 设置空变量 用于防止穿透判断
+                    CacheUtil.putNilFlag(cacheKey);
+                    return null;
                 }
+
+                return model;
             }else {
-                    // 设置空变量 用于防止穿透判断
-                    CacheUtil.putNilFlag("get::" + id);
+                // 如果没开启缓存 则直接查询数据库
+                model = WrapperUtil.transformInstance(IService.get(id), modelClazz);
+                if(model != null){
+                    return model;
                 }
             }
         }
-        if (model == null){
+
+        // 如果参数没传入ID 则创建一个空的对象
         try {
             // 创建泛型对象
             model = this.createModel();
@@ -190,7 +186,6 @@ public abstract class BaseRestController <T extends BaseEntity, E extends ApiWra
             log.error(CommonMsg.EXCEPTION_CONTROLLER_MODEL.toString()+" : {}",e.getMessage());
             throw new ServiceException(CommonMsg.EXCEPTION_CONTROLLER_MODEL);
         }
-        }
         return model;
     }