org.broadleafcommerce.common.vendor.service.monitor.ServiceMonitor Maven / Gradle / Ivy
/*
* #%L
* BroadleafCommerce Common Libraries
* %%
* Copyright (C) 2009 - 2013 Broadleaf Commerce
* %%
* 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.
* #L%
*/
package org.broadleafcommerce.common.vendor.service.monitor;
import org.apache.commons.logging.Log;
import org.apache.commons.logging.LogFactory;
import org.aspectj.lang.ProceedingJoinPoint;
import org.broadleafcommerce.common.vendor.service.monitor.handler.LogStatusHandler;
import org.broadleafcommerce.common.vendor.service.type.ServiceStatusType;
import java.util.HashMap;
import java.util.Map;
public class ServiceMonitor {
private static final Log LOG = LogFactory.getLog(ServiceMonitor.class);
protected Map serviceHandlers = new HashMap();
protected StatusHandler defaultHandler = new LogStatusHandler();
protected Map statusMap = new HashMap();
public synchronized void init() {
for (ServiceStatusDetectable statusDetectable : serviceHandlers.keySet()) {
checkService(statusDetectable);
}
}
public Object checkServiceAOP(ProceedingJoinPoint call) throws Throwable {
try {
checkService((ServiceStatusDetectable) call.getThis());
} catch (Throwable e) {
LOG.error("Could not check service status", e);
}
return call.proceed();
}
public void checkService(ServiceStatusDetectable statusDetectable) {
ServiceStatusType type = statusDetectable.getServiceStatus();
if (!statusMap.containsKey(statusDetectable)) {
statusMap.put(statusDetectable, type);
if (type.equals(ServiceStatusType.DOWN)) {
handleStatusChange(statusDetectable, type);
}
}
if (!statusMap.get(statusDetectable).equals(type)) {
handleStatusChange(statusDetectable, type);
statusMap.put(statusDetectable, type);
}
}
protected void handleStatusChange(ServiceStatusDetectable serviceStatus, ServiceStatusType serviceStatusType) {
if (serviceHandlers.containsKey(serviceStatus)) {
serviceHandlers.get(serviceStatus).handleStatus(serviceStatus.getServiceName(), serviceStatusType);
} else {
defaultHandler.handleStatus(serviceStatus.getServiceName(), serviceStatusType);
}
}
public Map getServiceHandlers() {
return serviceHandlers;
}
public void setServiceHandlers(Map serviceHandlers) {
this.serviceHandlers = serviceHandlers;
}
public StatusHandler getDefaultHandler() {
return defaultHandler;
}
public void setDefaultHandler(StatusHandler defaultHandler) {
this.defaultHandler = defaultHandler;
}
}