mongodb工具类

MongodbUtils .java工具类 

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;

import javax.annotation.PostConstruct;
import java.util.List;

/**
 * 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;
    }
}

测试类MoTestController.java

import com.jugan.dao.Api;
import org.springframework.web.bind.annotation.PostMapping;
import org.springframework.web.bind.annotation.RestController;

import java.util.List;
import java.util.UUID;

@RestController
public class MoTestController {
    @PostMapping(value="/mo101")
    public void saveTest101() throws Exception {
        MongodbTestModel mo = new MongodbTestModel();
        mo.setMid("1");
        mo.setName("MongodbTestModel1");
        mo.setAge("111");
        MongodbUtils.save(mo);
    }
    @PostMapping(value="/mo102")
    public void saveTest102() throws Exception {
        MongodbTestModel mo = new MongodbTestModel();
        mo.setMid("2");
        mo.setName("MongodbTestModel2");
        mo.setAge("222");
        MongodbUtils.save(mo);
    }
    @PostMapping(value="/mo103")
    public void saveTest103() throws Exception {
        MongodbTestModel mo = new MongodbTestModel();
        mo.setMid("3");
        mo.setName("MongodbTestModel3");
        mo.setAge("333");
        MongodbUtils.save(mo);
    }
    @PostMapping(value="/mo104")
    public void saveTest104() throws Exception {
        MongodbTestModel mo = new MongodbTestModel();
        mo.setMid("4");
        mo.setName("MongodbTestModel3");
        mo.setAge("333");
        MongodbUtils.save(mo);
    }

    @PostMapping(value="/qmo101")
    public void qsaveTest101() throws Exception {
        MongodbTestModel mo = new MongodbTestModel();
        mo.setMid("3");
        mo.setName("MongodbTestModel3");
        mo.setAge("333");
        String[] findKeys = { "mid", "name" };
        String[] findValues = { "3", "MongodbTestModel3" };
        List<MongodbTestModel> t = (List<MongodbTestModel>) MongodbUtils.find(mo,findKeys,findValues);
        System.out.println("========"+ Api.entitysToJson(t));
    }
    @PostMapping(value="/qmo102")
    public void qsaveTest102() throws Exception {
        MongodbTestModel mo = new MongodbTestModel();
        mo.setMid("3");
        mo.setName("MongodbTestModel3");
        mo.setAge("333");
        String[] findKeys = {"name" };
        String[] findValues = {"MongodbTestModel3" };
        List<MongodbTestModel> t = (List<MongodbTestModel>) MongodbUtils.find(mo,findKeys,findValues);
        System.out.println("========"+ Api.entitysToJson(t));
    }
    @PostMapping(value="/qmo103")
    public void qsaveTest103() throws Exception {
        MongodbTestModel mo = new MongodbTestModel();
        mo.setMid("3");
        mo.setName("MongodbTestModel3");
        mo.setAge("333");
        String[] findKeys = {"name" };
        String[] findValues = {"MongodbTestModel4" };
        List<MongodbTestModel> t = (List<MongodbTestModel>) MongodbUtils.find(mo,findKeys,findValues);
        if (t.size()==0){
            System.out.println("========0000"+ Api.entitysToJson(t));
        }
        System.out.println("========"+ Api.entitysToJson(t));
    }


    @PostMapping(value="/mo2")
    public void saveTest2() throws Exception {
        MongodbTestModel mo = new MongodbTestModel();
        mo.setMid(UUID.randomUUID().toString());
        mo.setName("MongodbTestModel");
        mo.setAge("22");
        MongodbUtils.save(mo, "test23");
    }
    @PostMapping(value="/mo3")
        public void testRemove1() {
        MongodbTestModel mo = new MongodbTestModel();
        mo.setMid("123");
        mo.setName("MongodbTestMode");
        mo.setAge("22");
        MongodbUtils.remove(mo);
    }

    @PostMapping(value="/mo4")
    public void testRemove2() {
        MongodbTestModel mo = new MongodbTestModel();
        mo.setMid("628fdbc8-1155-497e-9c6a-1264050f9fef");
        mo.setName("MongodbTestModel");
        mo.setAge("22");
        MongodbUtils.remove(mo, "test23");
    }

    @PostMapping(value="/mo5")
    public void testRemove3() {
        MongodbUtils.removeById("_id", "123", "mongodbTestModel");
    }

    @PostMapping(value="/mo6")
    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);
    }

    @PostMapping(value="/mo7")
    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);
    }

    @PostMapping(value="/mo8")
    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);
    }

    @PostMapping(value="/mo9")
    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);
    }

    @PostMapping(value="/mo10")
    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);
    }

   @PostMapping(value="/mo11")
    public void testFind4() {
        List<? extends Object> findAll = MongodbUtils.findAll(new MongodbTestModel());
        System.out.println(findAll);
    }

    @PostMapping(value="/mo12")
    public void testFind5() {
        List<? extends Object> findAll = MongodbUtils.findAll(new MongodbTestModel(),"test23");
        System.out.println(findAll);
    }

}

相关实体类

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 + "]";
    }
}

 

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值