sinvo/tianai-captcha
1
0
Fork 0
mirror of https://github.com/dromara/tianai-captcha.git synced 2026-05-07 06:04:34 +08:00
Code Issues Packages Projects Releases Wiki Activity

修复在并发场景下验证器生成多次的问题

Browse Source
This commit is contained in:
天爱有情
2023-04-12 08:44:26 +08:00
parent e1224446cb
commit 988ad07d34
1 changed files with 8 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/main/java/cloud/tianai/captcha/generator/impl/MultiImageCaptchaGenerator.java
+8 -9
View File
@@ -18,6 +18,7 @@ import lombok.Setter;
import java.util.HashMap;
import java.util.Map;
import java.util.concurrent.ConcurrentHashMap;
/**
* @Author: 天爱有情
@@ -26,7 +27,7 @@ import java.util.Map;
*/
public class MultiImageCaptchaGenerator extends AbstractImageCaptchaGenerator {
protected Map<String, ImageCaptchaGenerator> imageCaptchaGeneratorMap = new HashMap<>(4);
protected Map<String, ImageCaptchaGenerator> imageCaptchaGeneratorMap = new ConcurrentHashMap<>(4);
protected Map<String, ImageCaptchaGeneratorProvider> imageCaptchaGeneratorProviderMap = new HashMap<>(4);
@Setter
@@ -93,16 +94,14 @@ public class MultiImageCaptchaGenerator extends AbstractImageCaptchaGenerator {
}
public ImageCaptchaGenerator requireGetCaptchaGenerator(String type) {
ImageCaptchaGenerator imageCaptchaGenerator = imageCaptchaGeneratorMap.get(type);
if (imageCaptchaGenerator == null) {
ImageCaptchaGeneratorProvider provider = imageCaptchaGeneratorProviderMap.get(type);
ImageCaptchaGenerator imageCaptchaGenerator = imageCaptchaGeneratorMap.computeIfAbsent(type, t -> {
ImageCaptchaGeneratorProvider provider = imageCaptchaGeneratorProviderMap.get(t);
if (provider == null) {
throw new IllegalArgumentException("生成验证码失败，错误的type类型:" + type);
}
imageCaptchaGenerator = imageCaptchaGeneratorMap.computeIfAbsent(type, k ->
// get and init
provider.get(getImageResourceManager(), getImageTransform()).init(initDefaultResource));
throw new IllegalArgumentException("生成验证码失败，错误的type类型:" + t);
}
return provider.get(getImageResourceManager(), getImageTransform()).init(initDefaultResource);
});
return imageCaptchaGenerator;
}