REPORT-13202 包含插件控件的模板切换环境后存储出错

designer-base/src/main/java/com/fr/design/file/HistoryTemplateListCache.java

@@ -1,6 +1,8 @@
 package com.fr.design.file;
 
 import com.fr.base.chart.chartdata.CallbackEvent;
+import com.fr.base.io.BaseBook;
+import com.fr.base.io.IOFile;
 import com.fr.design.DesignerEnvManager;
 import com.fr.design.base.mode.DesignModeContext;
 import com.fr.design.data.DesignTableDataManager;
@@ -18,10 +20,14 @@ import com.fr.stable.CoreConstants;
 import com.fr.stable.StringUtils;
 import com.fr.third.org.apache.commons.io.FilenameUtils;
 
+import java.io.ByteArrayInputStream;
+import java.io.ByteArrayOutputStream;
 import java.util.ArrayList;
 import java.util.Collections;
+import java.util.HashMap;
 import java.util.List;
 import java.util.ListIterator;
+import java.util.Map;
 
 /**
  * 历史模板缓存
@@ -305,4 +311,73 @@ public class HistoryTemplateListCache implements CallbackEvent {
         }
         return true;
     }
+
+    /**
+     * 切换环境时暂存打开的模板内容，key 是在历史中的index，value 是模板xml 内容byte[]
+     */
+    private Map<Integer, byte[]> bytesMap;
+
+    /**
+     * 切换环境前将正在编辑的模板暂存起来，并且在新环境中重新读取一遍，暂存的不是模板文件的内容而是模板对象的内容
+     * <p>
+     * 防止新环境加载到的同名 Class 和模板对象中的 Class 不一致，在新环境存储失败
+     *
+     * @see HistoryTemplateListCache#load()
+     */
+    public void stash() {
+        FineLoggerFactory.getLogger().info("Env Change Template Stashing...");
+        if (bytesMap == null) {
+            bytesMap = new HashMap<Integer, byte[]>();
+        } else {
+            bytesMap.clear();
+        }
+        int size = historyList.size();
+        for (int i = 0; i < size; i++) {
+            JTemplate<?, ?> template = historyList.get(i);
+            try {
+                ByteArrayOutputStream outputStream = new ByteArrayOutputStream();
+                BaseBook target = template.getTarget();
+                if (target != null) {
+                    target.export(outputStream);
+                    bytesMap.put(i, outputStream.toByteArray());
+                }
+                // 如果 target == null 那么这个模板是被模板内存优化功能处理过的，不用处理
+            } catch (Exception e) {
+                FineLoggerFactory.getLogger().error(e.getMessage(), e);
+            }
+        }
+        FineLoggerFactory.getLogger().info("Env Change Template Stashed.");
+    }
+
+    /**
+     * 切换环境前将正在编辑的模板暂存起来后，在新环境重新读取一遍
+     * <p>
+     * 防止新环境加载到的同名 Class 和模板对象中的 Class 不一致，在新环境存储失败
+     *
+     * @see HistoryTemplateListCache#stash()
+     */
+    public void load() {
+        FineLoggerFactory.getLogger().info("Env Change Template Loading...");
+        if (bytesMap != null && bytesMap.size() != 0) {
+            int size = historyList.size();
+            for (int i = 0; i < size; i++) {
+                try {
+                    byte[] bytes = bytesMap.get(i);
+                    // 可能有模板 stash 失败的情况，在 load 的时候不更新它
+                    if (bytes == null) {
+                        continue;
+                    }
+                    ByteArrayInputStream inputStream = new ByteArrayInputStream(bytes);
+                    // todo readStream 这个行为应该上升到 BaseBook 上
+                    ((IOFile) historyList.get(i).getTarget()).readStream(inputStream);
+                } catch (Exception e) {
+                    FineLoggerFactory.getLogger().error(e.getMessage(), e);
+                }
+            }
+            bytesMap.clear();
+            MutilTempalteTabPane.getInstance().refreshOpenedTemplate(historyList);
+            MutilTempalteTabPane.getInstance().repaint();
+        }
+        FineLoggerFactory.getLogger().info("Env Change Template Loaded.");
+    }
 }

designer-base/src/main/java/com/fr/design/mainframe/JTemplate.java

@@ -522,19 +522,6 @@ public abstract class JTemplate<T extends BaseBook, U extends BaseUndoState<?>>
         if (editingFILE == null) {
             return false;
         }
-//        //检查一下连接是否成功
-//        try {
-//            if (FRContext.getCommonOperator() != null && !FRContext.getCommonOperator().testServerConnectionWithOutShowMessagePane()) {
-//                //连接不成功，提示
-//                JOptionPane.showMessageDialog(DesignerContext.getDesignerFrame(),
-//                        com.fr.design.i18n.Toolkit.i18nText(new String[]{"server_disconnected", "template_unsaved"}, new String[]{",", "!"})
-//                        , com.fr.design.i18n.Toolkit.i18nText("FR-Designer_Error"), JOptionPane.ERROR_MESSAGE);
-//                return false;
-//            }
-//        } catch (Exception e) {
-//            FineLoggerFactory.getLogger().error(e.getMessage(), e);
-//        }
-
 
         // 检查一下editingFILE是不是已存在的文件,如果不存在则用saveAs
         if (!editingFILE.exists()) {

designer-realize/src/main/java/com/fr/start/module/DesignerStartup.java

@@ -1,6 +1,7 @@
 package com.fr.start.module;
 
 
+import com.fr.design.file.HistoryTemplateListCache;
 import com.fr.design.mainframe.DesignerContext;
 import com.fr.event.Event;
 import com.fr.event.Listener;
@@ -28,13 +29,13 @@ public class DesignerStartup extends Activator {
     @Override
     @Metrics
     public void start() {
-    
+
         startSub(PreStartActivator.class);
         //启动基础部分
         startSub(BasicActivator.class);
         final String[] args = getModule().upFindSingleton(StartupArgs.class).get();
         final Designer designer = new Designer(args);
-    
+
         startSub(DesignerWorkspaceProvider.class);
         registerEnvListener();
         //启动env
@@ -49,7 +50,7 @@ public class DesignerStartup extends Activator {
             }
         });
         service.submit(new Runnable() {
-        
+
             @Override
             public void run() {
                 try {
@@ -63,46 +64,65 @@ public class DesignerStartup extends Activator {
         DesignerContext.getDesignerFrame().setVisible(true);
         //启动画面结束
         SplashContext.getInstance().hide();
-    
+
         DesignerContext.getDesignerFrame().getProgressDialog().setVisible(true);
         startSub(StartFinishActivator.class);
         FineRuntime.startFinish();
     }
-    
+
     private void browserDemo() {
-    
+
         if (getModule().leftFindSingleton(StartupArgs.class) != null && getModule().leftFindSingleton(StartupArgs.class).isDemo()) {
             ServerStarter.browserDemoURL();
         }
     }
-    
+
     /**
-     * 切换环境时，重新启动所有相关模块
+     * 注册切换环境前后事件监听
      */
     private void registerEnvListener() {
 
+
+        /*切换环境前，关闭所有相关模块*/
         listenEvent(WorkspaceEvent.BeforeSwitch, new Listener<Workspace>() {
 
             @Override
-            public void on(Event event, Workspace param) {
-
+            public void on(Event event, Workspace current) {
                 getSub(EnvBasedModule.class).stop();
             }
         });
+        /*切换环境后，重新启动所有相关模块*/
         listenEvent(WorkspaceEvent.AfterSwitch, new Listener<Workspace>(Integer.MAX_VALUE) {
 
             @Override
-            public void on(Event event, Workspace param) {
-
+            public void on(Event event, Workspace current) {
                 getSub(EnvBasedModule.class).start();
-                ExecutorService service = Executors.newSingleThreadExecutor();
-                service.submit(new Runnable() {
-                    @Override
-                    public void run() {
-                        FineEmbedServer.start();
-                    }
-                });
-                service.shutdown();
+                // 切换后的环境是本地环境才启动内置服务器
+                if (current.isLocal()) {
+                    ExecutorService service = Executors.newSingleThreadExecutor();
+                    service.submit(new Runnable() {
+                        @Override
+                        public void run() {
+                            FineEmbedServer.start();
+                        }
+                    });
+                    service.shutdown();
+                }
+            }
+        });
+        /*切换环境前，存储一下打开的所有文件对象，优先级高于默认优先级，要先于 关闭相关模块部分 被触发*/
+        listenEvent(WorkspaceEvent.BeforeSwitch, new Listener<Workspace>(1) {
+            @Override
+            public void on(Event event, Workspace workspace) {
+                HistoryTemplateListCache.getInstance().stash();
+            }
+        });
+
+        /*切换环境后，装载一下打开的所有文件对象，优先级低于默认优先级，要后于 启动相关模块部分 被触发*/
+        listenEvent(WorkspaceEvent.AfterSwitch, new Listener<Workspace>(-1) {
+            @Override
+            public void on(Event event, Workspace workspace) {
+                HistoryTemplateListCache.getInstance().load();
             }
         });
     }