sprinboot之mongodb

　　 <parent>
        <groupId>org.springframework.boot</groupId>
        <artifactId>spring-boot-starter-parent</artifactId>
        <version>2.0.6.RELEASE</version>
    </parent>

    <dependencies>
        <dependency>
            <groupId>org.springframework.boot</groupId>
            <artifactId>spring-boot-starter-web</artifactId>
            <version>2.0.6.RELEASE</version>
        </dependency>
        <dependency>
            <groupId>org.springframework.boot</groupId>
            <artifactId>spring-boot-starter-data-mongodb</artifactId>
            <version>2.0.6.RELEASE</version>
        </dependency>
    </dependencies>

/**
 * file 文件处理配置
 */
@Configuration
public class MongoConfiguration {

    @Bean
    public GridFSBucket gridFSBucket(MongoDbFactory mongoDbFactory) {
        //主要是配置GridFSBuckets用于下载
        return GridFSBuckets.create(mongoDbFactory.getDb());
    }
}

@Service
@Transactional
public class MongoServiceImpl implements IMongoService {

    @Autowired
    private GridFsTemplate gridFsTemplate;

    @Autowired
    private GridFSBucket gridFSBucket;

    @Autowired
    private MongoTemplate mongoTemplate;

    /**
     * 上传文件
     * @param file
     * @return
     * @throws FileNotFoundException
     */
    @Override
    public String uploadFile(File file) throws FileNotFoundException {
        ObjectId objectId = gridFsTemplate.store(new FileInputStream(file), file.getName());
        return objectId.toHexString();
    }

    /**
     * 上传文件
     * @param inputStream
     * @param fileName
     * @return
     */
    @Override
    public String uploadFile(InputStream inputStream, String fileName) {
        ObjectId objectId = gridFsTemplate.store(inputStream, fileName);
        return objectId.toHexString();
    }

    /**
     * 下载文件
     * @param mongoId
     * @param outputStream
     */
    @Override
    public void downloadFile(String mongoId, OutputStream outputStream) throws IOException {
//        GridFSFile gridFSFile = gridFsTemplate.findOne(Query.query(Criteria.where("_id").is(mongoId)));
//        GridFsResource gridFsResource = new GridFsResource(gridFSFile);
//        IOUtils.copy(gridFsResource.getInputStream(), outputStream);
        //上面这种方式也可以实现下载
        gridFSBucket.downloadToStream(new ObjectId(mongoId), outputStream);
    }

    /**
     * 删除文件
     * @param mongoId
     */
    @Override
    public void deleteFile(String mongoId) {
        gridFsTemplate.delete(Query.query(Criteria.where("_id").is(mongoId)));
    }

    /**
     * 保存对象
     * @param t
     * @param <T>
     */
    @Override
    public <T> void save(T t) {
        mongoTemplate.save(t);
    }

    /**
     * 查询对象
     * @param query
     * @param clazz
     * @param <T>
     * @return
     */
    @Override
    public <T> List<T> find(Query query, Class<T> clazz) {
        return mongoTemplate.find(query, clazz);
    }

    /**
     * 查询单个对象
     * @param query
     * @param clazz
     * @param <T>
     * @return
     */
    @Override
    public <T> T findOne(Query query, Class<T> clazz) {
        return mongoTemplate.findOne(query, clazz);
    }

    /**
     * 删除对象
     * @param query
     * @param clazz
     * @param <T>
     */
    @Override
    public <T> void delete(Query query, Class<T> clazz) {
        mongoTemplate.findAndRemove(query, clazz);
    }

}

@RestController
@RequestMapping(value = "/file")
public class FileController {

    private Logger logger = LoggerFactory.getLogger(FileController.class);

    @Autowired
    private IMongoService mongoService;

    @RequestMapping(value = "/uploadFile", method = RequestMethod.POST)
    public ResultInfo uploadFile(MultipartFile multipartFile) {
        ResultInfo resultInfo = new ResultInfo();
        try {
            resultInfo.setData(mongoService.uploadFile(multipartFile.getInputStream(), multipartFile.getOriginalFilename()));
        } catch (Exception e) {
            logger.error("upload file error:", e);
            resultInfo.setStatus(false);
            resultInfo.setFailCode(e.getMessage());
        }
        return resultInfo;
    }

    @RequestMapping(value = "/downloadFile", method = RequestMethod.POST)
    public ResultInfo downloadFile(@RequestBody Map<String, Object> map, HttpServletResponse response) {
        ResultInfo resultInfo = new ResultInfo();
        try {
            response.setContentType("multipart/form-data");
            response.setHeader("Content-Disposition", "attachment;fileName=test.mp3");
            mongoService.downloadFile(map.get("mongoId").toString(), response.getOutputStream());
        } catch (Exception e) {
            logger.error("download file error:", e);
            resultInfo.setStatus(false);
            resultInfo.setFailCode(e.getMessage());
        }
        return resultInfo;
    }

    @RequestMapping(value = "/deleteFile", method = RequestMethod.POST)
    public ResultInfo deleteFile(@RequestBody Map<String, Object> map) {
        ResultInfo resultInfo = new ResultInfo();
        try {
            mongoService.deleteFile(map.get("mongoId").toString());
        } catch (Exception e) {
            logger.error("delete file error:", e);
            resultInfo.setStatus(false);
            resultInfo.setFailCode(e.getMessage());
        }
        return resultInfo;
    }
}

/**
 * 对象存储，测试
 */
@Component
public class MongoTask implements CommandLineRunner {

    @Autowired
    private IMongoService mongoService;

    @Override
    public void run(String... args) throws Exception {
        mongoService.save(new User(1L, "test1", 25));
        mongoService.save(new User(2L, "test2", 25));
        System.out.println(mongoService.find(Query.query(Criteria.where("name").regex("test")), User.class));
        System.out.println(mongoService.findOne(Query.query(Criteria.where("name").is("test2")), User.class));
        mongoService.delete(Query.query(Criteria.where("name").is("test2")), User.class);
    }
}