/**
    * Logback: the reliable, generic, fast and flexible logging framework.
    * Copyright (C) 1999-2011, QOS.ch. All rights reserved.
    *
    * This program and the accompanying materials are dual-licensed under
    * either the terms of the Eclipse Public License v1.0 as published by
    * the Eclipse Foundation
    *
    *   or (per the licensee's choosing)
   *
   * under the terms of the GNU Lesser General Public License version 2.1
   * as published by the Free Software Foundation.
   */
   package ch.qos.logback.access.filter;
  
  import ch.qos.logback.core.filter.Filter;
  import ch.qos.logback.core.spi.FilterReply;
  
  import javax.management.MBeanServer;
  import javax.management.ObjectName;
  import javax.management.StandardMBean;
  import java.lang.management.ManagementFactory;
  
  public class CountingFilter extends Filter {
  
    long total = 0;
    final StatisticalViewImpl accessStatsImpl;
    
    String domain = "ch.qos.logback.access";
    
    public CountingFilter() {
      accessStatsImpl = new StatisticalViewImpl(this);
    }
    
    @Override
    public FilterReply decide(Object event) {
      total++;
      accessStatsImpl.update();
      return FilterReply.NEUTRAL;
    }
  
    public long getTotal() {
      return total;
    }
    
    
    @Override
    public void start() {
      MBeanServer mbs = ManagementFactory.getPlatformMBeanServer();
      try {
        ObjectName on = new ObjectName(domain+":Name="+getName());
        StandardMBean mbean = new StandardMBean(accessStatsImpl, StatisticalView.class);
        if (mbs.isRegistered(on)) {
            mbs.unregisterMBean(on);
        }
        mbs.registerMBean(mbean, on);
        super.start();
      } catch (Exception e) {
        addError("Failed to create mbean", e);
      }
    }
    
    @Override
    public void stop() {
      super.stop();
      try {
        MBeanServer mbs = ManagementFactory.getPlatformMBeanServer();
        ObjectName on = new ObjectName("totp:Filter=1");
        mbs.unregisterMBean(on);
      } catch(Exception e) {
        addError("Failed to unregister mbean", e);
      }
    }
  
    public String getDomain() {
      return domain;
    }
  
    public void setDomain(String domain) {
      this.domain = domain;
    }
    
  }