Line data    Source code

       1             : // Copyright (C) 2012 The Android Open Source Project
       2             : //
       3             : // Licensed under the Apache License, Version 2.0 (the "License");
       4             : // you may not use this file except in compliance with the License.
       5             : // You may obtain a copy of the License at
       6             : //
       7             : // http://www.apache.org/licenses/LICENSE-2.0
       8             : //
       9             : // Unless required by applicable law or agreed to in writing, software
      10             : // distributed under the License is distributed on an "AS IS" BASIS,
      11             : // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
      12             : // See the License for the specific language governing permissions and
      13             : // limitations under the License.
      14             : 
      15             : package com.google.gerrit.server.cache;
      16             : 
      17             : import com.google.common.base.Strings;
      18             : import com.google.common.cache.RemovalListener;
      19             : import com.google.common.cache.RemovalNotification;
      20             : import com.google.gerrit.extensions.registration.PluginName;
      21             : import com.google.gerrit.server.plugincontext.PluginSetContext;
      22             : import com.google.inject.Inject;
      23             : import com.google.inject.assistedinject.Assisted;
      24             : 
      25             : /**
      26             :  * This listener dispatches removal events to all other RemovalListeners attached via the DynamicSet
      27             :  * API.
      28             :  */
      29             : @SuppressWarnings("rawtypes")
      30             : public class ForwardingRemovalListener<K, V> implements RemovalListener<K, V> {
      31             :   public interface Factory {
      32             :     ForwardingRemovalListener create(String cacheName);
      33             :   }
      34             : 
      35             :   private final PluginSetContext<CacheRemovalListener> listeners;
      36             :   private final String cacheName;
      37         151 :   private String pluginName = PluginName.GERRIT;
      38             : 
      39             :   @Inject
      40             :   ForwardingRemovalListener(
      41         151 :       PluginSetContext<CacheRemovalListener> listeners, @Assisted String cacheName) {
      42         151 :     this.listeners = listeners;
      43         151 :     this.cacheName = cacheName;
      44         151 :   }
      45             : 
      46             :   @Inject(optional = true)
      47             :   void setPluginName(String name) {
      48         151 :     if (!Strings.isNullOrEmpty(name)) {
      49           0 :       this.pluginName = name;
      50             :     }
      51         151 :   }
      52             : 
      53             :   @Override
      54             :   @SuppressWarnings("unchecked")
      55             :   public void onRemoval(RemovalNotification<K, V> notification) {
      56         151 :     listeners.runEach(l -> l.onRemoval(pluginName, cacheName, notification));
      57         151 :   }
      58             : }