com.mongodb.FongoMongoCollection Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of fongo Show documentation
Fake in-memory mongo
The newest version!
package com.mongodb;

import com.github.fakemongo.Fongo;
import com.mongodb.client.MongoCollection;
import com.mongodb.client.model.CountOptions;
import com.mongodb.client.model.IndexModel;
import com.mongodb.client.model.UpdateOptions;
import com.mongodb.client.result.UpdateResult;
import com.mongodb.util.FongoJSON;
import java.util.ArrayList;
import java.util.List;
import org.bson.Document;
import org.bson.codecs.configuration.CodecRegistry;
import org.bson.conversions.Bson;

/**
 *
 */
public class FongoMongoCollection extends MongoCollectionImpl {

  private final Fongo fongo;

  private final DBCollection dbCollection;

  FongoMongoCollection(Fongo fongo, MongoNamespace namespace, Class tDocumentClass, CodecRegistry codecRegistry, ReadPreference readPreference, WriteConcern writeConcern, ReadConcern readConcern) {
    super(namespace, tDocumentClass, codecRegistry, readPreference, writeConcern, false, readConcern, new FongoOperationExecutor(fongo));
    this.fongo = fongo;
    this.dbCollection = fongo.getDB(namespace.getDatabaseName()).getCollection(namespace.getCollectionName());
  }

  @Override
  public  MongoCollection withDocumentClass(Class clazz) {
    return new FongoMongoCollection(this.fongo, super.getNamespace(), clazz, super.getCodecRegistry(), super.getReadPreference(), super.getWriteConcern(), super.getReadConcern());
  }

  @Override
  public MongoCollection withCodecRegistry(CodecRegistry codecRegistry) {
    return new FongoMongoCollection(this.fongo, super.getNamespace(), super.getDocumentClass(), codecRegistry, super.getReadPreference(), super.getWriteConcern(), super.getReadConcern());
  }

  @Override
  public MongoCollection withReadPreference(ReadPreference readPreference) {
    return new FongoMongoCollection(this.fongo, super.getNamespace(), super.getDocumentClass(), super.getCodecRegistry(), readPreference, super.getWriteConcern(), super.getReadConcern());
  }

  @Override
  public MongoCollection withWriteConcern(WriteConcern writeConcern) {
    return new FongoMongoCollection(this.fongo, super.getNamespace(), super.getDocumentClass(), super.getCodecRegistry(), super.getReadPreference(), writeConcern, super.getReadConcern());
  }

  @Override
  public MongoCollection withReadConcern(ReadConcern readConcern) {
    return new FongoMongoCollection(this.fongo, super.getNamespace(), super.getDocumentClass(), super.getCodecRegistry(), super.getReadPreference(), super.getWriteConcern(), readConcern);
  }

  @Override
  public long count(Bson filter, CountOptions options) {
    final DBObject query = dbObject(filter);
    final int limit = options.getLimit();
    final int skip = options.getSkip();

    return dbCollection.getCount(query, null, limit, skip);
  }

  //
//  @Override
//  public  DistinctIterable distinct(String fieldName, Class tResultClass) {
//    return new DistinctIterableImpl(getNamespace(), getDocumentClass(), tResultClass, getCodecRegistry(), getReadPreference(), fongo, fieldName){
//      @Override
//      public MongoCursor iterator() {
//        return super.iterator();
//      }
//    };
//  }


  @Override
  public List createIndexes(List indexes) {
    ArrayList names = new ArrayList(indexes.size());
    for (IndexModel indexModel : indexes) {
      this.dbCollection.createIndex(dbObject(indexModel.getKeys()), indexModel.getOptions().getName(), indexModel.getOptions().isUnique());
      names.add(indexModel.getOptions().getName());
    }
//    return super.createIndexes(indexes);
    return names;
  }

  private DBObject dbObject(Bson bson) {
    if (bson == null) {
      return null;
    }
    // TODO Performance killer
    return (DBObject) FongoJSON.parse(bson.toBsonDocument(Document.class, super.getCodecRegistry()).toString());
  }
}