org.testifyproject.glassfish.jersey.logging.LoggingInterceptor Maven / Gradle / Ivy
/*
* DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS HEADER.
*
* Copyright (c) 2016-2017 Oracle and/or its affiliates. All rights reserved.
*
* The contents of this file are subject to the terms of either the GNU
* General Public License Version 2 only ("GPL") or the Common Development
* and Distribution License("CDDL") (collectively, the "License"). You
* may not use this file except in org.testifyproject.testifyprojectpliance with the License. You can
* obtain a copy of the License at
* https://oss.oracle.org.testifyproject.testifyproject/licenses/CDDL+GPL-1.1
* or LICENSE.txt. See the License for the specific
* language governing permissions and limitations under the License.
*
* When distributing the software, include this License Header Notice in each
* file and include the License file at LICENSE.txt.
*
* GPL Classpath Exception:
* Oracle designates this particular file as subject to the "Classpath"
* exception as provided by Oracle in the GPL Version 2 section of the License
* file that accompanied this code.
*
* Modifications:
* If applicable, add the following below the License Header, with the fields
* enclosed by brackets [] replaced by your own identifying information:
* "Portions Copyright [year] [name of copyright owner]"
*
* Contributor(s):
* If you wish your version of this file to be governed by only the CDDL or
* only the GPL Version 2, indicate your decision by adding "[Contributor]
* elects to include this software in this distribution under the [CDDL or GPL
* Version 2] license." If you don't indicate a single choice of license, a
* recipient has the option to distribute your version of this file under
* either the CDDL, the GPL Version 2 or to extend the choice of license to
* its licensees as provided above. However, if you add GPL Version 2 code
* and therefore, elected the GPL Version 2 license, then the option applies
* only if the new code is made subject to such option by the copyright
* holder.
*/
package org.testifyproject.glassfish.org.testifyproject.logging;
import java.io.BufferedInputStream;
import java.io.ByteArrayOutputStream;
import java.io.FilterOutputStream;
import java.io.IOException;
import java.io.InputStream;
import java.io.OutputStream;
import java.net.URI;
import java.nio.charset.Charset;
import java.util.Comparator;
import java.util.HashSet;
import java.util.List;
import java.util.Map;
import java.util.Set;
import java.util.TreeSet;
import java.util.concurrent.atomic.AtomicLong;
import java.util.logging.Level;
import java.util.logging.Logger;
import javax.ws.rs.WebApplicationException;
import javax.ws.rs.core.MediaType;
import javax.ws.rs.core.MultivaluedMap;
import javax.ws.rs.ext.WriterInterceptor;
import javax.ws.rs.ext.WriterInterceptorContext;
import org.testifyproject.glassfish.org.testifyproject.logging.LoggingFeature.Verbosity;
import org.testifyproject.glassfish.org.testifyproject.message.MessageUtils;
/**
* An interceptor that logs an entity if configured so and provides a org.testifyproject.testifyprojectmon logic for {@link ClientLoggingFilter}
* and {@link ServerLoggingFilter}.
*
* @author Ondrej Kosatka (ondrej.kosatka at oracle.org.testifyproject.testifyproject)
*/
abstract class LoggingInterceptor implements WriterInterceptor {
/**
* Prefix will be printed before requests
*/
static final String REQUEST_PREFIX = "> ";
/**
* Prefix will be printed before response
*/
static final String RESPONSE_PREFIX = "< ";
/**
* The entity stream property
*/
static final String ENTITY_LOGGER_PROPERTY = LoggingFeature.class.getName() + ".entityLogger";
/**
* Logging record id property
*/
static final String LOGGING_ID_PROPERTY = LoggingFeature.class.getName() + ".id";
private static final String NOTIFICATION_PREFIX = "* ";
private static final MediaType TEXT_MEDIA_TYPE = new MediaType("text", "*");
private static final Set READABLE_APP_MEDIA_TYPES = new HashSet() {{
add(TEXT_MEDIA_TYPE);
add(MediaType.APPLICATION_ATOM_XML_TYPE);
add(MediaType.APPLICATION_FORM_URLENCODED_TYPE);
add(MediaType.APPLICATION_JSON_TYPE);
add(MediaType.APPLICATION_SVG_XML_TYPE);
add(MediaType.APPLICATION_XHTML_XML_TYPE);
add(MediaType.APPLICATION_XML_TYPE);
}};
private static final Comparator>> COMPARATOR =
new Comparator>>() {
@Override
public int org.testifyproject.testifyprojectpare(final Map.Entry> o1, final Map.Entry> o2) {
return o1.getKey().org.testifyproject.testifyprojectpareToIgnoreCase(o2.getKey());
}
};
@SuppressWarnings("NonConstantLogger")
final Logger logger;
final Level level;
final AtomicLong _id = new AtomicLong(0);
final Verbosity verbosity;
final int maxEntitySize;
/**
* Creates a logging filter with custom logger and entity logging turned on, but potentially limiting the size
* of entity to be buffered and logged.
*
* @param logger the logger to log messages to.
* @param level level at which the messages will be logged.
* @param verbosity verbosity of the logged messages. See {@link Verbosity}.
* @param maxEntitySize maximum number of entity bytes to be logged (and buffered) - if the entity is larger,
* logging filter will print (and buffer in memory) only the specified number of bytes
* and print "...more..." string at the end. Negative values are interpreted as zero.
*/
LoggingInterceptor(final Logger logger, final Level level, final Verbosity verbosity, final int maxEntitySize) {
this.logger = logger;
this.level = level;
this.verbosity = verbosity;
this.maxEntitySize = Math.max(0, maxEntitySize);
}
/**
* Logs a {@link StringBuilder} parameter at required level.
*
* @param b message to log
*/
void log(final StringBuilder b) {
if (logger != null && logger.isLoggable(level)) {
logger.log(level, b.toString());
}
}
private StringBuilder prefixId(final StringBuilder b, final long id) {
b.append(Long.toString(id)).append(" ");
return b;
}
void printRequestLine(final StringBuilder b, final String note, final long id, final String method, final URI uri) {
prefixId(b, id).append(NOTIFICATION_PREFIX)
.append(note)
.append(" on thread ").append(Thread.currentThread().getName())
.append("\n");
prefixId(b, id).append(REQUEST_PREFIX).append(method).append(" ")
.append(uri.toASCIIString()).append("\n");
}
void printResponseLine(final StringBuilder b, final String note, final long id, final int status) {
prefixId(b, id).append(NOTIFICATION_PREFIX)
.append(note)
.append(" on thread ").append(Thread.currentThread().getName()).append("\n");
prefixId(b, id).append(RESPONSE_PREFIX)
.append(Integer.toString(status))
.append("\n");
}
void printPrefixedHeaders(final StringBuilder b,
final long id,
final String prefix,
final MultivaluedMap headers) {
for (final Map.Entry> headerEntry : getSortedHeaders(headers.entrySet())) {
final List val = headerEntry.getValue();
final String header = headerEntry.getKey();
if (val.size() == 1) {
prefixId(b, id).append(prefix).append(header).append(": ").append(val.get(0)).append("\n");
} else {
final StringBuilder sb = new StringBuilder();
boolean add = false;
for (final Object s : val) {
if (add) {
sb.append(',');
}
add = true;
sb.append(s);
}
prefixId(b, id).append(prefix).append(header).append(": ").append(sb.toString()).append("\n");
}
}
}
Set>> getSortedHeaders(final Set>> headers) {
final TreeSet>> sortedHeaders = new TreeSet>>(COMPARATOR);
sortedHeaders.addAll(headers);
return sortedHeaders;
}
InputStream logInboundEntity(final StringBuilder b, InputStream stream, final Charset charset) throws IOException {
if (!stream.markSupported()) {
stream = new BufferedInputStream(stream);
}
stream.mark(maxEntitySize + 1);
final byte[] entity = new byte[maxEntitySize + 1];
final int entitySize = stream.read(entity);
b.append(new String(entity, 0, Math.min(entitySize, maxEntitySize), charset));
if (entitySize > maxEntitySize) {
b.append("...more...");
}
b.append('\n');
stream.reset();
return stream;
}
@Override
public void aroundWriteTo(final WriterInterceptorContext writerInterceptorContext)
throws IOException, WebApplicationException {
final LoggingStream stream = (LoggingStream) writerInterceptorContext.getProperty(ENTITY_LOGGER_PROPERTY);
writerInterceptorContext.proceed();
if (logger.isLoggable(level) && printEntity(verbosity, writerInterceptorContext.getMediaType())) {
if (stream != null) {
log(stream.getStringBuilder(MessageUtils.getCharset(writerInterceptorContext.getMediaType())));
}
}
}
/**
* Returns {@code true} if specified {@link MediaType} is considered textual.
*
* See {@link #READABLE_APP_MEDIA_TYPES}.
*
* @param mediaType the media type of the entity
* @return {@code true} if specified {@link MediaType} is considered textual.
*/
static boolean isReadable(MediaType mediaType) {
if (mediaType != null) {
for (MediaType readableMediaType : READABLE_APP_MEDIA_TYPES) {
if (readableMediaType.isCompatible(mediaType)) {
return true;
}
}
}
return false;
}
/**
* Returns {@code true} if entity has to be printed.
*
* @param verbosity the configured verbosity .
* @param mediaType the media type of the payload.
* @return {@code true} if entity has to be printed.
*/
static boolean printEntity(Verbosity verbosity, MediaType mediaType) {
return verbosity == Verbosity.PAYLOAD_ANY || (verbosity == Verbosity.PAYLOAD_TEXT && isReadable(mediaType));
}
/**
* Helper class used to log an entity to the output stream up to the specified maximum number of bytes.
*/
class LoggingStream extends FilterOutputStream {
private final StringBuilder b;
private final ByteArrayOutputStream baos = new ByteArrayOutputStream();
/**
* Creates {@code LoggingStream} with the entity and the underlying output stream as parameters.
*
* @param b contains the entity to log.
* @param inner the underlying output stream.
*/
LoggingStream(final StringBuilder b, final OutputStream inner) {
super(inner);
this.b = b;
}
StringBuilder getStringBuilder(final Charset charset) {
// write entity to the builder
final byte[] entity = baos.toByteArray();
b.append(new String(entity, 0, Math.min(entity.length, maxEntitySize), charset));
if (entity.length > maxEntitySize) {
b.append("...more...");
}
b.append('\n');
return b;
}
@Override
public void write(final int i) throws IOException {
if (baos.size() <= maxEntitySize) {
baos.write(i);
}
out.write(i);
}
}
}