com.arcadedb.mongo.MongoDBCollectionWrapper Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of arcadedb-mongodbw Show documentation
The newest version!
/*
 * Copyright © 2021-present Arcade Data Ltd ([email protected])
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 *
 * SPDX-FileCopyrightText: 2021-present Arcade Data Ltd ([email protected])
 * SPDX-License-Identifier: Apache-2.0
 */
package com.arcadedb.mongo;

import com.arcadedb.database.Database;
import com.arcadedb.database.MutableDocument;
import de.bwaldvogel.mongo.MongoCollection;
import de.bwaldvogel.mongo.MongoDatabase;
import de.bwaldvogel.mongo.backend.ArrayFilters;
import de.bwaldvogel.mongo.backend.Index;
import de.bwaldvogel.mongo.backend.QueryParameters;
import de.bwaldvogel.mongo.backend.QueryResult;
import de.bwaldvogel.mongo.bson.Document;
import de.bwaldvogel.mongo.bson.ObjectId;
import de.bwaldvogel.mongo.oplog.Oplog;

import java.util.*;
import java.util.stream.*;

public class MongoDBCollectionWrapper implements MongoCollection {
  private final Database database;
  //  private final int      collectionId;
  private final String   collectionName;
  private final UUID     uuid = UUID.randomUUID();

//  private static class ProjectingIterable implements Iterable {
//    private final Iterable iterable;
//    private final Document           fieldSelector;
//    private final String             idField;
//
//    ProjectingIterable(Iterable iterable, Document fieldSelector, String idField) {
//      this.iterable = iterable;
//      this.fieldSelector = fieldSelector;
//      this.idField = idField;
//    }
//
//    public Iterator iterator() {
//      return new ProjectingIterator(this.iterable.iterator(), this.fieldSelector, this.idField);
//    }
//  }
//
//  private static class ProjectingIterator implements Iterator {
//    private final Iterator iterator;
//    private final Document           fieldSelector;
//    private final String             idField;
//
//    ProjectingIterator(Iterator iterator, Document fieldSelector, String idField) {
//      this.iterator = iterator;
//      this.fieldSelector = fieldSelector;
//      this.idField = idField;
//    }
//
//    public boolean hasNext() {
//      return this.iterator.hasNext();
//    }
//
//    public Document next() {
//      Document document = this.iterator.next();
//      return MongoDBToSqlTranslator.projectDocument(document, this.fieldSelector, this.idField);
//    }
//
//    public void remove() {
//      this.iterator.remove();
//    }
//  }

  protected MongoDBCollectionWrapper(final Database database, final String collectionName) {
    this.database = database;
    this.collectionName = collectionName;
    //this.collectionId = database.getSchema().getType(collectionName).getBuckets(false).get(0).getId();
  }

//  protected Document getDocument(final Long aLong) {
//    final com.arcadedb.database.Document record = (com.arcadedb.database.Document) database.lookupByRID(new RID(database, collectionId, aLong), true);
//
//    final Document result = new Document();
//
//    for (String p : record.getPropertyNames())
//      result.put(p, record.get(p));
//
//    return result;
//  }

  @Override
  public UUID getUuid() {
    return uuid;
  }

  @Override
  public MongoDatabase getDatabase() {
    return null;
  }

  @Override
  public String getDatabaseName() {
    return database.getName();
  }

  @Override
  public String getFullName() {
    return null;
  }

  @Override
  public String getCollectionName() {
    return collectionName;
  }

  @Override
  public void addIndex(final Index index) {
    // TODO
  }

  @Override
  public void dropIndex(final String s) {
    // TODO

  }

  @Override
  public void renameTo(final MongoDatabase mongoDatabase, final String s) {
    // TODO
  }

  @Override
  public void addDocument(final Document document) {
    // TODO
  }

  @Override
  public void addDocuments(Stream documents) {
    MongoCollection.super.addDocuments(documents);
  }

  @Override
  public void removeDocument(final Document document) {
    // TODO
  }

  @Override
  public void addDocumentIfMissing(Document document) {
    MongoCollection.super.addDocumentIfMissing(document);
  }

  @Override
  public Iterable queryAll() {
    return MongoCollection.super.queryAll();
  }

  @Override
  public Stream queryAllAsStream() {
    return MongoCollection.super.queryAllAsStream();
  }

  @Override
  public Iterable handleQuery(Document query) {
    return MongoCollection.super.handleQuery(query);
  }

  @Override
  public Stream handleQueryAsStream(Document query) {
    return MongoCollection.super.handleQueryAsStream(query);
  }

  @Override
  public QueryResult handleQuery(Document query, int numberToSkip, int limit) {
    return MongoCollection.super.handleQuery(query, numberToSkip, limit);
  }

  @Override
  public QueryResult handleQuery(final QueryParameters queryParameters) {
    int numberToReturn = queryParameters.getLimit();
    if (numberToReturn < 0)
      numberToReturn = -numberToReturn;

    final int numberToSkip = queryParameters.getNumberToSkip();

    final Document queryObject = queryParameters.getQuerySelector();

    Document query = null;
    Document orderBy = null;
    if (queryObject != null) {
      if (queryObject.containsKey("query")) {
        query = (Document) queryObject.get("query");
      } else if (queryObject.containsKey("$query")) {
        query = (Document) queryObject.get("$query");
      } else {
        query = queryObject;
      }

      orderBy = (Document) queryObject.remove("$orderBy");
    }

    if (this.count() == 0)
      return new QueryResult();

    final Iterable objs = this.queryDocuments(query, orderBy, numberToSkip, numberToReturn);

    return new QueryResult(objs);
  }

  @Override
  public void insertDocuments(final List list) {
    database.begin();

    for (final Document d : list) {
      final MutableDocument record = database.newDocument(collectionName);

      for (final Map.Entry p : d.entrySet()) {
        final Object value = p.getValue();
        if (value instanceof ObjectId) {
          final byte[] var2 = ((ObjectId) value).toByteArray();
          final int var3 = var2.length;

          final StringBuilder s = new StringBuilder();
          for (int var4 = 0; var4 < var3; ++var4) {
            final byte b = var2[var4];
            s.append(String.format("%02x", b));
          }

          record.set(p.getKey(), s.toString());
        } else
          record.set(p.getKey(), value);
      }

      record.save();
    }

    database.commit();
  }

  @Override
  public List insertDocuments(final List list, final boolean b) {
    return null;
  }

  @Override
  public Document updateDocuments(final Document document, final Document document1, final ArrayFilters filters, final boolean b, final boolean b1,
      final Oplog opLog) {
    return null;
  }

  @Override
  public int deleteDocuments(final Document document, final int limit) {
    return 0;
  }

  @Override
  public int deleteDocuments(final Document document, final int i, final Oplog oplog) {
    return 0;
  }

  @Override
  public Document handleDistinct(final Document document) {
    return null;
  }

  @Override
  public Document getStats() {
    return null;
  }

  @Override
  public Document validate() {
    throw new UnsupportedOperationException();
  }

  @Override
  public Document findAndModify(final Document document) {
    return null;
  }

  @Override
  public int count(final Document document, final int i, final int i1) {
    return (int) database.countType(collectionName, false);
  }

  @Override
  public boolean isEmpty() {
    return MongoCollection.super.isEmpty();
  }

  @Override
  public int count() {
    return (int) database.countType(getCollectionName(), false);
  }

  @Override
  public int getNumIndexes() {
    return 0;
  }

  @Override
  public List> getIndexes() {
    return null;
  }

  @Override
  public void drop() {
    database.getSchema().dropType(collectionName);
  }

  private Iterable queryDocuments(final Document query, final Document orderBy, final int numberToSkip, final int numberToReturn) {
    final List result = new ArrayList<>();

    final Iterator it;

    if (query == null || query.isEmpty()) {
      // SCAN
      it = database.iterateType(collectionName, false);
    } else {
      // EXECUTE A SQL QUERY
      final StringBuilder sql = new StringBuilder("select from " + collectionName + " where ");

      MongoDBToSqlTranslator.buildExpression(sql, query);

      if (orderBy != null) {
        sql.append(" order by ");
        int i = 0;
        for (final String p : orderBy.keySet()) {
          if (i > 0)
            sql.append(", ");
          sql.append(p);
          sql.append(' ');
          sql.append(((Number) orderBy.get(p)).intValue() == 1 ? "asc" : "desc");
          ++i;
        }
      }

      it = database.query("SQL", sql.toString());
    }

    MongoDBToSqlTranslator.fillResultSet(numberToSkip, numberToReturn, result, it);

    return result;
  }
}