使用SpringBoot上传文件并存储至数据库

淩亂°似流年 2024-02-18 13:58 55阅读 0赞

### Pom文件 ###

<dependency>
                <groupId>org.springframework.boot</groupId>
                <artifactId>spring-boot-starter-web</artifactId>
            </dependency>
            <dependency>
                <groupId>org.projectlombok</groupId>
                <artifactId>lombok</artifactId>
            </dependency>
            <dependency>
                <groupId>org.springframework.boot</groupId>
                <artifactId>spring-boot-starter-jdbc</artifactId>
            </dependency>
            <dependency>
                <groupId>mysql</groupId>
                <artifactId>mysql-connector-java</artifactId>
            </dependency>
            <dependency>
                <groupId>org.springframework.boot</groupId>
                <artifactId>spring-boot-starter-data-jpa</artifactId>
            </dependency>

*由于是演示上传文件，并将文件存储进数据库，所以这里简单的使用JPA 进行单表处理*

### 建表语句 ###

CREATE TABLE `image` (
      `id` int(11) NOT NULL AUTO_INCREMENT COMMENT '主键',
      `image` longblob NOT NULL COMMENT '图片文件',
      PRIMARY KEY (`id`)
    ) ENGINE=InnoDB DEFAULT CHARSET=utf8

### 向数据库插入 ###

**domain类如下：**

@Table(name="image")
    @Entity
    @Data
    public class ImageFile {
        @Id
        @GeneratedValue(strategy = GenerationType.IDENTITY)
        private Integer id;
    
        @Column(name = "image",columnDefinition = "imageBlob")
        private byte[] imageBytes;
    
    }

这里简单的使用下JPA，如下：

@Repository
    public interface ImageRepository extends JpaRepository<ImageFile,Integer> {
    }

*JPA是基于反射的机制来动态构成SQL，执行SQL*

注意点：

*  上传文件需要指定类型为**MultiPartFile**，如果要上传多文件，使用**MultiPartFile\[\]**。
 *  文件对应到数据库类型是blob/longblob等，在处理的时候，获取文件的字节数据 直接存储即可。

**Controller控制器**

@Autowired
        private ImageRepository imageRepository;
        @PostMapping
        public void upload(@RequestParam("file") MultipartFile file, @RequestParam("id")Integer id, HttpServletRequest request) throws IOException {
            byte[] bytes = file.getBytes();
            ImageFile imageFile=new ImageFile();
            imageFile.setImageBytes(bytes);
    
            imageRepository.save(imageFile);
    
        }

**使用MyBatis处理，方式类似:**

<insert id="insert" parameterType="com.wojiushiwo.ImageFile">
    insert into image
    values(#{image})
    </insert>

至此 将文件上传存储至数据库即完成。

### 从数据库读取 ###

*下面展示如果从数据库中获取存储的二进制文件，并处理(这里还原二进制文件)*

@GetMapping
        public void getById(@RequestParam("id")Integer id, @Nullable Integer abc) throws IOException {
    
            Optional<ImageFile> imageFile = imageRepository.findById(id);
            if(imageFile.isPresent()){
                byte[] imageBytes = imageFile.get().getImageBytes();
                ByteArrayInputStream inputStream=new ByteArrayInputStream(imageBytes);
    			//将二进制字节数组 转为文件
                Files.copy(inputStream, Paths.get("/Users/wojiushiwo/Desktop/1.jpg"));
            }
    
        }

######  ######

**domain类:**

//这里是lombok jar包的注解
    @Data
    public class User {
    
        private Integer userId;
    
        private String name;
    
        private MultipartFile file;
    }

**controller类：**

@PostMapping(value = "/upload")
        public void upload(User user){
            // 这里就可以对传入的参数进行业务处理了
        }

问题1 Data too long for column ‘image’ at row 1’

原因及解决方式：上传的文件太大了，使用blob类型不行了，将数据库表中image的数据类型改为longblob

问题2 Table ‘…hibernate\_sequence’ doesn’t exist  
这个与主键的生成策略有关，只需要将id的生成策略由@GeneratedValue==>@GeneratedValue(strategy = GenerationType.IDENTITY)