gerrit.fd Code Review - hc2vpp.git/blob

   1 /*
   2  * Copyright (c) 2016 Cisco and/or its affiliates.
   3  *
   4  * Licensed under the Apache License, Version 2.0 (the "License");
   5  * you may not use this file except in compliance with the License.
   6  * You may obtain a copy of the License at:
   7  *
   8  *     http://www.apache.org/licenses/LICENSE-2.0
   9  *
  10  * Unless required by applicable law or agreed to in writing, software
  11  * distributed under the License is distributed on an "AS IS" BASIS,
  12  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  13  * See the License for the specific language governing permissions and
  14  * limitations under the License.
  15  */
  16
  17 package io.fd.hc2vpp.common.integration;
  18
  19 import com.google.inject.Inject;
  20 import io.fd.hc2vpp.common.translate.util.VppStatusListener;
  21 import io.fd.honeycomb.binding.init.ProviderTrait;
  22 import io.fd.honeycomb.data.init.ShutdownHandler;
  23 import io.fd.vpp.jvpp.JVppRegistry;
  24 import io.fd.vpp.jvpp.JVppRegistryImpl;
  25 import java.io.IOException;
  26 import org.slf4j.Logger;
  27 import org.slf4j.LoggerFactory;
  28
  29 /**
  30  * Provides JVppRegistry. Must be a singleton due to shutdown hook usage. Registers shutdown hook to disconnect from
  31  * VPP.
  32  */
  33 public final class JVppRegistryProvider extends ProviderTrait<JVppRegistry> {
  34
  35     private static final Logger LOG = LoggerFactory.getLogger(JVppRegistryProvider.class);
  36
  37     @Inject
  38     private VppConfigAttributes config;
  39     @Inject
  40     private VppStatusListener vppStatus;
  41     @Inject
  42     private ShutdownHandler shutdownHandler;
  43
  44     @Override
  45     protected JVppRegistryImpl create() {
  46         try {
  47             final JVppRegistryImpl registry = new JVppRegistryImpl(config.jvppConnectionName);
  48
  49             // Closing JVpp connection with shutdown hook to erase the connection from VPP so HC will be able
  50             // to connect next time. If JVM is force closed, this will not be executed and VPP connection
  51             // with name from config will stay open and prevent next startup of HC to success
  52             shutdownHandler.register("jvpp-registry", () -> {
  53                 LOG.info("Disconnecting from VPP");
  54                 if (vppStatus.isDown()) {
  55                     LOG.info("VPP is down. JVppRegistry cleanup is not needed. Exiting");
  56                     return;
  57                 }
  58                 try {
  59                     registry.close();
  60                     LOG.info("Successfully disconnected from VPP as {}", config.jvppConnectionName);
  61                 } catch (Exception e) {
  62                     LOG.warn("Unable to properly close jvpp registry", e);
  63                 }
  64             });
  65             LOG.info("JVpp connection opened successfully as: {}", config.jvppConnectionName);
  66             return registry;
  67         } catch (IOException e) {
  68             throw new IllegalStateException("Unable to open VPP management connection", e);
  69         }
  70     }
  71 }