mchades commented on code in PR #4925: URL: https://github.com/apache/gravitino/pull/4925#discussion_r1810015763
########## trino-connector/trino-connector/src/main/java/org/apache/gravitino/trino/connector/catalog/iceberg/ExpressionUtil.java: ########## @@ -0,0 +1,308 @@ +/* + * Licensed to the Apache Software Foundation (ASF) under one + * or more contributor license agreements. See the NOTICE file + * distributed with this work for additional information + * regarding copyright ownership. The ASF licenses this file + * to you under the Apache License, Version 2.0 (the + * "License"); you may not use this file except in compliance + * with the License. You may obtain a copy of the License at + * + * http://www.apache.org/licenses/LICENSE-2.0 + * + * Unless required by applicable law or agreed to in writing, + * software distributed under the License is distributed on an + * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY + * KIND, either express or implied. See the License for the + * specific language governing permissions and limitations + * under the License. + */ +package org.apache.gravitino.trino.connector.catalog.iceberg; + +import static org.apache.gravitino.trino.connector.GravitinoErrorCode.GRAVITINO_EXPRESSION_ERROR; + +import io.trino.spi.TrinoException; +import java.util.ArrayList; +import java.util.List; +import java.util.regex.MatchResult; +import java.util.regex.Matcher; +import java.util.regex.Pattern; +import org.apache.gravitino.rel.expressions.Expression; +import org.apache.gravitino.rel.expressions.NamedReference; +import org.apache.gravitino.rel.expressions.literals.Literal; +import org.apache.gravitino.rel.expressions.sorts.NullOrdering; +import org.apache.gravitino.rel.expressions.sorts.SortDirection; +import org.apache.gravitino.rel.expressions.sorts.SortOrder; +import org.apache.gravitino.rel.expressions.sorts.SortOrders; +import org.apache.gravitino.rel.expressions.transforms.Transform; +import org.apache.gravitino.rel.expressions.transforms.Transforms; + +/** This class is used to convert expression of bucket, sort_by, partition object to string */ +public class ExpressionUtil { + private static final String IDENTIFIER = "[a-zA-Z_][a-zA-Z0-9_]*"; + private static final String FUNCTION_ARG_INT = "(\\d+)"; + private static final String FUNCTION_ARG_IDENTIFIER = "(" + IDENTIFIER + ")"; + private static final Pattern YEAR_FUNCTION_PATTERN = + Pattern.compile("year\\(" + FUNCTION_ARG_IDENTIFIER + "\\)", Pattern.CASE_INSENSITIVE); + private static final Pattern MONTH_FUNCTION_PATTERN = + Pattern.compile("month\\(" + FUNCTION_ARG_IDENTIFIER + "\\)", Pattern.CASE_INSENSITIVE); + private static final Pattern DAY_FUNCTION_PATTERN = + Pattern.compile("day\\(" + FUNCTION_ARG_IDENTIFIER + "\\)", Pattern.CASE_INSENSITIVE); + private static final Pattern HOUR_FUNCTION_PATTERN = + Pattern.compile("hour\\(" + FUNCTION_ARG_IDENTIFIER + "\\)", Pattern.CASE_INSENSITIVE); + private static final Pattern BUCKET_FUNCTION_PATTERN = + Pattern.compile( + "bucket\\(" + FUNCTION_ARG_IDENTIFIER + ",\\s*" + FUNCTION_ARG_INT + "\\)", + Pattern.CASE_INSENSITIVE); + private static final Pattern TRUNCATE_FUNCTION_PATTERN = + Pattern.compile( + "truncate\\(" + FUNCTION_ARG_IDENTIFIER + ",\\s*" + FUNCTION_ARG_INT + "\\)", + Pattern.CASE_INSENSITIVE); + private static final Pattern IDENTIFILER_PATTERN = + Pattern.compile(IDENTIFIER, Pattern.CASE_INSENSITIVE); + + private static final String SORT_DIRECTION_ASC = "ASC"; + private static final String SORT_DIRECTION_DESC = "DESC"; + private static final String NULL_ORDERING_FIRST = "NULLS FIRST"; + private static final String NULL_ORDERING_LAST = "NULLS LAST"; + private static final String SORT_DIRECTION = + "(" + SORT_DIRECTION_ASC + "|" + SORT_DIRECTION_DESC + ")"; + private static final String NULL_ORDERING = + "(" + NULL_ORDERING_FIRST + "|" + NULL_ORDERING_LAST + ")"; + private static final Pattern SROT_ORDER_PATTERN = + Pattern.compile(IDENTIFIER, Pattern.CASE_INSENSITIVE); + private static final Pattern SORT_ORDER_WITH_SORT_DIRECTION_PATTERN = + Pattern.compile("(" + IDENTIFIER + ")\\s+" + SORT_DIRECTION, Pattern.CASE_INSENSITIVE); + private static final Pattern SORT_ORDER_WITH_SORT_DIRECTION_AND_NULL_ORDERING_PATTERN = + Pattern.compile( + "(" + IDENTIFIER + ")\\s+" + SORT_DIRECTION + "\\s+" + NULL_ORDERING, + Pattern.CASE_INSENSITIVE); + + public static List<String> expressionToPartitionFiled(Transform[] transforms) { + try { + List<String> partitionFields = new ArrayList<>(); + for (Transform transform : transforms) { + partitionFields.add(transFormToString(transform)); + } + return partitionFields; + } catch (IllegalArgumentException e) { + throw new TrinoException( + GRAVITINO_EXPRESSION_ERROR, + "Error to handle transform Expressions :" + e.getMessage(), + e); + } + } + + public static Transform[] partitionFiledToExpression(List<String> partitions) { + try { + List<Transform> partitionTransforms = new ArrayList<>(); + for (String partition : partitions) { + parseTransform(partitionTransforms, partition); + } + return partitionTransforms.toArray(new Transform[0]); + } catch (IllegalArgumentException e) { + throw new TrinoException( + GRAVITINO_EXPRESSION_ERROR, "Error parsing the partition field: " + e.getMessage(), e); + } + } + + public static List<String> expressionToSortOrderFiled(SortOrder[] orders) { + try { + List<String> orderFields = new ArrayList<>(); + for (SortOrder order : orders) { + orderFields.add(sortOrderToString(order)); + } + return orderFields; + } catch (IllegalArgumentException e) { + throw new TrinoException( + GRAVITINO_EXPRESSION_ERROR, + "Error to handle the sort order expressions : " + e.getMessage(), + e); + } + } + + public static SortOrder[] sortOrderFiledToExpression(List<String> orderFields) { + try { + List<SortOrder> sortOrders = new ArrayList<>(); + for (String orderField : orderFields) { + parseSortOrder(sortOrders, orderField); + } + return sortOrders.toArray(new SortOrder[0]); + } catch (IllegalArgumentException e) { + throw new TrinoException( + GRAVITINO_EXPRESSION_ERROR, "Error parsing the sort order field: " + e.getMessage(), e); + } + } + + private static void parseTransform(List<Transform> transforms, String value) { + boolean match = + false Review Comment: magic false? -- This is an automated message from the Apache Git Service. To respond to the message, please log on to GitHub and use the URL above to go to the specific comment. To unsubscribe, e-mail: [email protected] For queries about this service, please contact Infrastructure at: [email protected]
