You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
145 lines
3.9 KiB
Java
145 lines
3.9 KiB
Java
/*
|
|
* Copyright 2008 National Library of Sweden
|
|
*
|
|
* Licensed under the Apache License, Version 2.0 (the "License");
|
|
* you may not use this file except in compliance with the License.
|
|
* You may obtain a copy of the License at
|
|
*
|
|
* http://www.apache.org/licenses/LICENSE-2.0
|
|
*
|
|
* Unless required by applicable law or agreed to in writing, software
|
|
* distributed under the License is distributed on an "AS IS" BASIS,
|
|
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
* See the License for the specific language governing permissions and
|
|
* limitations under the License.
|
|
*/
|
|
|
|
package se.kb.oai.ore;
|
|
|
|
import javax.net.ssl.HostnameVerifier;
|
|
import javax.net.ssl.HttpsURLConnection;
|
|
import javax.net.ssl.SSLSession;
|
|
import java.io.ByteArrayOutputStream;
|
|
import java.io.IOException;
|
|
import java.io.InputStream;
|
|
import java.net.HttpURLConnection;
|
|
import java.net.URI;
|
|
import java.net.URISyntaxException;
|
|
import java.net.URL;
|
|
|
|
/**
|
|
* An aggregated resource is a resource, that together with other resources,
|
|
* create an {@link Aggregation}. This class has the actual content of a
|
|
* resource and also metadata about it.
|
|
*
|
|
* @author Oskar Grenholm, National Library of Sweden
|
|
*/
|
|
public class AggregatedResource extends AggregateBase {
|
|
|
|
private String mimetype;
|
|
|
|
/**
|
|
* Create an <code>AggreagatedResource</code> with the specified id.
|
|
*
|
|
* @param id the id
|
|
*
|
|
* @throws URISyntaxException
|
|
*/
|
|
public AggregatedResource(String id) throws URISyntaxException {
|
|
this(new URI(id));
|
|
}
|
|
|
|
|
|
/**
|
|
* Create an <code>AggreagatedResource</code> with the specified id.
|
|
*
|
|
* @param id the id
|
|
*/
|
|
public AggregatedResource(URI id) {
|
|
super(id);
|
|
}
|
|
/**
|
|
* Get the MIME type (if known) for this <code>AggreagatedResource</code>.
|
|
*
|
|
* @return the MIME type, or <code>null</code> if not known
|
|
*/
|
|
public String getMimeType() {
|
|
return mimetype;
|
|
}
|
|
|
|
/**
|
|
* Set the MIME type for this <code>AggreagatedResource</code>.
|
|
*
|
|
* @param mimetype the MIME type
|
|
*/
|
|
public void setMimeType(String mimetype) {
|
|
this.mimetype = mimetype;
|
|
}
|
|
|
|
/**
|
|
* Get an <code>InputStream</code> for the content held in this
|
|
* <code>AggreagatedResource</code>.
|
|
*
|
|
* @return a stream to the content
|
|
* @throws IOException
|
|
*/
|
|
public InputStream getContent() throws IOException, URISyntaxException {
|
|
URL url;
|
|
String uri_schema;
|
|
HttpURLConnection con = null;
|
|
boolean redirect = true;
|
|
|
|
HostnameVerifier hostnameVerifier = new HostnameVerifier() {
|
|
public boolean verify(String hostname, SSLSession session) {
|
|
return true;
|
|
}
|
|
};
|
|
|
|
while (redirect) {
|
|
redirect = false;
|
|
uri_schema = id.getScheme();
|
|
url = id.toURL();
|
|
if (uri_schema.equals("https")) {
|
|
HttpsURLConnection https = (HttpsURLConnection) url.openConnection();
|
|
https.setHostnameVerifier(hostnameVerifier);
|
|
con = https;
|
|
} else if (uri_schema.equals("http")) {
|
|
con = (HttpURLConnection) url.openConnection();
|
|
}
|
|
int status = con.getResponseCode();
|
|
if (status != HttpURLConnection.HTTP_OK) {
|
|
if (status == HttpURLConnection.HTTP_MOVED_TEMP
|
|
|| status == HttpURLConnection.HTTP_MOVED_PERM
|
|
|| status == HttpURLConnection.HTTP_SEE_OTHER) {
|
|
redirect = true;
|
|
}
|
|
}
|
|
|
|
if (redirect) {
|
|
new URI(con.getHeaderField("Location"));
|
|
}
|
|
}
|
|
return con.getInputStream();
|
|
}
|
|
|
|
/**
|
|
* Get a <code>String</code> containing the content for this
|
|
* <code>AggreagatedResource</code>.
|
|
*
|
|
* @return a stream to the content
|
|
* @throws IOException
|
|
*/
|
|
public String getContentAsString() throws IOException, URISyntaxException {
|
|
InputStream in = getContent();
|
|
ByteArrayOutputStream out = new ByteArrayOutputStream();
|
|
byte[] bytes = new byte[4 * 1024];
|
|
int read = in.read(bytes);
|
|
while (read != -1) {
|
|
out.write(bytes, 0, read);
|
|
read = in.read(bytes);
|
|
}
|
|
in.close();
|
|
return new String(out.toByteArray());
|
|
}
|
|
}
|