[CSV-68] Use the Builder pattern for CSVFormat.

garydgregory · garydgregory · commit db11c04d29dd · 2012-11-17T18:00:38.000Z
git-svn-id: https://svn.apache.org/repos/asf/commons/proper/csv/trunk@1410759 13f79535-47bb-0310-9956-ffa450edef68
diff --git a/src/main/java/org/apache/commons/csv/CSVFormat.java b/src/main/java/org/apache/commons/csv/CSVFormat.java
@@ -29,12 +29,13 @@
 import java.io.Reader;
 import java.io.Serializable;
 import java.io.StringWriter;
+import java.util.Arrays;
 
 /**
  * The format specification of a CSV file.
  *
  * This class is immutable.
- * 
+ *
  * @version $Id$
  */
 public class CSVFormat implements Serializable {
@@ -125,8 +126,8 @@ public class CSVFormat implements Serializable {
 
     /**
      * Creates a new CSV format builds.
-     * 
-     * @param delimiter 
+     *
+     * @param delimiter
      *            the char used for value separation, must not be a line break character
      * @throws IllegalArgumentException if the delimiter is a line break character
      */
@@ -137,7 +138,7 @@ public static CSVFormatBuilder newBuilder(final char delimiter) {
     public static CSVFormatBuilder newBuilder(final CSVFormat format) {
         return new CSVFormatBuilder(format);
     }
-    
+
     /**
      * Standard comma separated format, as for {@link #RFC4180} but allowing blank lines.
      * <ul>
@@ -158,7 +159,7 @@ public static CSVFormatBuilder newBuilder() {
      *            the char used for value separation, must not be a line break character
      * @param quoteChar
      *            the char used as value encapsulation marker
-     * @param quotePolicy 
+     * @param quotePolicy
      *            the quote policy
      * @param commentStart
      *            the char used for comment identification
@@ -174,10 +175,12 @@ public static CSVFormatBuilder newBuilder() {
      *            the header
      * @throws IllegalArgumentException if the delimiter is a line break character
      */
-    private CSVFormat(final char delimiter, final Character quoteChar, final Quote quotePolicy, final Character commentStart, final Character escape, final 
-                    boolean ignoreSurroundingSpaces, final boolean ignoreEmptyLines, final String lineSeparator, 
-            final String[] header) {
-        if (isLineBreak(delimiter)) {
+    private CSVFormat(final char delimiter, final Character quoteChar, final Quote quotePolicy, final Character commentStart, final Character escape, final
+                    boolean ignoreSurroundingSpaces, final boolean ignoreEmptyLines, final String lineSeparator,
+ final String[] header)
+    {
+        if (isLineBreak(delimiter))
+        {
             throw new IllegalArgumentException("The delimiter cannot be a line break");
         }
         this.delimiter = delimiter;
@@ -188,7 +191,7 @@ private CSVFormat(final char delimiter, final Character quoteChar, final Quote q
         this.ignoreSurroundingSpaces = ignoreSurroundingSpaces;
         this.ignoreEmptyLines = ignoreEmptyLines;
         this.recordSeparator = lineSeparator;
-        this.header = header;
+        this.header = header == null ? null : header.clone();
     }
 
     /**
@@ -373,7 +376,109 @@ public String toString() {
     public Quote getQuotePolicy() {
         return quotePolicy;
     }
-    
+
+    @Override
+    public int hashCode()
+    {
+        final int prime = 31;
+        int result = 1;
+
+        result = prime * result + delimiter;
+        result = prime * result + ((quotePolicy == null) ? 0 : quotePolicy.hashCode());
+        result = prime * result + ((quoteChar == null) ? 0 : quoteChar.hashCode());
+        result = prime * result + ((commentStart == null) ? 0 : commentStart.hashCode());
+        result = prime * result + ((escape == null) ? 0 : escape.hashCode());
+        result = prime * result + (ignoreSurroundingSpaces ? 1231 : 1237);
+        result = prime * result + (ignoreEmptyLines ? 1231 : 1237);
+        result = prime * result + ((recordSeparator == null) ? 0 : recordSeparator.hashCode());
+        result = prime * result + Arrays.hashCode(header);
+        return result;
+    }
+
+    @Override
+    public boolean equals(Object obj)
+    {
+        if (this == obj)
+        {
+            return true;
+        }
+        if (obj == null)
+        {
+            return false;
+        }
+        if (getClass() != obj.getClass())
+        {
+            return false;
+        }
+
+        CSVFormat other = (CSVFormat) obj;
+        if (delimiter != other.delimiter)
+        {
+            return false;
+        }
+        if (quotePolicy != other.quotePolicy)
+        {
+            return false;
+        }
+        if (quoteChar == null)
+        {
+            if (other.quoteChar != null)
+            {
+                return false;
+            }
+        }
+        else if (!quoteChar.equals(other.quoteChar))
+        {
+            return false;
+        }
+        if (commentStart == null)
+        {
+            if (other.commentStart != null)
+            {
+                return false;
+            }
+        }
+        else if (!commentStart.equals(other.commentStart))
+        {
+            return false;
+        }
+        if (escape == null)
+        {
+            if (other.escape != null)
+            {
+                return false;
+            }
+        }
+        else if (!escape.equals(other.escape))
+        {
+            return false;
+        }
+        if (!Arrays.equals(header, other.header))
+        {
+            return false;
+        }
+        if (ignoreSurroundingSpaces != other.ignoreSurroundingSpaces)
+        {
+            return false;
+        }
+        if (ignoreEmptyLines != other.ignoreEmptyLines)
+        {
+            return false;
+        }
+        if (recordSeparator == null)
+        {
+            if (other.recordSeparator != null)
+            {
+                return false;
+            }
+        }
+        else if (!recordSeparator.equals(other.recordSeparator))
+        {
+            return false;
+        }
+        return true;
+    }
+
     public static class CSVFormatBuilder {
 
         private char delimiter;
@@ -393,7 +498,7 @@ public static class CSVFormatBuilder {
          *            the char used for value separation, must not be a line break character
          * @param quoteChar
          *            the char used as value encapsulation marker
-         * @param quotePolicy 
+         * @param quotePolicy
          *            the quote policy
          * @param commentStart
          *            the char used for comment identification
@@ -410,8 +515,8 @@ public static class CSVFormatBuilder {
          * @throws IllegalArgumentException if the delimiter is a line break character
          */
         // package protected for use by test code
-        CSVFormatBuilder(final char delimiter, final Character quoteChar, final Quote quotePolicy, final Character commentStart, final Character escape, final 
-                        boolean ignoreSurroundingSpaces, final boolean ignoreEmptyLines, final String lineSeparator, 
+        CSVFormatBuilder(final char delimiter, final Character quoteChar, final Quote quotePolicy, final Character commentStart, final Character escape, final
+                        boolean ignoreSurroundingSpaces, final boolean ignoreEmptyLines, final String lineSeparator,
                 final String[] header) {
             if (isLineBreak(delimiter)) {
                 throw new IllegalArgumentException("The delimiter cannot be a line break");
@@ -426,11 +531,11 @@ public static class CSVFormatBuilder {
             this.recordSeparator = lineSeparator;
             this.header = header;
         }
-        
+
         /**
-         * 
+         *
          * Creates a CSVFormatBuilder, using the values of the given CSVFormat.
-         * 
+         *
          * @param format
          *            The format to use values from
          */
@@ -443,8 +548,8 @@ private CSVFormatBuilder(CSVFormat format) {
 
         /**
          * Creates a basic CSVFormatBuilder.
-         * 
-         * @param delimiter 
+         *
+         * @param delimiter
          *            the char used for value separation, must not be a line break character
          * @throws IllegalArgumentException if the delimiter is a line break character
          */
@@ -457,35 +562,35 @@ public CSVFormat build() {
             return new CSVFormat(delimiter, quoteChar, quotePolicy, commentStart, escape,
                                  ignoreSurroundingSpaces, ignoreEmptyLines, recordSeparator, header);
         }
-        
+
         /**
          * Verifies the consistency of the parameters and throws an IllegalStateException if necessary.
-         * 
+         *
          * @throws IllegalStateException
          */
         private void validate() throws IllegalStateException {
             if (quoteChar != null && delimiter == quoteChar.charValue()) {
                 throw new IllegalStateException("The quoteChar character and the delimiter cannot be the same ('" + quoteChar + "')");
             }
-        
+
             if (escape != null && delimiter == escape.charValue()) {
                 throw new IllegalStateException("The escape character and the delimiter cannot be the same ('" + escape + "')");
             }
-        
+
             if (commentStart != null && delimiter == commentStart.charValue()) {
-                throw new IllegalStateException("The comment start character and the delimiter cannot be the same ('" + commentStart + 
+                throw new IllegalStateException("The comment start character and the delimiter cannot be the same ('" + commentStart +
                         "')");
             }
-        
+
             if (quoteChar != null && quoteChar.equals(commentStart)) {
-                throw new IllegalStateException("The comment start character and the quoteChar cannot be the same ('" + commentStart + 
+                throw new IllegalStateException("The comment start character and the quoteChar cannot be the same ('" + commentStart +
                         "')");
             }
-        
+
             if (escape != null && escape.equals(commentStart)) {
                 throw new IllegalStateException("The comment start and the escape character cannot be the same ('" + commentStart + "')");
             }
-        
+
             if (escape == null && quotePolicy == Quote.NONE) {
                 throw new IllegalStateException("No quotes mode set but no escape character is set");
             }
@@ -504,7 +609,7 @@ public CSVFormatBuilder withDelimiter(final char delimiter) {
             if (isLineBreak(delimiter)) {
                 throw new IllegalArgumentException("The delimiter cannot be a line break");
             }
-            this.delimiter = delimiter; 
+            this.delimiter = delimiter;
             return this;
         }
 
@@ -625,7 +730,7 @@ public CSVFormatBuilder withHeader(final String... header) {
             this.header = header;
             return this;
         }
-        
+
         /**
          * Sets the trimming behavior of the format.
          *
diff --git a/src/test/java/org/apache/commons/csv/CSVFormatBuilderTest.java b/src/test/java/org/apache/commons/csv/CSVFormatBuilderTest.java
@@ -21,9 +21,9 @@
 import static org.apache.commons.csv.Constants.CR;
 import static org.apache.commons.csv.Constants.CRLF;
 import static org.apache.commons.csv.Constants.LF;
-import static org.junit.Assert.assertArrayEquals;
 import static org.junit.Assert.assertEquals;
 import static org.junit.Assert.assertFalse;
+import static org.junit.Assert.assertNotSame;
 import static org.junit.Assert.assertTrue;
 
 import org.apache.commons.csv.CSVFormat.CSVFormatBuilder;
@@ -159,7 +159,7 @@ public void testIgnoreEmptyLines() {
     @Test
     public void testCopiedFormatIsEqualToOriginal() {
         CSVFormat copyOfRCF4180 = CSVFormat.newBuilder(RFC4180).build();
-        assertEqualFormats(RFC4180, copyOfRCF4180);
+        assertEquals(RFC4180, copyOfRCF4180);
     }
 
     @Test
@@ -168,16 +168,14 @@ public void testCopiedFormatWithChanges() {
         assertTrue(newFormat.getDelimiter() != RFC4180.getDelimiter());
     }
     
-    // FIXME implement equals on CSVFormat to allow use of Assert.assertEquals()
-    private static void assertEqualFormats(CSVFormat expected, CSVFormat acutal) {
-        assertEquals(expected.getCommentStart(), acutal.getCommentStart());
-        assertEquals(expected.getDelimiter(), acutal.getDelimiter());
-        assertEquals(expected.getEscape(), acutal.getEscape());
-        assertArrayEquals(expected.getHeader(), acutal.getHeader());
-        assertEquals(expected.getIgnoreEmptyLines(), acutal.getIgnoreEmptyLines());
-        assertEquals(expected.getIgnoreSurroundingSpaces(), acutal.getIgnoreSurroundingSpaces());
-        assertEquals(expected.getQuoteChar(), acutal.getQuoteChar());
-        assertEquals(expected.getQuotePolicy(), acutal.getQuotePolicy());
-        assertEquals(expected.getRecordSeparator(), acutal.getRecordSeparator());
+    @Test
+    public void testHeaderReferenceCannotEscape() {
+        String[] header = new String[]{"one", "tow", "three"};
+        builder.withHeader(header);
+        
+        CSVFormat firstFormat = builder.build();
+        CSVFormat secondFormat = builder.build();
+        assertNotSame(header, firstFormat.getHeader());
+        assertNotSame(firstFormat, secondFormat.getHeader());
     }
 }
diff --git a/src/test/java/org/apache/commons/csv/CSVFormatTest.java b/src/test/java/org/apache/commons/csv/CSVFormatTest.java
