com.arangodb.ArangoClient Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of arangodb-java-driver Show documentation

ArangoDB Java Driver

There is a newer version: 7.15.0

/*
 * Copyright (C) 2012 tamtam180
 *
 * 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.
 */

package com.arangodb;

import java.util.ArrayList;
import java.util.Iterator;
import java.util.List;

import com.arangodb.entity.ImportResultEntity;

/**
 * @author tamtam180 - kirscheless at gmail.com
 *
 */
public class ArangoClient {
  
  public static final int DEFAULT_IMPORT_BUFFER_SIZE = 1000;
  
  protected ArangoDriver driver;
  
  public ArangoClient(ArangoConfigure configure) {
    driver = new ArangoDriver(configure);
  }


  private void importDocumentsImpl(String collectionName, boolean createCollection, List values, ImportResultEntity total) throws ArangoException {
    ImportResultEntity result = driver.importDocuments(collectionName, createCollection, values);
    total.setCreated(total.getCreated() + result.getCreated());
    total.setErrors(total.getErrors() + result.getErrors());
    total.setEmpty(total.getEmpty() + result.getEmpty());
  }
  
  public ImportResultEntity importRawJsonDocuments(String collectionName, boolean createCollection, Iterator itr, int bufferCount) throws ArangoException {
    
    if (bufferCount <= 0) {
      bufferCount = DEFAULT_IMPORT_BUFFER_SIZE;
    }
    
    ImportResultEntity total = new ImportResultEntity();
    
    ArrayList buffers = new ArrayList(bufferCount);
    while (itr.hasNext()) {
      buffers.add(itr.next());
      if (buffers.size() % bufferCount == 0) {
        importDocumentsImpl(collectionName, createCollection, buffers, total);        
        buffers.clear();
      }
    }
    if (!buffers.isEmpty()) {
      importDocumentsImpl(collectionName, createCollection, buffers, total);        
    }
    
    return total;
    
  }
  
}