com.rt.storage.api.client.http.UrlEncodedContent Maven / Gradle / Ivy
package com.rt.storage.api.client.http;
import com.rt.storage.api.client.util.Data;
import com.rt.storage.api.client.util.FieldInfo;
import com.rt.storage.api.client.util.Preconditions;
import com.rt.storage.api.client.util.Types;
import com.rt.storage.api.client.util.escape.CharEscapers;
import java.io.BufferedWriter;
import java.io.IOException;
import java.io.OutputStream;
import java.io.OutputStreamWriter;
import java.io.Writer;
import java.util.HashMap;
import java.util.Map;
/**
* Implements support for HTTP form content encoding serialization of type {@code
* application/x-www-form-urlencoded} as specified in the HTML 4.0 Specification.
*
* Sample usage:
*
*
* static void setContent(HttpRequest request, Object item) {
* request.setContent(new UrlEncodedContent(item));
* }
*
*
* Implementation is not thread-safe.
*
* @since 1.0
* @author Yaniv Inbar
*/
public class UrlEncodedContent extends AbstractHttpContent {
/** Key name/value data. */
private Object data;
/** @param data key name/value data */
public UrlEncodedContent(Object data) {
super(UrlEncodedParser.MEDIA_TYPE);
setData(data);
}
public void writeTo(OutputStream out) throws IOException {
Writer writer = new BufferedWriter(new OutputStreamWriter(out, getCharset()));
boolean first = true;
for (Map.Entry nameValueEntry : Data.mapOf(data).entrySet()) {
Object value = nameValueEntry.getValue();
if (value != null) {
String name = CharEscapers.escapeUri(nameValueEntry.getKey());
Class valueClass = value.getClass();
if (value instanceof Iterable || valueClass.isArray()) {
for (Object repeatedValue : Types.iterableOf(value)) {
first = appendParam(first, writer, name, repeatedValue);
}
} else {
first = appendParam(first, writer, name, value);
}
}
}
writer.flush();
}
@Override
public UrlEncodedContent setMediaType(HttpMediaType mediaType) {
super.setMediaType(mediaType);
return this;
}
/**
* Returns the key name/value data or {@code null} for none.
*
* @since 1.5
*/
public final Object getData() {
return data;
}
/**
* Sets the key name/value data.
*
* Overriding is only supported for the purpose of calling the super implementation and
* changing the return type, but nothing else.
*
* @since 1.5
*/
public UrlEncodedContent setData(Object data) {
this.data = Preconditions.checkNotNull(data);
return this;
}
/**
* Returns the URL-encoded content of the given HTTP request, or if none return and set as content
* a new instance of {@link UrlEncodedContent} (whose {@link #getData()} is an implementation of
* {@link Map}).
*
* @param request HTTP request
* @return URL-encoded content
* @throws ClassCastException if the HTTP request has a content defined that is not {@link
* UrlEncodedContent}
* @since 1.7
*/
public static UrlEncodedContent getContent(HttpRequest request) {
HttpContent content = request.getContent();
if (content != null) {
return (UrlEncodedContent) content;
}
UrlEncodedContent result = new UrlEncodedContent(new HashMap());
request.setContent(result);
return result;
}
private static boolean appendParam(boolean first, Writer writer, String name, Object value)
throws IOException {
// ignore nulls
if (value == null || Data.isNull(value)) {
return first;
}
// append value
if (first) {
first = false;
} else {
writer.write("&");
}
writer.write(name);
String stringValue =
CharEscapers.escapeUri(
value instanceof Enum ? FieldInfo.of((Enum) value).getName() : value.toString());
if (stringValue.length() != 0) {
writer.write("=");
writer.write(stringValue);
}
return first;
}
}