com.testvagrant.monitor.clients.ServiceClient Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of optimus-monitor Show documentation
The Java Client provides access to Optimus cloud for all java based projects.
The newest version!
package com.testvagrant.monitor.clients;

import com.jayway.awaitility.Awaitility;
import com.jayway.awaitility.Duration;
import io.restassured.response.Response;

import java.io.File;
import java.io.IOException;
import java.util.*;

import static io.restassured.RestAssured.given;

public class ServiceClient {
    private static Process serviceProcess;
    private static String port="";
    private static List commands = new ArrayList();

    public static void main(String[] args) throws IOException {
        ServiceClient serviceClient = new ServiceClient();
        List commands = serviceClient.initCommands(args);
        serviceClient.getPort(commands);
        serviceClient.startService();
    }
    public boolean isServiceRunning() {
        try {
            String uri = String.format("http://localhost:%s/v1",port);
            Response serviceResponse = given()
                    .header("Content-Type", "application/json")
                    .get(uri + "/ping");
            return serviceResponse.asString().equals("PONG");
        } catch (Exception e) {
            return false;
        }
    }

    public boolean isServiceDown() {
        return !isServiceRunning();
    }

    public void startService() throws IOException {
        if(isServiceDown()) {
            ProcessBuilder processBuilder = new ProcessBuilder(new String[0]);
            serviceProcess = processBuilder.command(commands).start();
            Awaitility.await().atMost(Duration.FIVE_MINUTES).until(this::isServiceRunning);
        }
    }

    private List initCommands(String[] args) {
        commands.add("java");
        commands.add("-jar");
        commands.add(System.getProperty("user.dir") + "/libs/"+getLatestServicesName());
        Collections.addAll(commands, args);
        return commands;
    }

    private String getLatestServicesName() {
        File file = new File(System.getProperty("user.dir") + "/libs");
        if(file.isDirectory()) {
            File[] files = file.listFiles();
            Optional first = Arrays.asList(files).stream().filter(serviceFile -> serviceFile.getName().contains("optimus-services")).findFirst();
            if(first.isPresent()) {
                return first.get().getName();
            }
        }
        return "optimus-services-1.0";
    }

    private String getPort(List commands) {
        System.out.println("Service commands "+commands);
        Optional first = commands.stream().filter(command -> command.contains("--port")).findFirst();
        if(first.isPresent()) {
            String s = first.get();
            port = s.split("=")[1];
            port = port.equals("")? "8080" : port;
        }
        return port;
    }

    public void stopService() {
        if(serviceProcess!=null)
            serviceProcess.destroy();
    }

}