archive/net.sourceforge.phpeclipse.quantum.sql/src/com/quantum/view/bookmark/EntityNode.java

   1 package com.quantum.view.bookmark;
   2
   3 import java.sql.SQLException;
   4 import java.util.ArrayList;
   5 import java.util.Collections;
   6 import java.util.HashMap;
   7 import java.util.Iterator;
   8 import java.util.List;
   9 import java.util.Map;
  10
  11 import com.quantum.model.Column;
  12 import com.quantum.model.Entity;
  13 import com.quantum.model.EntityHolder;
  14 import com.quantum.model.NotConnectedException;
  15 import com.quantum.model.SchemaHolder;
  16 import com.quantum.model.Table;
  17 import com.quantum.model.View;
  18
  19 /**
  20  * @author BC
  21  */
  22 public class EntityNode extends TreeNode implements EntityHolder {
  23
  24     private Entity entity;
  25     private boolean longFormName;
  26
  27     private List columns = Collections.synchronizedList(new ArrayList());
  28     private List foreignKeys = Collections.synchronizedList(new ArrayList());
  29     boolean initialized = false;
  30
  31     public EntityNode(TreeNode parent, Entity entity) {
  32         this(parent, entity, false);
  33     }
  34     public EntityNode(TreeNode parent, Entity entity, boolean longFormName) {
  35         super(parent);
  36         this.entity = entity;
  37         this.longFormName = longFormName;
  38
  39         if (parent instanceof SchemaHolder) {
  40             SchemaHolder schemaHolder = (SchemaHolder) parent;
  41             if (!schemaHolder.getSchema().getDisplayName().equals(entity.getSchema())) {
  42                 this.longFormName = true;
  43             }
  44         }
  45     }
  46
  47     public Object[] getChildren() throws NotConnectedException, SQLException {
  48          if (!isInitialized()) {
  49                 initializeChildren();
  50          }
  51
  52         if (this.children.size() > 0) {
  53             return (ColumnNode[]) this.children.toArray(new ColumnNode[this.children.size()]);
  54         } else {
  55             return BookmarkListNode.EMPTY_ARRAY;
  56         }
  57     }
  58
  59     protected synchronized void initializeChildren() throws NotConnectedException, SQLException {
  60         boolean wasInitialized = isInitialized();
  61         Map map = getChildrenAsMap();
  62         Column[] columns = this.entity.getColumns();
  63         this.children.clear();
  64         for (int i = 0, length = (columns == null) ? 0 : columns.length;
  65             i < length;
  66             i++) {
  67
  68                 ColumnNode node = (ColumnNode) map.get(columns[i].getName());
  69                 if (node == null) {
  70                         this.children.add(new ColumnNode(this, columns[i]));
  71                 } else {
  72                         node.setColumn(columns[i]);
  73                         this.children.add(node);
  74                 }
  75
  76                 if (wasInitialized) {
  77                         firePropertyChange("columns", null, null);
  78                 }
  79         }
  80     }
  81
  82     private Map getChildrenAsMap() {
  83         Map map = new HashMap();
  84         for (Iterator i = this.children.iterator(); i.hasNext();) {
  85                         TreeNode node = (TreeNode) i.next();
  86                         map.put(node.getName(), node);
  87                 }
  88         return map;
  89     }
  90
  91     public boolean hasChildren() {
  92         if (!isSequence()) {
  93             return true;
  94         } else {
  95             return false;
  96         }
  97     }
  98
  99     public Entity getEntity() {
 100         return this.entity;
 101     }
 102
 103     public String getName() {
 104         return this.entity.getName();
 105     }
 106
 107     public String getLabelName() {
 108         return (this.longFormName) ? this.entity.getQualifiedName() : this.entity.getName();
 109     }
 110
 111     public boolean isTable() {
 112         return Entity.TABLE_TYPE.equals(this.entity.getType());
 113     }
 114
 115     public boolean isView() {
 116         return Entity.VIEW_TYPE.equals(this.entity.getType());
 117     }
 118
 119     public boolean isSequence() {
 120         return Entity.SEQUENCE_TYPE.equals(this.entity.getType());
 121     }
 122
 123     protected String getImageName() {
 124         if (isSequence()) {
 125             return "sequence.gif";
 126         } else if (isView()) {
 127             return "view.gif";
 128         } else {
 129             return (this.entity.exists() == null || this.entity.exists().booleanValue())
 130                 ? "bigtable.gif" : "missingtable.gif";
 131         }
 132     }
 133
 134     public String getLabelDecorations(LabelDecorationInstructions instructions) {
 135         String decoration = null;
 136         if (instructions.isSizeVisible()) {
 137             Integer size = getSize();
 138             if (size != null) {
 139                 decoration = ((decoration == null) ? "" : decoration)
 140                     + "[" + size + "]";
 141             }
 142         }
 143         return decoration;
 144     }
 145
 146     private Integer getSize() {
 147         if (isTable()) {
 148             return ((Table) this.entity).getSize();
 149         } else if (isView()) {
 150             return ((View) this.entity).getSize();
 151         } else {
 152             return null;
 153         }
 154     }
 155
 156     public int compareTo(Object o) {
 157         if (o instanceof EntityNode) {
 158             EntityNode that = (EntityNode) o;
 159             return this.entity.getQualifiedName().compareTo(
 160                 that.entity.getQualifiedName());
 161         } else {
 162             return super.compareTo(o);
 163         }
 164     }
 165         void setEntity(Entity entity) {
 166                 this.entity = entity;
 167         }
 168 }