abs func

Author: WeiXinChan

src/main/java/com/alipay/oceanbase/rpc/protocol/payload/impl/column/ObGeneratedColumnAbsFunc.java

@@ -0,0 +1,88 @@
+/*-
+ * #%L
+ * OBKV Table Client Framework
+ * %%
+ * Copyright (C) 2025 OceanBase
+ * %%
+ * OBKV Table Client Framework is licensed under Mulan PSL v2.
+ * You can use this software according to the terms and conditions of the Mulan PSL v2.
+ * You may obtain a copy of Mulan PSL v2 at:
+ *          http://license.coscl.org.cn/MulanPSL2
+ * THIS SOFTWARE IS PROVIDED ON AN "AS IS" BASIS, WITHOUT WARRANTIES OF ANY KIND,
+ * EITHER EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO NON-INFRINGEMENT,
+ * MERCHANTABILITY OR FIT FOR A PARTICULAR PURPOSE.
+ * See the Mulan PSL v2 for more details.
+ * #L%
+ */
+
+package com.alipay.oceanbase.rpc.protocol.payload.impl.column;
+
+import com.alipay.oceanbase.rpc.protocol.payload.impl.ObCollationType;
+
+import java.util.ArrayList;
+import java.util.List;
+
+public class ObGeneratedColumnAbsFunc implements ObGeneratedColumnSimpleFunc {
+    private final List<String> refColumnNames;
+
+    /*
+     * Ob generated column refer func.
+     */
+    public ObGeneratedColumnAbsFunc(String refColumn) {
+        List<String> refColumnNameList = new ArrayList<String>(1);
+        refColumnNameList.add(refColumn);
+        this.refColumnNames = refColumnNameList;
+    }
+
+    /*
+     * Set parameters.
+     */
+    @Override
+    public void setParameters(List<Object> parameters) {
+        //ignore
+    }
+
+    /*
+     * Get min parameters.
+     */
+    @Override
+    public int getMinParameters() {
+        return 0;
+    }
+
+    /*
+     * Get max parameters.
+     */
+    @Override
+    public int getMaxParameters() {
+        return 0;
+    }
+
+    /*
+     * Get ref column names.
+     */
+    @Override
+    public List<String> getRefColumnNames() {
+        return refColumnNames;
+    }
+
+    /*
+     * Eval value.
+     */
+    @Override
+    public Object evalValue(ObCollationType collationType, Object... refs)
+                                                                          throws IllegalArgumentException {
+        if (refs == null || refs.length == 0) {
+            throw new IllegalArgumentException("Input references cannot be null or empty");
+        }
+        Object ref = refs[0];
+        if (ref instanceof Long) {
+            return Math.abs((Long) ref);
+        } else if (ref instanceof Integer) {
+            return Math.abs((Integer) ref);
+        } else {
+            throw new IllegalArgumentException("Object [" + ref + "] can not evaluate by"
+                                               + " ObGeneratedColumnNegateFunc");
+        }
+    }
+}

src/main/java/com/alipay/oceanbase/rpc/protocol/payload/impl/parser/ObGeneratedColumnExpressParser.java

@@ -18,10 +18,7 @@
 package com.alipay.oceanbase.rpc.protocol.payload.impl.parser;
 
 import com.alipay.oceanbase.rpc.exception.GenerateColumnParseException;
-import com.alipay.oceanbase.rpc.protocol.payload.impl.column.ObGeneratedColumnNegateFunc;
-import com.alipay.oceanbase.rpc.protocol.payload.impl.column.ObGeneratedColumnReferFunc;
-import com.alipay.oceanbase.rpc.protocol.payload.impl.column.ObGeneratedColumnSimpleFunc;
-import com.alipay.oceanbase.rpc.protocol.payload.impl.column.ObGeneratedColumnSubStrFunc;
+import com.alipay.oceanbase.rpc.protocol.payload.impl.column.*;
 
 import java.math.BigInteger;
 import java.util.ArrayList;
@@ -60,6 +57,8 @@ public ObGeneratedColumnSimpleFunc parse() throws GenerateColumnParseException {
                                 lexer.nextToken();
                                 listParameters(subStr);
                                 return subStr;
+                            case ABS:
+                                return handleAbsFunction();
                             default:
                                 throw new GenerateColumnParseException("");
                         }
@@ -81,6 +80,49 @@ public ObGeneratedColumnSimpleFunc parse() throws GenerateColumnParseException {
         }
     }
 
+    private ObGeneratedColumnSimpleFunc handleAbsFunction() throws GenerateColumnParseException {
+        lexer.nextToken(); // Consume LPAREN
+
+        // handle abs(T) or abs(`T`)
+        if (lexer.token() == IDENTIFIER) {
+            String refColumn = lexer.stringVal();
+            lexer.nextToken(); // Consume the identifier
+
+            // check next token is RPAREN or not
+            if (lexer.token() != RPAREN) {
+                throw new GenerateColumnParseException(
+                    "Expected closing parenthesis after ABS argument");
+            }
+            lexer.nextToken(); // Consume RPAREN
+            return new ObGeneratedColumnAbsFunc(refColumn);
+        } else if (lexer.token() == BACKTICK) { // abs(`T`)
+            lexer.nextToken(); // Consume BACKTICK
+            if (lexer.token() == IDENTIFIER) {
+                String refColumn = lexer.stringVal();
+                lexer.nextToken(); // Consume the identifier
+
+                if (lexer.token() != BACKTICK) {
+                    throw new GenerateColumnParseException(
+                        "Expected closing backtick after identifier");
+                }
+                lexer.nextToken(); // Consume BACKTICK
+
+                if (lexer.token() != RPAREN) {
+                    throw new GenerateColumnParseException(
+                        "Expected closing parenthesis after ABS argument");
+                }
+                lexer.nextToken(); // Consume RPAREN
+                return new ObGeneratedColumnAbsFunc(refColumn);
+            } else {
+                throw new GenerateColumnParseException(
+                    "Invalid argument for ABS function: expected IDENTIFIER after backtick");
+            }
+        } else {
+            throw new GenerateColumnParseException(
+                "Invalid argument for ABS function: expected IDENTIFIER or BACKTICK enclosed identifier");
+        }
+    }
+
     protected void listParameters(ObGeneratedColumnSimpleFunc func) {
         List<Object> args = new ArrayList<Object>();
         args.add(primaryParameters());

src/main/java/com/alipay/oceanbase/rpc/protocol/payload/impl/parser/ObGeneratedColumnFuncName.java

@@ -21,7 +21,7 @@
 import java.util.Map;
 
 public enum ObGeneratedColumnFuncName {
-    UNKNOWN("unknown"), SUB_STR("substr");
+    UNKNOWN("unknown"), SUB_STR("substr"), ABS("abs");
 
     public final String name;
 

src/main/java/com/alipay/oceanbase/rpc/protocol/payload/impl/parser/ObGeneratedColumnSimpleToken.java

@@ -19,7 +19,7 @@
 
 public enum ObGeneratedColumnSimpleToken {
 
-    COMMA(","), LPAREN("("), RPAREN(")"), SUB("-"), PLUS("+"), LITERAL_HEX, LITERAL_FLOAT, LITERAL_INT, IDENTIFIER, ERROR, EOF;
+    COMMA(","), LPAREN("("), RPAREN(")"), SUB("-"), PLUS("+"), BACKTICK("`"), LITERAL_HEX, LITERAL_FLOAT, LITERAL_INT, IDENTIFIER, ERROR, EOF;
 
     public final String name;