br.eti.clairton.migrator.rest.Uploader Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of migrator-rest Show documentation
Migrator Rest for migrator
The newest version!
package br.eti.clairton.migrator.rest;

import static java.nio.file.Files.readAllBytes;
import static java.util.logging.Level.INFO;
import static java.util.logging.Level.SEVERE;
import static java.util.logging.Logger.getLogger;

import java.io.BufferedInputStream;
import java.io.BufferedReader;
import java.io.DataOutputStream;
import java.io.File;
import java.io.IOException;
import java.io.InputStreamReader;
import java.net.HttpURLConnection;
import java.net.URL;
import java.util.logging.Logger;

public class Uploader {
	private static final Logger logger = getLogger(Uploader.class.getSimpleName());

	public boolean run(final String tenant, final File file, final String url) {
		return run(tenant, file, url, "");
	}

	public boolean run(final String tenant, final File file, final String url, final String token) {
		// https://blog.morizyun.com/blog/android-httpurlconnection-post-multipart/index.html
		final String crlf = "\r\n";
		final String twoHyphens = "--";
		try {
			final String boundary = "*****";
			final HttpURLConnection connection = (HttpURLConnection) new URL(url).openConnection();
			connection.setUseCaches(false);
			connection.setDoOutput(true);
			connection.setDoInput(true);
			connection.setRequestMethod("POST");
			connection.setRequestProperty("Connection", "Keep-Alive");
			connection.setRequestProperty("Cache-Control", "no-cache");
			connection.setRequestProperty("Authorization", "Bearer " + token);
			connection.setRequestProperty("Content-Type", "multipart/form-data;boundary=" + boundary);
			final DataOutputStream output = new DataOutputStream(connection.getOutputStream());
			output.writeBytes(twoHyphens + boundary + crlf);
			output.writeBytes("Content-Disposition: form-data; name=\"file\"; filename=\"changelog.zip\"");
			output.writeBytes(crlf);
			output.writeBytes("Content-Type: application/zip");
			output.writeBytes(crlf + crlf);
			byte[] bytes = readAllBytes(file.toPath());
			output.write(bytes);
			output.writeBytes(crlf);
			output.writeBytes(twoHyphens + boundary + crlf);
			output.writeBytes("Content-Disposition: form-data; name=\"tenant\"");
			output.writeBytes(crlf);
			output.writeBytes("Content-Type: text/plain");
			output.writeBytes(crlf + crlf);
			output.writeBytes(tenant);
			output.writeBytes(crlf);
			output.writeBytes(twoHyphens + boundary + twoHyphens + crlf);
			output.flush();
			output.close();
			int status = connection.getResponseCode();
			logger.log(INFO, "Http status {0} for {1} address", new Object[] { status, url });
			final boolean isSuccess = status == 200;
			if (!isSuccess) {
				final StringBuffer sb = new StringBuffer();
				final BufferedInputStream response = new BufferedInputStream(connection.getErrorStream());
				final BufferedReader br = new BufferedReader(new InputStreamReader(response));
				String inputLine = "";
				while ((inputLine = br.readLine()) != null) {
					sb.append(inputLine);
				}
				logger.log(SEVERE, sb.toString());
			}
			return isSuccess;
		} catch (final IOException e) {
			throw new RuntimeException(e);
		}
	}

}