br.unb.cic.bionimbuz.controller.elasticitycontroller.ElasticityController.java Source code

Introduction

Here is the source code for br.unb.cic.bionimbuz.controller.elasticitycontroller.ElasticityController.java

Source

/*
 * To change this license header, choose License Headers in Project Properties.
 * To change this template file, choose Tools | Templates
 * and open the template in the editor.
 */
package br.unb.cic.bionimbuz.controller.elasticitycontroller;

import java.io.IOException;
import java.util.concurrent.Executors;
import java.util.concurrent.ScheduledExecutorService;
import java.util.concurrent.TimeUnit;
import java.util.logging.Level;

import org.apache.commons.lang3.concurrent.BasicThreadFactory;
import org.apache.zookeeper.WatchedEvent;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import com.google.common.base.Preconditions;
import com.google.inject.Inject;

import br.unb.cic.bionimbuz.controller.Controller;
import br.unb.cic.bionimbuz.controller.usercontroller.UserController;
import br.unb.cic.bionimbuz.model.Log;
import br.unb.cic.bionimbuz.model.LogSeverity;
import br.unb.cic.bionimbuz.model.User;
import br.unb.cic.bionimbuz.model.Workflow;
import br.unb.cic.bionimbuz.persistence.dao.WorkflowLoggerDao;
import br.unb.cic.bionimbuz.services.messaging.CloudMessageService;
import br.unb.cic.bionimbuz.services.monitor.MonitoringService;

/**
 *
 * @author zoonimbus
 */
public class ElasticityController implements Controller, Runnable {

    private static final Logger LOGGER = LoggerFactory.getLogger(UserController.class);

    private final ScheduledExecutorService threadExecutor = Executors.newScheduledThreadPool(1,
            new BasicThreadFactory.Builder().namingPattern("ElasticityController-%d").build());
    protected CloudMessageService cms;
    private final WorkflowLoggerDao loggerDao;

    /**
     * Starts ElasticityController execution
     *
     * @param cms
     * @param rs
     */
    @Inject
    public ElasticityController(CloudMessageService cms) {
        Preconditions.checkNotNull(cms);
        this.cms = cms;
        LOGGER.info("ElasticityController started");
        this.loggerDao = new WorkflowLoggerDao();
    }

    @Override
    public void start() {
        this.threadExecutor.scheduleAtFixedRate(this, 0, 1, TimeUnit.MINUTES);
    }

    @Override
    public void shutdown() {
        throw new UnsupportedOperationException("Not supported yet."); // To change body of generated methods, choose Tools | Templates.
    }

    @Override
    public void getStatus() {
        throw new UnsupportedOperationException("Not supported yet."); // To change body of generated methods, choose Tools | Templates.
    }

    @Override
    public void verifyPlugins() {
        throw new UnsupportedOperationException("Not supported yet."); // To change body of generated methods, choose Tools | Templates.
    }

    @Override
    public void event(WatchedEvent eventType) {
        throw new UnsupportedOperationException("Not supported yet."); // To change body of generated methods, choose Tools | Templates.
    }

    @Override
    public void run() {
        LOGGER.info("Checking Elasticity Controller:");

    }

    public String createInstance(String provider, String type, String nameInstance) throws InterruptedException {
        final AmazonAPI amazonapi = new AmazonAPI();
        final GoogleAPI googleapi = new GoogleAPI();
        String IP = null;
        switch (provider) {
        case "Amazon": {
            try {
                // amazonapi.createinstance(type, nameInstance);
                amazonapi.createinstance("t2.micro", nameInstance);
            } catch (final IOException ex) {
                java.util.logging.Logger.getLogger(ElasticityController.class.getName()).log(Level.SEVERE, null,
                        ex);
            }
            System.out.println("Amazon IP:" + amazonapi.getIpInstance());

            while ((IP = amazonapi.getIpInstance()) == null) {
                Thread.sleep(1000);
            }
            //TODO: alterar pra receber o id do usuario e o id do workflow, pois assim fica sem referencia
            //loggerDao.log(new Log("New instance has been created:" + IP, 0, "test", LogSeverity.INFO));
            break;
        }
        case "Google": {
            try {
                // googleapi.createinstance(type, nameInstance);
                googleapi.createinstance("n1-standard-1", nameInstance);
            } catch (final IOException ex) {
                java.util.logging.Logger.getLogger(ElasticityController.class.getName()).log(Level.SEVERE, null,
                        ex);
            }
            System.out.println("Google IP:" + googleapi.getIpInstance());
            while ((IP = googleapi.getIpInstance()) == null) {
                Thread.sleep(1000);
            }
            //TODO: alterar pra receber o id do usuario e o id do workflow, pois assim fica sem referencia
            // loggerDao.log(new Log("New instance has been created:" + IP, 0, "test", LogSeverity.INFO));
            break;
        }
        default:
            break;
        }

        for (final User u : MonitoringService.getZkUsers()) {
            for (final Workflow work : u.getWorkflows()) {
                this.loggerDao.log(
                        new Log("Mquina criada com IP " + IP, work.getUserId(), work.getId(), LogSeverity.WARN));
            }
        }

        return IP;
    }
}