View Javadoc
1   /*
2    * Licensed to the Apache Software Foundation (ASF) under one
3    * or more contributor license agreements.  See the NOTICE file
4    * distributed with this work for additional information
5    * regarding copyright ownership.  The ASF licenses this file
6    * to you under the Apache License, Version 2.0 (the
7    * "License"); you may not use this file except in compliance
8    * with the License.  You may obtain a copy of the License at
9    *
10   *     http://www.apache.org/licenses/LICENSE-2.0
11   *
12   * Unless required by applicable law or agreed to in writing,
13   * software distributed under the License is distributed on an
14   * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
15   * KIND, either express or implied.  See the License for the
16   * specific language governing permissions and limitations
17   * under the License.
18   */
19  package org.apache.shiro.guice;
20  
21  import com.google.inject.TypeLiteral;
22  import com.google.inject.matcher.Matcher;
23  import com.google.inject.spi.InjectionListener;
24  import com.google.inject.spi.TypeEncounter;
25  import com.google.inject.spi.TypeListener;
26  import org.apache.shiro.util.Destroyable;
27  import org.apache.shiro.util.Initializable;
28  
29  
30  class LifecycleTypeListener implements TypeListener {
31      public static final Matcher<TypeLiteral> MATCHER = InitializableInjectionListener.MATCHER.or(DestroyableInjectionListener.MATCHER);
32      private DestroyableInjectionListener.DestroyableRegistry registry;
33  
34      public LifecycleTypeListener(DestroyableInjectionListener.DestroyableRegistry registry) {
35          this.registry = registry;
36      }
37  
38      public <I> void hear(TypeLiteral<I> type, TypeEncounter<I> encounter) {
39          if (InitializableInjectionListener.MATCHER.matches(type)) {
40              encounter.register(this.<I>castListener(new InitializableInjectionListener<Initializable>()));
41          }
42          if (DestroyableInjectionListener.MATCHER.matches(type)) {
43              encounter.register(this.<I>castListener(new DestroyableInjectionListener<Destroyable>(registry)));
44          }
45      }
46  
47      @SuppressWarnings({"unchecked"})
48      private <I> InjectionListener<? super I> castListener(InjectionListener<?> listener) {
49          return (InjectionListener<? super I>) listener;
50      }
51  }