cn.aradin.spring.actuator.starter.actuate.OnlineEndpoint Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of aradin-spring-actuator-starter Show documentation

Actuator And Customized HealthCheck

There is a newer version: 1.1.2

Show newest version

package cn.aradin.spring.actuator.starter.actuate;

import java.io.IOException;
import java.util.Collections;
import java.util.List;
import java.util.Map;

import org.apache.commons.collections.CollectionUtils;
import org.apache.commons.lang3.StringUtils;
import org.springframework.boot.actuate.endpoint.annotation.Endpoint;
import org.springframework.boot.actuate.endpoint.annotation.ReadOperation;

import cn.aradin.spring.actuator.starter.extension.IOnlineHandler;
import cn.aradin.spring.actuator.starter.properties.ActuatorOnlineProperties;
import lombok.extern.slf4j.Slf4j;

@Slf4j
@Endpoint(id = "online", enableByDefault = true)
public class OnlineEndpoint {
	
	private static final Map ONLINE_MESSAGE = Collections
			.unmodifiableMap(Collections.singletonMap("message", "Online Success"));
	
	private final ActuatorOnlineProperties actuatorOnlineProperties;
	
	private final List onlineHandlers;
	
	public OnlineEndpoint(ActuatorOnlineProperties actuatorOnlineProperties, List onlineHandlers) {
		// TODO Auto-generated constructor stub
		this.actuatorOnlineProperties = actuatorOnlineProperties;
		this.onlineHandlers = onlineHandlers;
	}
	
	@ReadOperation
	public Map online() {
		if (StringUtils.isNotBlank(actuatorOnlineProperties.getShell())) {
			String bashCommand = "sh "+actuatorOnlineProperties.getShell();
	    	Runtime runtime = Runtime.getRuntime();
			try {
				Process process = runtime.exec(bashCommand);
				int status = process.waitFor();
				if (status != 0) {
					if (log.isWarnEnabled()) {
						log.warn("Filebeat Start Error {}", status);
					}
				}else {
					if (log.isDebugEnabled()) {
						log.debug("Filebeat start OK");
					}
				}
			} catch (IOException e) {
				// TODO Auto-generated catch block
				e.printStackTrace();
			} catch (InterruptedException e) {
				// TODO Auto-generated catch block
				e.printStackTrace();
			} catch (Exception e) {
				// TODO: handle exception
				e.printStackTrace();
			}
		}
		try {
			if (CollectionUtils.isNotEmpty(onlineHandlers)) {
				onlineHandlers.forEach(handler->{
					handler.online();
				});
			}
		} catch (Exception e) {
			// TODO: handle exception
			e.printStackTrace();
		}
        return ONLINE_MESSAGE;
	}
}