ch.simas.monitor.HostController Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of simon Show documentation
SImple MONitoring
The newest version!
package ch.simas.monitor;

import ch.simas.monitor.xml.Hosts;
import java.io.File;
import java.io.IOException;
import java.net.HttpURLConnection;
import java.net.URL;
import javax.xml.bind.JAXBContext;
import javax.xml.bind.JAXBException;
import javax.xml.bind.Unmarshaller;
import org.jboss.logging.Logger;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.scheduling.annotation.Scheduled;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RestController;

@RestController
public class HostController {

    private final static Logger LOGGER = Logger.getLogger(HostController.class);
    
    private String config;

    public HostController(@Value("${simon.config.hosts}") String config) {
        this.config = config;
    }

    @RequestMapping("/check")
    @Scheduled(fixedRate = 5 * 60 * 1000)
    public Hosts check() {
        LOGGER.debug("check");
        
        try {
            Hosts hosts = loadConfiguration(config);

            for (Hosts.Group group : hosts.getGroup()) {
                for (Hosts.Group.Host host : group.getHost()) {
                    long start = System.currentTimeMillis();
                    URL obj = new URL(host.getUrl());
                    try {
                        HttpURLConnection con = (HttpURLConnection) obj.openConnection();
                        int responseCode = con.getResponseCode();
                        host.setStatus("" + responseCode);
                        host.setTime("" + (System.currentTimeMillis() - start));
                    } catch (Exception e) {
                        host.setStatus(e.getLocalizedMessage());
                        host.setTime("");
                    }
                }
            }
            return hosts;
        } catch (IOException e) {
            throw new RuntimeException(e);
        }
    }

    private Hosts loadConfiguration(String configFile) {
        try {
            File file = new File(configFile);
            JAXBContext jaxbContext = JAXBContext.newInstance(Hosts.class);

            Unmarshaller unmarshaller = jaxbContext.createUnmarshaller();
            Hosts hosts = (Hosts) unmarshaller.unmarshal(file);
            return hosts;
        } catch (JAXBException e) {
            throw new RuntimeException(e);
        }
    }

}