diff --git a/dolphinscheduler-task-plugin/dolphinscheduler-task-datax/src/main/java/org/apache/dolphinscheduler/plugin/task/datax/DataxTask.java b/dolphinscheduler-task-plugin/dolphinscheduler-task-datax/src/main/java/org/apache/dolphinscheduler/plugin/task/datax/DataxTask.java index 4b989a5503c5..d8aadffe96de 100644 --- a/dolphinscheduler-task-plugin/dolphinscheduler-task-datax/src/main/java/org/apache/dolphinscheduler/plugin/task/datax/DataxTask.java +++ b/dolphinscheduler-task-plugin/dolphinscheduler-task-datax/src/main/java/org/apache/dolphinscheduler/plugin/task/datax/DataxTask.java @@ -534,7 +534,7 @@ public String[] tryExecuteSqlResolveColumnNames(DbType sourceType, BaseConnectio int num = md.getColumnCount(); columnNames = new String[num]; for (int i = 1; i <= num; i++) { - columnNames[i - 1] = md.getColumnName(i).replace("t.", ""); + columnNames[i - 1] = md.getColumnLabel(i).replace("t.", ""); } } catch (SQLException | ExecutionException e) { log.error(e.getMessage(), e); diff --git a/dolphinscheduler-task-plugin/dolphinscheduler-task-datax/src/test/java/org/apache/dolphinscheduler/plugin/task/datax/DataxTaskTest.java b/dolphinscheduler-task-plugin/dolphinscheduler-task-datax/src/test/java/org/apache/dolphinscheduler/plugin/task/datax/DataxTaskTest.java index fbd5c59aa713..3b1dd82c9283 100644 --- a/dolphinscheduler-task-plugin/dolphinscheduler-task-datax/src/test/java/org/apache/dolphinscheduler/plugin/task/datax/DataxTaskTest.java +++ b/dolphinscheduler-task-plugin/dolphinscheduler-task-datax/src/test/java/org/apache/dolphinscheduler/plugin/task/datax/DataxTaskTest.java @@ -213,7 +213,7 @@ public void testTryExecuteSqlResolveColumnNames() throws Exception { ResultSetMetaData md = mock(ResultSetMetaData.class); when(md.getColumnCount()).thenReturn(1); - when(md.getColumnName(eq(1))).thenReturn("something"); + when(md.getColumnLabel(eq(1))).thenReturn("something"); ResultSet resultSet = mock(ResultSet.class); when(resultSet.getMetaData()).thenReturn(md);