org.bedework.inoutsched.BwInoutSched Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of bw-calendar-engine-inoutsched Show documentation
Main calendar engine code for bedework
There is a newer version: 4.1.4
Show newest version
/* ********************************************************************
    Licensed to Jasig under one or more contributor license
    agreements. See the NOTICE file distributed with this work
    for additional information regarding copyright ownership.
    Jasig licenses this file to you 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 org.bedework.inoutsched;

import org.bedework.calsvc.scheduling.hosts.BwHosts;
import org.bedework.util.jmx.ConfBase;

import javax.management.ObjectName;

/** JMX bean for bedework scheduling
 *
 * @author douglm
 *
 */
public class BwInoutSched extends ConfBase
        implements BwInoutSchedMBean {
  private static BwHosts isched;

  private class ProcessorThread extends Thread {
    private final InoutSched sched;

    /**
     * @param name - for the thread
     * @param sched - the scheduling object
     */
    public ProcessorThread(final String name, final InoutSched sched) {
      super(name);

      this.sched = sched;
    }

    @Override
    public void run() {
      try {
        sched.run();
      } catch (final Throwable t) {
        error(t.getMessage());
      }
    }
  }

  private ProcessorThread inProcessor;
  private ProcessorThread outProcessor;

  private int incomingRetryLimit = 10;

  private int outgoingRetryLimit = 10;

  private final Counts counts = new Counts();

  /**
   *
   */
  public BwInoutSched() {
    super("org.bedework.bwengine:service=BwInoutSched",
          (String)null,
          null);

    try {
      isched = new BwHosts();
      register(new ObjectName(isched.getServiceName()), isched);
      isched.loadConfigs();
    } catch (final Throwable t) {
      error("Failed to register hosts");
      throw new RuntimeException(t);
    }
  }

  @Override
  public String loadConfig() {
    return null;
  }

  @Override
  public void setIncomingRetryLimit(final int val) {
    incomingRetryLimit = val;

    if (inProcessor != null) {
      inProcessor.sched.setRetryLimit(val);
    }
  }

  @Override
  public int getIncomingRetryLimit() {
    return incomingRetryLimit;
  }

  @Override
  public void setOutgoingRetryLimit(final int val) {
    outgoingRetryLimit = val;

    if (outProcessor != null) {
      outProcessor.sched.setRetryLimit(val);
    }
  }

  @Override
  public int getOutgoingRetryLimit() {
    return outgoingRetryLimit;
  }

  @Override
  public Counts getCounts() {
    return counts;
  }

  @Override
  public String getName() {
    /* This apparently must be the same as the name attribute in the
     * jboss service definition
     */
    return getServiceName();
  }

  /* ========================================================================
   * Lifecycle
   * ======================================================================== */

  @Override
  public void create() {
  }

  @Override
  public synchronized void start() {
    if (outProcessor != null) {
      error("Already started");
      return;
    }

    info("************************************************************");
    info(" * Starting " + getName());
    info("************************************************************");

    try {
      inProcessor = new ProcessorThread(getName(),
                                        new InoutSched(counts.inCounts,
                                                       incomingRetryLimit,
                                                       true));
      outProcessor = new ProcessorThread(getName(),
                                         new InoutSched(counts.outCounts,
                                                        outgoingRetryLimit,
                                                        false));
    } catch (final Throwable t) {
      error("Error starting scheduler");
      error(t);
    }

    inProcessor.start();
    outProcessor.start();
  }

  @Override
  public synchronized void stop() {
    if (outProcessor == null) {
      error("Already stopped");
      return;
    }

    info("************************************************************");
    info(" * Stopping " + getName());
    info("************************************************************");

    stopProc(inProcessor);
    inProcessor = null;

    stopProc(outProcessor);
    outProcessor = null;

    info("************************************************************");
    info(" * " + getName() + " terminated");
    info("************************************************************");
  }

  @Override
  public boolean isStarted() {
    return (outProcessor != null) && outProcessor.isAlive();
  }

  @Override
  public void destroy() {
    try {
      getManagementContext().stop();
    } catch (Throwable t) {
      error("Failed to stop management context");
      error(t);
    }
  }

  /* ====================================================================
   *                   Private methods
   * ==================================================================== */

  private void stopProc(final ProcessorThread p) {
    if (p == null) {
      return;
    }

    p.interrupt();
    try {
      p.join();
    } catch (final InterruptedException ignored) {
    } catch (final Throwable t) {
      error("Error waiting for processor termination");
      error(t);
    }
  }
}