Springboot 工程中快速判断web应用服务器类型

文章目录

一、核心源码
二、典型应用

一、核心源码

java 复制代码

    boolean isTomcat;
    
    boolean isJetty;
    
    @Autowired
    ApplicationContext applicationContext;
    
    @PostConstruct
    private void init()
    {
        isTomcat = Stream.of(applicationContext.getBeanDefinitionNames()).anyMatch(name -> StringUtils.containsIgnoreCase(name, "EmbeddedTomcat"));
        isJetty = Stream.of(applicationContext.getBeanDefinitionNames()).anyMatch(name -> StringUtils.containsIgnoreCase(name, "EmbeddedJetty"));
        log.info("#### isTomcat: {}", isTomcat);
        log.info("#### isJetty: {}", isJetty);
    }

二、典型应用

java 复制代码

import java.io.File;
import java.io.IOException;
import java.util.stream.Stream;

import javax.annotation.PostConstruct;

import org.apache.commons.io.FileUtils;
import org.apache.commons.lang3.RandomUtils;
import org.apache.commons.lang3.StringUtils;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.context.ApplicationContext;
import org.springframework.web.bind.annotation.PostMapping;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RequestParam;
import org.springframework.web.bind.annotation.RestController;
import org.springframework.web.multipart.MultipartFile;

import com.fly.demo.common.JsonResult;

import io.swagger.annotations.Api;
import io.swagger.annotations.ApiOperation;
import lombok.extern.slf4j.Slf4j;

@Slf4j
@Api(tags = "文件上传(simple)")
@RestController
@RequestMapping("/simple/file")
public class SimpleFileController
{
    boolean isTomcat;
    
    boolean isJetty;
    
    @Autowired
    ApplicationContext applicationContext;
    
    @PostConstruct
    private void init()
    {
        isTomcat = Stream.of(applicationContext.getBeanDefinitionNames()).anyMatch(name -> StringUtils.containsIgnoreCase(name, "EmbeddedTomcat"));
        isJetty = Stream.of(applicationContext.getBeanDefinitionNames()).anyMatch(name -> StringUtils.containsIgnoreCase(name, "EmbeddedJetty"));
        log.info("#### isTomcat: {}", isTomcat);
        log.info("#### isJetty: {}", isJetty);
    }
    
    @ApiOperation("文件上传")
    @PostMapping("/upload")
    public JsonResult<?> upload(@RequestParam MultipartFile file)
        throws IOException
    {
        File rootDir = new File("upload");
        File dest;
        if (isTomcat)
        {
            if (RandomUtils.nextBoolean())
            {
                log.info("### transferTo");
                dest = new File(rootDir.getCanonicalPath() + File.separator + file.getOriginalFilename());
                file.transferTo(dest);
            }
            else
            {
                log.info("### copyInputStreamToFile");
                dest = new File(rootDir, file.getOriginalFilename());
                FileUtils.copyInputStreamToFile(file.getInputStream(), dest);
            }
        }
        else
        {
            log.info("### copyInputStreamToFile");
            dest = new File(rootDir, file.getOriginalFilename());
            FileUtils.copyInputStreamToFile(file.getInputStream(), dest);
        } 
        return JsonResult.success(dest.getName());
    }
}