forked from D-Net/dnet-hadoop
152 lines
4.6 KiB
Java
152 lines
4.6 KiB
Java
|
|
package eu.dnetlib.dhp.oa.graph.dump;
|
|
|
|
import java.io.*;
|
|
import java.io.IOException;
|
|
import java.nio.charset.StandardCharsets;
|
|
|
|
import org.apache.commons.io.FileUtils;
|
|
import org.apache.http.HttpEntity;
|
|
import org.apache.http.HttpResponse;
|
|
import org.apache.http.client.HttpClient;
|
|
import org.apache.http.client.methods.HttpPost;
|
|
import org.apache.http.client.methods.HttpPut;
|
|
import org.apache.http.entity.ByteArrayEntity;
|
|
import org.apache.http.entity.InputStreamEntity;
|
|
import org.apache.http.entity.StringEntity;
|
|
//import org.apache.http.entity.mime.MultipartEntityBuilder;
|
|
import org.apache.http.impl.client.CloseableHttpClient;
|
|
import org.apache.http.impl.client.DefaultHttpClient;
|
|
import org.apache.http.impl.client.HttpClients;
|
|
import org.apache.http.util.EntityUtils;
|
|
|
|
import com.google.gson.Gson;
|
|
|
|
import eu.dnetlib.dhp.oa.graph.dump.zenodo.ZenodoModel;
|
|
|
|
//import org.apache.http.entity.mime.MultipartEntityBuilder;
|
|
|
|
public class APIClient implements Serializable {
|
|
|
|
String urlString;
|
|
String bucket;
|
|
|
|
String deposition_id;
|
|
String access_token;
|
|
|
|
public String getUrlString() {
|
|
return urlString;
|
|
}
|
|
|
|
public void setUrlString(String urlString) {
|
|
this.urlString = urlString;
|
|
}
|
|
|
|
public String getBucket() {
|
|
return bucket;
|
|
}
|
|
|
|
public void setBucket(String bucket) {
|
|
this.bucket = bucket;
|
|
}
|
|
|
|
public APIClient(String urlString, String access_token) throws IOException {
|
|
|
|
this.urlString = urlString;
|
|
this.access_token = access_token;
|
|
}
|
|
|
|
public int connect() throws IOException {
|
|
|
|
String json = "{}";
|
|
|
|
HttpClient client = new DefaultHttpClient();
|
|
|
|
HttpPost post = new HttpPost(urlString);
|
|
|
|
StringEntity input = new StringEntity(json);
|
|
post.setEntity(input);
|
|
post.addHeader("Content-Type", "application/json");
|
|
post.setHeader("Authorization", "Bearer " + access_token);
|
|
|
|
HttpResponse response = client.execute(post);
|
|
|
|
json = EntityUtils.toString(response.getEntity());
|
|
|
|
ZenodoModel newSubmission = new Gson().fromJson(json, ZenodoModel.class);
|
|
this.bucket = newSubmission.getLinks().getBucket();
|
|
this.deposition_id = newSubmission.getId();
|
|
|
|
return response.getStatusLine().getStatusCode();
|
|
|
|
}
|
|
|
|
// public int upload(InputStream is, String file_name) throws IOException {
|
|
// HttpClient client = new DefaultHttpClient();
|
|
//
|
|
// HttpPut put = new HttpPut(bucket + "/" + file_name);
|
|
// put.setHeader("Authorization", "Bearer " + access_token);
|
|
// put.addHeader("Content-Type", "application/zip");
|
|
//
|
|
// HttpEntity data = MultipartEntityBuilder
|
|
// .create()
|
|
// // .addPart("file", new ByteArrayInputStream(is));
|
|
// .addBinaryBody(file_name, is, ContentType.APPLICATION_OCTET_STREAM, file_name)
|
|
// .build();
|
|
// put.setEntity(data);
|
|
//
|
|
// HttpResponse response = client.execute(put);
|
|
//
|
|
// return response.getStatusLine().getStatusCode();
|
|
// }
|
|
|
|
public int upload(File file, String file_name) throws IOException {
|
|
HttpClient client = new DefaultHttpClient();
|
|
|
|
HttpPut put = new HttpPut(bucket + "/" + file_name);
|
|
put.setHeader("Authorization", "Bearer " + access_token);
|
|
put.addHeader("Content-Type", "application/zip");
|
|
InputStreamEntity data = new InputStreamEntity(new FileInputStream(file), -1);
|
|
data.setContentType("binary/octet-stream");
|
|
data.setChunked(true); // Send in multiple parts if needed
|
|
// ByteArrayInputStream bais = new ByteArrayInputStream(FileUtils.readFileToByteArray(file));
|
|
// EntityUtils.toByteArray(new ByteArrayInputStream(FileUtils.readFileToByteArray(file)));
|
|
// InputStream targetStream = new FileInputStream(file);
|
|
// DataInputStream tmp = new DataInputStream(targetStream);
|
|
// HttpEntity data = new ByteArrayEntity(tmp.);
|
|
// HttpEntity data = MultipartEntityBuilder.create().addBinaryBody(file_name, file).build();
|
|
put.setEntity(data);
|
|
|
|
HttpResponse response = client.execute(put);
|
|
|
|
return response.getStatusLine().getStatusCode();
|
|
|
|
}
|
|
|
|
public int sendMretadata(String metadata) throws IOException {
|
|
|
|
HttpClient client = new DefaultHttpClient();
|
|
HttpPut post = new HttpPut(urlString + "/" + deposition_id);
|
|
post.setHeader("Authorization", "Bearer " + access_token);
|
|
post.addHeader("Content-Type", "application/json");
|
|
StringEntity entity = new StringEntity(metadata, StandardCharsets.UTF_8);
|
|
post.setEntity(entity);
|
|
|
|
HttpResponse response = client.execute(post);
|
|
|
|
return response.getStatusLine().getStatusCode();
|
|
|
|
}
|
|
|
|
public int publish() throws IOException {
|
|
HttpClient client = new DefaultHttpClient();
|
|
HttpPost post = new HttpPost(urlString + "/" + deposition_id + "/actions/publish");
|
|
post.setHeader("Authorization", "Bearer " + access_token);
|
|
|
|
HttpResponse response = client.execute(post);
|
|
|
|
return response.getStatusLine().getStatusCode();
|
|
}
|
|
|
|
}
|