UploadController

目录

1. UploadController

1.1. // 获取原始文件名称

1.2. // 生成新文件名

1.3. // 保存文件

1.4. // 返回结果

1.5. // 获取后缀

1.6. // 生成目录

1.7. // 判断目录是否存在

1.8. // 生成文件名

1. UploadController 

package com.goalias.controller;

import cn.hutool.core.io.FileUtil;

import cn.hutool.core.util.StrUtil;

import com.goalias.dto.Result;

import com.goalias.utils.SystemConstants;

import lombok.extern.slf4j.Slf4j;

import org.springframework.web.bind.annotation.*;

import org.springframework.web.multipart.MultipartFile;

import java.io.File;

import java.io.IOException;

import java.util.UUID;

@Slf4j

@RestController

@RequestMapping("upload")

public class UploadController {

    @PostMapping("blog")

    public Result uploadImage(@RequestParam("file") MultipartFile image) {

        try {

1. 1.             // 获取原始文件名称

            String originalFilename = image.getOriginalFilename();

1. 1.             // 生成新文件名

            String fileName = createNewFileName(originalFilename);

1. 1.             // 保存文件

            image.transferTo(new File(SystemConstants.IMAGE_UPLOAD_DIR, fileName));

1. 1.             // 返回结果

            log.debug("文件上传成功，{}", fileName);

            return Result.ok(fileName);

        } catch (IOException e) {

            throw new RuntimeException("文件上传失败", e);

        }

    }

    @GetMapping("/blog/delete")

    public Result deleteBlogImg(@RequestParam("name") String filename) {

        File file = new File(SystemConstants.IMAGE_UPLOAD_DIR, filename);

        if (file.isDirectory()) {

            return Result.fail("错误的文件名称");

        }

        FileUtil.del(file);

        return Result.ok();

    }

    private String createNewFileName(String originalFilename) {

1. 1.         // 获取后缀

        String suffix = StrUtil.subAfter(originalFilename, ".", true);

1. 1.         // 生成目录

        String name = UUID.randomUUID().toString();

        int hash = name.hashCode();

        int d1 = hash & 0xF;

        int d2 = (hash >> 4) & 0xF;

1. 1.         // 判断目录是否存在

        File dir = new File(SystemConstants.IMAGE_UPLOAD_DIR, StrUtil.format("/blogs/{}/{}", d1, d2));

        if (!dir.exists()) {

            dir.mkdirs();

        }

1. 1.         // 生成文件名

        return StrUtil.format("/blogs/{}/{}/{}.{}", d1, d2, name, suffix);

    }

}