org.mule.modules.FtpConnectionClosingStream.java Source code

Introduction

Here is the source code for org.mule.modules.FtpConnectionClosingStream.java

Source

package org.mule.modules;

import org.apache.commons.net.ftp.FTPClient;

import java.io.IOException;
import java.io.InputStream;

/**
 * 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.md file.
 *
 * Modifications copyright (c) 2017 VIAA vzw
 */
public class FtpConnectionClosingStream extends InputStream {

    private final FTPClient client;

    private final InputStream stream;

    public FtpConnectionClosingStream(FTPClient client, InputStream stream) {
        this.client = client;
        this.stream = stream;
    }

    @Override
    public int read() throws IOException {
        int result = this.stream.read();

        if (result == -1) {
            FtpUtils.disconnect(client);
        }
        return result;
    }

    @Override
    public int read(byte[] bytes) throws IOException {
        int result = stream.read(bytes);

        if (result == -1) {
            FtpUtils.disconnect(client);
        }
        return result;
    }

    @Override
    public int read(byte[] bytes, int i, int i2) throws IOException {
        int result = stream.read(bytes, i, i2);

        if (result == -1) {
            FtpUtils.disconnect(client);
        }
        return result;
    }

    @Override
    public long skip(long l) throws IOException {
        return stream.skip(l);
    }

    @Override
    public int available() throws IOException {
        return stream.available();
    }

    @Override
    public void close() throws IOException {
        FtpUtils.disconnect(client);
        stream.close();
    }

    @Override
    public synchronized void mark(int i) {
        stream.mark(i);
    }

    @Override
    public synchronized void reset() throws IOException {
        stream.reset();
    }

    @Override
    public boolean markSupported() {
        return stream.markSupported();
    }

    public InputStream getStream() {
        return stream;
    }
}