org.mule.module.pgp.TransformContinuouslyPolicy Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of mule-module-pgp Show documentation

A Mule implementation of PGP security using cryptix libraries.

There is a newer version: 3.9.0

/*
 * Copyright (c) MuleSoft, Inc.  All rights reserved.  http://www.mulesoft.com
 * The software in this package is published under the terms of the CPAL v1.0
 * license, a copy of which has been included with this distribution in the
 * LICENSE.txt file.
 */
package org.mule.module.pgp;

import java.io.PipedOutputStream;

/**
 * A {@link TransformPolicy} that copies the transformed bytes continuously into the {@link PipedOutputStream}
 * without taking into account about how many bytes the object has requested.
 */
public class TransformContinuouslyPolicy extends AbstractTransformPolicy
{

    public static final long DEFAULT_CHUNK_SIZE = 1 << 24;
    
    private long chunkSize;

    public TransformContinuouslyPolicy()
    {
        this(DEFAULT_CHUNK_SIZE);
    }

    public TransformContinuouslyPolicy(long chunkSize)
    {
        this.chunkSize = chunkSize;
    }

    /**
     * {@inheritDoc}
     */
    @Override
    public void readRequest(long length)
    {
        /**
         * Avoid calling super so that we don't add more bytes. 
         * The ContinuousWork will add the requested bytes as necessary
         * only start the copying thread
         */
        startCopyingThread();
    }
    
    /**
     * {@inheritDoc}
     */
    @Override
    protected Thread getCopyingThread()
    {
        return new ContinuousWork();
    }

    private class ContinuousWork extends TransformerWork
    {
        @Override
        protected void execute() throws Exception
        {
            getTransformer().initialize(getInputStream().getOut());
            
            boolean finishWriting = false;
            while (!finishWriting)
            {
                getBytesRequested().addAndGet(chunkSize);
                finishWriting = getTransformer().write(getInputStream().getOut(), getBytesRequested());
            }            
        }
    }
}