springcloud上传图片到数据库

约定不等于承诺〃 2022-10-26 12:58 197阅读 0赞

##### 实体类 #####

public class FileInfo{
    
        private Long filename;
    
        private String filename;
    
        private byte[] content;
    
    }

##### 上传图片到实体类 #####

public boolean importDada(MultipartFile file) {
            FileInfo fileInfo = new FileInfo();
            byte[] pic = null;
            if (file != null) {
                try {
                    //MultipartFile转InputStream
                    InputStream inputStream = file.getInputStream();
                    pic = new byte[inputStream.available()];
                    inputStream.read(pic);
                    inputStream.close();
                } catch (Exception e) {
                    new CustomException("团片转化失败", 400);
                }
                fileInfo.setContent(pic);
                fileInfo.setFilename(file.getOriginalFilename());
                
            }
          
            return fileInfoService.save(fileInfo);
        }

##### 下载图片 #####

public void findSalesPic(Long id,HttpServletResponse response) {
            FileInfo fileInfo = fileInfoMapper.selectOne(new LambdaUpdateWrapper<FileInfo>().eq(FileInfo::getId, id));
            byte[] data = fileInfo.getContent();
            response.setContentType("image/jpeg");
            response.setCharacterEncoding("UTF-8");
            OutputStream outputSream = null;
            try {
                outputSream = response.getOutputStream();
                InputStream in = new ByteArrayInputStream(data);
                int len = 0;
                byte[] buf = new byte[1024];
                while ((len = in.read(buf, 0, 1024)) != -1) {
                    outputSream.write(buf, 0, len);
                }
                outputSream.close();
            } catch (IOException e) {
                new CustomException("团片转化失败", 400);
    
            }
        }