Examples with ValueType com.alibaba.maxgraph.compiler.tree.value.ValueType used on opensource projects

Example 6 with ValueType

use of com.alibaba.maxgraph.compiler.tree.value.ValueType in project GraphScope by alibaba.

the class TreeBuilder method visitSelectStep.

private TreeNode visitSelectStep(SelectStep step, TreeNode prev) {
    List<String> selectKeyList = ReflectionUtils.getFieldValue(SelectStep.class, step, "selectKeys");
    Pop pop = step.getPop();
    List<Traversal.Admin<?, ?>> ringTraversalList = step.getLocalChildren();
    if (selectKeyList.size() < 2) {
        throw new IllegalArgumentException("select key size < 2 for select operator");
    }
    Map<String, List<TreeNode>> labelTreeNodeList = Maps.newHashMap();
    selectKeyList.forEach(v -> {
        if (treeNodeLabelManager.getLabelIndexList().containsKey(v)) {
            labelTreeNodeList.put(v, treeNodeLabelManager.getTreeNodeList(v));
        }
    });
    SelectTreeNode selectTreeNode = new SelectTreeNode(prev, selectKeyList, pop, labelTreeNodeList, schema);
    boolean saveFlag = rootPathFlag;
    rootPathFlag = false;
    if (!ringTraversalList.isEmpty()) {
        Set<ValueType> valueTypeList = selectKeyList.stream().map(v -> treeNodeLabelManager.getValueType(v, pop)).collect(Collectors.toSet());
        ValueType selectValueType = valueTypeList.size() > 1 ? new VarietyValueType(valueTypeList) : valueTypeList.iterator().next();
        ringTraversalList.forEach(v -> {
            SourceDelegateNode sourceDelegateNode = new SourceDelegateNode(selectTreeNode, schema);
            sourceDelegateNode.setDelegateOutputValueType(selectValueType);
            selectTreeNode.addTraversalTreeNode(travelTraversalAdmin(v, sourceDelegateNode));
        });
    }
    rootPathFlag = saveFlag;
    return selectTreeNode;
}