org.apache.dubbo.remoting.http12.HttpUtils Maven / Gradle / Ivy
Go to download
Show more of this group Show more artifacts with this name
Show all versions of dubbo Show documentation
Show all versions of dubbo Show documentation
The all in one project of dubbo
The newest version!
/*
* Licensed to the Apache Software Foundation (ASF) under one or more
* contributor license agreements. See the NOTICE file distributed with
* this work for additional information regarding copyright ownership.
* The ASF licenses this file to You 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 org.apache.dubbo.remoting.http12;
import org.apache.dubbo.common.io.StreamUtils;
import org.apache.dubbo.common.utils.StringUtils;
import org.apache.dubbo.remoting.http12.exception.DecodeException;
import java.io.IOException;
import java.io.InputStream;
import java.nio.charset.Charset;
import java.util.ArrayList;
import java.util.Collections;
import java.util.List;
import java.util.Locale;
import io.netty.buffer.ByteBuf;
import io.netty.buffer.ByteBufAllocator;
import io.netty.buffer.ByteBufInputStream;
import io.netty.buffer.ByteBufOutputStream;
import io.netty.buffer.UnpooledByteBufAllocator;
import io.netty.handler.codec.http.DefaultFullHttpRequest;
import io.netty.handler.codec.http.DefaultHttpHeaders;
import io.netty.handler.codec.http.HttpHeaders;
import io.netty.handler.codec.http.HttpMethod;
import io.netty.handler.codec.http.HttpResponseStatus;
import io.netty.handler.codec.http.HttpVersion;
import io.netty.handler.codec.http.cookie.Cookie;
import io.netty.handler.codec.http.cookie.CookieHeaderNames.SameSite;
import io.netty.handler.codec.http.cookie.DefaultCookie;
import io.netty.handler.codec.http.cookie.ServerCookieDecoder;
import io.netty.handler.codec.http.cookie.ServerCookieEncoder;
import io.netty.handler.codec.http.multipart.Attribute;
import io.netty.handler.codec.http.multipart.DefaultHttpDataFactory;
import io.netty.handler.codec.http.multipart.FileUpload;
import io.netty.handler.codec.http.multipart.HttpDataFactory;
import io.netty.handler.codec.http.multipart.HttpPostRequestDecoder;
import io.netty.handler.codec.http.multipart.InterfaceHttpData;
public final class HttpUtils {
public static final ByteBufAllocator HEAP_ALLOC = new UnpooledByteBufAllocator(false, false);
public static final HttpDataFactory DATA_FACTORY = new DefaultHttpDataFactory(DefaultHttpDataFactory.MINSIZE);
public static final String CHARSET_PREFIX = "charset=";
private HttpUtils() {}
public static String getStatusMessage(int status) {
return HttpResponseStatus.valueOf(status).reasonPhrase();
}
public static String toStatusString(int statusCode) {
if (statusCode == 200) {
return HttpStatus.OK.getStatusString();
}
if (statusCode == 500) {
return HttpStatus.INTERNAL_SERVER_ERROR.getStatusString();
}
return Integer.toString(statusCode);
}
public static List decodeCookies(String value) {
List cookies = new ArrayList<>();
for (Cookie c : ServerCookieDecoder.LAX.decodeAll(value)) {
cookies.add(new HttpCookie(c.name(), c.value()));
}
return cookies;
}
public static String encodeCookie(HttpCookie cookie) {
DefaultCookie c = new DefaultCookie(cookie.name(), cookie.value());
c.setPath(cookie.path());
c.setDomain(cookie.domain());
c.setMaxAge(cookie.maxAge());
c.setSecure(cookie.secure());
c.setHttpOnly(cookie.httpOnly());
c.setSameSite(SameSite.valueOf(cookie.sameSite()));
return ServerCookieEncoder.LAX.encode(c);
}
public static List parseAccept(String header) {
if (header == null) {
return Collections.emptyList();
}
List> mediaTypes = new ArrayList<>();
for (String item : StringUtils.tokenize(header, ',')) {
int index = item.indexOf(';');
mediaTypes.add(new Item<>(StringUtils.substring(item, 0, index), parseQuality(item, index)));
}
return Item.sortAndGet(mediaTypes);
}
public static float parseQuality(String expr, int index) {
float quality = 1.0F;
if (index != -1) {
int qStart = expr.indexOf("q=", index + 1);
if (qStart != -1) {
qStart += 2;
int qEnd = expr.indexOf(',', qStart);
String qString = qEnd == -1
? expr.substring(qStart)
: expr.substring(qStart, qEnd).trim();
try {
quality = Float.parseFloat(qString);
} catch (NumberFormatException ignored) {
}
}
}
return quality;
}
public static List parseAcceptLanguage(String header) {
List> locales = new ArrayList<>();
if (header == null) {
return Collections.emptyList();
}
for (String item : StringUtils.tokenize(header, ',')) {
String[] pair = StringUtils.tokenize(item, ';');
locales.add(new Item<>(parseLocale(pair[0]), pair.length > 1 ? Float.parseFloat(pair[1]) : 1.0F));
}
return Item.sortAndGet(locales);
}
public static List parseContentLanguage(String header) {
List locales = new ArrayList<>();
if (header == null) {
return Collections.emptyList();
}
for (String item : StringUtils.tokenize(header, ',')) {
locales.add(parseLocale(item));
}
return locales;
}
public static Locale parseLocale(String locale) {
String[] parts = StringUtils.tokenize(locale, '-', '_');
switch (parts.length) {
case 2:
return new Locale(parts[0], parts[1]);
case 3:
return new Locale(parts[0], parts[1], parts[2]);
default:
return new Locale(parts[0]);
}
}
@SuppressWarnings("deprecation")
public static HttpPostRequestDecoder createPostRequestDecoder(
HttpRequest request, InputStream inputStream, String charset) {
ByteBuf data;
boolean canMark = inputStream.markSupported();
try {
if (canMark) {
inputStream.mark(Integer.MAX_VALUE);
}
if (inputStream.available() == 0) {
return null;
} else {
data = HEAP_ALLOC.buffer();
ByteBufOutputStream os = new ByteBufOutputStream(data);
StreamUtils.copy(inputStream, os);
}
} catch (IOException e) {
throw new DecodeException("Error while reading post data: " + e.getMessage(), e);
} finally {
try {
if (canMark) {
inputStream.reset();
} else {
inputStream.close();
}
} catch (IOException ignored) {
}
}
DefaultFullHttpRequest nRequest = new DefaultFullHttpRequest(
HttpVersion.HTTP_1_1,
HttpMethod.POST,
request.uri(),
data,
new DefaultHttpHeaders(false),
new DefaultHttpHeaders(false));
HttpHeaders headers = nRequest.headers();
request.headers().forEach(e -> headers.add(e.getKey(), e.getValue()));
if (charset == null) {
return new HttpPostRequestDecoder(DATA_FACTORY, nRequest);
} else {
return new HttpPostRequestDecoder(DATA_FACTORY, nRequest, Charset.forName(charset));
}
}
public static String readPostValue(InterfaceHttpData item) {
try {
return ((Attribute) item).getValue();
} catch (IOException e) {
throw new DecodeException("Error while reading post value: " + e.getMessage(), e);
}
}
public static HttpRequest.FileUpload readUpload(InterfaceHttpData item) {
return new DefaultFileUploadAdapter((FileUpload) item);
}
private static final class DefaultFileUploadAdapter implements HttpRequest.FileUpload {
private final FileUpload fu;
private InputStream inputStream;
DefaultFileUploadAdapter(FileUpload fu) {
this.fu = fu;
}
@Override
public String name() {
return fu.getName();
}
@Override
public String filename() {
return fu.getFilename();
}
@Override
public String contentType() {
return fu.getContentType();
}
@Override
public int size() {
return (int) fu.length();
}
@Override
public InputStream inputStream() {
if (inputStream == null) {
inputStream = new ByteBufInputStream(fu.content());
}
return inputStream;
}
}
private static final class Item implements Comparable> {
private final V value;
private final float q;
public Item(V value, float q) {
this.value = value;
this.q = q;
}
@Override
public int compareTo(Item o) {
return Float.compare(o.q, q);
}
public static List sortAndGet(List> items) {
int size = items.size();
if (size == 1) {
return Collections.singletonList(items.get(0).value);
}
Collections.sort(items);
List values = new ArrayList<>(size);
for (int i = 0; i < size; i++) {
values.add(items.get(i).value);
}
return values;
}
}
}