01: package net.sourceforge.squirrel_sql.plugins.db2.exp;
02:
03: /*
04: * Copyright (C) 2007 Rob Manning
05: * manningr@users.sourceforge.net
06: *
07: * This library is free software; you can redistribute it and/or
08: * modify it under the terms of the GNU Lesser General Public
09: * License as published by the Free Software Foundation; either
10: * version 2.1 of the License, or (at your option) any later version.
11: *
12: * This library is distributed in the hope that it will be useful,
13: * but WITHOUT ANY WARRANTY; without even the implied warranty of
14: * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
15: * Lesser General Public License for more details.
16: *
17: * You should have received a copy of the GNU Lesser General Public
18: * License along with this library; if not, write to the Free Software
19: * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
20: */
21: import java.util.ArrayList;
22: import java.util.List;
23:
24: import net.sourceforge.squirrel_sql.client.session.ISession;
25: import net.sourceforge.squirrel_sql.client.session.mainpanel.objecttree.INodeExpander;
26: import net.sourceforge.squirrel_sql.client.session.mainpanel.objecttree.ObjectTreeNode;
27: import net.sourceforge.squirrel_sql.fw.sql.DatabaseObjectInfo;
28: import net.sourceforge.squirrel_sql.fw.sql.DatabaseObjectType;
29: import net.sourceforge.squirrel_sql.fw.sql.IDatabaseObjectInfo;
30: import net.sourceforge.squirrel_sql.fw.sql.SQLDatabaseMetaData;
31:
32: /**
33: * This class is an expander for the schema nodes. It will add Sequence Object
34: * Type nodes to the schema node.
35: *
36: * @author manningr
37: */
38: public class SchemaExpander implements INodeExpander {
39:
40: /** whether or not we are connected to OS/400 */
41: private boolean isOS400 = false;
42:
43: /**
44: * Ctor.
45: */
46: public SchemaExpander(boolean isOS400) {
47: super ();
48: this .isOS400 = isOS400;
49: }
50:
51: /**
52: * Create the child nodes for the passed parent node and return them. Note
53: * that this method should <B>not</B> actually add the child nodes to the
54: * parent node as this is taken care of in the caller.
55: *
56: * @param session Current session.
57: * @param node Node to be expanded.
58: *
59: * @return A list of <TT>ObjectTreeNode</TT> objects representing the child
60: * nodes for the passed node.
61: */
62: public List<ObjectTreeNode> createChildren(ISession session,
63: ObjectTreeNode parentNode) {
64: final List<ObjectTreeNode> childNodes = new ArrayList<ObjectTreeNode>();
65: final IDatabaseObjectInfo parentDbinfo = parentNode
66: .getDatabaseObjectInfo();
67: final SQLDatabaseMetaData md = session.getSQLConnection()
68: .getSQLMetaData();
69: final String catalogName = parentDbinfo.getCatalogName();
70: final String schemaName = parentDbinfo.getSimpleName();
71:
72: IDatabaseObjectInfo seqInfo = new DatabaseObjectInfo(
73: catalogName, schemaName, "SEQUENCE",
74: DatabaseObjectType.SEQUENCE_TYPE_DBO, md);
75: ObjectTreeNode node = new ObjectTreeNode(session, seqInfo);
76: node.addExpander(new SequenceParentExpander(isOS400));
77: childNodes.add(node);
78:
79: IDatabaseObjectInfo udfInfo = new DatabaseObjectInfo(
80: catalogName, schemaName, "UDF",
81: DatabaseObjectType.UDF_TYPE_DBO, md);
82: ObjectTreeNode udfnode = new ObjectTreeNode(session, udfInfo);
83: udfnode.addExpander(new UDFParentExpander(isOS400));
84: childNodes.add(udfnode);
85:
86: return childNodes;
87: }
88:
89: }
|