com.google.sitebricks.slf4j.Slf4jInjectionTypeListener Maven / Gradle / Ivy

Show more of this group Show more artifacts with this name
Show all versions of sitebricks-slf4j Show documentation

Slf4j logging support for Guice applications (does not require sitebricks)

There is a newer version: 0.8.11

package com.google.sitebricks.slf4j;

import com.google.inject.ProvisionException;
import com.google.inject.TypeLiteral;
import com.google.inject.spi.InjectionListener;
import com.google.inject.spi.TypeEncounter;
import com.google.inject.spi.TypeListener;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import java.lang.reflect.Field;

/**
 * @author [email protected] (Dhanji R. Prasanna)
 */
class Slf4jInjectionTypeListener implements TypeListener {
  @Override
  public  void hear(final TypeLiteral type, TypeEncounter encounter) {
    final Field field = getLoggerField(type.getRawType());

    if (field != null) {
      encounter.register(new InjectionListener() {
        @Override
        public void afterInjection(I injectee) {
          try {
            boolean b = field.isAccessible();
            if (!b) field.setAccessible(true);
            field.set(injectee,
                LoggerFactory.getLogger(type.getRawType()));
            if (!b) field.setAccessible(false);
          } catch (IllegalAccessException e) {
            throw new ProvisionException(
                "Unable to inject SLF4J logger", e);
          }
        }
      });
    }
  }

  protected Field getLoggerField(Class clazz) {
    // search for Logger in current class and return it if found
    for (final Field field : clazz.getDeclaredFields()) {
      final Class typeOfField = field.getType();
      if (Logger.class.isAssignableFrom(typeOfField)) {

        return field;
      }
    }

    // search for Logger in superclass if not found in this class
    if (clazz.getSuperclass() != null) {
      return getLoggerField(clazz.getSuperclass());
    }

    // not in current class and not having superclass, return null
    return null;
  }
}