Bomba! DAO + Relacionamento 1..N

class EmpresaDao {

public void update(Empresa x) {
  PrestadorDao pd = new PrestadorDao();
  for(Prestador p  : this.prestadores) {
    if(p.isUnsaved())
     pd.save(p);
    else if(p.isModified())
     pd.update(p);
   }
}

}

class SmartSet extends HashSet {
 private Set removidas = new HashSet();
 private Set novas = new HashSet();
 

 public boolean add(Object o) {
   boolean b = super.add(o);
   if(b) novas.add(o); 
   return b;
 }

 public boolean remove(Object o) {
   boolean b = super.remove(o);
   if(b) removidas.add(o); 
   return b;
 }
}

/*
 * Created on 12/07/2005
 *
 * sabes
 */
package br.umc.sabes.util.collection;

import java.util.ArrayList;
import java.util.Collections;
import java.util.List;

/**
 * @author thiago
 *
 * Lista que armazena internamente objetos excluídos
 */
public class SmartArrayList extends ArrayList {
	
	/**
	 * Armazena objetos adicionados
	 * 
	 */
	private List objetosAdicionados = new ArrayList();
	
	/**
	 * Armazena objetos alterados
	 * 
	 */
	private List objetosAlterados = new ArrayList();
	
	/**
	 * Armazena objetos excluidos
	 * 
	 */
	private List objetosExcluidos = new ArrayList();	
	
	/**
	 * Adiciona objeto na lista
	 * 
	 */
	public boolean add(Object o) {
		boolean b = super.add(o);
		if (b) objetosAdicionados.add(o);
		return b;
	}
	
	/**
	 * Marca um objeto para update
	 * 
	 */
	public boolean update(Object o) {
		boolean b = super.contains(o);
		if (b) objetosAlterados.add(o);
		return b;
	}
	
	/**
	 * Remove objeto da lista
	 * 
	 */
	public boolean remove(Object o) {
		boolean b = super.remove(o);
		if (b) objetosExcluidos.add(o);
		return b;
	}
	
	

	/**
	 * @return Returns the objetosExcluidos.
	 * 
	 */
	public List getObjetosExcluidos() {
		return Collections.unmodifiableList(objetosExcluidos);
	}	
	
	/**
	 * @return Returns the objetosAdicionados.
	 * 
	 */
	public List getObjetosAdicionados() {
		return Collections.unmodifiableList(objetosAdicionados);
	}
	
	/**
	 * @return Returns the objetosAlterados.
	 * 
	 */
	public List getObjetosAlterados() {
		return Collections.unmodifiableList(objetosAlterados);
	}
}

/**
	 * Teste do relacionamento Empresa - ServicoPrestado
     * Teste update de Serviços Prestados
     * 
	 * @throws DAOException
	 */
	public void testUpdateServicoPrestadoAPartirDaEmpresa() throws DAOException {
		Empresa empresa = new Empresa(new Cnpj("1"), "Empresa");
		dao.create(empresa);
		
		empresa.addServicoPrestado(new ServicoPrestado("Titulo1", "Descricao1"));
		empresa.addServicoPrestado(new ServicoPrestado("Titulo2", "Descricao2"));
		dao.update(empresa);
		
		List servicos = empresa.getServicoPrestado();		
		ServicoPrestado sp1 = (ServicoPrestado) servicos.get(1);
		sp1.setTitulo("Titulo 3");
		sp1.setDescricao("Descricao 3");
		empresa.updateServicoPrestado(sp1); // marcando um objeto para update
		dao.update(empresa);
		
		long spIdPK = sp1.getIdPK();
		sp1 = null;
		
		sp1 = new ServicoPrestadoDAO().load(spIdPK);		
		assertEquals("Titulo 3", sp1.getTitulo());
		assertEquals("Descricao 3", sp1.getDescricao());		
	}