opendaylight/config/netty-event-executor-config/src/main/java/org/opendaylight/controller/config/yang/netty/eventexecutor/AutoCloseableEventExecutor.java

   1 /*
   2  * Copyright (c) 2013 Cisco Systems, Inc. and others.  All rights reserved.
   3  *
   4  * This program and the accompanying materials are made available under the
   5  * terms of the Eclipse Public License v1.0 which accompanies this distribution,
   6  * and is available at http://www.eclipse.org/legal/epl-v10.html
   7  */
   8 package org.opendaylight.controller.config.yang.netty.eventexecutor;
   9
  10 import com.google.common.reflect.AbstractInvocationHandler;
  11 import com.google.common.reflect.Reflection;
  12 import io.netty.util.concurrent.EventExecutor;
  13 import io.netty.util.concurrent.GlobalEventExecutor;
  14 import io.netty.util.concurrent.ImmediateEventExecutor;
  15 import java.lang.reflect.Method;
  16 import java.util.concurrent.TimeUnit;
  17
  18 public interface AutoCloseableEventExecutor extends EventExecutor, AutoCloseable {
  19
  20
  21     public static class CloseableEventExecutorMixin implements AutoCloseable {
  22         public static final int DEFAULT_SHUTDOWN_SECONDS = 1;
  23         private final EventExecutor eventExecutor;
  24
  25         public CloseableEventExecutorMixin(EventExecutor eventExecutor) {
  26             this.eventExecutor = eventExecutor;
  27         }
  28
  29         @Override
  30         public void close() {
  31             eventExecutor.shutdownGracefully(0, DEFAULT_SHUTDOWN_SECONDS, TimeUnit.SECONDS);
  32         }
  33
  34
  35         private static AutoCloseableEventExecutor createCloseableProxy(final EventExecutor eventExecutor) {
  36             final CloseableEventExecutorMixin closeableGlobalEventExecutorMixin =
  37                     new CloseableEventExecutorMixin(eventExecutor);
  38             return Reflection.newProxy(AutoCloseableEventExecutor.class, new AbstractInvocationHandler() {
  39                 @Override
  40                 protected Object handleInvocation(Object proxy, Method method, Object[] args) throws Throwable {
  41                     if (method.getName().equals("close")) {
  42                         closeableGlobalEventExecutorMixin.close();
  43                         return null;
  44                     } else {
  45                         return method.invoke(eventExecutor, args);
  46                     }
  47                 }
  48             });
  49         }
  50
  51         public static AutoCloseableEventExecutor globalEventExecutor() {
  52             return createCloseableProxy(GlobalEventExecutor.INSTANCE);
  53         }
  54
  55         public static AutoCloseableEventExecutor immediateEventExecutor() {
  56             return createCloseableProxy(ImmediateEventExecutor.INSTANCE);
  57         }
  58     }
  59 }