org.apache.wicket.request.handler.resource.ResourceStreamRequestHandler.java Source code

Java tutorial

Introduction

Here is the source code for org.apache.wicket.request.handler.resource.ResourceStreamRequestHandler.java

Source

/*
 * 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.wicket.request.handler.resource;

import org.apache.wicket.core.request.handler.logger.ResourceStreamLogData;
import org.apache.wicket.request.ILoggableRequestHandler;
import org.apache.wicket.request.IRequestCycle;
import org.apache.wicket.request.IRequestHandler;
import org.apache.wicket.request.resource.ContentDisposition;
import org.apache.wicket.request.resource.IResource.Attributes;
import org.apache.wicket.request.resource.ResourceStreamResource;
import org.apache.wicket.util.lang.Args;
import org.apache.wicket.util.resource.IResourceStream;
import org.apache.wicket.util.string.Strings;
import org.apache.wicket.util.time.Duration;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

/**
 * Request target that responds by sending its resource stream.
 * 
 * @author Eelco Hillenius
 */
public class ResourceStreamRequestHandler implements IRequestHandler, ILoggableRequestHandler {
    /** Logger */
    private static final Logger log = LoggerFactory.getLogger(ResourceStreamRequestHandler.class);

    /**
     * Optional filename, used to set the content disposition header. Only meaningful when using
     * with web requests.
     */
    private String fileName;
    private ContentDisposition contentDisposition;

    /**
     * The duration fow which the resource will be cached by the browser.
     * <p>
     * By default is {@code null} and
     * {@link org.apache.wicket.settings.ResourceSettings#getDefaultCacheDuration()} is used.
     * </p>
     */
    private Duration cacheDuration;

    /** the resource stream for the response. */
    private final IResourceStream resourceStream;

    private ResourceStreamLogData logData;

    /**
     * Construct.
     * 
     * @param resourceStream
     *            the resource stream for the response
     */
    public ResourceStreamRequestHandler(IResourceStream resourceStream) {
        this(resourceStream, null);
    }

    /**
     * Construct.
     * 
     * @param resourceStream
     *            the resource stream for the response
     * @param fileName
     */
    public ResourceStreamRequestHandler(IResourceStream resourceStream, String fileName) {
        Args.notNull(resourceStream, "resourceStream");

        this.resourceStream = resourceStream;
        this.fileName = fileName;
    }

    @Override
    public void detach(IRequestCycle requestCycle) {
        if (logData == null)
            logData = getResourceStream() == null ? new ResourceStreamLogData(this)
                    : new ResourceStreamLogData(this, getResourceStream());
    }

    /** {@inheritDoc} */
    @Override
    public ResourceStreamLogData getLogData() {
        return logData;
    }

    /**
     * @return Optional filename, used to set the content disposition header. Only meaningful when
     *         using with web requests.
     */
    public final String getFileName() {
        return fileName;
    }

    /**
     * Gets the resource stream for the response.
     * 
     * @return the resource stream for the response
     */
    public final IResourceStream getResourceStream() {
        return resourceStream;
    }

    /**
     * Responds by sending the contents of the resource stream.
     * 
     * @see org.apache.wicket.request.IRequestHandler#respond(org.apache.wicket.request.IRequestCycle)
     */
    @Override
    public void respond(IRequestCycle requestCycle) {
        Attributes attributes = new Attributes(requestCycle.getRequest(), requestCycle.getResponse());

        ResourceStreamResource resource = new ResourceStreamResource(resourceStream);

        configure(resource);

        resource.respond(attributes);
    }

    /**
     * Configures the ResourceStreamResource used by this request handler
     *
     * @param resource
     *          the resource to configure
     */
    protected void configure(ResourceStreamResource resource) {
        resource.setFileName(fileName);
        if (contentDisposition != null) {
            resource.setContentDisposition(contentDisposition);
        } else {
            resource.setContentDisposition(
                    Strings.isEmpty(fileName) ? ContentDisposition.INLINE : ContentDisposition.ATTACHMENT);
        }

        final Duration cacheDuration = getCacheDuration();
        if (cacheDuration != null) {
            resource.setCacheDuration(cacheDuration);
        }
    }

    @Override
    public int hashCode() {
        final int prime = 31;
        int result = 1;
        result = prime * result + ((contentDisposition == null) ? 0 : contentDisposition.hashCode());
        result = prime * result + ((fileName == null) ? 0 : fileName.hashCode());
        result = prime * result + resourceStream.hashCode();
        return result;
    }

    @Override
    public boolean equals(Object obj) {
        if (this == obj)
            return true;
        if (obj == null)
            return false;
        if (getClass() != obj.getClass())
            return false;
        ResourceStreamRequestHandler other = (ResourceStreamRequestHandler) obj;
        if (contentDisposition != other.contentDisposition)
            return false;
        if (fileName == null) {
            if (other.fileName != null)
                return false;
        } else if (!fileName.equals(other.fileName))
            return false;
        if (!resourceStream.equals(other.resourceStream))
            return false;
        return true;
    }

    /**
     * @param fileName
     *            Optional filename, used to set the content disposition header. Only meaningful
     *            when using with web requests.
     * 
     * @return The this.
     */
    public final ResourceStreamRequestHandler setFileName(String fileName) {
        this.fileName = fileName;
        return this;
    }

    /**
     * @see java.lang.Object#toString()
     */
    @Override
    public String toString() {
        return "[ResourceStreamRequestTarget[resourceStream=" + resourceStream + ",fileName=" + fileName
                + ", contentDisposition=" + contentDisposition + "]";
    }

    /**
     * @return ContentDisposition
     */
    public final ContentDisposition getContentDisposition() {
        return contentDisposition;
    }

    /**
     * @param contentDisposition
     * @return this
     */
    public final ResourceStreamRequestHandler setContentDisposition(ContentDisposition contentDisposition) {
        this.contentDisposition = contentDisposition;
        return this;
    }

    /**
     * @return the duration for which the resource will be cached by the browser
     */
    public Duration getCacheDuration() {
        return cacheDuration;
    }

    /**
     * @param cacheDuration
     *            the duration for which the resource will be cached by the browser
     * @return this component
     */
    public ResourceStreamRequestHandler setCacheDuration(Duration cacheDuration) {
        this.cacheDuration = cacheDuration;
        return this;
    }

}