<?xml version="1.0"?>

<!DOCTYPE hibernate-mapping PUBLIC
    "-//Hibernate/Hibernate Mapping DTD 2.0//EN"
    "http://hibernate.sourceforge.net/hibernate-mapping-2.0.dtd">

<hibernate-mapping>
    <class
        name="hibernate.ontoitself.TreeNode"
        table="treenode"
        dynamic-update="false"
        dynamic-insert="false"
    >

        <id
            name="id"
            column="id"
            type="java.lang.Long"
        >
            <generator class="identity">
            </generator>
        </id>

        <many-to-one
            name="next"
            class="hibernate.ontoitself.TreeNode"
            cascade="none"
            outer-join="auto"
            update="true"
            insert="true"
            column="next"
        />

        <set
            name="previous"
            lazy="false"
            inverse="true"
            cascade="all"
            sort="unsorted"
        >
              <key
                  column="next"
              />

              <one-to-many
                  class="hibernate.ontoitself.TreeNode"
             
              />
        </set>

    </class>
   
</hibernate-mapping>

package hibernate.ontoitself;

import java.util.Set;

public class TreeNode {

    protected Long id;
   
    protected TreeNode next;
    protected Set previous;
   
    public Long getId () {
        return id;
    }
    public void setId (Long id) {
        this.id = id;
    }
    public TreeNode getNext () {
        return next;
    }
    public void setNext (TreeNode next) {
        this.next = next;
    }
    public Set getPrevious () {
        return previous;
    }
    public void setPrevious (Set previous) {
        this.previous = previous;
    }
}

drop table if exists treenode;

create table treenode (

   id            bigint not null auto_increment,
   next          bigint,
   primary  key (id)
   
);

package hibernate.ontoitself;

import java.util.Iterator;
import java.util.Set;

import hibernate.testing.HibernateSession;
import net.sf.hibernate.HibernateException;
import net.sf.hibernate.Session;
import net.sf.hibernate.SessionFactory;
import net.sf.hibernate.cfg.Configuration;
import junit.framework.TestCase;

public class TreeNodeTests extends TestCase {

    protected SessionFactory sf;
   
    /*
     * @see TestCase#setUp()
     */
    protected void setUp () throws Exception {
        super.setUp ();
       
        try {
           
            sf = new Configuration()
                      .configure ("/hibernate/ontoitself/hibernate.cfg.xml")
                      .buildSessionFactory ();
           
            HibernateSession.sf = sf;
           
        } catch (Exception e) {
            e.printStackTrace ();
            throw e;
        }
       
    }

    /*
     * @see TestCase#tearDown()
     */
    protected void tearDown () throws Exception {
       
        super.tearDown ();
        try { sf.close (); } catch (Throwable t) {}
       
    }

    public void testReadiness () {}
   
    public void testInserts () {
        Session s = null;
        try {
            s = sf.openSession ();

            TreeNode t2 [] = new TreeNode [20];
            TreeNode t1 [] = new TreeNode [4];
            TreeNode t0 [] = new TreeNode [2];
            TreeNode root = new TreeNode ();
           
            for (int i = 0; i < t0.length; i++ ) {
                t0 [i] = new TreeNode ();
                t0 [i].setNext (root);
            }
           
            for (int i = 0; i < t1.length; i++) {
                t1 [i] = new TreeNode ();
                t1 [i].setNext (t0 [i % 2]);
            }
           
            for (int i = 0; i < t2.length; i++) {
                t2 [i] = new TreeNode ();
                t2 [i].setNext (t1 [i%4]);
            }
           
            s.saveOrUpdate (root);
            for (int i = 0; i < t0.length; i++) { s.saveOrUpdate (t0 [i]); }
            for (int i = 0; i < t1.length; i++) { s.saveOrUpdate (t1 [i]); }
            for (int i = 0; i < t2.length; i++) { s.saveOrUpdate (t2 [i]); }
           
            s.flush ();
        } catch (HibernateException he) {
            he.printStackTrace ();
        } finally {
            try { s.close (); } catch (Throwable t) {}
        }
    }
   
    public void testSelects () {
        Session s = null;
        try {
            s = sf.openSession ();

            TreeNode root = (TreeNode) s.find ("from hibernate.ontoitself.TreeNode t where t.next is null").get (0);
            // level 0
            Set level0 = root.getPrevious ();
            assertEquals ("level 0 size ", 2, level0.size ());
            for (Iterator iter0 = level0.iterator(); iter0.hasNext (); ) {
                TreeNode t0 = (TreeNode) iter0.next ();
                assertEquals ("level 0 next", t0.getNext ().getId (), root.getId ());
               
                // level 1
                Set level1 = t0.getPrevious ();
                for (Iterator iter1 = level1.iterator (); iter1.hasNext (); ) {
                    TreeNode t1 = (TreeNode) iter1.next ();
                    assertEquals ("level 1 next", t1.getNext ().getId (), t0.getId ());
                   
                    // level 2
                    Set level2 = t1.getPrevious ();
                    for (Iterator iter2 = level2.iterator (); iter2.hasNext (); ) {
                        TreeNode t2 = (TreeNode) iter2.next ();
                        assertEquals ("level 2 next", t2.getNext ().getId (), t1.getId ());
                    }
                }
            }
   
        } catch (HibernateException he) {
            he.printStackTrace ();
        } finally {
            try { s.close (); } catch (Throwable t) {}
        }
    }   
   
}