org.mule.model.streaming.CallbackOutputStream.java Source code

Introduction

Here is the source code for org.mule.model.streaming.CallbackOutputStream.java

Source

/*
 * $Id$
 * --------------------------------------------------------------------------------------
 * 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.model.streaming;

import java.io.IOException;
import java.io.OutputStream;

import org.apache.commons.logging.Log;
import org.apache.commons.logging.LogFactory;

public class CallbackOutputStream extends OutputStream {
    private static final Log logger = LogFactory.getLog(CallbackOutputStream.class);

    public static interface Callback {
        public void onClose() throws Exception;
    }

    private OutputStream delegate;
    private Callback callback;

    public CallbackOutputStream(OutputStream delegate, Callback callback) {
        this.delegate = delegate;
        this.callback = callback;
    }

    @Override
    public void write(int b) throws IOException {
        delegate.write(b);
    }

    @Override
    public void write(byte b[]) throws IOException {
        delegate.write(b);
    }

    @Override
    public void write(byte b[], int off, int len) throws IOException {
        delegate.write(b, off, len);
    }

    @Override
    public void close() throws IOException {
        try {
            delegate.close();
        } finally {
            closeCallback();
        }
    }

    private void closeCallback() {
        if (null != callback) {
            try {
                callback.onClose();
            } catch (Exception e) {
                logger.debug("Suppressing exception while releasing resources: " + e.getMessage());
            }
        }

    }
}