package xgg.springboot.base.controller;

import io.swagger.annotations.ApiOperation;
import lombok.RequiredArgsConstructor;
import lombok.extern.slf4j.Slf4j;
import org.apache.commons.io.FileUtils;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.web.bind.annotation.PostMapping;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RestController;
import org.springframework.web.multipart.MultipartFile;
import xgg.springboot.common.util.GetIpAddressUtil;

import javax.servlet.ServletContext;
import java.io.File;
import java.io.IOException;
import java.io.InputStream;
import java.util.HashMap;
import java.util.Map;
import java.util.UUID;

/**
 * @author xiegege
 * @date 2020/12/7 14:03
 */
@Slf4j
@RestController
@RequestMapping("upload")
@RequiredArgsConstructor(onConstructor = @__(@Autowired))
public class UploadFileController {

    private final ServletContext context;

    @PostMapping("upload-image")
    public Map<String, String> upload(MultipartFile file) throws IOException {
        //参数校验
        if (file == null) {
            throw new IllegalArgumentException("file or filename object is null");
        }
        // 获取文件类型
        String contentType = file.getContentType();
        // 获取文件后缀
        String ext = contentType.substring(contentType.lastIndexOf("/") + 1);
        // uuid生成文件名 + 后缀
        String fileName = UUID.randomUUID() + "." + ext;
        // 返回url路径
        String url = "http://" + GetIpAddressUtil.getIpAddress() + ":8000/statics/attachment/" + fileName;
        // 本地磁盘存储路径
        String filePath = context.getRealPath("/") + "statics/attachment/" + fileName;
        log.info("入库本地磁盘路径：" + filePath);
        // 写入文件
        write(file.getInputStream(), filePath);
        // 返回数据
        Map<String, String> dataMap = new HashMap<>(3);
        dataMap.put("name", fileName);
        dataMap.put("url", url);
        dataMap.put("ext", ext);
        return dataMap;
    }

    /**
     * 将inputStream写入文件
     *
     * @param stream 文件流
     * @param path   要写入的文件路径
     */
    private static void write(InputStream stream, String path) {
        try {
            FileUtils.copyInputStreamToFile(stream, new File(path));
        } catch (IOException e) {
            e.printStackTrace();
        }
    }
}

GetIpAddressUtil

public class GetIpAddressUtil {

    public static void main(String[] args) {
        System.out.println("本机IP:" + getIpAddress());
    }

    public static String getIpAddress() {
        try {
            Enumeration<NetworkInterface> allNetInterfaces = NetworkInterface.getNetworkInterfaces();
            InetAddress ip;
            while (allNetInterfaces.hasMoreElements()) {
                NetworkInterface netInterface = allNetInterfaces.nextElement();
                if (netInterface.isLoopback() || netInterface.isVirtual() || !netInterface.isUp()) {
                    continue;
                } else {
                    Enumeration<InetAddress> addresses = netInterface.getInetAddresses();
                    while (addresses.hasMoreElements()) {
                        ip = addresses.nextElement();
                        if (ip instanceof Inet4Address) {
                            return ip.getHostAddress();
                        }
                    }
                }
            }
        } catch (Exception e) {
            System.err.println("IP地址获取失败" + e.toString());
        }
        return "";
    }
}