封装同步锁，测试并发

xumiao
1 parent ce4d674c
Showing 3 changed files with 75 additions and 0 deletions
src/main/java/com/huaheng/framework/config/ShiroConfig.java
src/main/java/com/huaheng/framework/web/controller/BaseController.java
src/main/java/com/huaheng/framework/web/controller/Test2Controller.java
@@ -267,6 +267,7 @@ public class ShiroConfig {
         filterChainDefinitionMap.put("/endpoint/**", "anon");
         filterChainDefinitionMap.put("/manager/**", "anon");
         filterChainDefinitionMap.put("/API/WMS/v2/**","anon");
+        filterChainDefinitionMap.put("/test","anon");
         // 系统权限列表
 //        filterChainDefinitionMap.putAll(SpringUtils.getBean(IMenuService.class).selectPermsAll());
  
@@ -3,7 +3,10 @@ package com.huaheng.framework.web.controller;
 import java.util.Date;
 import java.util.List;
 import java.text.SimpleDateFormat;
+import java.util.Map;
+import java.util.concurrent.ConcurrentHashMap;
 import java.util.concurrent.Semaphore;
+import java.util.concurrent.locks.ReentrantLock;
  
 import com.github.pagehelper.PageHelper;
 import com.github.pagehelper.PageInfo;
@@ -227,6 +230,37 @@ public class BaseController
         return ajaxResult;
     }
  
+    Map<Object, ReentrantLock> mutexCache= new ConcurrentHashMap<>();
+
+    public AjaxResult exec(Object key,MultiProcessListener multiProcessListener){
+        ReentrantLock mutex4Key = null;
+        ReentrantLock mutexIncache;
+        do {
+            if (mutex4Key != null) {
+                mutex4Key.unlock();
+            }
+            mutex4Key = mutexCache.computeIfAbsent(key, k -> new ReentrantLock());
+            mutex4Key.lock();
+            mutexIncache = mutexCache.get(key);
+            /**
+             * 执行到这里：问题有两种情况
+             * 1、 mutexIncache ==null，说明刚拿到锁，就被remove了
+             * 2、 mutexIncache !=mutex4Key，说明刚拿到锁，就被新进来的线程创建的锁覆盖了
+             */
+        }while (mutexIncache ==null|| mutexIncache !=mutex4Key);
+        try {
+            multiProcessListener.doProcess();
+        }catch (Exception e){
+            e.printStackTrace();
+        }finally {
+            if (mutex4Key.getQueueLength()==0) {
+                mutexCache.remove(key);
+            }
+            mutex4Key.unlock();
+        }
+        return AjaxResult.success();
+    }
+
     public interface MultiProcessListener {
         AjaxResult doProcess();
     }
+package com.huaheng.framework.web.controller;
+
+
+import com.huaheng.api.wcs.domain.TaskFinishDomain;
+import com.huaheng.framework.web.domain.AjaxResult;
+import org.slf4j.Logger;
+import org.slf4j.LoggerFactory;
+import org.springframework.web.bind.annotation.PathVariable;
+import org.springframework.web.bind.annotation.PostMapping;
+import org.springframework.web.bind.annotation.RequestBody;
+import org.springframework.web.bind.annotation.RestController;
+
+import java.util.concurrent.ThreadLocalRandom;
+import java.util.concurrent.TimeUnit;
+
+/**
+ * 测试并发
+ */
+@RestController
+public class Test2Controller extends BaseController{
+    private static final Logger log=  LoggerFactory.getLogger(Test2Controller.class);
+    @PostMapping("/test")
+    public void test(@RequestBody TaskFinishDomain taskFinishDomain){
+        exec(taskFinishDomain, new MultiProcessListener() {
+            @Override
+            public AjaxResult doProcess() {
+                int i = ThreadLocalRandom.current().nextInt(10000);
+                log.info("并发测试：[{}]：开始",taskFinishDomain+"*starting begin*&第"+i+"day!");
+                try {
+                    TimeUnit.MILLISECONDS.sleep(ThreadLocalRandom.current().nextInt(5000));
+                } catch (InterruptedException e) {
+                    e.printStackTrace();
+                }
+                log.info("并发测试：[{}]：结束",taskFinishDomain+"*starting begin*&第"+i+"day!");
+                return AjaxResult.success();
+            }
+        });
+
+    }
+}