org.openbaton.nfvo.vnfm_reg.impl.register.RestRegister.java Source code

Introduction

Here is the source code for org.openbaton.nfvo.vnfm_reg.impl.register.RestRegister.java

Source

/*
 * Copyright (c) 2015 Fraunhofer FOKUS
 *
 * 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.openbaton.nfvo.vnfm_reg.impl.register;

import org.openbaton.catalogue.nfvo.EndpointType;
import org.openbaton.catalogue.nfvo.VnfmManagerEndpoint;
import org.openbaton.exceptions.AlreadyExistingException;
import org.openbaton.nfvo.vnfm_reg.VnfmRegister;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.http.HttpStatus;
import org.springframework.http.MediaType;
import org.springframework.scheduling.annotation.Scheduled;
import org.springframework.web.bind.annotation.*;

import java.io.IOException;
import java.net.InetSocketAddress;
import java.net.MalformedURLException;
import java.net.Socket;
import java.net.URL;

/**
 * Created by lto on 27/05/15.
 */
@RestController
@RequestMapping("/admin/v1")
public class RestRegister extends VnfmRegister {

    private Logger log = LoggerFactory.getLogger(this.getClass());

    private static boolean pingHost(String host, int port, int timeout) {
        try (Socket socket = new Socket()) {
            socket.connect(new InetSocketAddress(host, port), timeout);
            return true;
        } catch (IOException ignored) {
            return false; // Either timeout or unreachable or failed DNS lookup.
        }
    }

    @RequestMapping(value = "/vnfm-register", method = RequestMethod.POST, consumes = MediaType.APPLICATION_JSON_VALUE, produces = MediaType.APPLICATION_JSON_VALUE)
    @ResponseStatus(HttpStatus.CREATED)
    public void receiveRegister(@RequestBody VnfmManagerEndpoint endpoint) {
        addManagerEndpoint(gson.toJson(endpoint));
    }

    public void addManagerEndpoint(String endpoint) {
        log.debug("Received: " + endpoint);
        try {
            this.register(gson.fromJson(endpoint, VnfmManagerEndpoint.class));
        } catch (AlreadyExistingException e) {
            log.warn(e.getLocalizedMessage());
        }
    }

    @RequestMapping(value = "/vnfm-unregister", method = RequestMethod.POST, consumes = MediaType.APPLICATION_JSON_VALUE, produces = MediaType.APPLICATION_JSON_VALUE)
    @ResponseStatus(HttpStatus.OK)
    public void receiveUnregister(@RequestBody VnfmManagerEndpoint endpoint) {
        removeManagerEndpoint(gson.toJson(endpoint));
    }

    public void removeManagerEndpoint(String endpoint) {
        log.debug("Unregistering endpoint: " + endpoint);
        this.unregister(gson.fromJson(endpoint, VnfmManagerEndpoint.class));
    }

    @Scheduled(initialDelay = 15000, fixedDelay = 20000)
    public void checkHeartBeat() throws MalformedURLException {
        for (VnfmManagerEndpoint endpoint : vnfmEndpointRepository.findAll()) {
            if (endpoint.getEndpointType().ordinal() == EndpointType.REST.ordinal()) {
                if (endpoint.isEnabled()) {
                    try {
                        URL url = new URL(endpoint.getEndpoint());
                        if (!pingHost(url.getHost(), url.getPort(), 2)) {
                            if (endpoint.isActive()) {
                                log.info("Set endpoint " + endpoint.getType() + " to unactive");
                                endpoint.setActive(false);
                                vnfmEndpointRepository.save(endpoint);
                            }
                        } else {
                            if (!endpoint.isActive()) {
                                log.info("Set endpoint " + endpoint.getType() + " to active");
                                endpoint.setActive(true);
                                vnfmEndpointRepository.save(endpoint);
                            }
                        }
                    } catch (MalformedURLException ignored) {
                        if (endpoint.isActive()) {
                            log.warn("Not able to check endpoint: " + endpoint.getEndpoint());
                            endpoint.setActive(false);
                            vnfmEndpointRepository.save(endpoint);
                        }
                    }
                }
            }
        }
    }
}