优化

2026-05-07 06:04:34 +08:00 · 2022-05-07 11:14:15 +08:00
parent d802fd481a
commit afbb88c67d
46 changed files with 376 additions and 260 deletions
@@ -0,0 +1,111 @@
+package cloud.tianai.captcha.resource.impl;
+
+import cloud.tianai.captcha.resource.ImageCaptchaResourceManager;
+import cloud.tianai.captcha.resource.impl.provider.URLResourceProvider;
+import cloud.tianai.captcha.resource.ImageCaptchaResourceManager;
+import cloud.tianai.captcha.resource.ResourceProvider;
+import cloud.tianai.captcha.resource.ResourceStore;
+import cloud.tianai.captcha.resource.common.model.dto.Resource;
+import cloud.tianai.captcha.resource.impl.provider.ClassPathResourceProvider;
+import cloud.tianai.captcha.resource.impl.provider.FileResourceProvider;
+import cloud.tianai.captcha.resource.impl.provider.URLResourceProvider;
+
+import java.io.InputStream;
+import java.util.ArrayList;
+import java.util.Collections;
+import java.util.List;
+import java.util.Map;
+
+/**
+ * @Author: 天爱有情
+ * @date 2021/8/7 15:35
+ * @Description 默认的滑块验证码资源管理
+ */
+public class DefaultImageCaptchaResourceManager implements ImageCaptchaResourceManager {
+
+    /** 资源存储. */
+    private ResourceStore resourceStore;
+    /** 资源转换 转换为stream流. */
+    private final List<ResourceProvider> resourceProviderList = new ArrayList<>(8);
+
+
+    public DefaultImageCaptchaResourceManager() {
+        init();
+    }
+
+    public DefaultImageCaptchaResourceManager(ResourceStore resourceStore) {
+        this.resourceStore = resourceStore;
+        init();
+    }
+
+    private void init() {
+        if (this.resourceStore == null) {
+            this.resourceStore = new DefaultResourceStore();
+        }
+        // 注入一些默认的提供者
+        registerResourceProvider(new URLResourceProvider());
+        registerResourceProvider(new ClassPathResourceProvider());
+        registerResourceProvider(new FileResourceProvider());
+    }
+
+    @Override
+    public Map<String, Resource> randomGetTemplate(String type) {
+        Map<String, Resource> resourceMap = resourceStore.randomGetTemplateByType(type);
+        if (resourceMap == null) {
+            throw new IllegalStateException("随机获取模板错误，store中模板为空, type:" + type);
+        }
+        return resourceMap;
+    }
+
+    @Override
+    public Resource randomGetResource(String type) {
+        Resource resource = resourceStore.randomGetResource(type);
+        if (resource == null) {
+            throw new IllegalStateException("随机获取资源错误，store中资源为空, type:" + type);
+        }
+        return resource;
+    }
+
+    @Override
+    public InputStream getResourceInputStream(Resource resource) {
+        for (ResourceProvider resourceProvider : resourceProviderList) {
+            if (resourceProvider.supported(resource.getType())) {
+                InputStream resourceInputStream = resourceProvider.getResourceInputStream(resource);
+                if (resourceInputStream == null) {
+                    throw new IllegalArgumentException("滑块验证码 ResourceProvider 读到的图片资源为空,providerName=["
+                            + resourceProvider.getName() + "], resource=[" + resource + "]");
+                }
+                return resourceInputStream;
+            }
+        }
+        throw new IllegalStateException("没有找到Resource [" + resource.getType() + "]对应的资源提供者");
+    }
+
+    @Override
+    public List<ResourceProvider> listResourceProviders() {
+        return Collections.unmodifiableList(resourceProviderList);
+    }
+
+    @Override
+    public void registerResourceProvider(ResourceProvider resourceProvider) {
+        deleteResourceProviderByName(resourceProvider.getName());
+        resourceProviderList.add(resourceProvider);
+    }
+
+    @Override
+    public boolean deleteResourceProviderByName(String name) {
+        return resourceProviderList.removeIf(r -> r.getName().equals(name));
+    }
+
+    @Override
+    public void setResourceStore(ResourceStore resourceStore) {
+        this.resourceStore = resourceStore;
+    }
+
+    @Override
+    public ResourceStore getResourceStore() {
+        return resourceStore;
+    }
+
+
+}