com.houkunlin.system.common.aop.SystemCommonAopPOIAutoConfiguration Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of system-common-aop-starter Show documentation
常用的 AOP 注解功能。 Commonly used AOP annotation features.
The newest version!
package com.houkunlin.system.common.aop;

import com.alibaba.excel.EasyExcel;
import com.deepoove.poi.XWPFTemplate;
import jakarta.servlet.http.HttpServletResponse;
import lombok.RequiredArgsConstructor;
import org.apache.poi.POIDocument;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.boot.autoconfigure.condition.ConditionalOnClass;
import org.springframework.boot.autoconfigure.condition.ConditionalOnMissingBean;
import org.springframework.context.ApplicationContext;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;
import org.springframework.lang.NonNull;

import java.io.IOException;
import java.io.InputStream;

/**
 * Excel 导出下载配置
 *
 * @author HouKunLin
 */
@ConditionalOnClass(POIDocument.class)
@Configuration(proxyBeanMethods = false)
@RequiredArgsConstructor
public class SystemCommonAopPOIAutoConfiguration {

    @Bean
    @ConditionalOnMissingBean
    public DownloadPoiHandler downloadPoiHandler() {
        return new DownloadPoiHandler() {
            private static final Logger logger = LoggerFactory.getLogger(DownloadPoiHandler.class);

            @Override
            public InputStream getTemplate(@NonNull String templateName) throws IOException {
                if (templateName.isBlank()) {
                    return null;
                }
                InputStream inputStream = ClassPathUtil.getResourceAsStream(templateName);
                if (inputStream == null) {
                    logger.warn("使用默认的 Excel/Word 模板处理器，不支持读取 ClassPath 之外的文件模板，需要自行实现 DownloadPoiHandler 接口功能。当前读取模板：{}", templateName);
                }
                return inputStream;
            }
        };
    }

    @ConditionalOnClass(EasyExcel.class)
    @Configuration(proxyBeanMethods = false)
    public static class POIExcelAutoConfiguration {
        @Bean
        public DownloadExcelAspect downloadExcelAspect(
                TemplateParser templateParser,
                DownloadPoiHandler downloadPoiHandler,
                HttpServletResponse response,
                ApplicationContext applicationContext
        ) {
            return new DownloadExcelAspect(templateParser, downloadPoiHandler, response, applicationContext);
        }
    }

    @ConditionalOnClass(XWPFTemplate.class)
    @Configuration(proxyBeanMethods = false)
    public static class POIWordAutoConfiguration {
        @Bean
        public DownloadWordAspect downloadWordAspect(
                TemplateParser templateParser,
                DownloadPoiHandler downloadPoiHandler,
                HttpServletResponse response
        ) {
            return new DownloadWordAspect(templateParser, downloadPoiHandler, response);
        }
    }
}