package com.firefly.utils.log;
import java.io.File;
import java.io.IOException;
import java.io.InputStream;
import java.net.URISyntaxException;
import java.util.HashMap;
import java.util.Map;
import java.util.Map.Entry;
import java.util.Properties;
import com.firefly.utils.StringUtils;
import com.firefly.utils.log.file.FileLog;
import com.firefly.utils.log.file.FileLogTask;
import com.firefly.utils.time.SafeSimpleDateFormat;
public class LogFactory {
private Map<String, Log> logMap = new HashMap<String, Log>();
private Map<String, Integer> levelMap = new HashMap<String, Integer>();
public static final SafeSimpleDateFormat dayDateFormat = new SafeSimpleDateFormat(
"yyyy-MM-dd");
private LogTask logTask = new FileLogTask();
private static class Holder {
private static LogFactory instance = new LogFactory();
}
public static LogFactory getInstance() {
return Holder.instance;
}
private LogFactory() {
levelMap.put("TRACE", Log.TRACE);
levelMap.put("DEBUG", Log.DEBUG);
levelMap.put("INFO", Log.INFO);
levelMap.put("WARN", Log.WARN);
levelMap.put("ERROR", Log.ERROR);
defaultLog();
File configFile = null;
try {
configFile = new File(LogFactory.class.getClassLoader()
.getResource("firefly-log.properties").toURI());
} catch (URISyntaxException e1) {
e1.printStackTrace();
}
if (configFile != null && configFile.exists()) {
loadProperties();
}
logTask.start();
}
private void defaultLog() {
String name = "firefly-system";
FileLog fileLog = new FileLog();
fileLog.setName(name);
fileLog.setLevel(2);
fileLog.setFileOutput(false);
fileLog.setConsoleOutput(true);
System.out.println(name + "|console");
logMap.put(name, fileLog);
}
private Properties loadProperties0() throws IOException{
Properties properties = new Properties();
InputStream input = null;
try {
input = LogFactory.class.getClassLoader().getResourceAsStream("firefly-log.properties");
properties.load(input);
} finally {
if(input != null)
input.close();
}
return properties;
}
@SuppressWarnings("resource")
private void loadProperties() {
Properties properties = null;
try {
properties = loadProperties0();
} catch (IOException e) {
e.printStackTrace();
}
for (Entry<Object, Object> entry : properties.entrySet()) {
String name = (String) entry.getKey();
String value = (String) entry.getValue();
System.out.println(name + "|" + value);
String[] strs = StringUtils.split(value, ',');
if (strs.length < 2)
throw new LogException("config format error");
int level = levelMap.get(strs[0]);
String path = strs[1];
FileLog fileLog = new FileLog();
fileLog.setName(name);
fileLog.setLevel(level);
if ("console".equalsIgnoreCase(path)) {
fileLog.setFileOutput(false);
fileLog.setConsoleOutput(true);
} else {
File file = new File(path);
if (!file.exists()) {
boolean mkdirRet = file.mkdirs();
if (!mkdirRet) {
throw new LogException("create dir " + path + " failure");
}
}
if (!file.isDirectory()) {
throw new LogException(path + " is not directory");
}
fileLog.setPath(path);
fileLog.setFileOutput(true);
if (strs.length > 2)
fileLog.setConsoleOutput("console".equalsIgnoreCase(strs[2]));
}
logMap.put(name, fileLog);
}
}
public void flush() {
for (Entry<String, Log> entry : logMap.entrySet()) {
Log log = entry.getValue();
if (log instanceof FileLog) {
((FileLog) log).flush();
// System.out.println(">>> flush all " + fileLog.getName());
}
}
}
public Log getLog(String name) {
return logMap.get(name);
}
public LogTask getLogTask() {
return logTask;
}
public void shutdown() {
logTask.shutdown();
for (Entry<String, Log> entry : logMap.entrySet()) {
Log log = entry.getValue();
if (log instanceof FileLog) {
try {
((FileLog) log).close();
} catch (IOException e) {
e.printStackTrace();
}
}
}
}
public void start() {
logTask.start();
}
}