Mongodb批量插入数据

 
 
import com.mongodb.client.MongoCollection;
import org.bson.Document;
import org.junit.Test;
import org.junit.runner.RunWith;
import org.springframework.boot.test.SpringApplicationConfiguration;
import org.springframework.test.context.junit4.SpringJUnit4ClassRunner;

import javax.annotation.Resource;
import java.math.BigDecimal;
import java.util.ArrayList;
import java.util.Date;
import java.util.List;

/**
 * Created by Administrator on 2018/1/23. 13:07
 *
 * @Author:
 */

@RunWith(SpringJUnit4ClassRunner.class) // SpringJUnit支持,由此引入Spring-Test框架支持!
@SpringApplicationConfiguration(classes = Bootstrap.class) // 指定我们SpringBoot工程的Application启动类
public class BinlogTest {
    @Resource
    private MongoClient mongoClient;
    private static long count = 2000000;
    @Test
    public void testBatchInsert() {
        List<Document> list = new ArrayList<>();
        MongoCollection<Document> collection = mongoClient.getDatabase("test").getCollection("book");
        for (int i = 0; i < 2200000; i++){
            Document document = new Document();
            document.append("id", String.valueOf(count));
            Date date = new Date();
            document.append("add_time", date);
            document.append("title", "mongo"+count++);
            list.add(document);
            if (i != 0 && i %10000 == 0){
                System.out.println("插入" + " " + i + "本书");
                collection.insertMany(list);
                list.clear();
            }
        }


    }
}
 

猜你喜欢

转载自blog.csdn.net/y534560449/article/details/79716951