Examples with TraverserManager com.baidu.hugegraph.driver.TraverserManager used on opensource projects

Example 1 with TraverserManager

use of com.baidu.hugegraph.driver.TraverserManager in project incubator-hugegraph-toolchain by apache.

the class BackupManager method backupEdgeShard.

private void backupEdgeShard(Shard shard) {
    String desc = String.format("backing up edges[shard %s]", shard);
    Edges edges = null;
    String page = this.initPage();
    TraverserManager g = client.traverser();
    do {
        try {
            String p = page;
            if (page == null) {
                edges = retry(() -> g.edges(shard), desc);
            } else {
                edges = retry(() -> g.edges(shard, p), desc);
            }
        } catch (ToolsException e) {
            this.exceptionHandler(e, HugeType.EDGE, shard);
        }
        if (edges == null) {
            return;
        }
        List<Edge> edgeList = edges.results();
        if (edgeList == null || edgeList.isEmpty()) {
            return;
        }
        long count = this.backup(HugeType.EDGE, suffix.get(), edgeList);
        this.edgeCounter.getAndAdd(count);
        Printer.printInBackward(this.edgeCounter.get());
    } while ((page = edges.page()) != null);
}

Example 2 with TraverserManager

use of com.baidu.hugegraph.driver.TraverserManager in project incubator-hugegraph-toolchain by apache.

the class OltpAlgoService method shortestPath.

public GremlinResult shortestPath(int connId, ShortestPath body) {
    HugeClient client = this.getClient(connId);
    TraverserManager traverser = client.traverser();
    Path result = traverser.shortestPath(body.getSource(), body.getTarget(), body.getDirection(), body.getLabel(), body.getMaxDepth(), body.getMaxDegree(), body.getSkipDegree(), body.getCapacity());
    JsonView jsonView = new JsonView();
    jsonView.setData(result.objects());
    Date createTime = HubbleUtil.nowDate();
    TableView tableView = this.buildPathTableView(result);
    GraphView graphView = this.buildPathGraphView(result);
    // Insert execute history
    ExecuteStatus status = ExecuteStatus.SUCCESS;
    ExecuteHistory history;
    history = new ExecuteHistory(null, connId, 0L, ExecuteType.ALGORITHM, body.toString(), status, AsyncTaskStatus.UNKNOWN, -1L, createTime);
    this.historyService.save(history);
    return GremlinResult.builder().type(GremlinResult.Type.PATH).jsonView(jsonView).tableView(tableView).graphView(graphView).build();
}

Example 3 with TraverserManager

use of com.baidu.hugegraph.driver.TraverserManager in project incubator-hugegraph-toolchain by apache.

the class BackupManager method backupVertexShard.

private void backupVertexShard(Shard shard) {
    String desc = String.format("backing up vertices[shard:%s]", shard);
    Vertices vertices = null;
    String page = this.initPage();
    TraverserManager g = client.traverser();
    do {
        String p = page;
        try {
            if (page == null) {
                vertices = retry(() -> g.vertices(shard), desc);
            } else {
                vertices = retry(() -> g.vertices(shard, p), desc);
            }
        } catch (ToolsException e) {
            this.exceptionHandler(e, HugeType.VERTEX, shard);
        }
        if (vertices == null) {
            return;
        }
        List<Vertex> vertexList = vertices.results();
        if (vertexList == null || vertexList.isEmpty()) {
            return;
        }
        long count = this.backup(HugeType.VERTEX, suffix.get(), vertexList);
        this.vertexCounter.getAndAdd(count);
        Printer.printInBackward(this.vertexCounter.get());
    } while ((page = vertices.page()) != null);
}