Add Mycilla code to odlguice

[odlguice.git] / inject / inject-guice-extensions / injection / src / main / java / org / opendaylight / odlguice / inject / guice / extensions / injection / MemberInjectorTypeListener.java

diff --git a/inject/inject-guice-extensions/injection/src/main/java/org/opendaylight/odlguice/inject/guice/extensions/injection/MemberInjectorTypeListener.java b/inject/inject-guice-extensions/injection/src/main/java/org/opendaylight/odlguice/inject/guice/extensions/injection/MemberInjectorTypeListener.java
new file mode 100755 (executable)
index 0000000..0d54192
--- /dev/null
+++ b/inject/inject-guice-extensions/injection/src/main/java/org/opendaylight/odlguice/inject/guice/extensions/injection/MemberInjectorTypeListener.java
@@ -0,0 +1,101 @@
+/*
+ * Copyright (C) 2010 Mycila (mathieu.carbou@gmail.com)
+ *
+ * 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.opendaylight.odlguice.inject.guice.extensions.injection;
+
+import com.google.common.collect.Lists;
+import com.google.inject.Injector;
+import com.google.inject.Key;
+import com.google.inject.MembersInjector;
+import com.google.inject.Provider;
+import com.google.inject.TypeLiteral;
+import com.google.inject.spi.TypeEncounter;
+import com.google.inject.spi.TypeListener;
+import java.lang.annotation.Annotation;
+import java.lang.reflect.Field;
+import java.security.AccessController;
+import java.security.PrivilegedAction;
+import java.util.List;
+
+/**
+ * This code originated in https://github.com/mycila/guice and was forked into
+ * OpenDaylight.
+ * @author Mathieu Carbou (mathieu.carbou@gmail.com)
+ */
+public class MemberInjectorTypeListener<A extends Annotation> implements TypeListener {
+
+    private final Class<A> annotationType;
+    private final Class<? extends KeyProvider<A>> providerClass;
+
+    public MemberInjectorTypeListener(Class<A> annotationType, Class<? extends KeyProvider<A>> providerClass) {
+        this.annotationType = annotationType;
+        this.providerClass = providerClass;
+    }
+
+    @SuppressWarnings("checkstyle:illegalCatch")
+    @Override
+    public <I> void hear(final TypeLiteral<I> injectableType, TypeEncounter<I> encounter) {
+        final Provider<? extends KeyProvider<A>> provider = encounter.getProvider(providerClass);
+        final Provider<Injector> injectorProvider = encounter.getProvider(Injector.class);
+        final List<Field> fields = Lists
+                .newLinkedList(Reflect.findAllAnnotatedFields(injectableType.getRawType(), annotationType));
+        final List<MethodInvoker> methods = Lists
+                .newLinkedList(Reflect.findAllAnnotatedInvokables(injectableType.getRawType(), annotationType));
+        if (!fields.isEmpty() || !methods.isEmpty()) {
+            encounter.register(new MembersInjector<I>() {
+                @SuppressWarnings("unchecked")
+                @Override
+                public void injectMembers(I injectee) {
+                    KeyProvider<A> keyProvider = provider.get();
+                    // inject fields
+                    for (Field field : fields) {
+                        Object value = injectorProvider.get()
+                                .getProvider(
+                                        keyProvider.getKey(injectableType, field, field.getAnnotation(annotationType)))
+                                .get();
+                        if (!field.isAccessible()) {
+//                            field.setAccessible(true);
+                            AccessController.doPrivileged((PrivilegedAction) () -> {
+                                field.setAccessible(true);
+                                return null;
+                            });
+                        }
+                        try {
+                            field.set(injectee, value);
+                        } catch (IllegalAccessException e) {
+                            throw new IllegalStateException(
+                                    "Failed to inject field " + field + ". Reason: " + e.getMessage(), e);
+                        }
+                    }
+                    // inject methods
+                    for (MethodInvoker invokable : methods) {
+                        List<Key<?>> parameterKeys = keyProvider.getParameterKeys(injectableType, invokable.method,
+                                invokable.getAnnotation(annotationType));
+                        Object[] parameters = new Object[parameterKeys.size()];
+                        for (int i = 0; i < parameters.length; i++) {
+                            parameters[i] = injectorProvider.get().getProvider(parameterKeys.get(i)).get();
+                        }
+                        try {
+                            invokable.invoke(injectee, parameters);
+                        } catch (Exception ex) {
+                            throw MycilaGuiceException.toRuntime(ex);
+                        }
+                    }
+                }
+            });
+        }
+    }
+
+}