fix bug: 数据库连接配置页面NPE

Author: sagexueqi

src/main/java/com/zzg/mybatis/generator/controller/DbConnectionController.java

@@ -34,6 +34,7 @@ public class DbConnectionController extends BaseFXController {
     @FXML
     protected ChoiceBox<String> dbTypeChoice;
     protected MainUIController mainUIController;
+    protected TabPaneController tabPaneController;
     protected boolean isUpdate = false;
     protected Integer primayKey;
 
@@ -48,7 +49,7 @@ final void saveConnection() {
         }
         try {
             ConfigHelper.saveDatabaseConfig(this.isUpdate, primayKey, config);
-            getDialogStage().close();
+            this.tabPaneController.getDialogStage().close();
             mainUIController.loadLeftDBTree();
         } catch (Exception e) {
             _LOG.error(e.getMessage(), e);
@@ -61,6 +62,10 @@ void setMainUIController(MainUIController controller) {
         super.setDialogStage(mainUIController.getDialogStage());
     }
 
+    public void setTabPaneController(TabPaneController tabPaneController) {
+        this.tabPaneController = tabPaneController;
+    }
+
     public DatabaseConfig extractConfigForUI() {
         String name = nameField.getText();
         String host = hostField.getText();

src/main/java/com/zzg/mybatis/generator/controller/OverSshController.java

@@ -202,7 +202,7 @@ public void saveConfig() {
         }
         try {
             ConfigHelper.saveDatabaseConfig(this.isUpdate, primayKey, databaseConfig);
-            getDialogStage().close();
+            this.tabPaneController.getDialogStage().close();
             mainUIController.loadLeftDBTree();
         } catch (Exception e) {
             logger.error(e.getMessage(), e);

src/main/java/com/zzg/mybatis/generator/controller/TabPaneController.java

@@ -51,7 +51,9 @@ public void initialize(URL location, ResourceBundle resources) {
     public void setMainUIController(MainUIController mainUIController) {
         this.mainUIController = mainUIController;
         this.tabControlAController.setMainUIController(mainUIController);
+        this.tabControlAController.setTabPaneController(this);
         this.tabControlBController.setMainUIController(mainUIController);
+        this.tabControlBController.setTabPaneController(this);
     }
 
     public void setConfig(DatabaseConfig selectedConfig) {