org.jclouds.logging.internal.Wire.java Source code

Introduction

Here is the source code for org.jclouds.logging.internal.Wire.java

Source

/**
 *
 * Copyright (C) 2009 Cloud Conscious, LLC. <info@cloudconscious.com>
 *
 * ====================================================================
 * 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.jclouds.logging.internal;

import static com.google.common.base.Preconditions.checkNotNull;

import java.io.ByteArrayInputStream;
import java.io.File;
import java.io.FileInputStream;
import java.io.FileNotFoundException;
import java.io.IOException;
import java.io.InputStream;
import java.io.PipedInputStream;
import java.io.PipedOutputStream;
import java.util.concurrent.ExecutorService;

import javax.annotation.Resource;
import javax.inject.Inject;

import org.apache.commons.io.IOUtils;
import org.apache.commons.io.input.TeeInputStream;
import org.jclouds.concurrent.SingleThreaded;
import org.jclouds.logging.Logger;

/**
 * Logs data to the wire LOG.
 * 
 * @author Adrian Cole
 * @see org.apache.HttpWire.impl.conn.Wire
 */
public abstract class Wire {

    @Resource
    protected Logger logger = Logger.NULL;
    protected final ExecutorService exec;

    @Inject
    public Wire(ExecutorService exec) {
        this.exec = checkNotNull(exec, "executor");
    }

    protected abstract Logger getWireLog();

    private void wire(String header, InputStream instream) {
        StringBuilder buffer = new StringBuilder();
        int ch;
        try {
            while ((ch = instream.read()) != -1) {
                if (ch == 13) {
                    buffer.append("[\\r]");
                } else if (ch == 10) {
                    buffer.append("[\\n]\"");
                    buffer.insert(0, "\"");
                    buffer.insert(0, header);
                    getWireLog().debug(buffer.toString());
                    buffer.setLength(0);
                } else if ((ch < 32) || (ch > 127)) {
                    buffer.append("[0x");
                    buffer.append(Integer.toHexString(ch));
                    buffer.append("]");
                } else {
                    buffer.append((char) ch);
                }
            }
            if (buffer.length() > 0) {
                buffer.append('\"');
                buffer.insert(0, '\"');
                buffer.insert(0, header);
                getWireLog().debug(buffer.toString());
            }
        } catch (IOException e) {
            logger.error(e, "Error tapping line");
        }
    }

    public boolean enabled() {
        return getWireLog().isDebugEnabled();
    }

    public InputStream copy(final String header, InputStream instream) {
        try {
            byte[] data = IOUtils.toByteArray(instream);
            wire(header, new ByteArrayInputStream(data));
            return new ByteArrayInputStream(data);
        } catch (IOException e) {
            throw new RuntimeException("Error tapping line", e);
        } finally {
            IOUtils.closeQuietly(instream);
        }
    }

    public InputStream tapAsynch(final String header, InputStream instream) {
        PipedOutputStream out = new PipedOutputStream();
        InputStream toReturn = new TeeInputStream(instream, out, true);
        final InputStream line;
        try {
            line = new PipedInputStream(out);
            exec.submit(new Runnable() {
                public void run() {
                    try {
                        wire(header, line);
                    } finally {
                        IOUtils.closeQuietly(line);
                    }
                }
            });
        } catch (IOException e) {
            logger.error(e, "Error tapping line");
        }
        return toReturn;
    }

    public InputStream input(InputStream instream) {
        return copy("<< ", checkNotNull(instream, "input"));
    }

    @SuppressWarnings("unchecked")
    public <T> T output(T data) {
        checkNotNull(data, "data");
        if (data instanceof InputStream) {
            if (exec.getClass().isAnnotationPresent(SingleThreaded.class))
                return (T) copy(">> ", (InputStream) data);
            else
                return (T) tapAsynch(">> ", (InputStream) data);
        } else if (data instanceof byte[]) {
            output((byte[]) data);
            return data;
        } else if (data instanceof String) {
            output((String) data);
            return data;
        } else if (data instanceof File) {
            output(((File) data));
            return data;
        } else {
            throw new UnsupportedOperationException("Content not supported " + data.getClass());
        }
    }

    private void output(final File out) {
        checkNotNull(out, "output");
        exec.submit(new Runnable() {
            public void run() {
                InputStream in = null;
                try {
                    in = new FileInputStream(out);
                    wire(">> ", in);
                } catch (FileNotFoundException e) {
                    logger.error(e, "Error tapping file: %s", out);
                } finally {
                    IOUtils.closeQuietly(in);
                }
            }
        });
    }

    private void output(byte[] b) {
        wire(">> ", new ByteArrayInputStream(checkNotNull(b, "output")));
    }

    private void output(final String s) {
        output(checkNotNull(s, "output").getBytes());
    }

}