com.vesoft.nebula.driver.graph.scan.ScanEdgeResult Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of client Show documentation
The newest version!
package com.vesoft.nebula.driver.graph.scan;

import com.vesoft.nebula.driver.graph.data.Relationship;
import com.vesoft.nebula.driver.graph.data.ResultSet;
import com.vesoft.nebula.driver.graph.data.ValueWrapper;
import com.vesoft.nebula.driver.graph.data.Vertex;
import java.util.ArrayList;
import java.util.List;
import java.util.Map;

public class ScanEdgeResult extends ScanResult {

    private List propNames = new ArrayList<>();
    private List returnColumns = new ArrayList<>();

    public ScanEdgeResult(List results, List propNames) {
        super(results);
        this.propNames = propNames;
        returnColumns.add("src");
        returnColumns.add("dst");
        returnColumns.addAll(propNames);
    }

    /**
     * get node table row's column names
     *
     * @return list of row column names
     */
    public List getPropNames() {
        return returnColumns;
    }

    protected void convertResultToRow() {
        if (isEmpty) {
            return;
        }
        if (tableRows.isEmpty()) {
            for (ResultSet resultSet : results) {

                while (resultSet.hasNext()) {
                    ResultSet.Record   record    = resultSet.next();
                    List rowValues = new ArrayList<>();

                    Vertex       srcNode = record.get("Src").asNode();
                    Vertex       dstNode = record.get("Dst").asNode();
                    Relationship edge    = record.get("Edge").asEdge();
                    for (String key : srcNode.getProperties().keySet()) {
                        rowValues.add(srcNode.getProperties().get(key));
                    }
                    for (String key : dstNode.getProperties().keySet()) {
                        rowValues.add(dstNode.getProperties().get(key));
                    }
                    Map properties = edge.getProperties();
                    for (String propName : propNames) {
                        rowValues.add(properties.get(propName));
                    }
                    tableRows.add(new TableRow(rowValues));
                }
            }
        }
    }
}