From 21ad45ba352ca9ee0370ce69dcb57acc7b69a02f Mon Sep 17 00:00:00 2001
From: "Destiny.Lin" <Destiny.Lin@fanruan.com>
Date: Tue, 11 Apr 2023 19:59:23 +0800
Subject: [PATCH] =?UTF-8?q?REPORT-93649=20=E5=85=B3=E9=97=AD=E6=95=B0?=
 =?UTF-8?q?=E6=8D=AE=E8=BF=9E=E6=8E=A5=E6=9D=83=E9=99=90=EF=BC=8C=E6=A8=A1?=
 =?UTF-8?q?=E6=9D=BF=E6=95=B0=E6=8D=AE=E9=9B=86=E6=95=B0=E6=8D=AE=E4=BB=8D?=
 =?UTF-8?q?=E7=84=B6=E5=8F=AF=E4=BB=A5=E9=A2=84=E8=A7=88?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../design/data/BasicTableDataTreePane.java   | 46 ++++++++++++++++++-
 1 file changed, 45 insertions(+), 1 deletion(-)

diff --git a/designer-base/src/main/java/com/fr/design/data/BasicTableDataTreePane.java b/designer-base/src/main/java/com/fr/design/data/BasicTableDataTreePane.java
index b3a076a92..af9850172 100644
--- a/designer-base/src/main/java/com/fr/design/data/BasicTableDataTreePane.java
+++ b/designer-base/src/main/java/com/fr/design/data/BasicTableDataTreePane.java
@@ -5,14 +5,17 @@ import com.fr.base.TableData;
 import com.fr.base.svg.IconUtils;
 import com.fr.data.MultiResultTableData;
 import com.fr.data.TableDataSource;
+import com.fr.data.impl.DBTableData;
 import com.fr.design.DesignModelAdapter;
 import com.fr.design.actions.UpdateAction;
 import com.fr.design.data.datapane.TableDataCreatorProducer;
 import com.fr.design.data.datapane.TableDataNameObjectCreator;
 import com.fr.design.data.datapane.TableDataSourceOP;
 import com.fr.design.data.datapane.TableDataTree;
+import com.fr.design.data.datapane.auth.TableDataAuthHelper;
 import com.fr.design.data.tabledata.ResponseDataSourceChange;
 import com.fr.design.data.tabledata.tabledatapane.AbstractTableDataPane;
+import com.fr.design.data.tabledata.tabledatapane.loading.TableDataLoadingPane;
 import com.fr.design.data.tabledata.wrapper.MultiResultTableDataWrapper;
 import com.fr.design.data.tabledata.wrapper.TableDataWrapper;
 import com.fr.design.data.tabledata.wrapper.TemplateTableDataWrapper;
@@ -42,6 +45,8 @@ import javax.swing.DefaultCellEditor;
 import javax.swing.Icon;
 import javax.swing.JComponent;
 import javax.swing.JTree;
+import javax.swing.SwingUtilities;
+import javax.swing.SwingWorker;
 import javax.swing.event.CellEditorListener;
 import javax.swing.event.ChangeEvent;
 import javax.swing.tree.TreeCellEditor;
@@ -51,6 +56,7 @@ import java.awt.event.FocusAdapter;
 import java.awt.event.FocusEvent;
 import java.awt.event.KeyAdapter;
 import java.awt.event.KeyEvent;
+import java.util.Collection;
 import java.util.EventObject;
 import java.util.HashMap;
 import java.util.HashSet;
@@ -449,7 +455,45 @@ public abstract class BasicTableDataTreePane extends DockingView implements Resp
                         storeProcedureDataWrapper.previewData(MultiResultTableDataWrapper.PREVIEW_ALL);
                     }
                 } else {
-                    ((TableDataWrapper) data).previewData();
+                    TableDataWrapper wrapper = ((TableDataWrapper) data);
+                    if (TableDataAuthHelper.needCheckAuthWhenEdit(wrapper.getTableData())) {
+                        // 先打开一个Loading面板
+                        TableDataLoadingPane loadingPane = new TableDataLoadingPane();
+                        BasicDialog loadingDialog = loadingPane.showLargeWindow(SwingUtilities.getWindowAncestor(BasicTableDataTreePane.this), null);
+                        // 查询权限
+                        new SwingWorker<Boolean, Void>() {
+                            @Override
+                            protected Boolean doInBackground() throws Exception {
+                                // 获取无权限连接名称集合
+                                Collection<String> noAuthConnections = TableDataAuthHelper.getNoAuthConnections();
+                                // 获取当前数据集对应的数据连接名称
+                                String connectionName = TableDataAuthHelper.getConnectionNameByDBTableData((DBTableData) wrapper.getTableData());
+                                return !noAuthConnections.contains(connectionName);
+                            }
+
+                            @Override
+                            protected void done() {
+                                try {
+                                    Boolean hasAuth = get();
+                                    if (hasAuth) {
+                                        // 有权限时，关闭Loading面板，打开编辑面板
+                                        loadingDialog.setVisible(false);
+                                        wrapper.previewData();
+                                    } else {
+                                        // 无权限时，给出无权限提示
+                                        loadingPane.switchTo(TableDataLoadingPane.NO_AUTH_PANE_NAME);
+                                    }
+                                } catch (Exception e) {
+                                    FineLoggerFactory.getLogger().error("loading connection error in remote design", e.getMessage());
+                                    // 查询权限失败时，给出报错提示
+                                    loadingPane.switchTo(TableDataLoadingPane.ERROR_NAME);
+                                }
+                            }
+                        }.execute();
+                        loadingDialog.setVisible(true);
+                    } else {
+                        wrapper.previewData();
+                    }
                 }
 
             } catch (Exception ex) {