//
// A Structured Logger for Fluent
//
// Copyright (C) 2011 - 2013 OZAWA Tsuyoshi
//
// Licensed 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.fluentd.logger;
import java.lang.reflect.Constructor;
import java.lang.reflect.InvocationTargetException;
import java.util.Map;
import java.util.Properties;
import java.util.WeakHashMap;
import org.fluentd.logger.sender.ExponentialDelayReconnector;
import org.fluentd.logger.sender.RawSocketSender;
import org.fluentd.logger.sender.Reconnector;
import org.fluentd.logger.sender.Sender;
public class FluentLoggerFactory {
private final Map<String, FluentLogger> loggers;
public FluentLoggerFactory() {
loggers = new WeakHashMap<String, FluentLogger>();
}
public FluentLogger getLogger(String tagPrefix) {
return getLogger(tagPrefix, "localhost", 24224);
}
public FluentLogger getLogger(String tagPrefix, String host, int port) {
return getLogger(tagPrefix, host, port, 3 * 1000, 1 * 1024 * 1024, new ExponentialDelayReconnector());
}
public FluentLogger getLogger(String tagPrefix, String host, int port, int timeout, int bufferCapacity) {
return getLogger(tagPrefix, host, port, timeout, bufferCapacity, new ExponentialDelayReconnector());
}
public synchronized FluentLogger getLogger(String tagPrefix, String host, int port, int timeout, int bufferCapacity,
Reconnector reconnector) {
String key = String.format("%s_%s_%d_%d_%d", new Object[] { tagPrefix, host, port, timeout, bufferCapacity });
if (loggers.containsKey(key)) {
return loggers.get(key);
} else {
Sender sender = null;
Properties props = System.getProperties();
if (!props.containsKey(Config.FLUENT_SENDER_CLASS)) {
// create default sender object
sender = new RawSocketSender(host, port, timeout, bufferCapacity, reconnector);
} else {
String senderClassName = props.getProperty(Config.FLUENT_SENDER_CLASS);
try {
sender = createSenderInstance(senderClassName, new Object[] { host, port, timeout, bufferCapacity });
} catch (Exception e) {
throw new RuntimeException(e);
}
}
FluentLogger logger = new FluentLogger(tagPrefix, sender);
loggers.put(key, logger);
return logger;
}
}
@SuppressWarnings("unchecked")
private Sender createSenderInstance(final String className, final Object[] params) throws ClassNotFoundException,
SecurityException, NoSuchMethodException, IllegalArgumentException, InstantiationException,
IllegalAccessException, InvocationTargetException {
Class<Sender> cl = (Class<Sender>) FluentLogger.class.getClassLoader().loadClass(className);
Constructor<Sender> cons = cl.getDeclaredConstructor(new Class[] { String.class, int.class, int.class,
int.class });
return (Sender) cons.newInstance(params);
}
/**
* the method is for testing
*/
Map<String, FluentLogger> getLoggers() {
return loggers;
}
public synchronized void closeAll() {
for (FluentLogger logger : loggers.values()) {
logger.close();
}
loggers.clear();
}
public synchronized void flushAll() {
for (FluentLogger logger : loggers.values()) {
logger.flush();
}
}
}