- jar包依赖
pom.xml:
<!-- mongodb -->
<dependency>
<groupId>org.springframework.boot</groupId>
<artifactId>spring-boot-starter-data-mongodb</artifactId>
</dependency>
- application.yml配置:
spring:
data:
mongodb:
database: test
host: 127.0.0.1
port: 27017
logging:
level:
org.springframework.data.mongodb.core.MongoTemplate: DEBUG #配置MongoTemplate日志
- MongoTemplate工具类:
MongodbUtils.java
import java.util.List;
import javax.annotation.PostConstruct;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.data.domain.Sort;
import org.springframework.data.domain.Sort.Direction;
import org.springframework.data.mongodb.core.MongoTemplate;
import org.springframework.data.mongodb.core.query.Criteria;
import org.springframework.data.mongodb.core.query.Query;
import org.springframework.data.mongodb.core.query.Update;
import org.springframework.stereotype.Component;
/**
* mongodb curd 工具类
*
* @author ttan
*
*/
@Component
public class MongodbUtils {
public static MongodbUtils mongodbUtils;
@PostConstruct
public void init() {
mongodbUtils = this;
mongodbUtils.mongoTemplate = this.mongoTemplate;
}
@Autowired
private MongoTemplate mongoTemplate;
/**
* 保存数据对象,集合为数据对象中@Document 注解所配置的collection
*
* @param obj
* 数据对象
*/
public static void save(Object obj) {
mongodbUtils.mongoTemplate.save(obj);
}
/**
* 指定集合保存数据对象
*
* @param obj
* 数据对象
* @param collectionName
* 集合名
*/
public static void save(Object obj, String collectionName) {
mongodbUtils.mongoTemplate.save(obj, collectionName);
}
/**
* 根据数据对象中的id删除数据,集合为数据对象中@Document 注解所配置的collection
*
* @param obj
* 数据对象
*/
public static void remove(Object obj) {
mongodbUtils.mongoTemplate.remove(obj);
}
/**
* 指定集合 根据数据对象中的id删除数据
*
* @param obj
* 数据对象
* @param collectionName
* 集合名
*/
public static void remove(Object obj, String collectionName) {
mongodbUtils.mongoTemplate.remove(obj, collectionName);
}
/**
* 根据key,value到指定集合删除数据
*
* @param key
* 键
* @param value
* 值
* @param collectionName
* 集合名
*/
public static void removeById(String key, Object value, String collectionName) {
Criteria criteria = Criteria.where(key).is(value);
criteria.and(key).is(value);
Query query = Query.query(criteria);
mongodbUtils.mongoTemplate.remove(query, collectionName);
}
/**
* 指定集合 修改数据,且仅修改找到的第一条数据
*
* @param accordingKey
* 修改条件 key
* @param accordingValue
* 修改条件 value
* @param updateKeys
* 修改内容 key数组
* @param updateValues
* 修改内容 value数组
* @param collectionName
* 集合名
*/
public static void updateFirst(String accordingKey, Object accordingValue, String[] updateKeys, Object[] updateValues,
String collectionName) {
Criteria criteria = Criteria.where(accordingKey).is(accordingValue);
Query query = Query.query(criteria);
Update update = new Update();
for (int i = 0; i < updateKeys.length; i++) {
update.set(updateKeys[i], updateValues[i]);
}
mongodbUtils.mongoTemplate.updateFirst(query, update, collectionName);
}
/**
* 指定集合 修改数据,且修改所找到的所有数据
*
* @param accordingKey
* 修改条件 key
* @param accordingValue
* 修改条件 value
* @param updateKeys
* 修改内容 key数组
* @param updateValues
* 修改内容 value数组
* @param collectionName
* 集合名
*/
public static void updateMulti(String accordingKey, Object accordingValue, String[] updateKeys, Object[] updateValues,
String collectionName) {
Criteria criteria = Criteria.where(accordingKey).is(accordingValue);
Query query = Query.query(criteria);
Update update = new Update();
for (int i = 0; i < updateKeys.length; i++) {
update.set(updateKeys[i], updateValues[i]);
}
mongodbUtils.mongoTemplate.updateMulti(query, update, collectionName);
}
/**
* 根据条件查询出所有结果集 集合为数据对象中@Document 注解所配置的collection
*
* @param obj
* 数据对象
* @param findKeys
* 查询条件 key
* @param findValues
* 查询条件 value
* @return
*/
public static List<? extends Object> find(Object obj, String[] findKeys, Object[] findValues) {
Criteria criteria = null;
for (int i = 0; i < findKeys.length; i++) {
if (i == 0) {
criteria = Criteria.where(findKeys[i]).is(findValues[i]);
} else {
criteria.and(findKeys[i]).is(findValues[i]);
}
}
Query query = Query.query(criteria);
List<? extends Object> resultList = mongodbUtils.mongoTemplate.find(query, obj.getClass());
return resultList;
}
/**
* 指定集合 根据条件查询出所有结果集
*
* @param obj
* 数据对象
* @param findKeys
* 查询条件 key
* @param findValues
* 查询条件 value
* @param collectionName
* 集合名
* @return
*/
public static List<? extends Object> find(Object obj, String[] findKeys, Object[] findValues, String collectionName) {
Criteria criteria = null;
for (int i = 0; i < findKeys.length; i++) {
if (i == 0) {
criteria = Criteria.where(findKeys[i]).is(findValues[i]);
} else {
criteria.and(findKeys[i]).is(findValues[i]);
}
}
Query query = Query.query(criteria);
List<? extends Object> resultList = mongodbUtils.mongoTemplate.find(query, obj.getClass(), collectionName);
return resultList;
}
/**
* 指定集合 根据条件查询出所有结果集 并排倒序
*
* @param obj
* 数据对象
* @param findKeys
* 查询条件 key
* @param findValues
* 查询条件 value
* @param collectionName
* 集合名
* @param sort
* 排序字段
* @return
*/
public static List<? extends Object> find(Object obj, String[] findKeys, Object[] findValues, String collectionName ,String sort) {
Criteria criteria = null;
for (int i = 0; i < findKeys.length; i++) {
if (i == 0) {
criteria = Criteria.where(findKeys[i]).is(findValues[i]);
} else {
criteria.and(findKeys[i]).is(findValues[i]);
}
}
Query query = Query.query(criteria);
query.with(new Sort(Direction.DESC, sort));
List<? extends Object> resultList = mongodbUtils.mongoTemplate.find(query, obj.getClass(), collectionName);
return resultList;
}
/**
* 根据条件查询出符合的第一条数据 集合为数据对象中 @Document 注解所配置的collection
*
* @param obj
* 数据对象
* @param findKeys
* 查询条件 key
* @param findValues
* 查询条件 value
* @return
*/
public static Object findOne(Object obj, String[] findKeys, Object[] findValues) {
Criteria criteria = null;
for (int i = 0; i < findKeys.length; i++) {
if (i == 0) {
criteria = Criteria.where(findKeys[i]).is(findValues[i]);
} else {
criteria.and(findKeys[i]).is(findValues[i]);
}
}
Query query = Query.query(criteria);
Object resultObj = mongodbUtils.mongoTemplate.findOne(query, obj.getClass());
return resultObj;
}
/**
* 指定集合 根据条件查询出符合的第一条数据
*
* @param obj
* 数据对象
* @param findKeys
* 查询条件 key
* @param findValues
* 查询条件 value
* @param collectionName
* 集合名
* @return
*/
public static Object findOne(Object obj, String[] findKeys, Object[] findValues, String collectionName) {
Criteria criteria = null;
for (int i = 0; i < findKeys.length; i++) {
if (i == 0) {
criteria = Criteria.where(findKeys[i]).is(findValues[i]);
} else {
criteria.and(findKeys[i]).is(findValues[i]);
}
}
Query query = Query.query(criteria);
Object resultObj = mongodbUtils.mongoTemplate.findOne(query, obj.getClass(), collectionName);
return resultObj;
}
/**
* 查询出所有结果集 集合为数据对象中 @Document 注解所配置的collection
*
* @param obj
* 数据对象
* @return
*/
public static List<? extends Object> findAll(Object obj) {
List<? extends Object> resultList = mongodbUtils.mongoTemplate.findAll(obj.getClass());
return resultList;
}
/**
* 指定集合 查询出所有结果集
*
* @param obj
* 数据对象
* @param collectionName
* 集合名
* @return
*/
public static List<? extends Object> findAll(Object obj, String collectionName) {
List<? extends Object> resultList = mongodbUtils.mongoTemplate.findAll(obj.getClass(), collectionName);
return resultList;
}
}
- 测试类:
MongodbTest.java
import java.util.List;
import java.util.UUID;
import org.junit.Test;
import org.junit.runner.RunWith;
import org.springframework.boot.test.context.SpringBootTest;
import org.springframework.test.context.junit4.SpringRunner;
import com.hwsoft.invoicecloud.common.mongodb.MongodbUtils;
import com.hwsoft.invoicecloud.model.mongodb.MongodbTestModel;
@RunWith(SpringRunner.class)
@SpringBootTest
public class MongodbTest {
@Test
public void testSave1() {
MongodbTestModel mo = new MongodbTestModel();
mo.setMid("123");
mo.setName("MongodbTestModel");
mo.setAge("22");
MongodbUtils.save(mo);
}
@Test
public void testSave2() {
MongodbTestModel mo = new MongodbTestModel();
mo.setMid(UUID.randomUUID().toString());
mo.setName("MongodbTestModel");
mo.setAge("22");
MongodbUtils.save(mo, "test23");
}
@Test
public void testRemove1() {
MongodbTestModel mo = new MongodbTestModel();
mo.setMid("5eb1091b-b2a8-4a92-b1a3-4d515e837823");
mo.setName("MongodbTestModel");
mo.setAge("22");
MongodbUtils.remove(mo);
}
@Test
public void testRemove2() {
MongodbTestModel mo = new MongodbTestModel();
mo.setMid("a27a0652-a0ed-4d7d-a100-d5f11fe1b844");
mo.setName("MongodbTestModel");
mo.setAge("22");
MongodbUtils.remove(mo, "test23");
}
@Test
public void testRemove3() {
MongodbUtils.removeById("_id", "123", "mongodbTestModel");
}
@Test
public void testUpdate1() {
String accordingKey = "_id";
String accordingValue = "e933f48a-cc68-4993-9069-820ff3adab34";
String[] updateKeys = { "name", "age" };
Object[] updateValues = { "tat", "22222" };
String collectionName = "test23";
MongodbUtils.updateFirst(accordingKey, accordingValue, updateKeys, updateValues, collectionName);
}
@Test
public void testUpdate2() {
String accordingKey = "name";
String accordingValue = "tat";
String[] updateKeys = { "age" };
Object[] updateValues = { "000000" };
String collectionName = "test23";
MongodbUtils.updateMulti(accordingKey, accordingValue, updateKeys, updateValues, collectionName);
}
@Test
public void testFind1() {
MongodbTestModel obj = new MongodbTestModel();
String[] findKeys = { "age", "name" };
String[] findValues = { "22", "MongodbTestModel" };
List<? extends Object> find = MongodbUtils.find(obj, findKeys, findValues);
System.out.println(find);
}
@Test
public void testFind2() {
MongodbTestModel obj = new MongodbTestModel();
String[] findKeys = { "name" };
String[] findValues = { "tat" };
String collectionName = "test23";
List<? extends Object> find = MongodbUtils.find(obj, findKeys, findValues, collectionName);
System.out.println(find);
}
@Test
public void testFind3() {
MongodbTestModel obj = new MongodbTestModel();
String[] findKeys = { "name" };
String[] findValues = { "tat" };
String collectionName = "test23";
Object findOne = MongodbUtils.findOne(obj, findKeys, findValues, collectionName);
System.out.println(findOne);
}
@Test
public void testFind4() {
List<? extends Object> findAll = MongodbUtils.findAll(new MongodbTestModel());
System.out.println(findAll);
}
@Test
public void testFind5() {
List<? extends Object> findAll = MongodbUtils.findAll(new MongodbTestModel(),"test23");
System.out.println(findAll);
}
}
- 测试实体类:
MongodbTestModel.java
package com.hwsoft.invoicecloud.model.mongodb;
import org.springframework.data.annotation.Id;
import org.springframework.data.mongodb.core.mapping.Document;
//@Document(collection="mongodbTestModel") //collection:集合名 类似于关系型数据里面的 tableName
@Document(collection="mongodbTestModel")
public class MongodbTestModel {
@Id
private String mid;
private String name;
private String age;
public String getMid() {
return mid;
}
public void setMid(String mid) {
this.mid = mid;
}
public String getName() {
return name;
}
public void setName(String name) {
this.name = name;
}
public String getAge() {
return age;
}
public void setAge(String age) {
this.age = age;
}
@Override
public String toString() {
return "MongodbTestModel [mid=" + mid + ", name=" + name + ", age=" + age + "]";
}
}
由于MongoTemplate方法众多,工具类可以有很多扩展,在此仅使用到了一些常用的api。