/*
    * Copyright 2010 Capgemini and Contributors
    *
    * 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.
   */
  package net.sf.appstatus.core.check.impl;
  
  import java.lang.management.ManagementFactory;
  import java.lang.management.MemoryMXBean;
  import java.lang.management.MemoryUsage;
  import java.util.Locale;
  import java.util.Properties;
  
  import net.sf.appstatus.core.check.AbstractCheck;
  import net.sf.appstatus.core.check.CheckResultBuilder;
  import net.sf.appstatus.core.check.ICheckResult;
  
  /**
   * @author Nicolas Richeton
   *
   */
  public class JvmCheck extends AbstractCheck {
  
  	private int limitError = 95;
  	private int limitWarn = 80;
  
  	@Override
  	public ICheckResult checkStatus(Locale locale) {
  		MemoryMXBean memory = ManagementFactory.getMemoryMXBean();
  		MemoryUsage heap = memory.getHeapMemoryUsage();
  		long heapRatio = heap.getUsed() * 100 / heap.getMax();
  		CheckResultBuilder result = result(this).messageBundle("net.sf.appstatus.core.check.impl.JvmCheck_msg", locale);
  		if (heapRatio > limitError) {
  			result.code(ICheckResult.ERROR).fatal().resolutionSteps("resolutionSteps.fatal", new Object[] {});
  		} else if (heapRatio > limitWarn) {
  			result.code(ICheckResult.ERROR).resolutionSteps("resolutionSteps.warn", new Object[] {});
  		} else {
  			result.code(ICheckResult.OK);
  		}
  		result.description("description", new Object[] { heapRatio });
  		return result.build();
  	}
  
  	public String getGroup() {
  		return "JVM";
  	}
  
  	public String getName() {
  		return "Heap usage";
  	}
  
  	@Override
  	public void setConfiguration(Properties configuration) {
  		super.setConfiguration(configuration);
  
  		String error = getConfiguration().getProperty("jvmCheck.limitError");
  		if (error != null) {
  			limitError = Integer.valueOf(error);
  		}
  
  		String warn = getConfiguration().getProperty("jvmCheck.limitWarn");
  		if (warn != null) {
  			limitWarn = Integer.valueOf(warn);
  		}
  
  	}
  
  }