Improve #883

AsyncHttpClient · May 15, 2015 · 2c0e384 · 2c0e384
1 parent 2858f8f
commit 2c0e384
Showing 1 changed file with 6 additions and 5 deletions.
diff --git a/api/src/main/java/org/asynchttpclient/multipart/PartBase.java b/api/src/main/java/org/asynchttpclient/multipart/PartBase.java
@@ -18,8 +18,9 @@
 import java.io.IOException;
 import java.io.OutputStream;
 import java.nio.charset.Charset;
+import java.util.ArrayList;
+import java.util.List;
 
-import org.asynchttpclient.FluentStringsMap;
 import org.asynchttpclient.Param;
 
 public abstract class PartBase implements Part {
@@ -57,7 +58,7 @@ public abstract class PartBase implements Part {
     /**
      * Additional part headers
      */
-    private FluentStringsMap customHeaders;
+    private List<Param> customHeaders;
 
     /**
      * Constructor.
@@ -127,7 +128,7 @@ protected void visitContentIdHeader(PartVisitor visitor) throws IOException {
 
     protected void visitCustomHeaders(PartVisitor visitor) throws IOException {
         if (isNonEmpty(customHeaders)) {
-            for (Param param: customHeaders.toParams()) {
+            for (Param param: customHeaders) {
                 visitor.withBytes(CRLF_BYTES);
                 visitor.withBytes(param.getName().getBytes(US_ASCII));
                 visitor.withBytes(param.getValue().getBytes(US_ASCII));
@@ -251,8 +252,8 @@ public void setDispositionType(String dispositionType) {
 
     public void addCustomHeader(String name, String value) {
         if (customHeaders == null) {
-            customHeaders = new FluentStringsMap();
+            customHeaders = new ArrayList<Param>(2);
         }
-        customHeaders.add(name, value);
+        customHeaders.add(new Param(name, value));
     }
 }