mongodb java 多线程执行

import com.mongodb.BasicDBObject;
import com.mongodb.DB;
import com.mongodb.DBCollection;
import com.mongodb.DBObject;
import com.mongodb.MongoClient;
import com.mongodb.MongoClientURI;
import com.mongodb.ReadPreference;

import java.net.UnknownHostException;
import java.util.Random;
import java.util.concurrent.ExecutorService;
import java.util.concurrent.Executors;


public class MultiThreadedExample {

    public static final int NUM_DOCUMENTS = 10000;
    public static final int NUM_THREADS = 100;

    public static void main(String[] args) throws UnknownHostException {
        MongoClientURI uri = args.length > 0 ? new MongoClientURI(args[0])
                : new MongoClientURI("mongodb://localhost");
        MongoClient mongoClient = new MongoClient(uri);

        DB db = mongoClient.getDB(uri.getDatabase() != null ? uri.getDatabase()
                : "test");

        final DBCollection collection = db.getCollection("test");
        collection.drop();

        ExecutorService executorService = Executors
                .newFixedThreadPool(NUM_THREADS);

        for (int i = 0; i < NUM_THREADS; i++) {
            executorService.submit(new Runnable() {
                @Override
                public void run() {
                    Random random = new Random();
                    while (true) {
                        int i = random.nextInt(NUM_DOCUMENTS);
                        try {
                            DBObject document = collection
                                    .find(new BasicDBObject("i", i))
                                    .setReadPreference(
                                            ReadPreference.secondaryPreferred())
                                    .one();
                            if (document == null) {
                                collection.insert(new BasicDBObject("i", i));
                            } else {
                                collection.update(new BasicDBObject("_id",
                                        document.get("_id")),
                                        new BasicDBObject("$set",
                                                new BasicDBObject("i", i + 1)));
                            }
                        } catch (Exception e) {
                            e.printStackTrace();
                        }
                    }
                }
            });
        }
    }
}
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值