使用GridFsTemplate在Mongo中存取文件

 

Maven依赖（还有一些springboot需要的）

    <parent>
        <groupId>org.springframework.boot</groupId> <artifactId>spring-boot-starter-parent</artifactId> <version>1.5.6.RELEASE</version> </parent> <properties> <java.version>1.8</java.version> </properties> <dependencies> <dependency> <groupId>org.springframework.boot</groupId> <artifactId>spring-boot-starter-data-mongodb</artifactId> </dependency> <dependency> <groupId>org.springframework.boot</groupId> <artifactId>spring-boot-starter-test</artifactId> <scope>test</scope> </dependency> </dependencies> <build> <plugins> <plugin> <groupId>org.springframework.boot</groupId> <artifactId>spring-boot-maven-plugin</artifactId> </plugin> </plugins> </build> <repositories> <repository> <id>spring-releases</id> <name>Spring Releases</name> <url>https://repo.spring.io/libs-release</url> </repository> </repositories> <pluginRepositories> <pluginRepository> <id>spring-releases</id> <name>Spring Releases</name> <url>https://repo.spring.io/libs-release</url> </pluginRepository> </pluginRepositories> 

配置GridFsTemplate

import com.mongodb.Mongo;
import com.mongodb.MongoClient;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration; import org.springframework.context.annotation.PropertySource; import org.springframework.core.env.Environment; import org.springframework.data.mongodb.MongoDbFactory; import org.springframework.data.mongodb.config.AbstractMongoConfiguration; import org.springframework.data.mongodb.core.SimpleMongoDbFactory; import org.springframework.data.mongodb.gridfs.GridFsTemplate; import org.springframework.data.mongodb.repository.config.EnableMongoRepositories; @Configuration @EnableMongoRepositories // 读取配置文件的，通过Environment读取 @PropertySource("classpath:mongo.yml") public class GridFsConfiguration extends AbstractMongoConfiguration { @Autowired Environment env; @Bean public GridFsTemplate gridFsTemplate() throws Exception { return new GridFsTemplate(mongoDbFactory(), mappingMongoConverter()); } @Override protected String getDatabaseName() { return env.getProperty("database"); } @Override public Mongo mongo() throws Exception { return new MongoClient(env.getProperty("host")); } @Bean public MongoDbFactory mongoDbFactory() throws Exception { return new SimpleMongoDbFactory(new MongoClient(env.getProperty("host"), env.getProperty("port", Integer.class)), getDatabaseName()); } } 

mongo.yml

mongo:
  host: 127.0.0.1
  port: 27017 database: filecenter 

使用GridFsTemplate存取文件

import com.mongodb.gridfs.GridFSDBFile; import org.junit.Test; import org.junit.runner.RunWith; import org.springframework.beans.factory.annotation.Autowired; import org.springframework.boot.test.context.SpringBootTest; import org.springframework.core.io.ClassPathResource; import org.springframework.core.io.Resource; import org.springframework.data.mongodb.gridfs.GridFsResource; import org.springframework.data.mongodb.gridfs.GridFsTemplate; import org.springframework.test.context.junit4.SpringRunner; import java.io.File; import java.io.IOException; import java.util.List; import static org.springframework.data.mongodb.core.query.Query.query; import static org.springframework.data.mongodb.gridfs.GridFsCriteria.whereFilename; @RunWith(SpringRunner.class) @SpringBootTest(classes = GridFsConfiguration.class) public class MongoTest { @Autowired GridFsTemplate gridFsTemplate; // 存文件 @Test public void storeFileInGridFs() { Resource file = new ClassPathResource("mongo.yml"); // Resource file = new FileSystemResource("C:\\Users\\Chenggaowei\\Downloads\\Adblock.crx"); try { gridFsTemplate.store(file.getInputStream(), file.getFilename(), "yml"); } catch (IOException e) { e.printStackTrace(); } } // 下载文件 @Test public void findFilesInGridFs() throws IOException { List<GridFSDBFile> result = gridFsTemplate.find(query(whereFilename().is("mongo.yml"))); GridFSDBFile gridFSDBFile = result.get(0); gridFSDBFile.writeTo(new File("C:\\Users\\Chenggaowei\\Downloads\\" + gridFSDBFile.getFilename())); } // 所有文件 @Test public void readFilesFromGridFs() { GridFsResource[] txtFiles = gridFsTemplate.getResources("*"); for (GridFsResource txtFile : txtFiles) { System.out.println(txtFile.getFilename()); } } } 

参考文献

https://docs.spring.io/spring-data/mongodb/docs/current/reference/html/#gridfs