Setting different tree nodes icons

Question

Hello.
I know this thread was discussed and i read it but it did not helped me.
I need a tree that might look like this:
+hostname
+---database1
--table1
--table2
--table3
+---database2
--table1
--table2
--table3
This is a database tree.
Now my code to set different icons is:
public class DMSTreeCellRenderer extends DefaultTreeCellRenderer {
    private DatabaseInfo databaseInfo;
    private String dbHost;
     /** Creates a new instance of DMSTreeCellRenderer */
    public DMSTreeCellRenderer(String dbHost) {
        this.dbHost = dbHost;
    public Component getTreeCellRendererComponent(JTree tree, Object value, boolean sel, boolean expanded, boolean leaf, int row, boolean hasFocus) {
        ImageIcon rootIcon = new ImageIcon("resources/images/database.png");
        ImageIcon tableIcon = new ImageIcon("resources/images/database_table.png");
        if(tree.getModel().getRoot().toString().equals(dbHost) ) {
            setIcon(new ImageIcon("resources/images/server_database.png"));
            isFirstSetup = false;
        if (rootIcon != null && tableIcon !=null) {
            setClosedIcon(rootIcon);
            setOpenIcon(rootIcon);
            setLeafIcon(tableIcon);
        return super.getTreeCellRendererComponent(tree,value,sel,expanded,leaf,row,hasFocus);
}Anyway it does not work. like it should. The root nod has the same icon like the childs :(

Answer 1

Thank you for your help, i solve my problem like this:
public Component getTreeCellRendererComponent(JTree tree, Object value, boolean sel, boolean expanded, boolean leaf, int row, boolean hasFocus) {
        super.getTreeCellRendererComponent(tree,value,sel,expanded,leaf,row,hasFocus);
        ImageIcon rootIcon = new ImageIcon("resources/images/database.png");
        ImageIcon tableIcon = new ImageIcon("resources/images/database_table.png");
        if(value.toString().equals(dbHost) ){
            setIcon(new ImageIcon("resources/images/server_database.png"));
        }else {
            if (rootIcon != null && tableIcon !=null) {
                if(leaf){
                    setIcon(tableIcon);
                }else {
                    setIcon(rootIcon);   
        return this;
    }