LogUtil.java

/**
 * 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 com.facebook.infrastructure.utils;

import java.io.*;
import java.text.*;
import java.util.*;
import org.apache.log4j.*;
import org.apache.log4j.spi.LoggerFactory;
import org.apache.log4j.xml.DOMConfigurator;

public class LogUtil
{
    private LogUtil() {}

    public static Logger getLogger(String name)
    {
        return Logger.getLogger(name);
    }
    
    public static String stackTrace(Throwable e)
    {
        StringWriter sw = new StringWriter();
        PrintWriter pw = new PrintWriter(sw);
        e.printStackTrace(pw);
        return sw.toString();
    }

    public static String getTimestamp()
    {
        Date date = new Date();
        DateFormat df = new SimpleDateFormat("MM/dd/yyyy HH:mm:ss");
        return df.format(date);
    }
    
    public static String throwableToString(Throwable e)
    {
        StringBuffer sbuf = new StringBuffer("");
        String trace = stackTrace(e);
        sbuf.append((new StringBuilder()).append("Exception was generated at : ").append(getTimestamp()).append(" on thread ").append(Thread.currentThread().getName()).toString());
        sbuf.append(System.getProperty("line.separator"));
        String message = e.getMessage();
        if(message != null)
            sbuf.append(message);
        sbuf.append(System.getProperty("line.separator"));
        sbuf.append(trace);
        return sbuf.toString();
    }

    public static String getLogMessage(String message)
    {
        StringBuffer sbuf = new StringBuffer((new StringBuilder()).append("Log started at : ").append(getTimestamp()).toString());
        sbuf.append(System.getProperty("line.separator"));
        sbuf.append(message);
        return sbuf.toString();
    }

    public static void setLogLevel(String logger, String level)
    {        
        Logger loggerObj = LogManager.getLogger(logger);
        if(null == loggerObj)
            return;
        level = level.toUpperCase();
        if(level.equals("DEBUG"))
            loggerObj.setLevel(Level.DEBUG);
        else
        if(level.equals("ERROR"))
            loggerObj.setLevel(Level.ERROR);
        else
        if(level.equals("FATAL"))
            loggerObj.setLevel(Level.FATAL);
        else
        if(level.equals("INFO"))
            loggerObj.setLevel(Level.INFO);
        else
        if(level.equals("OFF"))
            loggerObj.setLevel(Level.OFF);
        else
        if(level.equals("WARN"))
            loggerObj.setLevel(Level.WARN);
        else
            loggerObj.setLevel(Level.ALL);
    }
}