All Downloads are FREE. Search and download functionalities are using the official Maven repository.

com.scudata.ide.spl.base.PanelSelectField Maven / Gradle / Ivy

Go to download

SPL(Structured Process Language) A programming language specially for structured data computing.

There is a newer version: 20241126
Show newest version
package com.scudata.ide.spl.base;

import java.awt.Dimension;
import java.awt.GridBagConstraints;
import java.awt.GridBagLayout;
import java.awt.HeadlessException;
import java.awt.Insets;
import java.awt.datatransfer.Clipboard;
import java.awt.datatransfer.StringSelection;
import java.awt.event.ActionEvent;
import java.awt.event.MouseAdapter;
import java.awt.event.MouseEvent;
import java.sql.Connection;
import java.sql.DatabaseMetaData;
import java.sql.ResultSet;
import java.sql.SQLException;
import java.util.ArrayList;
import java.util.Arrays;
import java.util.HashMap;
import java.util.LinkedHashMap;
import java.util.List;
import java.util.Map;
import java.util.Vector;

import javax.swing.JButton;
import javax.swing.JCheckBox;
import javax.swing.JLabel;
import javax.swing.JPanel;
import javax.swing.JScrollPane;
import javax.swing.JTree;
import javax.swing.SwingUtilities;
import javax.swing.tree.DefaultTreeModel;
import javax.swing.tree.TreePath;

import com.scudata.app.common.Section;
import com.scudata.common.DBConfig;
import com.scudata.common.MessageManager;
import com.scudata.common.StringUtils;
import com.scudata.ide.common.ConfigOptions;
import com.scudata.ide.common.DataSource;
import com.scudata.ide.common.GC;
import com.scudata.ide.common.GM;
import com.scudata.ide.common.GV;
import com.scudata.ide.common.control.IconTreeNode;
import com.scudata.ide.common.control.IconTreeRender;
import com.scudata.ide.common.resources.IdeCommonMessage;
import com.scudata.ide.spl.GVSpl;
import com.scudata.ide.spl.SheetSpl;
import com.scudata.ide.spl.ToolBarProperty;
import com.scudata.ide.spl.control.ContentPanel;

public class PanelSelectField extends JPanel {
	private static final long serialVersionUID = 1L;

	/**
	 * Common??Դ??????
	 */
	private MessageManager mm = IdeCommonMessage.get();

	/**
	 * ???ư?ť
	 */
	private JButton jBCopy = new JButton();

	/**
	 * ???Ӱ?ť
	 */
	private JButton jBAdd = new JButton();

	/**
	 * Ϊ?ֶ??????ӱ???ǰ׺
	 */
	private JCheckBox jCBOpt = new JCheckBox(mm.getMessage("panelselectfield.tablepre"));

	/**
	 * ???ڵ?
	 */
	private IconTreeNode root = new IconTreeNode(mm.getMessage("panelselectfield.dsname"));

	/**
	 * ??ģ??
	 */
	private DefaultTreeModel treeModel = new DefaultTreeModel(root);

	/**
	 * ???ؼ?
	 */
	private JTree mTree = new JTree(treeModel);

	/**
	 * ?ֶν??ļ???
	 */
	private final byte LEVEL_FIELD = 3;

	/**
	 * ??ʾ??Ϣ?ı?ǩ?ؼ?
	 */
	private JLabel labelMsg = new JLabel();

	/**
	 * ???ݿ?ȡ?ṹ???߳?
	 */
	private Thread dbThread = null;

	/**
	 * ?߳??Ƿ?ֹͣ
	 */
	private boolean stoped = false;

	/**
	 * ???캯??
	 */
	public PanelSelectField() {
		init();
		resetEnv();
	}

	/**
	 * ???û????? ????????жϣ?һ???п????ӵ?????Դ??
	 */
	public void resetEnv() {
		if (dbThread != null) {
			stoped = true;
		}
		root.removeAllChildren();
		treeModel.nodeStructureChanged(root);

		if (!ConfigOptions.bShowDBStruct) // ?????????ݿ?ṹ
			return;

		dbThread = new Thread() {
			public void run() {
				stoped = false;
				DataSource ds;
				String dsName, tableName;
				Vector tables, fields;
				final List dsNames = new ArrayList();
				final ArrayList> dsList = new ArrayList>();
				try {
					labelMsg.setVisible(true);
					labelMsg.setText(mm.getMessage("panelselectfield.loading") + "...");
					for (int i = 0; i < GV.dsModel.getSize(); i++) {
						if (stoped) {
							return;
						}
						ds = (DataSource) GV.dsModel.get(i);
						if (ds != null && !ds.isClosed()) {
							Connection con = null;
							try {
								con = (Connection) ds.getDBSession().getSession();
							} catch (Throwable e1) {
								continue;
							}
							if (con == null)
								continue;
							if (ds.getDBInfo() instanceof DBConfig) {
								DBConfig dbConfig = ((DBConfig) ds.getDBInfo());
								if (dbConfig.getDriver().toLowerCase().startsWith("com.scudata.datahub")) {
									// datahub
									ResultSet rs = null;
									DatabaseMetaData dbmd = null;
									List splNames = new ArrayList();
									try {
										dbmd = con.getMetaData();
										rs = dbmd.getProcedures(null, null, null);
										while (rs.next()) {
											String splName = rs.getString("PROCEDURE_NAME");
											splNames.add(splName);
										}
									} catch (SQLException e) {
									} finally {
										if (rs != null)
											try {
												rs.close();
											} catch (SQLException e) {
											}
									}

									if (!splNames.isEmpty() && dbmd != null) {
										ArrayList tableList = new ArrayList();
										dsList.add(tableList);
										for (int d = 0, splSize = splNames.size(); d < splSize; d++) {
											String splName = (String) splNames.get(d);
											rs = null;
											try {
												// ??δʵ??
												rs = dbmd.getProcedureColumns(null, null, splName, null);
												TableConfig tc = new TableConfig();
												tc.tableName = splName;
												tc.fields = null;
												tableList.add(tc);
											} catch (SQLException e) {
											} finally {
												if (rs != null)
													try {
														rs.close();
													} catch (SQLException e) {
													}
											}
										}
									}
									continue;
								}
							}
							dsName = ds.getName();
							dsNames.add(dsName);
							tables = GM.listTableNames(dsName);
							if (tables == null || tables.isEmpty()) {
								continue;
							}
							String[] tableNames = new String[tables.size()];
							for (int s = 0; s < tables.size(); s++) {
								tableNames[s] = (String) tables.get(s);
							}
							Arrays.sort(tableNames);

							Map> tableFields = null;
							if (tables.size() > 300 && !GM.isDataLogicDS(ds)) {
								try {
									tableFields = getTableFields(con);
								} catch (Exception e) {
									e.printStackTrace();
								}
							}

							ArrayList tableList = new ArrayList();
							dsList.add(tableList);
							TableConfig tc;
							for (int t = 0; t < tableNames.length; t++) {
								if (stoped) {
									return;
								}
								tc = new TableConfig();
								tableList.add(tc);
								tableName = tableNames[t];
								tc.tableName = tableName;
								fields = tableFields == null ? null : (Vector) tableFields.get(tableName);
								if (fields == null) {
									fields = GM.listColumnNames(dsName, tableName);
								}
								tc.fields = fields;
								try {
									tc.fkMap = getTableFKMap(ds, null, tableName, con);
								} catch (Throwable e) {
									tc.fkMap = null;
								}
							}
						}
					}
					if (stoped) {
						return;
					}
				} finally {
					labelMsg.setVisible(false);
				}
				SwingUtilities.invokeLater(new Thread() {
					public void run() {
						rebuildTree(dsNames, dsList);
					}
				});
			}
		};
		dbThread.start();
	}

	/**
	 * ???¹??????ؼ?
	 * 
	 * @param dsNames ????Դ????
	 * @param dsList  ???????б?
	 */
	private synchronized void rebuildTree(List dsNames, List> dsList) {
		root = new IconTreeNode(mm.getMessage("panelselectfield.dsname"));
		if (dsList != null && dsNames != null) {
			IconTreeNode dsNode, tableNode, fieldNode, pkNode;
			Map fkMap;
			TableConfig tc;
			String dsName, fieldName;
			List tableList;
			Vector fields;
			for (int i = 0; i < dsNames.size(); i++) {
				dsName = (String) dsNames.get(i);
				dsNode = new IconTreeNode(dsName);
				root.add(dsNode);
				if (dsList.size() <= i)
					continue;
				tableList = (ArrayList) dsList.get(i);
				for (int t = 0; t < tableList.size(); t++) {
					tc = tableList.get(t);
					tableNode = new IconTreeNode(tc.tableName, IconTreeNode.TYPE_TABLE);
					dsNode.add(tableNode);
					fields = tc.fields;
					if (fields != null) {
						fkMap = tc.fkMap;
						for (int c = 0; c < fields.size(); c++) {
							fieldName = fields.get(c) == null ? "" : (String) fields.get(c);
							fieldNode = new IconTreeNode(fieldName, IconTreeNode.TYPE_COLUMN);
							tableNode.add(fieldNode);
							if (fkMap != null) {
								Object pkName = fkMap.get(fieldName);
								if (pkName != null) {
									pkNode = new IconTreeNode((String) pkName, IconTreeNode.TYPE_TABLE);
									fieldNode.add(pkNode);
								}
							}
						}
					}
					treeModel.nodeStructureChanged(tableNode);
					treeModel.nodeStructureChanged(dsNode);
				}
			}
		}
		treeModel = new DefaultTreeModel(root);
		mTree.setModel(treeModel);
		mTree.setCellRenderer(new IconTreeRender());
	}

	/**
	 * ??????
	 */
	class TableConfig {
		/**
		 * ????
		 */
		String tableName;
		/**
		 * ?ֶ??б?
		 */
		Vector fields;
		/**
		 * ???ӳ??
		 */
		Map fkMap;

		public TableConfig() {
		}
	}

	/**
	 * ȡ?????ֶ????б?
	 * 
	 * @param con
	 * @return
	 * @throws Exception
	 */
	public static Map> getTableFields(Connection con) throws Exception {
		DatabaseMetaData md = con.getMetaData();
		ResultSet rs = null;
		Map> tableFields = new LinkedHashMap>();
		try {
			rs = md.getColumns(null, null, null, "%");
			while (rs.next()) {
				String tableName = rs.getString("TABLE_NAME");
				Vector columns = (Vector) tableFields.get(tableName);
				if (columns == null) {
					columns = new Vector();
					tableFields.put(tableName, columns);
				}
				String columnName = rs.getString("COLUMN_NAME");
				columns.add(columnName);
			}
		} finally {
			if (rs != null)
				rs.close();
		}
		if (tableFields.isEmpty()) {
			return null;
		}
		return tableFields;
	}

	/**
	 * ȡѡ???ֵ
	 * 
	 * @return
	 */
	private String getSelectedValues() {
		Section sec = new Section();
		TreePath[] paths = mTree.getSelectionPaths();
		if (paths == null) {
			return null;
		}
		IconTreeNode node;
		String val;
		for (int i = 0; i < paths.length; i++) {
			node = (IconTreeNode) paths[i].getLastPathComponent();
			val = getNodeValue(node);
			if (val != null) {
				sec.addSection(val);
			}
		}
		if (sec.size() == 0) {
			return null;
		}
		return sec.toString();
	}

	/**
	 * ȡ????ֵ
	 * 
	 * @param node ?????
	 * @return
	 */
	private String getNodeValue(IconTreeNode node) {
		IconTreeNode pNode;
		String name = node.getName();
		if (StringUtils.isValidString(name)) {
			if (node.getLevel() == LEVEL_FIELD) {
				if (jCBOpt.isSelected()) {
					pNode = (IconTreeNode) node.getParent();
					if (pNode.getType() == IconTreeNode.TYPE_TABLE && StringUtils.isValidString(pNode.getName())) {
						name = pNode.getName() + "." + name;
					}
				}
			}
		}
		return name;
	}

	/**
	 * ??ʼ??
	 */
	private void init() {
		GridBagConstraints gbc;
		this.setLayout(new GridBagLayout());
		gbc = GM.getGBC(1, 1);
		gbc.insets = new Insets(3, 8, 3, 3);
		this.add(jBAdd, gbc);
		gbc = GM.getGBC(1, 2);
		gbc.insets = new Insets(3, 3, 3, 3);
		this.add(jBCopy, gbc);
		gbc = GM.getGBC(1, 3, true);
		gbc.insets = new Insets(3, 3, 3, 3);
		this.add(jCBOpt, gbc);
		gbc = GM.getGBC(4, 1, true, true);
		gbc.gridwidth = 3;
		JScrollPane jScrollPane1 = new JScrollPane();
		this.add(jScrollPane1, gbc);
		gbc = GM.getGBC(5, 1, true);
		gbc.gridwidth = 3;
		this.add(labelMsg, gbc);
		jScrollPane1.getViewport().add(mTree);
		jBAdd.setIcon(GM.getImageIcon(GC.IMAGES_PATH + "m_pmtundo.gif"));
		jBCopy.setIcon(GM.getImageIcon(GC.IMAGES_PATH + "m_copy.gif"));
		jBAdd.setToolTipText(mm.getMessage("panelselectfield.addtf"));
		jBCopy.setToolTipText(mm.getMessage("panelselectfield.copyf"));
		initButton(jBAdd);
		initButton(jBCopy);
		jBAdd.addActionListener(new PanelSelectField_jBAdd_actionAdapter(this));
		jBCopy.addActionListener(new PanelSelectField_jBCopy_actionAdapter(this));

		mTree.addMouseListener(new MouseAdapter() {
			public void mouseClicked(MouseEvent e) {
				if (e.getClickCount() != 2) {
					return;
				}
				TreePath path = mTree.getSelectionPath();
				if (path == null) {
					return;
				}
				IconTreeNode node = (IconTreeNode) path.getLastPathComponent();
				if (node.isRoot()) {
					return;
				}
				String val = getNodeValue(node);
				if (val != null) {
					add(val);
				}
			}
		});
		labelMsg.setVisible(false);
	}

	/**
	 * ??ʼ????ť
	 * 
	 * @param b
	 */
	public void initButton(JButton b) {
		Dimension d = new Dimension(22, 22);
		b.setMaximumSize(d);
		b.setMinimumSize(d);
		b.setPreferredSize(d);
	}

	/**
	 * ????ֵ
	 * 
	 * @param vals
	 */
	private void add(String vals) {
		if (vals == null || GVSpl.splEditor == null || !(GV.appSheet instanceof SheetSpl)) {
			return;
		}
		if (GV.isCellEditing) {
			ContentPanel cp = GVSpl.splEditor.getComponent().getContentPanel();
			if (cp.isEditing()) {
				cp.addText(vals);
			}
		} else {
			((ToolBarProperty) GV.toolBarProperty).addText(vals);
		}
	}

	/**
	 * ȡ???????ӳ??
	 * 
	 * @param ds     ????Դ
	 * @param schema ģʽ??
	 * @param table  ????
	 * @param con    ????
	 * @return
	 * @throws Throwable
	 */
	private Map getTableFKMap(DataSource ds, String schema, String table, Connection con)
			throws Throwable {
		DatabaseMetaData md = con.getMetaData();
		String[] schemaTable = GM.getRealSchemaTable(con, schema, table);
		schema = schemaTable[0];
		table = schemaTable[1];
		ResultSet rs = null;
		Map fkMap = new HashMap();
		try {
			rs = md.getImportedKeys(null, schema, table);
			String pkTableName, pkColName, colName;
			while (rs.next()) {
				pkTableName = rs.getString("PKTABLE_NAME");
				pkColName = rs.getString("PKCOLUMN_NAME");
				colName = rs.getString("FKCOLUMN_NAME");
				if (!pkColName.startsWith(pkTableName + ".")) {
					pkColName = pkTableName + "." + pkColName;
				}
				fkMap.put(colName, pkColName);
			}
		} finally {
			rs.close();
		}
		return fkMap;
	}

	/**
	 * ?????¼?
	 * 
	 * @param e
	 */
	void jBAdd_actionPerformed(ActionEvent e) {
		add(getSelectedValues());
	}

	/**
	 * ?????¼?
	 * 
	 * @param e
	 */
	void jBCopy_actionPerformed(ActionEvent e) {
		String vals = getSelectedValues();
		if (vals == null) {
			return;
		}
		try {
			Clipboard cb = java.awt.Toolkit.getDefaultToolkit().getSystemClipboard();
			cb.setContents(new StringSelection(vals), null);
		} catch (HeadlessException e1) {
		}
	}
}

class PanelSelectField_jBAdd_actionAdapter implements java.awt.event.ActionListener {
	PanelSelectField adaptee;

	PanelSelectField_jBAdd_actionAdapter(PanelSelectField adaptee) {
		this.adaptee = adaptee;
	}

	public void actionPerformed(ActionEvent e) {
		adaptee.jBAdd_actionPerformed(e);
	}
}

class PanelSelectField_jBCopy_actionAdapter implements java.awt.event.ActionListener {
	PanelSelectField adaptee;

	PanelSelectField_jBCopy_actionAdapter(PanelSelectField adaptee) {
		this.adaptee = adaptee;
	}

	public void actionPerformed(ActionEvent e) {
		adaptee.jBCopy_actionPerformed(e);
	}
}




© 2015 - 2024 Weber Informatics LLC | Privacy Policy