CSVRecord.java

/*
 * $Source$
 * $Revision$
 *
 * Part of Melati (http://melati.org), a framework for the rapid
 * development of clean, maintainable web applications.
 *
 *  Copyright (C) 2001 Myles Chippendale
 *
 * Melati is free software; Permission is granted to copy, distribute
 * and/or modify this software under the terms either:
 *
 * a) the GNU General Public License as published by the Free Software
 *    Foundation; either version 2 of the License, or (at your option)
 *    any later version,
 *
 *    or
 *
 * b) any version of the Melati Software License, as published
 *    at http://melati.org
 *
 * You should have received a copy of the GNU General Public License and
 * the Melati Software License along with this program;
 * if not, write to the Free Software Foundation, Inc.,
 * 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA to obtain the
 * GNU General Public License and visit http://melati.org to obtain the
 * Melati Software License.
 *
 * Feel free to contact the Developers of Melati (http://melati.org),
 * if you would like to work out a different arrangement than the options
 * outlined here.  It is our intention to allow Melati to be used by as
 * wide an audience as possible.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * Contact details for copyright holder:
 *
 *     Myles Chippendale <mylesc At paneris.org>
 *
 *
 * ------
 *  Note
 * ------
 *
 * I will assign copyright to PanEris (http://paneris.org) as soon as
 * we have sorted out what sort of legal existence we need to have for
 * that to make sense. 
 * In the meantime, if you want to use Melati on non-GPL terms,
 * contact me!
 */

package org.melati.poem.csv;

import java.util.Vector;

import org.melati.poem.Persistent;
import org.melati.poem.PoemThread;
import org.melati.poem.Table;

/**
 * A record within a CSV File.
 */
public class CSVRecord {

  private Vector<CSVField> fields;
  
  /** The value of the primary key of this record, from the csv file */
  String primaryKeyValue = null;

  /** The table this record wants to be written to */
  Table<?> table = null;

  /** The Poem Persistent corresponding to writing this record to the db */
  Persistent poemRecord = null;
  
  /** The line number of the CSV file. */
  private int lineNo;
  
  /** The record number of the CSV file. */
  private int recordNo;
  
  /**
   * Constructor.
   */
  public CSVRecord(Table<?> table) {
    super();
    this.table = table;
    this.fields = new Vector<CSVField>();
  }

  /**
   * Add a field to this record.
   */
  public synchronized void addField(CSVField field) {
    if (field.column.isPrimaryKey)
      primaryKeyValue = field.value;
    fields.addElement(field);
  }

  /**
   * Write the data in this record into a new Persistent.
   * @throws CSVWriteDownException
   */
  private void createPersistent() 
      throws NoPrimaryKeyInCSVTableException, CSVWriteDownException {
    if (poemRecord != null)
      return;
    Persistent newObj = table.newPersistent();

    for(int j = 0; j < fields.size(); j++) {
      CSVColumn col = ((CSVField)fields.elementAt(j)).column;
      String csvValue = ((CSVField)fields.elementAt(j)).value;
      if (col.foreignTable == null) {
        if (col.poemName != null) {
          try {
            if (csvValue != null && !csvValue.equals("")) {
              newObj.setRawString(col.poemName, csvValue);
            }
          } catch (Exception e) {
            throw new CSVWriteDownException(table.getName(), getLineNo(), 
                new RuntimeException("Problem processing column " + 
                col.poemName + 
                " of table " + 
                table.getName() +
                " on line " + getLineNo() + 
                " value :" + csvValue + 
                ": " + e.toString()));
          }
        }
      }
      // Lookup up value in the foreign Table
      else {
        if (csvValue == null || csvValue.trim().equals("")) {
          if (!newObj.getTable().getColumn(col.poemName).getColumnInfo().
             getNullable().booleanValue()) {
            throw new RuntimeException("CSV value missing for required column " + 
              col.poemName + 
              " of table " + 
              table.getName() +
              " on line " + getLineNo() 
              );
          }
        } else { 
          Persistent lookup = col.foreignTable.getRecordWithID(csvValue);
          if (lookup == null) {
            throw new RuntimeException("No persistent found with primary key " + 
                csvValue +
                " for column " + 
                col.poemName + 
                " of table " + 
                table.getName() +
                " on line " + getLineNo() 
                );
          }
          newObj.setCooked(col.poemName, lookup);
        }
      }
    }
    poemRecord = newObj;
  }
  
 /**
  * Retreive the Persistent corresponding to this CSVRecord, if there
  * is one.
  * @return the existing or newly created Poem Persistent 
  * @throws NoPrimaryKeyInCSVTableException
  * @throws CSVWriteDownException
  */
  public Persistent getPersistent() 
      throws NoPrimaryKeyInCSVTableException, CSVWriteDownException {
    if (poemRecord != null)
      return poemRecord;
    createPersistent();
    return poemRecord;
  }

 /**
  * Make sure this record is written to the database.
  */
  void makePersistent() 
      throws NoPrimaryKeyInCSVTableException, CSVWriteDownException {
    try {
      getPersistent();
      poemRecord.makePersistent();
    } catch (NoPrimaryKeyInCSVTableException e1) {
      throw e1;
    } catch (CSVWriteDownException e2) {
      throw e2;
    } catch (Exception e) {
      e.printStackTrace(System.err);
      throw new RuntimeException(e.toString()
                 + 
                " in table " + 
                table.getName() +
                " on line " + getLineNo() 
                );
    }    

    if (PoemThread.inSession())
      PoemThread.writeDown();
    PoemThread.commit();
  }

 /**
  * @param recordNo The recordNo to set.
  */
  public void setRecordNo(int recordNo) {
    this.recordNo = recordNo;
  }

 /**
  * @return Returns the recordNo.
  */
  public int getRecordNo() {
    return recordNo;
  }

 /**
  * @param lineNo The lineNo to set.
  */
  public void setLineNo(int lineNo) {
   this.lineNo = lineNo;
  }

  /**
   * @return Returns the lineNo.
   */
  public int getLineNo() {
    return lineNo;
  }

  /**
   * @return the fields 
   */
  public Vector<CSVField> getFields() {
    return fields;
  }

}