org.apache.dubbo.remoting.http12.HttpUtils Maven / Gradle / Ivy
/*
* 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.Unpooled;
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) {
List> mediaTypes = new ArrayList<>();
if (header == null) {
return Collections.emptyList();
}
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) {
data = Unpooled.EMPTY_BUFFER;
} 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;
}
}
}
© 2015 - 2025 Weber Informatics LLC | Privacy Policy