java mongodb 文件上传，下载，删除方法

矫情吗；* 2024-02-18 17:56 37阅读 0赞

package com.stylefeng.guns.config.mongo;
    
    import java.io.File;
    import java.io.FileInputStream;
    import java.io.IOException;
    import org.apache.log4j.Logger;
    import org.bson.types.ObjectId;
    import org.springframework.beans.factory.annotation.Autowired;
    import org.springframework.data.mongodb.core.MongoTemplate;
    import org.springframework.stereotype.Service;
    import com.mongodb.DB;
    import com.mongodb.gridfs.GridFS;
    import com.mongodb.gridfs.GridFSDBFile;
    import com.mongodb.gridfs.GridFSInputFile;
    /**
     * 
     * 类描述：mongodb文件处理方法类
     * @author lh
     * 2018年1月25日 下午2:59:08
     */
    
    @Service
    public class MongoFileService {
    private static Logger logger = Logger.getLogger(MongoFileService.class);
    @Autowired
    private MongoTemplate mongoTemplate;
    /**
    * 上传
    * @param file 
    * @param fileName 文件名
    * @param type 表名
    * @return mongoid
    */
    public String saveFile(File file, String fileName, MongoDbTypeEnum type) {
    String id = "";
    FileInputStream fileInputStream = null;
    try {
    GridFS fs = getGridFS(type);
    GridFSInputFile mongofile = null;
    fileInputStream = new FileInputStream(file);
    mongofile = fs.createFile(fileInputStream, fileName);
    mongofile.setFilename(fileName);
    mongofile.put("lastModified", file.lastModified());
    mongofile.save();
    id = this.generalDbKey(mongofile.getId().toString(), type);
    } catch (Exception e) {
    logger.error("saveFile  error=======>" + e);
    } finally {
    if (!(null == fileInputStream)) {
    try {
    fileInputStream.close();
    } catch (IOException e) {
    e.printStackTrace();
    }
    }
    }
    return id;
    }
    
    /**
    *  下载文件到指定路径
    * @param type 表名
    * @param id 主键
    * @param Path 下载路径
    * @return
    */
    public int downloadFile(MongoDbTypeEnum type, String id, File Path) {
    //result=1代表成功，0失败
    int result=1;
    try {
    // 获取fs的根节点
    GridFS fs = getGridFS(type);
    // 解析id
    String[] key = parseDbKey(id);
    // 取objectId
    ObjectId objectId = new ObjectId(key[0]);
    // 取表名
    // String CollectionsName=key[1];
    // 通过id查询
    GridFSDBFile file = fs.find(objectId);
    // 通过文件名称查询
    // GridFSDBFile file = fs.findOne("lvxing");
    // 输出文件
    file.writeTo(Path + File.separator + file.getFilename());
    } catch (Exception e) {
    result=0;
    logger.error("downloadFile  error=======>" + e);
    }
    return result;
    }
    // 删除文件
    public int deleteFile(MongoDbTypeEnum type, String id) {
    //result=1代表成功，0失败
    int result=1;
    try {
    // 获取fs的根节点
    GridFS fs = getGridFS(type);
    // 解析id
    String[] key = parseDbKey(id);
    // 取objectId
    ObjectId objectId = new ObjectId(key[0]);
    // 取表名
    // String CollectionsName=key[1];
    // 删除文件
    fs.remove(objectId);
    } catch (Exception e) {
    result=0;
    logger.error("deleteFile  error=======>" + e);
    }
    return result;
    }
    /**
    * 获取数据库.
    * 
    * @param type
    * @return
    */
    private GridFS getGridFS(MongoDbTypeEnum type) {
    DB db = mongoTemplate.getDb();
    String dbName = db.getName();
    GridFS gridFS = new GridFS(db, type.toString());
    return gridFS;
    }
    
    /**
    * 生成标记数据库的dbkey.
    * 
    * @param objId
    * @param type
    * @return
    */
    private String generalDbKey(String objId, MongoDbTypeEnum type) {
    return objId + "-" + type;
    }
    
    /**
    * 解析dbkey.
    * 
    * @param dbKey
    * @return
    */
    private String[] parseDbKey(String dbKey) {
    return dbKey.split("-");
    }
    
    }
    
    /**
     * 
     * 类描述：MongoDb数据库分类枚举.
     * @author lh
     * 2018年1月19日 下午3:11:24
     */
    public enum MongoDbTypeEnum {
    //测试用的表
    test
    
    }

如果你是湖南的 欢迎加入 湖南人在深圳-Java群：557651502