package org.paneris.jammyjoes.mvp;

 import java.util.Date;

 import java.util.Enumeration;

 import org.melati.poem.Table;

 import org.paneris.jammyjoes.model.Affiliate;

 import org.paneris.jammyjoes.util.JammyJoesUtil;

 public class HitsSelection implements Selection {

   private Table _hits;

   private Affiliate _affiliate;

   private Date _startDate;

   private final Date endDate;

   public HitsSelection(Table hits, Affiliate affiliate, Date startDate, Date endDate) {

     _hits = hits;

     _affiliate = affiliate;

     _startDate = startDate;

     this.endDate = endDate;

   }

   public void visitEach(Command command) {

     String selection = "affiliate="+ _affiliate.getTroid();

     if (_startDate != null) {

       selection += " and hit >= " + JammyJoesUtil.formatDateForPostgresSQL(_startDate);

     }

     if (endDate != null) {

       selection += " and hit < " + JammyJoesUtil.formatDateForPostgresSQL(endDate);

     }

     Enumeration enumeration = _hits.selection(selection);

     while (enumeration.hasMoreElements()) {

       command.visit(enumeration.nextElement());

     }

   }

 }