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

U 优化代码

Browse Source
This commit is contained in:
天爱有情
2022-05-12 09:35:50 +08:00
parent 7c6fec6ea3
commit 0d8e3df607
1 changed files with 8 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/main/java/cloud/tianai/captcha/generator/impl/MultiImageCaptchaGenerator.java
+8 -4
View File
@@ -21,8 +21,8 @@ import java.util.function.BiFunction;
*/ */
public class MultiImageCaptchaGenerator extends AbstractImageCaptchaGenerator { public class MultiImageCaptchaGenerator extends AbstractImageCaptchaGenerator {
private Map<String, ImageCaptchaGenerator> imageCaptchaGeneratorMap = new HashMap<>(4); protected Map<String, ImageCaptchaGenerator> imageCaptchaGeneratorMap = new HashMap<>(4);
private Map<String, BiFunction<String, MultiImageCaptchaGenerator, ImageCaptchaGenerator>> imageCaptchaGeneratorProviderMap = new HashMap<>(4); protected Map<String, BiFunction<String, MultiImageCaptchaGenerator, ImageCaptchaGenerator>> imageCaptchaGeneratorProviderMap = new HashMap<>(4);
@Setter @Setter
@Getter @Getter
@@ -79,6 +79,11 @@ public class MultiImageCaptchaGenerator extends AbstractImageCaptchaGenerator {
param.setType(defaultCaptcha); param.setType(defaultCaptcha);
type = defaultCaptcha; type = defaultCaptcha;
} }
ImageCaptchaGenerator imageCaptchaGenerator = requireGetCaptchaGenerator(type);
return imageCaptchaGenerator.generateCaptchaImage(param);
}
public ImageCaptchaGenerator requireGetCaptchaGenerator(String type) {
ImageCaptchaGenerator imageCaptchaGenerator = imageCaptchaGeneratorMap.get(type); ImageCaptchaGenerator imageCaptchaGenerator = imageCaptchaGeneratorMap.get(type);
if (imageCaptchaGenerator == null) { if (imageCaptchaGenerator == null) {
BiFunction<String, MultiImageCaptchaGenerator, ImageCaptchaGenerator> provider = imageCaptchaGeneratorProviderMap.get(type); BiFunction<String, MultiImageCaptchaGenerator, ImageCaptchaGenerator> provider = imageCaptchaGeneratorProviderMap.get(type);
@@ -87,7 +92,6 @@ public class MultiImageCaptchaGenerator extends AbstractImageCaptchaGenerator {
} }
imageCaptchaGenerator = imageCaptchaGeneratorMap.computeIfAbsent(type, k -> provider.apply(k, this)); imageCaptchaGenerator = imageCaptchaGeneratorMap.computeIfAbsent(type, k -> provider.apply(k, this));
} }
return imageCaptchaGenerator;
return imageCaptchaGenerator.generateCaptchaImage(param);
} }
} }