[DM-41951] BufferedOutputStreams for everything

This mimics a change I'm working on in lsst-tap-service where we stop using outputstream to try to avoid out of memory errors.
lsst-sqre · Dec 5, 2023 · dcf0f4b · dcf0f4b
1 parent 733299e
commit dcf0f4b
Showing 1 changed file with 9 additions and 6 deletions.
diff --git a/tap/src/main/java/ca/nrc/cadc/sample/ResultStoreImpl.java b/tap/src/main/java/ca/nrc/cadc/sample/ResultStoreImpl.java
@@ -80,7 +80,7 @@
 import com.google.cloud.storage.StorageOptions;
 
 import java.io.File;
-import java.io.OutputStream;
+import java.io.BufferedOutputStream;
 import java.io.IOException;
 import java.nio.channels.Channels;
 import java.net.MalformedURLException;
@@ -100,7 +100,7 @@ public class ResultStoreImpl implements ResultStore {
     public URL put(final ResultSet resultSet,
                    final TableWriter<ResultSet> resultSetTableWriter)
         throws IOException {
-        OutputStream os = getOutputStream();
+        BufferedOutputStream os = getOutputStream();
         resultSetTableWriter.write(resultSet, os);
         os.close();
         return getURL();
@@ -109,7 +109,7 @@ public URL put(final ResultSet resultSet,
     @Override
     public URL put(Throwable throwable, TableWriter tableWriter)
         throws IOException {
-        OutputStream os = getOutputStream();
+        BufferedOutputStream os = getOutputStream();
         tableWriter.write(throwable, os);
         os.close();
         return getURL();
@@ -119,7 +119,7 @@ public URL put(Throwable throwable, TableWriter tableWriter)
     public URL put(final ResultSet resultSet,
                    final TableWriter<ResultSet> resultSetTableWriter,
                    final Integer integer) throws IOException {
-        OutputStream os = getOutputStream();
+        BufferedOutputStream os = getOutputStream();
 
         if (integer == null) {
             resultSetTableWriter.write(resultSet, os);
@@ -132,12 +132,15 @@ public URL put(final ResultSet resultSet,
         return getURL();
     }
 
-    private OutputStream getOutputStream() {
+    private BufferedOutputStream getOutputStream() {
         Storage storage = StorageOptions.getDefaultInstance().getService();
         BlobId blobId = BlobId.of(bucket, filename);
         BlobInfo blobInfo = BlobInfo.newBuilder(blobId).setContentType("application/x-votable+xml").build();
         Blob blob = storage.create(blobInfo);
-        return Channels.newOutputStream(blob.writer());
+        return new BufferedOutputStream(
+            Channels.newOutputStream(blob.writer()),
+	    32768
+        );
     }
 
     private URL getURL() throws MalformedURLException {