girdFS 存儲照片，保存照片到硬盤

學習girdFS筆記，主要是girdFS存儲照片到MongoDB·，保存照片到硬盤

0，GridFSBucket 配置類，用於打開下載流對象

@Configuration
public class MongoConfig {
    @Value("${spring.data.mongodb.database}")
    String db;

    @Bean
    public GridFSBucket getGridFSBucket(MongoClient mongoClient){
        MongoDatabase database = mongoClient.getDatabase(db);
        GridFSBucket bucket = GridFSBuckets.create(database);
        return bucket;
    }
}

1，存儲照片

 @Autowired
    GridFsTemplate gridFsTemplate;

    @Autowired
    GridFSBucket gridFSBucket;

    @Test
    public void addImage() {
        try {
            //要存儲的文件
            File file = new File("E:\\360Downloads\\1.jpg");
            //定義輸入流
            FileInputStream inputStram = new FileInputStream(file);
            //向GridFS存儲文件
            ObjectId objectId = gridFsTemplate.store(inputStram, "img測試");
            //得到文件ID
            String fileId = objectId.toString();
            System.out.println(fileId);//5d7e000fd3e38128a8481c4f
        } catch (FileNotFoundException e) {
            e.printStackTrace();
        }
    }

2， 下載圖片

    @Test
    public void getImg() throws IOException {
        //獲取id值
        String fileId = "5d7e000fd3e38128a8481c4f";
        //取出文件
        GridFSFile gridFSFile = gridFsTemplate.findOne(query(Criteria.where("_id").is(fileId)));
       
        //打開下載流對象（方式1）
        GridFSDownloadStream gridFSDownloadStream = gridFSBucket.openDownloadStream(gridFSFile.getObjectId());
        //創建gridFsResource，用於獲取流對象
        GridFsResource gridFsResource = new GridFsResource(gridFSFile, gridFSDownloadStream);
        
        /**打開下載流對象（方式2）
         GridFsResource resource = gridFsTemplate.getResource(gridFSFile);*/
      
        //獲取輸入流
        InputStream fis = gridFsResource.getInputStream();
        //獲取輸出流
        FileOutputStream fos=new FileOutputStream("d:\\4.jpg");

        byte[] bytes=new byte[1024];
        int len=0;
        while ((len=fis.read(bytes))!=-1){
            fos.write(bytes,0,len);
        }
        fos.close();
        fis.close();
    }