com.isomorphic.maven.util.LoggingCountingOutputStream.java Source code

Introduction

Here is the source code for com.isomorphic.maven.util.LoggingCountingOutputStream.java

Source

package com.isomorphic.maven.util;

/*
 * 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.
 */

import org.apache.commons.io.FileUtils;
import org.apache.commons.io.output.CountingOutputStream;
import org.apache.commons.lang3.StringUtils;

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

/**
 * @author Daniel Johansson
 * @since 2014-02-17 09:04
 */
public class LoggingCountingOutputStream extends CountingOutputStream {

    private final long expectedByteCount;
    private long snapshotStart = System.currentTimeMillis();
    private long snapshotBytes;
    private long bytesPerSecond;

    /**
     * Constructs a new CountingOutputStream.
     *
     * @param out the OutputStream to write to
     */
    public LoggingCountingOutputStream(final OutputStream out, final long expectedByteCount) {
        super(out);
        this.expectedByteCount = expectedByteCount;
    }

    @Override
    protected void afterWrite(final int bytesWritten) throws IOException {
        super.afterWrite(bytesWritten);

        final long byteCount = getByteCount();
        final double progress = ((double) byteCount / (double) expectedByteCount) * 100.0;

        snapshotBytes += bytesWritten;

        // Lets grab the number of bytes written over the last second and use that as our
        // bytesPerSecond gauge giving the user an indication of speed.
        if (System.currentTimeMillis() - snapshotStart >= 1000) {
            snapshotStart = System.currentTimeMillis();
            bytesPerSecond = snapshotBytes;
            snapshotBytes = 0;
        }

        // Using \r in this print out will cause the console to render this line of text on the
        // same line in order to avoid console spam.
        System.out.print("\r" + StringUtils.rightPad(FileUtils.byteCountToDisplaySize(byteCount) + " / "
                + FileUtils.byteCountToDisplaySize(expectedByteCount) + " ("
                + FileUtils.byteCountToDisplaySize(bytesPerSecond) + "/second) "
                + String.format("%.1f%%", progress), 60, " "));

        if (progress >= 100.0) {
            System.out.println("Done!");
        }
    }
}