Help on importDocuments api method about arangodb-java-driver HOT 15 CLOSED

db._create('test')
db.test.save({'_key': 'foo' , 'bla': 'bar'})

db.test.save({'_key': 'foo' , 'bla': 'bar'})

db._query('UPDATE "foo" WITH { blarg: "blub"} IN test')

db.test.save({'_key': 'foo' , 'bla': 'bar'})

package com.arangodb;

import java.util.ArrayList;
import java.util.Arrays;
import java.util.Collection;
import java.util.Map;

import org.junit.AfterClass;
import org.junit.Assert;
import org.junit.BeforeClass;
import org.junit.Test;

import com.arangodb.entity.DocumentEntity;
import com.arangodb.util.ImportOptions;
import com.arangodb.util.MapBuilder;

public class ImportDocumentTest {

    private static final String COLLECTION_NAME = "test";
    private static final String KEY = "foo";
    private static ArangoConfigure configure;
    private static ArangoDriver driver;

    @BeforeClass
    public static void setup() {
        configure = new ArangoConfigure();
        configure.init();
        driver = new ArangoDriver(configure);

        try {
            driver.deleteCollection(COLLECTION_NAME);
        } catch (final ArangoException e) {
        }
        try {
            driver.createCollection(COLLECTION_NAME);
        } catch (final ArangoException e) {
        }
    }

    @AfterClass
    public static void cleanup() {
        try {
            driver.deleteCollection(COLLECTION_NAME);
        } catch (final ArangoException e) {
        }
        configure.shutdown();
    }

    @SuppressWarnings("rawtypes")
    @Test
    public void process() throws ArangoException {
        // initial create the document
        driver.createDocument(COLLECTION_NAME, new MapBuilder().put("_key", KEY).put("bla", "bar").get());

        // update the document
        driver.updateDocument(COLLECTION_NAME + "/" + KEY, new MapBuilder().put("blarg", "blub").get());

        // delete the document
        driver.deleteDocument(COLLECTION_NAME + "/" + KEY);

        // import same document
        final ImportOptions options = new ImportOptions();
        final Collection<Collection<?>> headerValues = new ArrayList<Collection<?>>();
        headerValues.add(Arrays.asList("_key", "bla", "blarg"));
        headerValues.add(Arrays.asList(KEY, "bar", "blub"));
        driver.importDocumentsByHeaderValues(COLLECTION_NAME, headerValues, options);

        // get the document
        final DocumentEntity<Map> doc = driver.getDocument(COLLECTION_NAME + "/" + KEY, Map.class);
        Assert.assertFalse(doc.isError());
        Assert.assertEquals(COLLECTION_NAME + "/" + KEY, doc.getDocumentHandle());
        Assert.assertEquals("bar", doc.getEntity().get("bla"));
        Assert.assertEquals("blub", doc.getEntity().get("blarg"));

        // query the document via AQL
        final String query = "RETURN DOCUMENT(@docHandle)";
        final Map<String, Object> bindVars = new MapBuilder().put("docHandle", COLLECTION_NAME + "/" + KEY).get();
        final DocumentCursor<Map> result = driver.executeDocumentQuery(query, bindVars, null, Map.class);
        final DocumentEntity<Map> uniqueResult = result.getUniqueResult();
        Assert.assertNotNull(uniqueResult);
        Assert.assertEquals(COLLECTION_NAME + "/" + KEY, uniqueResult.getDocumentHandle());
        Assert.assertEquals("bar", uniqueResult.getEntity().get("bla"));
        Assert.assertEquals("blub", uniqueResult.getEntity().get("blarg"));
    }

}

//Updating the document
        List<List<?>> values = new ArrayList<List<?>>();
        values.add(Arrays.asList("_key","firstName", "lastName", "age", "gender"));
        values.add(Arrays.asList("test1","Joe", "Public", 42, "male", 10)); // error
        values.add(Arrays.asList("test2","Jane", "Doe", 31, "female"));

        ImportOptions options = new ImportOptions();
        options.setOnDuplicate(OnDuplicate.UPDATE);

        final ImportResultEntity result = driver.importDocumentsByHeaderValues(UT_IMPORT_TEST, values,options);

        //Updating the same document with age given as string

        List<List<?>> values = new ArrayList<List<?>>();
        values.add(Arrays.asList("_key","firstName", "lastName", "age", "gender"));
        values.add(Arrays.asList("test1","Joe", "Public", "42", "male", 10)); // error
        values.add(Arrays.asList("test2","Jane", "Doe", "31", "female"));

        //Query age as just 42 or 32 will show no records, but document will come up if we query with "42" or "32"