clazz) {
+ try {
+ return clazz.cast(actual.get(nth).value);
+ } catch (ClassCastException e) {
+ throw new TypeException(null, clazz, actual.get(nth).type, actual.get(nth).value);
+ }
+ }
+
+ public interface Arity {
+ void check(String methodName, List params);
+
+ Arity NULLARY = Arity.of(0);
+ Arity UNARY = Arity.of(1);
+ Arity BINARY = Arity.of(2);
+
+ static Arity of(int count) {
+ return (methodName, params) -> {
+ if (params.size() != count) {
+ throw new ArityException(null, methodName, count, params.size());
+ }
+ };
+ }
+
+ static Arity min(int count) {
+ return (methodName, params) -> {
+ if (params.size() < count) {
+ throw new ArityException(null, "wrong number of arguments in call to '" + methodName
+ + "'. Expected at least " + count + " got " + params.size() + ".");
+ }
+ };
+ }
+
+ static Arity max(int count) {
+ return (methodName, params) -> {
+ if (params.size() > count) {
+ throw new ArityException(null, "wrong number of arguments in call to '" + methodName
+ + "'. Expected at most " + count + " got " + params.size() + ".");
+ }
+ };
+ }
+ }
+}
diff --git a/fe/fe-core/src/main/java/org/apache/doris/plsql/objects/PlClass.java b/fe/fe-core/src/main/java/org/apache/doris/plsql/objects/PlClass.java
new file mode 100644
index 00000000000000..38f8a39b2db23a
--- /dev/null
+++ b/fe/fe-core/src/main/java/org/apache/doris/plsql/objects/PlClass.java
@@ -0,0 +1,27 @@
+// 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.
+// This file is copied from
+// https://github.com/apache/hive/blob/master/hplsql/src/main/java/org/apache/hive/hplsql/objects/HplClass.java
+// and modified by Doris
+
+package org.apache.doris.plsql.objects;
+
+public interface PlClass {
+ PlObject newInstance();
+
+ org.apache.doris.plsql.objects.MethodDictionary methodDictionary();
+}
diff --git a/fe/fe-core/src/main/java/org/apache/doris/plsql/objects/PlObject.java b/fe/fe-core/src/main/java/org/apache/doris/plsql/objects/PlObject.java
new file mode 100644
index 00000000000000..1216cd319bc241
--- /dev/null
+++ b/fe/fe-core/src/main/java/org/apache/doris/plsql/objects/PlObject.java
@@ -0,0 +1,25 @@
+// 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.
+// This file is copied from
+// https://github.com/apache/hive/blob/master/hplsql/src/main/java/org/apache/hive/hplsql/objects/HplObject.java
+// and modified by Doris
+
+package org.apache.doris.plsql.objects;
+
+public interface PlObject {
+ PlClass plClass();
+}
diff --git a/fe/fe-core/src/main/java/org/apache/doris/plsql/objects/Table.java b/fe/fe-core/src/main/java/org/apache/doris/plsql/objects/Table.java
new file mode 100644
index 00000000000000..f24bff36e1216a
--- /dev/null
+++ b/fe/fe-core/src/main/java/org/apache/doris/plsql/objects/Table.java
@@ -0,0 +1,225 @@
+// 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.
+// This file is copied from
+// https://github.com/apache/hive/blob/master/hplsql/src/main/java/org/apache/hive/hplsql/objects/Table.java
+// and modified by Doris
+
+package org.apache.doris.plsql.objects;
+
+import org.apache.doris.common.AnalysisException;
+import org.apache.doris.plsql.ColumnDefinition;
+import org.apache.doris.plsql.Row;
+import org.apache.doris.plsql.Var;
+import org.apache.doris.plsql.executor.QueryResult;
+
+import java.util.HashMap;
+import java.util.Map;
+
+/**
+ * Oracle's PL/SQL Table/associative array.
+ *
+ * Tables can be modelled after a corresponding Hive table or they can be created manually.
+ *
+ * 1. Model the table after the emp Hive table
+ * TYPE t_tab IS TABLE OF emp%ROWTYPE INDEX BY BINARY_INTEGER;
+ *
+ * 2. Model the table after a column of a Hive table (emp.name). This table will hold a single column only.
+ * TYPE t_tab IS TABLE OF emp.name%TYPE INDEX BY BINARY_INTEGER;
+ *
+ * 3. Or you can specify the column manually. This table will hold one column only.
+ * TYPE t_tab IS TABLE OF NUMBER INDEX BY BINARY_INTEGER;
+ *
+ * In the first case the values will be records where each key in the record matches the columns to the corresponding
+ * table.
+ * tab(key).col_name;
+ *
+ * In the last two cases the values will represent scalars, but they stored in a record with a single key.
+ * tab(key)
+ *
+ * Iteration logic uses the first/last next and prior methods.
+ * First/last returns a key, next/prior gives back the next or previous key based on the key passed in.
+ */
+public class Table implements PlObject {
+ private final TableClass plClass;
+ private final Map rows = new HashMap<>();
+ private Object lastKey = null;
+ private Object firstKey = null;
+
+ public Table(TableClass plClass) {
+ this.plClass = plClass;
+ }
+
+ public void populate(QueryResult query, long rowIndex, int columnIndex) throws AnalysisException {
+ if (plClass().rowType()) {
+ putRow(rowIndex, query);
+ } else {
+ putColumn(rowIndex, query, columnIndex);
+ }
+ }
+
+ public void putRow(Object key, QueryResult result) throws AnalysisException {
+ put(key, readRow(result));
+ }
+
+ public void putColumn(Object key, QueryResult query, int columnIndex) throws AnalysisException {
+ put(key, readColumn(query, columnIndex));
+ }
+
+ public void put(Object key, Row row) {
+ Value existing = rows.get(key);
+ if (existing != null) {
+ existing.row = row;
+ } else {
+ if (lastKey != null) {
+ rows.get(lastKey).nextKey = key;
+ }
+ rows.put(key, new Value(row, lastKey));
+ lastKey = key;
+ if (firstKey == null) {
+ firstKey = key;
+ }
+ }
+ }
+
+ private Row readRow(QueryResult result) throws AnalysisException {
+ Row row = new Row();
+ int idx = 0;
+ for (ColumnDefinition column : plClass.columns()) {
+ Var var = new Var(column.columnName(), column.columnType().typeString(), (Integer) null, null, null);
+ var.setValue(result, idx);
+ row.addColumn(column.columnName(), column.columnTypeString(), var);
+ idx++;
+ }
+ return row;
+ }
+
+ private Row readColumn(QueryResult result, int columnIndex) throws AnalysisException {
+ Row row = new Row();
+ ColumnDefinition column = plClass.columns().get(0);
+ Var var = new Var(column.columnName(), column.columnType().typeString(), (Integer) null, null, null);
+ var.setValue(result, columnIndex);
+ row.addColumn(column.columnName(), column.columnTypeString(), var);
+ return row;
+ }
+
+ public Row at(Object key) {
+ Value value = rows.get(key);
+ return value == null ? null : value.row;
+ }
+
+ public boolean removeAt(Object key) {
+ Value value = rows.remove(key);
+ if (value != null) {
+ updateLinks(key, value.nextKey, value.prevKey);
+ }
+ return value != null;
+ }
+
+ private void updateLinks(Object deletedKey, Object nextKey, Object prevKey) {
+ if (prevKey != null) {
+ rows.get(prevKey).nextKey = nextKey;
+ }
+ if (nextKey != null) {
+ rows.get(nextKey).prevKey = prevKey;
+ }
+ if (deletedKey.equals(firstKey)) {
+ firstKey = nextKey;
+ }
+ if (deletedKey.equals(lastKey)) {
+ lastKey = prevKey;
+ }
+ }
+
+ public void removeFromTo(Object fromKey, Object toKey) {
+ Object current = fromKey;
+ while (current != null && !current.equals(toKey)) {
+ Object next = nextKey(current);
+ removeAt(current);
+ current = next;
+ }
+ if (current != null) {
+ removeAt(current);
+ }
+ }
+
+ public void removeAll() {
+ lastKey = null;
+ firstKey = null;
+ rows.clear();
+ }
+
+ public Object nextKey(Object key) {
+ Value value = rows.get(key);
+ return value == null ? null : value.nextKey;
+ }
+
+ public Object priorKey(Object key) {
+ Value value = rows.get(key);
+ return value == null ? null : value.prevKey;
+ }
+
+ public Object firstKey() {
+ return firstKey;
+ }
+
+ public Object lastKey() {
+ return lastKey;
+ }
+
+ public boolean existsAt(Object key) {
+ return rows.get(key) != null;
+ }
+
+ public int count() {
+ return rows.size();
+ }
+
+ @Override
+ public TableClass plClass() {
+ return plClass;
+ }
+
+ private static class Value {
+ private Row row;
+ private Object prevKey;
+ private Object nextKey;
+
+ public Value(Row row, Object prevKey) {
+ this.row = row;
+ this.prevKey = prevKey;
+ }
+
+ public void setPrevKey(Object prevKey) {
+ this.prevKey = prevKey;
+ }
+
+ public void setNextKey(Object nextKey) {
+ this.nextKey = nextKey;
+ }
+ }
+
+ @Override
+ public String toString() {
+ final StringBuilder sb = new StringBuilder("Table{");
+ sb.append("plClass=").append(plClass.getClass());
+ sb.append(", size=").append(count());
+ sb.append(", lastKey=").append(lastKey);
+ sb.append(", firstKey=").append(firstKey);
+ sb.append('}');
+ return sb.toString();
+ }
+}
diff --git a/fe/fe-core/src/main/java/org/apache/doris/plsql/objects/TableClass.java b/fe/fe-core/src/main/java/org/apache/doris/plsql/objects/TableClass.java
new file mode 100644
index 00000000000000..3995e4908b44de
--- /dev/null
+++ b/fe/fe-core/src/main/java/org/apache/doris/plsql/objects/TableClass.java
@@ -0,0 +1,135 @@
+// 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.
+// This file is copied from
+// https://github.com/apache/hive/blob/master/hplsql/src/main/java/org/apache/hive/hplsql/objects/TableClass.java
+// and modified by Doris
+
+package org.apache.doris.plsql.objects;
+
+import org.apache.doris.plsql.ColumnDefinition;
+import org.apache.doris.plsql.Row;
+import org.apache.doris.plsql.Var;
+import org.apache.doris.plsql.objects.MethodParams.Arity;
+
+import java.util.List;
+
+
+public class TableClass implements PlClass {
+ private static final MethodDictionary methodDictionary = new MethodDictionary<>();
+ private final String typeName;
+ private final List columns;
+ private final boolean rowType;
+
+ static {
+ methodDictionary.put("first", (self, args) -> {
+ Arity.NULLARY.check("first", args);
+ return wrap(self.firstKey());
+ });
+ methodDictionary.put("last", (self, args) -> {
+ Arity.NULLARY.check("last", args);
+ return wrap(self.lastKey());
+ });
+ methodDictionary.put("next", (self, args) -> {
+ Long key = new MethodParams("next", args, Arity.UNARY).longAt(0);
+ return wrap(self.nextKey(key));
+ });
+ methodDictionary.put("prior", (self, args) -> {
+ Long key = new MethodParams("prior", args, Arity.UNARY).longAt(0);
+ return wrap(self.priorKey(key));
+ });
+ methodDictionary.put("count", (self, args) -> {
+ Arity.NULLARY.check("count", args);
+ return new Var(Long.valueOf(self.count()));
+ });
+ methodDictionary.put("exists", (self, args) -> {
+ Long key = new MethodParams("exists", args, Arity.UNARY).longAt(0);
+ return new Var(self.existsAt(key));
+ });
+ methodDictionary.put("delete", (self, args) -> {
+ Arity.max(3).check("delete", args);
+ if (args.isEmpty()) {
+ self.removeAll();
+ } else if (args.size() == 1) {
+ self.removeAt(args.get(0).value);
+ } else {
+ self.removeFromTo(args.get(0).value, args.get(1).value);
+ }
+ return null;
+ });
+ methodDictionary.put(MethodDictionary.__GETITEM__, (self, args) -> {
+ Long key = new MethodParams(MethodDictionary.__GETITEM__, args, Arity.UNARY).longAt(0);
+ Row row = self.at(key);
+ if (row == null) {
+ return Var.Null;
+ }
+ if (self.plClass().rowType()) {
+ Var var = new Var();
+ var.setType(Var.Type.ROW.name());
+ var.setValue(row);
+ return var;
+ }
+ return row.getValue(0);
+ });
+ methodDictionary.put(MethodDictionary.__SETITEM__, (self, args) -> {
+ MethodParams params = new MethodParams(MethodDictionary.__SETITEM__, args, Arity.BINARY);
+ long key = params.longAt(0);
+ if (self.plClass().rowType()) {
+ self.put(key, params.rowAt(1));
+ } else { // single column
+ Row row = new Row();
+ row.addColumn(
+ self.plClass().columns().get(0).columnName(),
+ self.plClass().columns.get(0).columnTypeString(),
+ args.get(1));
+ self.put(key, row);
+ }
+ return Var.Null;
+ });
+ }
+
+ private static Var wrap(Object result) {
+ return result != null ? new Var((Long) result) : Var.Null;
+ }
+
+ public TableClass(String typeName, List columns, boolean rowType) {
+ this.typeName = typeName;
+ this.columns = columns;
+ this.rowType = rowType;
+ }
+
+ public String typeName() {
+ return typeName;
+ }
+
+ public List columns() {
+ return columns;
+ }
+
+ @Override
+ public Table newInstance() {
+ return new Table(this);
+ }
+
+ @Override
+ public MethodDictionary methodDictionary() {
+ return methodDictionary;
+ }
+
+ public boolean rowType() {
+ return rowType;
+ }
+}
diff --git a/fe/fe-core/src/main/java/org/apache/doris/plsql/objects/UtlFile.java b/fe/fe-core/src/main/java/org/apache/doris/plsql/objects/UtlFile.java
new file mode 100644
index 00000000000000..25aeade870756f
--- /dev/null
+++ b/fe/fe-core/src/main/java/org/apache/doris/plsql/objects/UtlFile.java
@@ -0,0 +1,78 @@
+// 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.
+// This file is copied from
+// https://github.com/apache/hive/blob/master/hplsql/src/main/java/org/apache/hive/hplsql/objects/UtlFile.java
+// and modified by Doris
+
+package org.apache.doris.plsql.objects;
+
+import org.apache.doris.plsql.File;
+
+import java.io.EOFException;
+import java.io.IOException;
+
+public class UtlFile implements PlObject {
+ private final UtlFileClass plClass;
+
+ public UtlFile(UtlFileClass plClass) {
+ this.plClass = plClass;
+ }
+
+ @Override
+ public PlClass plClass() {
+ return plClass;
+ }
+
+ public File fileOpen(String dir, String name, boolean write, boolean overwrite) {
+ File file = new File();
+ if (write) {
+ file.create(dir, name, overwrite);
+ } else {
+ file.open(dir, name);
+ }
+ return file;
+ }
+
+ public void fileClose(File file) {
+ file.close();
+ }
+
+ public String getLine(File file) {
+ StringBuilder out = new StringBuilder();
+ try {
+ while (true) {
+ char c = file.readChar();
+ if (c == '\n') {
+ break;
+ }
+ out.append(c);
+ }
+ } catch (IOException e) {
+ if (!(e instanceof EOFException)) {
+ out.setLength(0);
+ }
+ }
+ return out.toString();
+ }
+
+ public void put(File file, String str, boolean newLine) {
+ file.writeString(str);
+ if (newLine) {
+ file.writeString("\n");
+ }
+ }
+}
diff --git a/fe/fe-core/src/main/java/org/apache/doris/plsql/objects/UtlFileClass.java b/fe/fe-core/src/main/java/org/apache/doris/plsql/objects/UtlFileClass.java
new file mode 100644
index 00000000000000..19d70823f9e846
--- /dev/null
+++ b/fe/fe-core/src/main/java/org/apache/doris/plsql/objects/UtlFileClass.java
@@ -0,0 +1,80 @@
+// 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.
+// This file is copied from
+// https://github.com/apache/hive/blob/master/hplsql/src/main/java/org/apache/hive/hplsql/objects/UtlFileClass.java
+// and modified by Doris
+
+package org.apache.doris.plsql.objects;
+
+import org.apache.doris.plsql.File;
+import org.apache.doris.plsql.Var;
+import org.apache.doris.plsql.objects.MethodParams.Arity;
+
+public class UtlFileClass implements PlClass {
+ public static final UtlFileClass INSTANCE = new UtlFileClass();
+ private final MethodDictionary methodDictionary = new MethodDictionary();
+
+ private UtlFileClass() {
+ methodDictionary.put("fopen", (self, args) -> {
+ MethodParams params = new MethodParams("fopen", args, Arity.min(2));
+ String dir = params.stringAt(0);
+ String name = params.stringAt(1);
+ boolean write = true;
+ boolean overwrite = false;
+ if (args.size() > 2) {
+ String mode = params.stringAt(2);
+ if (mode.equalsIgnoreCase("r")) {
+ write = false;
+ } else if (mode.equalsIgnoreCase("w")) {
+ write = true;
+ overwrite = true;
+ }
+ }
+ File file = self.fileOpen(dir, name, write, overwrite);
+ return new Var(Var.Type.FILE, file);
+ });
+ methodDictionary.put("get_line", (self, args) -> {
+ MethodParams params = new MethodParams("get_line", args, Arity.UNARY);
+ return new Var(self.getLine(params.fileAt(0)));
+ });
+ methodDictionary.put("put_line", (self, args) -> {
+ MethodParams params = new MethodParams("put_line", args, Arity.BINARY);
+ self.put(params.fileAt(0), params.stringAt(1), true);
+ return null;
+ });
+ methodDictionary.put("put", (self, args) -> {
+ MethodParams params = new MethodParams("put", args, Arity.BINARY);
+ self.put(params.fileAt(0), params.stringAt(1), false);
+ return null;
+ });
+ methodDictionary.put("fclose", (self, args) -> {
+ File file = new MethodParams("fclose", args, Arity.UNARY).fileAt(0);
+ self.fileClose(file);
+ return null;
+ });
+ }
+
+ @Override
+ public UtlFile newInstance() {
+ return new UtlFile(this);
+ }
+
+ @Override
+ public MethodDictionary methodDictionary() {
+ return methodDictionary;
+ }
+}
diff --git a/fe/fe-core/src/main/java/org/apache/doris/plsql/packages/DorisPackageRegistry.java b/fe/fe-core/src/main/java/org/apache/doris/plsql/packages/DorisPackageRegistry.java
new file mode 100644
index 00000000000000..ae0222cc40531c
--- /dev/null
+++ b/fe/fe-core/src/main/java/org/apache/doris/plsql/packages/DorisPackageRegistry.java
@@ -0,0 +1,95 @@
+// 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.
+// This file is copied from
+// https://github.com/apache/hive/blob/master/hplsql/src/main/java/org/apache/hive/hplsql/packages/HmsPackageRegistry.java
+// and modified by Doris
+
+package org.apache.doris.plsql.packages;
+
+import org.apache.doris.plsql.metastore.PlsqlMetaClient;
+import org.apache.doris.plsql.metastore.PlsqlPackage;
+import org.apache.doris.qe.ConnectContext;
+
+import org.apache.commons.lang3.StringUtils;
+import org.apache.thrift.TException;
+
+import java.util.Optional;
+
+public class DorisPackageRegistry implements PackageRegistry {
+ private final PlsqlMetaClient client;
+
+ public DorisPackageRegistry(PlsqlMetaClient client) {
+ this.client = client;
+ }
+
+ @Override
+ public Optional getPackage(String name) {
+ try {
+ PlsqlPackage pkg = findPackage(name);
+ return pkg == null
+ ? Optional.empty()
+ : Optional.of(pkg.getHeader() + ";\n" + pkg.getBody());
+ } catch (TException e) {
+ throw new RuntimeException(e);
+ }
+ }
+
+ @Override
+ public void createPackageHeader(String name, String header, boolean replace) {
+ try {
+ PlsqlPackage existing = findPackage(name);
+ if (existing != null && !replace) {
+ throw new RuntimeException("Package " + name + " already exists");
+ }
+ savePackage(name, header, "");
+ } catch (TException e) {
+ throw new RuntimeException(e);
+ }
+ }
+
+ @Override
+ public void createPackageBody(String name, String body, boolean replace) {
+ try {
+ PlsqlPackage existing = findPackage(name);
+ if (existing == null || StringUtils.isEmpty(existing.getHeader())) {
+ throw new RuntimeException("Package header does not exists " + name);
+ }
+ if (StringUtils.isNotEmpty(existing.getBody()) && !replace) {
+ throw new RuntimeException("Package body " + name + " already exists");
+ }
+ savePackage(name, existing.getHeader(), body);
+ } catch (TException e) {
+ throw new RuntimeException(e);
+ }
+ }
+
+ private PlsqlPackage findPackage(String name) throws TException {
+ return client.getPlsqlPackage(name.toUpperCase(), ConnectContext.get().getCurrentCatalog().getName(),
+ ConnectContext.get().getDatabase());
+ }
+
+ @Override
+ public void dropPackage(String name) {
+ client.dropPlsqlPackage(name, ConnectContext.get().getCurrentCatalog().getName(),
+ ConnectContext.get().getDatabase());
+ }
+
+ private void savePackage(String name, String header, String body) {
+ client.addPlsqlPackage(name.toUpperCase(), ConnectContext.get().getCurrentCatalog().getName(),
+ ConnectContext.get().getDatabase(), ConnectContext.get().getQualifiedUser(), header, body);
+ }
+}
diff --git a/fe/fe-core/src/main/java/org/apache/doris/plsql/packages/InMemoryPackageRegistry.java b/fe/fe-core/src/main/java/org/apache/doris/plsql/packages/InMemoryPackageRegistry.java
new file mode 100644
index 00000000000000..af2c1edbbca712
--- /dev/null
+++ b/fe/fe-core/src/main/java/org/apache/doris/plsql/packages/InMemoryPackageRegistry.java
@@ -0,0 +1,74 @@
+// 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.
+// This file is copied from
+// https://github.com/apache/hive/blob/master/hplsql/src/main/java/org/apache/hive/hplsql/packages/InMemoryPackageRegistry.java
+// and modified by Doris
+
+package org.apache.doris.plsql.packages;
+
+import org.apache.commons.lang3.StringUtils;
+
+import java.util.HashMap;
+import java.util.Map;
+import java.util.Optional;
+
+public class InMemoryPackageRegistry implements PackageRegistry {
+ private Map registry = new HashMap<>();
+
+ @Override
+ public Optional getPackage(String name) {
+ Source src = registry.get(name.toUpperCase());
+ return src == null
+ ? Optional.empty()
+ : Optional.of(src.header + ";\n" + src.body);
+ }
+
+ @Override
+ public void createPackageHeader(String name, String header, boolean replace) {
+ if (registry.containsKey(name) && !replace) {
+ throw new RuntimeException("Package " + name + " already exits");
+ }
+ registry.put(name, new Source(header, ""));
+ }
+
+ @Override
+ public void createPackageBody(String name, String body, boolean replace) {
+ Source existing = registry.get(name);
+ if (existing == null || StringUtils.isEmpty(existing.header)) {
+ throw new RuntimeException("Package header does not exists " + name);
+ }
+ if (StringUtils.isNotEmpty(existing.body) && !replace) {
+ throw new RuntimeException("Package body " + name + " already exits");
+ }
+ registry.getOrDefault(name, new Source("", "")).body = body;
+ }
+
+ @Override
+ public void dropPackage(String name) {
+ registry.remove(name);
+ }
+
+ private static class Source {
+ String header;
+ String body;
+
+ public Source(String header, String body) {
+ this.header = header;
+ this.body = body;
+ }
+ }
+}
diff --git a/fe/fe-core/src/main/java/org/apache/doris/plsql/packages/PackageRegistry.java b/fe/fe-core/src/main/java/org/apache/doris/plsql/packages/PackageRegistry.java
new file mode 100644
index 00000000000000..2e3ac3e662f97d
--- /dev/null
+++ b/fe/fe-core/src/main/java/org/apache/doris/plsql/packages/PackageRegistry.java
@@ -0,0 +1,33 @@
+// 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.
+// This file is copied from
+// https://github.com/apache/hive/blob/master/hplsql/src/main/java/org/apache/hive/hplsql/packages/PackageRegistry.java
+// and modified by Doris
+
+package org.apache.doris.plsql.packages;
+
+import java.util.Optional;
+
+public interface PackageRegistry {
+ Optional getPackage(String name);
+
+ void createPackageHeader(String name, String header, boolean replace);
+
+ void createPackageBody(String name, String body, boolean replace);
+
+ void dropPackage(String name);
+}
diff --git a/fe/fe-core/src/main/java/org/apache/doris/qe/AutoCloseConnectContext.java b/fe/fe-core/src/main/java/org/apache/doris/qe/AutoCloseConnectContext.java
index e9aaea7415e4a9..ffebe97e706548 100644
--- a/fe/fe-core/src/main/java/org/apache/doris/qe/AutoCloseConnectContext.java
+++ b/fe/fe-core/src/main/java/org/apache/doris/qe/AutoCloseConnectContext.java
@@ -29,6 +29,11 @@ public AutoCloseConnectContext(ConnectContext connectContext) {
connectContext.setThreadLocalInfo();
}
+ public void call() {
+ // try (AutoCloseConnectContext autoCloseCtx = new AutoCloseConnectContext(context)) {
+ // will report autoCloseCtx is not used, so call an empty method.
+ }
+
@Override
public void close() {
ConnectContext.remove();
diff --git a/fe/fe-core/src/main/java/org/apache/doris/qe/ConnectContext.java b/fe/fe-core/src/main/java/org/apache/doris/qe/ConnectContext.java
index 95f0c5831a0bde..15e25cfdf3a088 100644
--- a/fe/fe-core/src/main/java/org/apache/doris/qe/ConnectContext.java
+++ b/fe/fe-core/src/main/java/org/apache/doris/qe/ConnectContext.java
@@ -50,6 +50,8 @@
import org.apache.doris.nereids.StatementContext;
import org.apache.doris.nereids.stats.StatsErrorEstimator;
import org.apache.doris.nereids.trees.expressions.literal.Literal;
+import org.apache.doris.plsql.Exec;
+import org.apache.doris.plsql.executor.PlSqlOperation;
import org.apache.doris.plugin.audit.AuditEvent.AuditEventBuilder;
import org.apache.doris.resource.Tag;
import org.apache.doris.service.arrowflight.results.FlightSqlChannel;
@@ -197,6 +199,8 @@ public enum ConnectType {
// If set to false, the system will not restrict query resources.
private boolean isResourceTagsSet = false;
+ private PlSqlOperation plSqlOperation = null;
+
private String sqlHash;
private JSONObject minidump = null;
@@ -226,6 +230,8 @@ public enum ConnectType {
// but the internal implementation will call the logic of `AlterTable`.
// In this case, `skipAuth` needs to be set to `true` to skip the permission check of `AlterTable`
private boolean skipAuth = false;
+ private Exec exec;
+ private boolean runProcedure = false;
public void setUserQueryTimeout(int queryTimeout) {
if (queryTimeout > 0) {
@@ -344,6 +350,18 @@ public ConnectContext(StreamConnection connection) {
init();
}
+ public ConnectContext cloneContext() {
+ ConnectContext context = new ConnectContext();
+ context.mysqlChannel = mysqlChannel;
+ context.setSessionVariable(VariableMgr.cloneSessionVariable(sessionVariable)); // deep copy
+ context.setEnv(env);
+ context.setDatabase(currentDb);
+ context.setQualifiedUser(qualifiedUser);
+ context.setCurrentUserIdentity(currentUserIdentity);
+ context.setProcedureExec(exec);
+ return context;
+ }
+
public boolean isTxnModel() {
return txnEntry != null && txnEntry.isTxnModel();
}
@@ -776,6 +794,13 @@ public StmtExecutor getExecutor() {
return executor;
}
+ public PlSqlOperation getPlSqlOperation() {
+ if (plSqlOperation == null) {
+ plSqlOperation = new PlSqlOperation();
+ }
+ return plSqlOperation;
+ }
+
protected void closeChannel() {
if (mysqlChannel != null) {
mysqlChannel.close();
@@ -1135,6 +1160,22 @@ public void setSkipAuth(boolean skipAuth) {
this.skipAuth = skipAuth;
}
+ public boolean isRunProcedure() {
+ return runProcedure;
+ }
+
+ public void setRunProcedure(boolean runProcedure) {
+ this.runProcedure = runProcedure;
+ }
+
+ public void setProcedureExec(Exec exec) {
+ this.exec = exec;
+ }
+
+ public Exec getProcedureExec() {
+ return exec;
+ }
+
public int getNetReadTimeout() {
return this.sessionVariable.getNetReadTimeout();
}
diff --git a/fe/fe-core/src/main/java/org/apache/doris/qe/ConnectProcessor.java b/fe/fe-core/src/main/java/org/apache/doris/qe/ConnectProcessor.java
index b297d192c422c1..442b53669e02bc 100644
--- a/fe/fe-core/src/main/java/org/apache/doris/qe/ConnectProcessor.java
+++ b/fe/fe-core/src/main/java/org/apache/doris/qe/ConnectProcessor.java
@@ -93,6 +93,10 @@ public ConnectProcessor(ConnectContext context) {
this.ctx = context;
}
+ public ConnectContext getConnectContext() {
+ return ctx;
+ }
+
// change current database of this session.
protected void handleInitDb(String fullDbName) {
String catalogName = null;
@@ -173,6 +177,14 @@ protected void auditAfterExec(String origStmt, StatementBase parsedStmt,
// only throw an exception when there is a problem interacting with the requesting client
protected void handleQuery(MysqlCommand mysqlCommand, String originStmt) {
+ try {
+ executeQuery(mysqlCommand, originStmt);
+ } catch (Exception ignored) {
+ // saved use handleQueryException
+ }
+ }
+
+ public void executeQuery(MysqlCommand mysqlCommand, String originStmt) throws Exception {
if (MetricRepo.isInit) {
MetricRepo.COUNTER_REQUEST_ALL.increase(1L);
}
@@ -274,11 +286,9 @@ protected void handleQuery(MysqlCommand mysqlCommand, String originStmt) {
handleQueryException(throwable, auditStmt, executor.getParsedStmt(),
executor.getQueryStatisticsForAuditLog());
// execute failed, skip remaining stmts
- break;
+ throw throwable;
}
-
}
-
}
private String convertOriginStmt(String originStmt) {
diff --git a/fe/fe-core/src/main/java/org/apache/doris/qe/StmtExecutor.java b/fe/fe-core/src/main/java/org/apache/doris/qe/StmtExecutor.java
index 173a4c1d53f6f9..c579c894b6804f 100644
--- a/fe/fe-core/src/main/java/org/apache/doris/qe/StmtExecutor.java
+++ b/fe/fe-core/src/main/java/org/apache/doris/qe/StmtExecutor.java
@@ -1561,6 +1561,13 @@ private void sendResult(boolean isOutfileQuery, boolean isSendFields, Queriable
}
return;
}
+
+ if (context.isRunProcedure()) {
+ // plsql will get the returned results without sending them to mysql client.
+ // see org/apache/doris/plsql/executor/DorisRowResult.java
+ return;
+ }
+
while (true) {
// register the fetch result time.
profile.getSummaryProfile().setTempStartTime();
@@ -2833,6 +2840,18 @@ private List convertResultBatchToResultRows(TResultBatch batch) {
return resultRows;
}
+ public Coordinator getCoord() {
+ return coord;
+ }
+
+ public List getColumns() {
+ return parsedStmt.getColLabels();
+ }
+
+ public List getReturnTypes() {
+ return exprToType(parsedStmt.getResultExprs());
+ }
+
public SummaryProfile getSummaryProfile() {
return profile.getSummaryProfile();
}
diff --git a/fe/fe-core/src/main/java/org/apache/doris/service/FrontendServiceImpl.java b/fe/fe-core/src/main/java/org/apache/doris/service/FrontendServiceImpl.java
index ca57b35c0880aa..798d3900ce8e71 100644
--- a/fe/fe-core/src/main/java/org/apache/doris/service/FrontendServiceImpl.java
+++ b/fe/fe-core/src/main/java/org/apache/doris/service/FrontendServiceImpl.java
@@ -82,6 +82,9 @@
import org.apache.doris.persist.gson.GsonUtils;
import org.apache.doris.planner.OlapTableSink;
import org.apache.doris.planner.StreamLoadPlanner;
+import org.apache.doris.plsql.metastore.PlsqlPackage;
+import org.apache.doris.plsql.metastore.PlsqlProcedureKey;
+import org.apache.doris.plsql.metastore.PlsqlStoredProcedure;
import org.apache.doris.qe.ConnectContext;
import org.apache.doris.qe.ConnectContext.ConnectType;
import org.apache.doris.qe.ConnectProcessor;
@@ -110,6 +113,8 @@
import org.apache.doris.task.StreamLoadTask;
import org.apache.doris.thrift.FrontendService;
import org.apache.doris.thrift.FrontendServiceVersion;
+import org.apache.doris.thrift.TAddPlsqlPackageRequest;
+import org.apache.doris.thrift.TAddPlsqlStoredProcedureRequest;
import org.apache.doris.thrift.TAutoIncrementRangeRequest;
import org.apache.doris.thrift.TAutoIncrementRangeResult;
import org.apache.doris.thrift.TBackend;
@@ -131,6 +136,8 @@
import org.apache.doris.thrift.TDescribeTableResult;
import org.apache.doris.thrift.TDescribeTablesParams;
import org.apache.doris.thrift.TDescribeTablesResult;
+import org.apache.doris.thrift.TDropPlsqlPackageRequest;
+import org.apache.doris.thrift.TDropPlsqlStoredProcedureRequest;
import org.apache.doris.thrift.TExecPlanFragmentParams;
import org.apache.doris.thrift.TFeResult;
import org.apache.doris.thrift.TFetchResourceResult;
@@ -186,6 +193,8 @@
import org.apache.doris.thrift.TOlapTablePartition;
import org.apache.doris.thrift.TPipelineFragmentParams;
import org.apache.doris.thrift.TPipelineWorkloadGroup;
+import org.apache.doris.thrift.TPlsqlPackageResult;
+import org.apache.doris.thrift.TPlsqlStoredProcedureResult;
import org.apache.doris.thrift.TPrivilegeCtrl;
import org.apache.doris.thrift.TPrivilegeHier;
import org.apache.doris.thrift.TPrivilegeStatus;
@@ -2934,6 +2943,113 @@ private TRestoreSnapshotResult restoreSnapshotImpl(TRestoreSnapshotRequest reque
return result;
}
+ @Override
+ public TPlsqlStoredProcedureResult addPlsqlStoredProcedure(TAddPlsqlStoredProcedureRequest request) {
+ TPlsqlStoredProcedureResult result = new TPlsqlStoredProcedureResult();
+ TStatus status = new TStatus(TStatusCode.OK);
+ result.setStatus(status);
+ if (!Env.getCurrentEnv().isMaster()) {
+ status.setStatusCode(TStatusCode.NOT_MASTER);
+ status.addToErrorMsgs(NOT_MASTER_ERR_MSG);
+ LOG.error("failed to addPlsqlStoredProcedure:{}, request:{}, backend:{}",
+ NOT_MASTER_ERR_MSG, request, getClientAddrAsString());
+ return result;
+ }
+
+ if (!request.isSetPlsqlStoredProcedure()) {
+ status.setStatusCode(TStatusCode.INVALID_ARGUMENT);
+ status.addToErrorMsgs("missing stored procedure.");
+ return result;
+ }
+ try {
+ Env.getCurrentEnv().getPlsqlManager()
+ .addPlsqlStoredProcedure(PlsqlStoredProcedure.fromThrift(request.getPlsqlStoredProcedure()),
+ request.isSetIsForce() && request.isIsForce());
+ } catch (RuntimeException e) {
+ status.setStatusCode(TStatusCode.ALREADY_EXIST);
+ status.addToErrorMsgs(e.getMessage());
+ return result;
+ }
+ return result;
+ }
+
+ @Override
+ public TPlsqlStoredProcedureResult dropPlsqlStoredProcedure(TDropPlsqlStoredProcedureRequest request) {
+ TPlsqlStoredProcedureResult result = new TPlsqlStoredProcedureResult();
+ TStatus status = new TStatus(TStatusCode.OK);
+ result.setStatus(status);
+ if (!Env.getCurrentEnv().isMaster()) {
+ status.setStatusCode(TStatusCode.NOT_MASTER);
+ status.addToErrorMsgs(NOT_MASTER_ERR_MSG);
+ LOG.error("failed to dropPlsqlStoredProcedure:{}, request:{}, backend:{}",
+ NOT_MASTER_ERR_MSG, request, getClientAddrAsString());
+ return result;
+ }
+
+ if (!request.isSetPlsqlProcedureKey()) {
+ status.setStatusCode(TStatusCode.INVALID_ARGUMENT);
+ status.addToErrorMsgs("missing stored key.");
+ return result;
+ }
+
+ Env.getCurrentEnv().getPlsqlManager().dropPlsqlStoredProcedure(PlsqlProcedureKey.fromThrift(
+ request.getPlsqlProcedureKey()));
+ return result;
+ }
+
+ @Override
+ public TPlsqlPackageResult addPlsqlPackage(TAddPlsqlPackageRequest request) throws TException {
+ TPlsqlPackageResult result = new TPlsqlPackageResult();
+ TStatus status = new TStatus(TStatusCode.OK);
+ result.setStatus(status);
+ if (!Env.getCurrentEnv().isMaster()) {
+ status.setStatusCode(TStatusCode.NOT_MASTER);
+ status.addToErrorMsgs(NOT_MASTER_ERR_MSG);
+ LOG.error("failed to addPlsqlPackage:{}, request:{}, backend:{}",
+ NOT_MASTER_ERR_MSG, request, getClientAddrAsString());
+ return result;
+ }
+
+ if (!request.isSetPlsqlPackage()) {
+ status.setStatusCode(TStatusCode.INVALID_ARGUMENT);
+ status.addToErrorMsgs("missing plsql package.");
+ return result;
+ }
+
+ try {
+ Env.getCurrentEnv().getPlsqlManager().addPackage(PlsqlPackage.fromThrift(request.getPlsqlPackage()),
+ request.isSetIsForce() && request.isIsForce());
+ } catch (RuntimeException e) {
+ status.setStatusCode(TStatusCode.ALREADY_EXIST);
+ status.addToErrorMsgs(e.getMessage());
+ return result;
+ }
+ return result;
+ }
+
+ @Override
+ public TPlsqlPackageResult dropPlsqlPackage(TDropPlsqlPackageRequest request) throws TException {
+ TPlsqlPackageResult result = new TPlsqlPackageResult();
+ TStatus status = new TStatus(TStatusCode.OK);
+ result.setStatus(status);
+ if (!Env.getCurrentEnv().isMaster()) {
+ status.setStatusCode(TStatusCode.NOT_MASTER);
+ status.addToErrorMsgs(NOT_MASTER_ERR_MSG);
+ LOG.error("failed to dropPlsqlPackage:{}, request:{}, backend:{}",
+ NOT_MASTER_ERR_MSG, request, getClientAddrAsString());
+ return result;
+ }
+
+ if (!request.isSetPlsqlProcedureKey()) {
+ status.setStatusCode(TStatusCode.INVALID_ARGUMENT);
+ status.addToErrorMsgs("missing stored key.");
+ return result;
+ }
+
+ Env.getCurrentEnv().getPlsqlManager().dropPackage(PlsqlProcedureKey.fromThrift(request.getPlsqlProcedureKey()));
+ return result;
+ }
+
public TGetMasterTokenResult getMasterToken(TGetMasterTokenRequest request) throws TException {
String clientAddr = getClientAddrAsString();
LOG.debug("receive get master token request: {}", request);
diff --git a/gensrc/thrift/FrontendService.thrift b/gensrc/thrift/FrontendService.thrift
index 4c0ec0019e1e12..ea0300d23e5939 100644
--- a/gensrc/thrift/FrontendService.thrift
+++ b/gensrc/thrift/FrontendService.thrift
@@ -1157,6 +1157,55 @@ struct TRestoreSnapshotResult {
2: optional Types.TNetworkAddress master_address
}
+struct TPlsqlStoredProcedure {
+ 1: optional string name
+ 2: optional string catalogName
+ 3: optional string dbName
+ 4: optional string ownerName
+ 5: optional string source
+}
+
+struct TPlsqlPackage {
+ 1: optional string name
+ 2: optional string catalogName
+ 3: optional string dbName
+ 4: optional string ownerName
+ 5: optional string header
+ 6: optional string body
+}
+
+struct TPlsqlProcedureKey {
+ 1: optional string name
+ 2: optional string catalogName
+ 3: optional string dbName
+}
+
+struct TAddPlsqlStoredProcedureRequest {
+ 1: optional TPlsqlStoredProcedure plsqlStoredProcedure
+ 2: optional bool isForce
+}
+
+struct TDropPlsqlStoredProcedureRequest {
+ 1: optional TPlsqlProcedureKey plsqlProcedureKey
+}
+
+struct TPlsqlStoredProcedureResult {
+ 1: optional Status.TStatus status
+}
+
+struct TAddPlsqlPackageRequest {
+ 1: optional TPlsqlPackage plsqlPackage
+ 2: optional bool isForce
+}
+
+struct TDropPlsqlPackageRequest {
+ 1: optional TPlsqlProcedureKey plsqlProcedureKey
+}
+
+struct TPlsqlPackageResult {
+ 1: optional Status.TStatus status
+}
+
struct TGetMasterTokenRequest {
1: optional string cluster
2: optional string user
@@ -1400,6 +1449,11 @@ service FrontendService {
TGetTabletReplicaInfosResult getTabletReplicaInfos(1: TGetTabletReplicaInfosRequest request)
+ TPlsqlStoredProcedureResult addPlsqlStoredProcedure(1: TAddPlsqlStoredProcedureRequest request)
+ TPlsqlStoredProcedureResult dropPlsqlStoredProcedure(1: TDropPlsqlStoredProcedureRequest request)
+ TPlsqlPackageResult addPlsqlPackage(1: TAddPlsqlPackageRequest request)
+ TPlsqlPackageResult dropPlsqlPackage(1: TDropPlsqlPackageRequest request)
+
TGetMasterTokenResult getMasterToken(1: TGetMasterTokenRequest request)
TGetBinlogLagResult getBinlogLag(1: TGetBinlogLagRequest request)
diff --git a/regression-test/data/plsql_p0/test_plsql_loop_cursor.out b/regression-test/data/plsql_p0/test_plsql_loop_cursor.out
new file mode 100644
index 00000000000000..008dacb74dbc30
--- /dev/null
+++ b/regression-test/data/plsql_p0/test_plsql_loop_cursor.out
@@ -0,0 +1,4 @@
+-- This file is automatically generated. You should know what you did if you want to edit this
+-- !select --
+777 4
+
diff --git a/regression-test/suites/plsql_p0/test_plsql.groovy b/regression-test/suites/plsql_p0/test_plsql.groovy
new file mode 100644
index 00000000000000..e976dcd37e3a75
--- /dev/null
+++ b/regression-test/suites/plsql_p0/test_plsql.groovy
@@ -0,0 +1,74 @@
+// 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.
+
+suite("test_plsql") {
+ // TODO:
+ // 1. doris parser support declare var
+ // 2. Stmt.statement() support insert into var, impl Stmt.getIntoCount(), Stmt.populateVariable()
+
+ // def tbl = "plsql_tbl"
+ // sql "DROP TABLE IF EXISTS ${tbl}"
+ // sql """
+ // create table ${tbl} (id int, name varchar(20)) DUPLICATE key(`id`) distributed by hash (`id`) buckets 4
+ // properties ("replication_num"="1");
+ // """
+
+ // sql "declare id INT default = 0;"
+ // sql """
+ // CREATE OR REPLACE PROCEDURE procedure_insert(IN name STRING, OUT result int)
+ // BEGIN
+ // select k1 into result from test_query_db.test where k7 = name;
+ // END;
+ // """
+ // sql "call procedure_insert('wangynnsf', id)"
+ // qt_select "select * from test_query_db.test where k1 = id"
+
+ // sql """
+ // CREATE OR REPLACE PROCEDURE cursor_demo()
+ // BEGIN
+ // DECLARE a CHAR(32);
+ // DECLARE b, c INT;
+ // DECLARE cur1 CURSOR FOR SELECT k7, k3 FROM test_query_db.test where k3 > 0 order by k3, k7;
+ // DECLARE cur2 CURSOR FOR SELECT k4 FROM test_query_db.baseall where k4 between 0 and 21011903 order by k4;
+
+ // OPEN cur1;
+ // OPEN cur2;
+
+ // read_loop: LOOP
+ // FETCH cur1 INTO a, b;
+ // IF(SQLCODE != 0) THEN
+ // LEAVE read_loop;
+ // END IF;
+ // FETCH cur2 INTO c;
+ // IF(SQLCODE != 0) THEN
+ // LEAVE read_loop;
+ // END IF;
+ // IF b < c THEN
+ // INSERT INTO ${tbl} (`name`,`id`) VALUES (a,b);
+ // ELSE
+ // INSERT INTO ${tbl} (`name`, `id`) VALUES (a,c);
+ // END IF;
+ // END LOOP;
+
+ // CLOSE cur1;
+ // CLOSE cur2;
+ // END;
+ // """
+
+ // sql "call cursor_demo()"
+ // qt_select """select * from ${tbl} order by 1, 2""";
+}
diff --git a/regression-test/suites/plsql_p0/test_plsql_loop_cursor.groovy b/regression-test/suites/plsql_p0/test_plsql_loop_cursor.groovy
new file mode 100644
index 00000000000000..0faa8228fa773e
--- /dev/null
+++ b/regression-test/suites/plsql_p0/test_plsql_loop_cursor.groovy
@@ -0,0 +1,62 @@
+// 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.
+
+suite("test_plsql_loop_cursor") {
+ def tableName = "plsql_tbl"
+ sql "DROP TABLE IF EXISTS ${tableName}"
+ sql """
+ create table ${tableName} (id int, name varchar(20)) DUPLICATE key(`id`) distributed by hash (`id`) buckets 4
+ properties ("replication_num"="1");
+ """
+
+ sql """
+ CREATE OR REPLACE PROCEDURE procedure_insert(IN id int, IN name STRING)
+ BEGIN
+ INSERT INTO ${tableName} VALUES(id, name);
+ END;
+ """
+ sql """call procedure_insert(111, "plsql111")"""
+ sql """call procedure_insert(222, "plsql222")"""
+ sql """call procedure_insert(333, "plsql333")"""
+ sql """call procedure_insert(111, "plsql333")"""
+ qt_select "select sum(id), count(1) from ${tableName}"
+
+ sql """
+ CREATE OR REPLACE PROCEDURE procedure_cursor_select(IN id_arg INT, IN name_arg STRING)
+ BEGIN
+ DECLARE a INT;
+ DECLARE b, c STRING;
+
+ DECLARE cur1 CURSOR FOR select * from ${tableName} where id=id_arg limit 5;
+ OPEN cur1;
+ read_loop: LOOP
+ FETCH cur1 INTO a, b;
+ IF(SQLCODE != 0) THEN
+ LEAVE read_loop;
+ END IF;
+ print a, b;
+ END LOOP;
+
+ CLOSE cur1;
+
+ END;
+ """
+
+ // TODO support print
+ sql """call procedure_cursor_select(111, "plsql111")"""
+ sql """call procedure_cursor_select(111, "plsql333")"""
+}