MongoDB增删改查API案例

MongoDB增删改查API案例

1、连接MongoDB数据库
2、创建集合
3、获取集合
4、插入文档
5、检索文档
6、删除文档

import com.mongodb.MongoClient;
import com.mongodb.client.FindIterable;
import com.mongodb.client.MongoCollection;
import com.mongodb.client.MongoCursor;
import com.mongodb.client.MongoDatabase;
import com.mongodb.client.model.Filters;
import org.bson.Document;
import org.junit.Before;
import org.junit.Test;

import java.util.ArrayList;
import java.util.List;


public class MongoClientTest {

    private MongoClient mongoClient;
    private MongoDatabase mongoDatabase;

    /**
     * 连接MongoDB数据库
     */
    @Before
    public void getConnection(){
      try{
         // 连接到 mongodb 服务
         mongoClient = new MongoClient( "localhost" , 27017 );
         // 连接到数据库
         mongoDatabase = mongoClient.getDatabase("t_member");
         System.out.println("Connect to database successfully");
      }catch(Exception e){
        System.err.println( e.getClass().getName() + ": " + e.getMessage() );
     }
   }

    /**
     * 创建集合
     */
    @Test
    public void createCollection() {
        try{
            mongoDatabase.createCollection("extra_info2");
            System.out.println("集合创建成功");
        }catch(Exception e){
            System.err.println( e.getClass().getName() + ": " + e.getMessage() );
        }
    }


    /**
     * 获取集合
     */
    @Test
    public void getCollection() {
        try{
            MongoCollection<Document> collection = mongoDatabase.getCollection("t_member");
            System.out.println("集合" + collection.getNamespace().getCollectionName() + "选择成功");
        }catch(Exception e){
            System.err.println( e.getClass().getName() + ": " + e.getMessage() );
        }
    }


    /**
     * 插入文档
     */
    @Test
    public void insertDocument() {
        try{
            MongoCollection<Document> collection = mongoDatabase.getCollection("base_info");
            System.out.println("集合选择成功");
            //插入文档
            /**
             * 1. 创建文档 org.bson.Document 参数为key-value的格式
             * 2. 创建文档集合List<Document>
             * 3. 将文档集合插入数据库集合中 mongoCollection.insertMany(List<Document>) 插入单个文档可以用 mongoCollection.insertOne(Document)
             * */
            Document document = new Document("title", "MongoDB").
                    append("description", "database").
                    append("likes", 100).
                    append("other", "Fly");
            List<Document> documents = new ArrayList<Document>();
            documents.add(document);
            collection.insertMany(documents);
            System.out.println("文档插入成功");
        }catch(Exception e){
            System.err.println( e.getClass().getName() + ": " + e.getMessage() );
        }
    }

    /**
     * 检索所有文档
     */
    @Test
    public void selectDocument() {
        try{
            MongoCollection<Document> collection = mongoDatabase.getCollection("base_info");
            System.out.println("集合选择成功");

            //检索所有文档
            /**
             * 1. 获取迭代器FindIterable<Document>
             * 2. 获取游标MongoCursor<Document>
             * 3. 通过游标遍历检索出的文档集合
             * */
            FindIterable<Document> findIterable = collection.find();
            MongoCursor<Document> mongoCursor = findIterable.iterator();
            while(mongoCursor.hasNext()){
                System.out.println(mongoCursor.next());
            }

        }catch(Exception e){
            System.err.println( e.getClass().getName() + ": " + e.getMessage() );
        }
    }

    /**
     * 更新文档
     */
    @Test
    public void updateDocument() {
        try{

            MongoCollection<Document> collection = mongoDatabase.getCollection("base_info");

            System.out.println("集合 " + collection.getNamespace().getCollectionName() + " 选择成功");

            //更新文档   将文档中likes=100的文档修改为likes=200
            collection.updateMany(Filters.eq("likes", 700), new Document("$set",new Document("likes",400)));
            //检索查看结果
            FindIterable<Document> findIterable = collection.find();
            MongoCursor<Document> mongoCursor = findIterable.iterator();
            while(mongoCursor.hasNext()){
                System.out.println(mongoCursor.next());
            }

        }catch(Exception e){
            System.err.println( e.getClass().getName() + ": " + e.getMessage() );
        }
    }

    @Test
    public void removeDocument() {
        try{

            MongoCollection<Document> collection = mongoDatabase.getCollection("base_info");
            System.out.println("集合 " + collection.getNamespace().getCollectionName() +" 选择成功");

            //删除符合条件的第一个文档
            collection.deleteOne(Filters.eq("likes", 300));
            //删除所有符合条件的文档
            collection.deleteMany (Filters.eq("likes", 400));
            //检索查看结果
            FindIterable<Document> findIterable = collection.find();
            MongoCursor<Document> mongoCursor = findIterable.iterator();
            while(mongoCursor.hasNext()){
                System.out.println(mongoCursor.next());
            }

        }catch(Exception e){
            System.err.println( e.getClass().getName() + ": " + e.getMessage() );
        }
    }

}