Skip to content

Commit

Permalink
avoid caching keystore if local file
Browse files Browse the repository at this point in the history
  • Loading branch information
apucher committed Jan 28, 2022
1 parent 52f058c commit 9d4950c
Showing 1 changed file with 7 additions and 1 deletion.
Original file line number Diff line number Diff line change
Expand Up @@ -24,6 +24,7 @@
import java.io.InputStream;
import java.io.OutputStream;
import java.net.URI;
import java.net.URL;
import java.nio.file.Files;
import java.util.ArrayList;
import java.util.Arrays;
Expand Down Expand Up @@ -264,10 +265,15 @@ public static String toString(Collection<? extends ListenerConfig> listenerConfi

private static File cacheInTempFile(String sourceUrl) {
try {
URL url = TlsUtils.makeKeyStoreUrl(sourceUrl);
if ("file".equals(url.getProtocol())) {
return new File(url.getPath());
}

File tempFile = Files.createTempFile("pinot-keystore-", null).toFile();
tempFile.deleteOnExit();

try (InputStream is = TlsUtils.makeKeyStoreUrl(sourceUrl).openStream();
try (InputStream is = url.openStream();
OutputStream os = new FileOutputStream(tempFile)) {
IOUtils.copy(is, os);
}
Expand Down

0 comments on commit 9d4950c

Please sign in to comment.