4rfc 发表于 2014-1-16 08:59:07

MongoDB Java 连接

Java 连接MongoDB
package mymaven;

import java.net.UnknownHostException;
import java.util.Set;

import com.mongodb.BasicDBObject;
import com.mongodb.DB;
import com.mongodb.DBCollection;
import com.mongodb.DBCursor;
import com.mongodb.DBObject;
import com.mongodb.Mongo;
import com.mongodb.MongoException;

public class Test {
    /**
   * @author ithomer.net
   * @datetime 2013-11-11
   */
    public static void main(String[] args) throws UnknownHostException, MongoException {
      Mongo mongo = new Mongo("172.27.9.104", 27017);   // 连接数据库
      DB db = mongo.getDB("mytestdb");                  // 数据库
      Set<String> cols = db.getCollectionNames();         // 获取数据库中所有的集合(类似于获取关系数据库中的表)

      // 打印出数据库中的集合,这里应当为null
      for (String s : cols) {
            System.out.println(s);
      }

      DBCollection collection = db.getCollection("mytestcoll");       // 创建一个集合
      collection.drop();                                              // 删除集合,插入数据时自动重建
      BasicDBObject obj = new BasicDBObject();                        // 初始化一个基本DB对象,插入数据库的就是DB对象

      obj.put("from", "blog.ithomer.net");      // 放入几个键值对
      obj.put("to", "forum.ithomer.net");
      obj.put("subject", "ithomer.net");
      collection.insert(obj);                     // 插入对象

      DBObject dbobj = collection.findOne();      // 查看一条记录,findOne()=find().limit(1);
      System.out.println(dbobj);                  // 打印出刚才插入的数据

      // 插入10条{ranking:i}的数据
      for (int i = 0; i < 10; i++) {
            collection.insert(new BasicDBObject().append("ranking", i));
      }
      System.out.println("count: " + collection.getCount());      // 打印集合中的数据总数


      DBCursor cursor = collection.find();      // 然后我们使用这个游标来遍历集合
      while (cursor.hasNext()) {
            System.out.println(cursor.next());
      }
      // 简单的条件查询,查询ranking为1的记录
      BasicDBObject query = new BasicDBObject();
      query.put("ranking", 1);
      cursor = collection.find(query);
      System.out.println("collection find({\"ranking\":1}):");
      while (cursor.hasNext()) {
            System.out.println(cursor.next());
      }

      // 复杂的条件查询,查询ranking大于等于5小于9的记录
      query = new BasicDBObject();
      query.put("ranking", new BasicDBObject("$gte", 5).append("$lt", 9));
      cursor = collection.find(query);
      System.out.println("collection find({\"ranking\":[5-9)}):");
      while (cursor.hasNext()) {
            System.out.println(cursor.next());
      }

//      mongo.dropDatabase("mytestdb");         // 最后删除我们的测试数据库
    }
}

运行结果:
mytestcoll
system.indexes
{ "_id" : { "$oid" : "52c62ed8e4b0f4de3dd10107"} , "from" : "blog.ithomer.net" , "to" : "forum.ithomer.net" , "subject" : "ithomer.net"}
count: 11
{ "_id" : { "$oid" : "52c62ed8e4b0f4de3dd10107"} , "from" : "blog.ithomer.net" , "to" : "forum.ithomer.net" , "subject" : "ithomer.net"}
{ "_id" : { "$oid" : "52c62ed8e4b0f4de3dd10108"} , "ranking" : 0}
{ "_id" : { "$oid" : "52c62ed8e4b0f4de3dd10109"} , "ranking" : 1}
{ "_id" : { "$oid" : "52c62ed8e4b0f4de3dd1010a"} , "ranking" : 2}
{ "_id" : { "$oid" : "52c62ed8e4b0f4de3dd1010b"} , "ranking" : 3}
{ "_id" : { "$oid" : "52c62ed8e4b0f4de3dd1010c"} , "ranking" : 4}
{ "_id" : { "$oid" : "52c62ed8e4b0f4de3dd1010d"} , "ranking" : 5}
{ "_id" : { "$oid" : "52c62ed8e4b0f4de3dd1010e"} , "ranking" : 6}
{ "_id" : { "$oid" : "52c62ed8e4b0f4de3dd1010f"} , "ranking" : 7}
{ "_id" : { "$oid" : "52c62ed8e4b0f4de3dd10110"} , "ranking" : 8}
{ "_id" : { "$oid" : "52c62ed8e4b0f4de3dd10111"} , "ranking" : 9}
collection find({"ranking":1}):
{ "_id" : { "$oid" : "52c62ed8e4b0f4de3dd10109"} , "ranking" : 1}
collection find({"ranking":[5-9)}):
{ "_id" : { "$oid" : "52c62ed8e4b0f4de3dd1010d"} , "ranking" : 5}
{ "_id" : { "$oid" : "52c62ed8e4b0f4de3dd1010e"} , "ranking" : 6}
{ "_id" : { "$oid" : "52c62ed8e4b0f4de3dd1010f"} , "ranking" : 7}
{ "_id" : { "$oid" : "52c62ed8e4b0f4de3dd10110"} , "ranking" : 8}
源码下载(maven)


mongodb数组示例:
@SuppressWarnings("unchecked")
public static void loadMediaTags(List<MediaEntity> mediaEntityList) {
    mediaEntityList.clear();

    try {
      Mongo mongo = new Mongo(CosineCluster.gMongo_HOST, CosineCluster.gMongo_PORT);
      DB db = mongo.getDB(CosineCluster.gMongo_DB);
      DBCollection collection = db.getCollection(CosineCluster.gMongo_Coll_Media);

      DBCursor cursor = collection.find();
      int index = 0;
      long startTime = System.currentTimeMillis();
      while(cursor.hasNext()){
            BasicDBObject obj = (BasicDBObject) cursor.next();

            long id = obj.getLong("_id");
            ArrayList<String> tagList = (ArrayList<String>) obj.get("tag");               // tag
            ArrayList<String> keywordList = (ArrayList<String>)obj.get("keyword");          // keyword
            ArrayList<Integer> copyrightList = (ArrayList<Integer>)obj.get("copyright");    // copyright


            MediaEntity mediaEntity = new MediaEntity();

            mediaEntity.setId(id);

            // tag
            for(int j=0; j<tagList.size(); j++) {
                mediaEntity.addTag(tagList.get(j));
                int tagId = getTagInt(tagList.get(j));
                mediaEntity.addTag(tagId);
            }

            // actors
            ArrayList<DBObject> actorsObjList = (ArrayList<DBObject>)obj.get("actors");   // actors
            for(int j=0; j<actorsObjList.size(); j++) {
               mediaEntity.addActor((String) actorsObjList.get(j).get("name"));   
               int actorId = getActorInt((String)actorsObjList.get(j).get("name"));
               mediaEntity.addActor(actorId);
            }

            // director
            ArrayList<DBObject> directorObjList = (ArrayList<DBObject>)obj.get("director"); // director
            for(int j=0; j<directorObjList.size(); j++) {
               mediaEntity.addDirector((String) directorObjList.get(j).get("name"));   
               int directorId = getDirectorInt((String) directorObjList.get(j).get("name"));
               mediaEntity.addDirector(directorId);
            }

            // keyword
            for(int j=0; j<keywordList.size(); j++) {
                mediaEntity.addKeyword(keywordList.get(j));
                int keywordId = getKeywordInt(keywordList.get(j));
                mediaEntity.addKeyword(keywordId);
            }

            // copyright
            for(int j=0; j<copyrightList.size(); j++) {
                mediaEntity.addCopyright(copyrightList.get(j));
            }

            mediaEntityList.add(mediaEntity);

            index++;
            if(index > 100) {
                break;
            }
            System.out.println(index + " --- mediaEntity : " + mediaEntity.toString());
      }
      long costTime = System.currentTimeMillis() - startTime;
      System.out.println("load data costTime = " + index + "; costTime = " + costTime/1000f);

    } catch (Exception e) {
      e.printStackTrace();
    }
}

public static int getTagInt(String tag) {
    int tagIntId = -1;

    try {
      MongoClient mongo = new MongoClient(CosineCluster.gMongo_HOST, CosineCluster.gMongo_PORT);
      DB db = mongo.getDB(CosineCluster.gMongo_DB);
      DBCollection tagmapCollection = db.getCollection("recommend_tag_map");

      DBCursor cursor = tagmapCollection.find(new BasicDBObject("name", tag));      
      if(cursor == null || cursor.toArray().size() <= 0) {   // 处理小于2或n的映射关键字,下同
            return tagIntId;
      }
      DBObject obj = cursor.toArray().get(0);


      String name = tag;
      tagIntId = (Integer) obj.get("id");
      int num = (Integer) obj.get("num");

      mongo.close();
    } catch (UnknownHostException e) {
      e.printStackTrace();
    }

    return tagIntId;
}

public static int getActorInt(String actor) {
    int actorIntId = -1;
    try {
      MongoClient mongo = new MongoClient(CosineCluster.gMongo_HOST, CosineCluster.gMongo_PORT);
      DB db = mongo.getDB(CosineCluster.gMongo_DB);
      DBCollection tagmapCollection = db.getCollection("recommend_actor_map");

      DBCursor cursor = tagmapCollection.find(new BasicDBObject("name", actor));
      if(cursor == null || cursor.toArray().size() <= 0) {
            return actorIntId;
      }
      DBObject obj = cursor.toArray().get(0);

      String name = actor;
      actorIntId = (Integer) obj.get("id");
      int num = (Integer) obj.get("num");

      mongo.close();
    } catch (UnknownHostException e) {
      e.printStackTrace();
    }

    return actorIntId;
}


页: [1]
查看完整版本: MongoDB Java 连接