import com.arangodb.ArangoDBException; //导入依赖的package包/类
public <T> CompletableFuture<T> execute(
	final Request request,
	final ResponseDeserializer<T> responseDeserializer,
	final HostHandle hostHandle) {
	final CompletableFuture<T> result = new CompletableFuture<>();
	communication.execute(request, hostHandle).whenComplete((response, ex) -> {
		if (response != null) {
			try {
				result.complete(responseDeserializer.deserialize(response));
			} catch (final VPackException | ArangoDBException e) {
				result.completeExceptionally(e);
			}
		} else if (ex != null) {
			result.completeExceptionally(ex);
		} else {
			result.cancel(true);
		}
	});
	return result;
}
 

import com.arangodb.ArangoDBException; //导入依赖的package包/类
@Test
public void queryOutboundInbound() throws ArangoDBException, InterruptedException, ExecutionException {
	String queryString = "FOR v IN 1..3 OUTBOUND 'circles/E' GRAPH 'traversalGraph' return v._key";
	ArangoCursorAsync<String> cursor = db.query(queryString, null, null, String.class).get();
	Collection<String> result = cursor.asListRemaining();
	assertThat(result.size(), is(1));
	assertThat(result, hasItems("F"));

	queryString = "FOR v IN 1..3 INBOUND 'circles/E' GRAPH 'traversalGraph' return v._key";
	cursor = db.query(queryString, null, null, String.class).get();
	result = cursor.asListRemaining();
	assertThat(result.size(), is(2));
	assertThat(result, hasItems("B", "A"));

	queryString = "FOR v IN 1..3 ANY 'circles/E' GRAPH 'traversalGraph' return v._key";
	cursor = db.query(queryString, null, null, String.class).get();

	result = cursor.asListRemaining();
	assertThat(result.size(), is(6));
	assertThat(result, hasItems("F", "B", "C", "D", "A", "G"));
}
 

import com.arangodb.ArangoDBException; //导入依赖的package包/类
protected static void init(String dbName) {

        AbstractArangoDbInterpreterTest.dbName = dbName;

        arango = new ArangoDB.Builder()
                .host(HOST, PORT)
                .user(USERNAME).password(PASSWORD)
                .build();

        try {
            arango.db(dbName).drop();
        }
        catch (final ArangoDBException e) {}

        arango.createDatabase(dbName);

        final Properties props = new Properties();
        props.put(ArangoDbInterpreter.ARANGODB_HOST, HOST);
        props.put(ArangoDbInterpreter.ARANGODB_PORT, "" + PORT);
        props.put(ArangoDbInterpreter.ARANGODB_DATABASE, dbName);
        props.put(ArangoDbInterpreter.ARANGODB_USER, USERNAME);
        props.put(ArangoDbInterpreter.ARANGODB_PWD, PASSWORD);
        interpreter = new ArangoDbInterpreter(props);
        interpreter.open();
    }
 

import com.arangodb.ArangoDBException; //导入依赖的package包/类
@BeforeClass
public static void populate() throws ArangoDBException, IOException {
    init(DB_NAME);

    arango.db(DB_NAME).createCollection(COLL_NAME);

    for (Integer i = 0; i < 50; i++) {
        final BaseDocument log = new BaseDocument();
        log.setKey(i.toString());
        log.addAttribute("date", new Date());
        log.addAttribute("status", STATUS[RandomUtils.nextInt(STATUS.length)]);
        log.addAttribute("content_length", RandomUtils.nextInt(2000));

        final Map<String, Object> req = new HashMap<>();
        req.put("method", METHODS[RandomUtils.nextInt(METHODS.length)]);
        req.put("url", "/zeppelin/" + UUID.randomUUID().toString());
        req.put("headers", Arrays.asList("Accept: *.*", "Host: apache.org"));
        log.addAttribute("request", req);

        arango.db(DB_NAME).collection(COLL_NAME).insertDocument(log);
    }
}
 

import com.arangodb.ArangoDBException; //导入依赖的package包/类
@Override
public Map<String, RedListSettings> getRedListSettings(String territory) throws FloraOnException {
    Map<String, Object> bindVars;
    Iterator<RedListSettings> it;
    Map<String, RedListSettings> out = new HashMap<>();
    try {
        if(territory == null) {
            it = database.query(AQLRedListQueries.getString("redlistdata.9a"), null
                    , null, RedListSettings.class);
        } else {
            bindVars = new HashMap<>();
            bindVars.put("terr", territory);
            it = database.query(AQLRedListQueries.getString("redlistdata.9"), bindVars
                    , null, RedListSettings.class);
        }
    } catch (ArangoDBException e) {
        throw new DatabaseException(e.getMessage());
    }
    while(it.hasNext()) {
        RedListSettings rls = it.next();
        out.put(rls.getTerritory(), rls);
    }
    return out;
}
 

import com.arangodb.ArangoDBException; //导入依赖的package包/类
@Override
public Set<String> getRedListTags(String territory) throws DatabaseException {
    Set<String> out = new HashSet<>();
    Object tmp;
    try {
        Iterator<Object> it = database.query(AQLRedListQueries.getString("redlistdata.5", territory), null
                , null, Object.class);
        while(it.hasNext()) {
            tmp = it.next();
            if(tmp != null) out.add((String) tmp);
        }
    } catch (ArangoDBException e) {
        throw new DatabaseException(e.getMessage());
    }
    return out;
}
 

import com.arangodb.ArangoDBException; //导入依赖的package包/类
public int addTagToRedListDataEntities(String territory, String[] taxEntIds, String tag) throws FloraOnException {
        Map<String, Object> bp = new HashMap<>();
        bp.put("ids", taxEntIds);
        Gson gs = new GsonBuilder().setPrettyPrinting().create();
/*
        System.out.println(gs.toJson(bp));
        System.out.println(AQLRedListQueries.getString("redlistdata.8", territory, tag));
*/
        try {
            ArangoCursor<String> c = database.query(AQLRedListQueries.getString("redlistdata.8", territory, tag)
                    , bp, new AqlQueryOptions().count(true), String.class);
            return c.getCount();
        } catch (ArangoDBException e) {
            throw new DatabaseException(e.getMessage());
        }
    }
 

import com.arangodb.ArangoDBException; //导入依赖的package包/类
@Override
public Iterator<Inventory> getOccurrencesOfObserver(INodeKey authorId, Integer offset, Integer count) throws DatabaseException {
    if(authorId == null) return getOccurrencesOfMaintainer(null, false, offset, count);
    Map<String, Object> bindVars = new HashMap<>();
    bindVars.put("observer", authorId.toString());
    bindVars.put("off", offset == null ? 0 : offset);
    bindVars.put("cou", count == null ? 999999 : count);

    try {
        return database.query(
                AQLOccurrenceQueries.getString("occurrencequery.2")
                , bindVars, null, Inventory.class);
    } catch (ArangoDBException e) {
        throw new DatabaseException(e.getMessage());
    }
}
 

import com.arangodb.ArangoDBException; //导入依赖的package包/类
@Override
public Iterator<Inventory> getOccurrencesOfObserverWithinDates(INodeKey authorId, Date from, Date to, Integer offset, Integer count) throws DatabaseException {
    if(authorId == null) return getOccurrencesOfMaintainer(null, false, offset, count);
    DateFormat df = Constants.dateFormatYMD.get();
    Map<String, Object> bindVars = new HashMap<>();
    bindVars.put("observer", authorId.toString());
    bindVars.put("off", offset == null ? 0 : offset);
    bindVars.put("cou", count == null ? 999999 : count);
    bindVars.put("from", df.format(from));
    bindVars.put("to", df.format(to));

    try {
        return database.query(
                AQLOccurrenceQueries.getString("occurrencequery.2.date")
                , bindVars, null, Inventory.class);
    } catch (ArangoDBException e) {
        throw new DatabaseException(e.getMessage());
    }
}
 

import com.arangodb.ArangoDBException; //导入依赖的package包/类
@Override
public Iterator<Inventory> findInventoriesByFilter(String filter, INodeKey userId, Integer offset, Integer count) throws FloraOnException {
    Map<String, Object> bindVars = new HashMap<>();
    if(offset == null) offset = 0;
    if(count == null) count = 999999;
    bindVars.put("query", "%" + filter + "%");
    bindVars.put("offset", offset);
    bindVars.put("count", count);
    if(userId != null)
        bindVars.put("user", userId.toString());

    try {
        return database.query(
                AQLOccurrenceQueries.getString(userId == null ? "occurrencequery.9a" : "occurrencequery.9")
                , bindVars, null, Inventory.class);
    } catch (ArangoDBException e) {
        throw new DatabaseException(e.getMessage());
    }
}
 

import com.arangodb.ArangoDBException; //导入依赖的package包/类
@Override
    public int deleteInventoriesOrOccurrences(String[] inventoryId, String[] uuid) throws FloraOnException {
        int count = 0;
        for (int i = 0; i < inventoryId.length; i++) {
            try {
                if(StringUtils.isArrayEmpty(uuid) || uuid[i].trim().equals("")) {
//                    driver.getNodeWorkerDriver().deleteDocument(driver.asNodeKey(inventoryId[i]));  // FIXME: check for connected links
                    driver.getNodeWorkerDriver().deleteVertexOrEdge(driver.asNodeKey(inventoryId[i]));
                } else {
                    Inventory inv = database.query(
                            AQLOccurrenceQueries.getString("occurrencequery.3", inventoryId[i], uuid[i])
                            , null, null, Inventory.class).next();
                    // if the inventory became empty, delete the inventory
                    if(inv._getOccurrences().size() == 0)
                        driver.getNodeWorkerDriver().deleteDocument(driver.asNodeKey(inventoryId[i]));
                }
            } catch (ArangoDBException e) {
                e.printStackTrace();
                continue;
            }
            count++;
        }
        return count;
    }
 

import com.arangodb.ArangoDBException; //导入依赖的package包/类
@Override
public Iterator<Inventory> findOccurrencesByFilter(String filter, INodeKey userId, Integer offset, Integer count) throws FloraOnException {
    Map<String, Object> bindVars = new HashMap<>();
    if(offset == null) offset = 0;
    if(count == null) count = 999999;
    bindVars.put("query", "%" + filter + "%");
    bindVars.put("offset", offset);
    bindVars.put("count", count);
    if(userId != null)
        bindVars.put("user", userId.toString());

    try {
        return database.query(
                AQLOccurrenceQueries.getString(userId == null ? "occurrencequery.8a" : "occurrencequery.8")
                , bindVars, null, Inventory.class);
    } catch (ArangoDBException e) {
        throw new DatabaseException(e.getMessage());
    }
}
 

import com.arangodb.ArangoDBException; //导入依赖的package包/类
@Override
public Iterator<TaxEnt> getTaxaWithTag(INodeKey userId, Date from, Date to, String territory, String tag, boolean withPhoto) throws DatabaseException {
    Map<String, Object> bindVars = new HashMap<>();
    DateFormat df = Constants.dateFormatYMD.get();
    bindVars.put("user", userId.toString());
    bindVars.put("from", df.format(from));
    bindVars.put("to", df.format(to));
    bindVars.put("tag", tag);
    bindVars.put("@redlistcollection", "redlist_" + territory);
    try {
        return database.query(AQLOccurrenceQueries.getString(
                withPhoto ? "occurrencereportquery.4" : "occurrencereportquery.3"), bindVars
                , null, TaxEnt.class);
    } catch (ArangoDBException e) {
        throw new DatabaseException(e.getMessage());
    }
}
 

import com.arangodb.ArangoDBException; //导入依赖的package包/类
@Override
public Iterator<StatisticPerTaxon> getTaxaWithTagCollected(INodeKey userId, Date from, Date to, String territory, String tag) throws DatabaseException {
    Map<String, Object> bindVars = new HashMap<>();
    DateFormat df = Constants.dateFormatYMD.get();
    bindVars.put("user", userId.toString());
    bindVars.put("from", df.format(from));
    bindVars.put("to", df.format(to));
    bindVars.put("tag", tag);
    bindVars.put("@redlistcollection", "redlist_" + territory);
    try {
        return database.query(AQLOccurrenceQueries.getString("occurrencereportquery.5"), bindVars
                , null, StatisticPerTaxon.class);
    } catch (ArangoDBException e) {
        throw new DatabaseException(e.getMessage());
    }
}
 

import com.arangodb.ArangoDBException; //导入依赖的package包/类
@Override
public Iterator<StatisticPerTaxon> getTaxaWithTagEstimates(INodeKey userId, Date from, Date to, String territory, String tag) throws DatabaseException {
    Map<String, Object> bindVars = new HashMap<>();
    DateFormat df = Constants.dateFormatYMD.get();
    bindVars.put("user", userId.toString());
    bindVars.put("from", df.format(from));
    bindVars.put("to", df.format(to));
    bindVars.put("tag", tag);
    bindVars.put("@redlistcollection", "redlist_" + territory);
    try {
        return database.query(AQLOccurrenceQueries.getString("occurrencereportquery.7"), bindVars
                , null, StatisticPerTaxon.class);
    } catch (ArangoDBException e) {
        throw new DatabaseException(e.getMessage());
    }
}
 

import com.arangodb.ArangoDBException; //导入依赖的package包/类
@Override
public Iterator<TaxEnt> getRedListSheetsIsCollaborator(INodeKey userId, String territory, @NotNull TypeOfCollaboration typeOfCollaboration) throws DatabaseException {
    Map<String, Object> bindVars = new HashMap<>();
    bindVars.put("user", userId.toString());
    bindVars.put("@redlistcollection", "redlist_" + territory);
    if(typeOfCollaboration == null) return Collections.emptyIterator();
    try {
        switch (typeOfCollaboration) {
            case TEXTAUTHOR:
                return database.query(AQLOccurrenceQueries.getString("reportquery.1"), bindVars, null, TaxEnt.class);

            case ASSESSOR:
                return database.query(AQLOccurrenceQueries.getString("reportquery.1a"), bindVars, null, TaxEnt.class);

            case REVIEWER:
                return database.query(AQLOccurrenceQueries.getString("reportquery.1b"), bindVars, null, TaxEnt.class);

            default: return Collections.emptyIterator();
        }
    } catch (ArangoDBException e) {
        throw new DatabaseException(e.getMessage());
    }
}
 

import com.arangodb.ArangoDBException; //导入依赖的package包/类
@Override
public User getUser(String name) throws FloraOnException {
    if(name == null) return null;
    ArangoCursor<User> cur;
    Map<String, Object> bind = new HashMap<>();
    bind.put("name", name);
    try {
        cur = database.query(AQLQueries.getString("Administration.4"), bind
                , new AqlQueryOptions().count(true), User.class);
        if(!cur.hasNext()) return null;
        if(cur.getCount() > 1) throw new FloraOnException(Messages.getString("error.1", name));
        return cur.next();
    } catch (ArangoDBException e) {
        throw new FloraOnException(e.getMessage());
    }
}
 

import com.arangodb.ArangoDBException; //导入依赖的package包/类
/**
	 * Initializes a new database from scratch. Creates collections, graphs, etc.
	 * @param dbname
	 * @throws ArangoDBException
	 */
	private void initializeNewDatabase(String dbname) throws ArangoDBException {
		System.out.println("Initializing a fresh new database");
		/*				UserEntity ue;
		ue=new UserEntity();*/
		UserEntity[] ue=new UserEntity[0];
		driver.createDatabase(dbname);
		database = driver.db(dbname);

		checkCollections();
		createTaxonomicGraph();

//		database.collection(NodeTypes.specieslist.toString()).createGeoIndex(Arrays.asList("location"), new GeoIndexOptions().geoJson(false));
		database.collection(NodeTypes.inventory.toString()).createGeoIndex(Arrays.asList("latitude", "longitude"), new GeoIndexOptions().geoJson(false));
//		database.collection(NodeTypes.author.toString()).createHashIndex(Arrays.asList("idAut"), new HashIndexOptions().unique(true).sparse(false));
		database.collection(NodeTypes.taxent.toString()).createHashIndex(Arrays.asList("oldId"), new HashIndexOptions().unique(false).sparse(true));
		database.collection(NodeTypes.taxent.toString()).createHashIndex(Arrays.asList("rank"), new HashIndexOptions().unique(false).sparse(true));
		database.collection(NodeTypes.taxent.toString()).createHashIndex(Arrays.asList("isSpeciesOrInf"), new HashIndexOptions().unique(false).sparse(false));
		database.collection(NodeTypes.taxent.toString()).createHashIndex(Arrays.asList("name"), new HashIndexOptions().unique(false).sparse(true));
		database.collection(NodeTypes.taxent.toString()).createFulltextIndex(Arrays.asList("name"), new FulltextIndexOptions());
		database.collection(NodeTypes.territory.toString()).createHashIndex(Arrays.asList("shortName"), new HashIndexOptions().unique(true).sparse(false));
		database.collection(NodeTypes.user.toString()).createHashIndex(Arrays.asList("userName"), new HashIndexOptions().unique(true).sparse(false));
		database.collection(NodeTypes.toponym.toString()).createFulltextIndex(Collections.singleton("locality"), new FulltextIndexOptions().minLength(1));
	}
 

import com.arangodb.ArangoDBException; //导入依赖的package包/类
@Override
public int setNativeStatus(INodeKey territory, NativeStatus status, OccurrenceStatus occurrenceStatus, AbundanceLevel abundanceLevel, PlantIntroducedStatus introducedStatus, PlantNaturalizationDegree naturalizationDegree, Boolean uncertainOccurrenceStatus) throws FloraOnException {
	String query;
	if(status == null) {	// remove the EXISTS_IN link, if it exists
		query=String.format(
			AQLQueries.getString("TaxEntWrapperDriver.6")
			,thisNode.toString()
			,territory.toString());
	} else {				// create or update the EXISTS_IN link
		EXISTS_IN eIn=new EXISTS_IN(status, occurrenceStatus, abundanceLevel, introducedStatus, naturalizationDegree, uncertainOccurrenceStatus, thisNode.toString(), territory.toString());
		query=String.format(
			AQLQueries.getString("TaxEntWrapperDriver.7")
			,thisNode.toString()
			,territory.toString()
			,eIn.toJson().toString());
	}

	try {
		return database.query(query,null,null,Integer.class).next();
	} catch (ArangoDBException | NoSuchElementException e) {
		throw new DatabaseException(e.getMessage());
	}
}
 

import com.arangodb.ArangoDBException; //导入依赖的package包/类
@Override
	public GraphUpdateResult createRelationshipTo(INodeKey parent,RelTypes type) throws FloraOnException, NoSuchMethodException, SecurityException, InstantiationException, IllegalAccessException, IllegalArgumentException, InvocationTargetException {
		String baseId=node.toString();
		String parentId=parent.toString();

		// checks whether there is already a relation of this type between these two nodes
		String query=String.format(
			"FOR e IN %3$s FILTER e._from=='%1$s' && e._to=='%2$s' COLLECT WITH COUNT INTO l RETURN l"
			,baseId,parentId,type.toString());
		
		try {
			Integer nrel=database.query(query, null, null, Integer.class).next();
			if(nrel==0) {
				return new GraphUpdateResult(driver, new String[] {
//					dbDriver.createEdge(type.toString(), type.getEdge(), baseId, parentId, false).getDocumentHandle()
						database.graph(Constants.TAXONOMICGRAPHNAME).edgeCollection(type.toString()).insertEdge(type.getEdge(baseId, parentId)).getId()
						, baseId, parentId
				});
			} else return GraphUpdateResult.emptyResult();
		} catch (ArangoDBException | NoSuchElementException e) {
			throw new DatabaseException(e.getMessage());
		}
	}