dbeaver#22963 Return datasource in CubridProcedureParameter.java file

longhaseng52 · Mar 21, 2024 · 3aa2618 · 3aa2618
1 parent 7289e3f
commit 3aa2618
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/.../org.jkiss.dbeaver.ext.cubrid/src/org/jkiss/dbeaver/ext/cubrid/model/CubridProcedure.java b/.../org.jkiss.dbeaver.ext.cubrid/src/org/jkiss/dbeaver/ext/cubrid/model/CubridProcedure.java
@@ -128,7 +128,7 @@ public void loadProcedureColumns(@NotNull DBRProgressMonitor monitor) throws DBE
                         String dataType = JDBCUtils.safeGetString(dbResult, "data_type");
                         String mode = JDBCUtils.safeGetString(dbResult, "mode");
                         String comment = JDBCUtils.safeGetString(dbResult, "comment");
-                        addColumn(new CubridProcedureParameter(getName(), argName, dataType, mode, comment));
+                        addColumn(new CubridProcedureParameter(this, getName(), argName, dataType, mode, comment));
                     }
                 }
             }

diff --git a/...s.dbeaver.ext.cubrid/src/org/jkiss/dbeaver/ext/cubrid/model/CubridProcedureParameter.java b/...s.dbeaver.ext.cubrid/src/org/jkiss/dbeaver/ext/cubrid/model/CubridProcedureParameter.java
@@ -34,11 +34,13 @@ public class CubridProcedureParameter implements DBSProcedureParameter
     private CubridProcedure procedure;
 
     public CubridProcedureParameter(
+            @NotNull CubridProcedure procedure,
             @NotNull String procName,
             @NotNull String argName,
             @NotNull String dataType,
             @NotNull String mode,
             @Nullable String comment) {
+        this.procedure = procedure;
         this.procName = procName;
         this.argName = argName;
         this.dataType = dataType;
@@ -99,7 +101,7 @@ public DBSProcedureParameterKind getParameterKind() {
     @Nullable
     @Override
     public DBPDataSource getDataSource() {
-        return null;
+        return procedure.getDataSource();
     }
 
     @NotNull