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();

}

}

}

});

}

}

}