From 6058f5f8d2ed19ca3c0461e7080680a093834823 Mon Sep 17 00:00:00 2001
From: Jon Marius Venstad <jonmv@users.noreply.github.com>
Date: Mon, 2 May 2022 21:08:17 +0200
Subject: Revert "Jonmv/remove last controller jersey client [run-systemtest]"

---
 .../ai/vespa/hosted/api/MultiPartStreamer.java     | 29 +++++++++++-----------
 1 file changed, 14 insertions(+), 15 deletions(-)

(limited to 'hosted-api/src/main')

diff --git a/hosted-api/src/main/java/ai/vespa/hosted/api/MultiPartStreamer.java b/hosted-api/src/main/java/ai/vespa/hosted/api/MultiPartStreamer.java
index 8c858437ad7..f155cbbce07 100644
--- a/hosted-api/src/main/java/ai/vespa/hosted/api/MultiPartStreamer.java
+++ b/hosted-api/src/main/java/ai/vespa/hosted/api/MultiPartStreamer.java
@@ -1,8 +1,6 @@
 // Copyright Yahoo. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root.
 package ai.vespa.hosted.api;
 
-import com.yahoo.yolean.Exceptions;
-
 import java.io.BufferedInputStream;
 import java.io.ByteArrayInputStream;
 import java.io.IOException;
@@ -53,26 +51,26 @@ public class MultiPartStreamer {
 
     /** Adds the given data as a named part in this, using the given content type. */
     public MultiPartStreamer addData(String name, String type, String data) {
-        return addData(name, type, null, () -> asStream(data));
+        streams.add(() -> separator(name, type));
+        streams.add(() -> asStream(data));
+
+        return this;
     }
 
     /** Adds the given data as a named part in this, using the {@code "application/octet-stream" content type}. */
     public MultiPartStreamer addBytes(String name, byte[] bytes) {
-        return addData(name, "application/octet-stream", null, () -> new ByteArrayInputStream(bytes));
-    }
-
-    /** Adds the given data as a named part in this, using the given content type. */
-    public MultiPartStreamer addData(String name, String type, String filename, Supplier<InputStream> data) {
-        streams.add(() -> separator(name, filename, type));
-        streams.add(data);
+        streams.add(() -> separator(name, "application/octet-stream"));
+        streams.add(() -> new ByteArrayInputStream(bytes));
 
         return this;
     }
 
     /** Adds the contents of the file at the given path as a named part in this. */
     public MultiPartStreamer addFile(String name, Path path) {
-        String type = Exceptions.uncheck(() -> Files.probeContentType(path));
-        return addData(name, type != null ? type : "application/octet-stream", path.getFileName().toString(), () -> asStream(path));
+        streams.add(() -> separator(name, path));
+        streams.add(() -> asStream(path));
+
+        return this;
     }
 
     /**
@@ -109,15 +107,16 @@ public class MultiPartStreamer {
     }
 
     /** Returns the separator to put between one part and the next, when this is a string. */
-    private InputStream separator(String name, String filename, String contentType) {
-        return asStream(disposition(name) + (filename == null ? "" : "; filename=\"" + filename + "\"") + type(contentType));
+    private InputStream separator(String name, String contentType) {
+        return asStream(disposition(name) + type(contentType));
     }
 
     /** Returns the separator to put between one part and the next, when this is a file. */
     private InputStream separator(String name, Path path) {
         try {
             String contentType = Files.probeContentType(path);
-            return separator(name, path.getFileName().toString(), contentType != null ? contentType : "application/octet-stream");
+            return asStream(disposition(name) + "; filename=\"" + path.getFileName() + "\"" +
+                            type(contentType != null ? contentType : "application/octet-stream"));
         }
         catch (IOException e) {
             throw new UncheckedIOException(e);
-- 
cgit v1.2.3