net.kuujo.copycat.internal.util.Quorum Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of copycat Show documentation
The newest version!
/*
 * Copyright 2014 the original author or authors.
 *
 * 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.kuujo.copycat.internal.util;

import java.util.function.Consumer;

/**
 * Quorum helper.
 *
 * @author Jordan Halterman
 */
public class Quorum {
  private int succeeded;
  private int failed;
  private int quorum;
  private Consumer callback;
  private boolean complete;

  public Quorum(int quorum, Consumer callback) {
    this.quorum = quorum;
    this.callback = callback;
  }

  /**
   * Counts the current node in the quorum.
   */
  public Quorum countSelf() {
    return succeed();
  }

  private void checkComplete() {
    if (!complete && callback != null) {
      if (succeeded >= quorum) {
        complete = true;
        callback.accept(true);
      } else if (failed >= quorum) {
        complete = true;
        callback.accept(false);
      }
    }
  }

  /**
   * Indicates that a call in the quorum succeeded.
   */
  public Quorum succeed() {
    succeeded++;
    checkComplete();
    return this;
  }

  /**
   * Indicates that a call in the quorum failed.
   */
  public Quorum fail() {
    failed++;
    checkComplete();
    return this;
  }

  /**
   * Cancels the quorum. Once this method has been called, the quorum
   * will be marked complete and the handler will never be called.
   */
  public void cancel() {
    callback = null;
    complete = true;
  }

}