com.davidbracewell.stream.SparkMAccumulator Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of mango Show documentation

A set of utilities and tools to speed up and ease programming in Java.

There is a newer version: 0.5

package com.davidbracewell.stream;

import lombok.NonNull;
import org.apache.spark.Accumulator;

import java.io.Serializable;

/**
 * @author David B. Bracewell
 */
public class SparkMAccumulator implements MAccumulator, Serializable {

  private final Accumulator accumulator;


  public SparkMAccumulator(@NonNull Accumulator accumulator) {
    this.accumulator = accumulator;
  }

  @Override
  public void increment(double amount) {
    accumulator.add(amount);
  }

  @Override
  public void decrement(double amount) {
    accumulator.add(-amount);
  }

  @Override
  public double value() {
    return accumulator.localValue();
  }

  @Override
  public void setValue(double value) {
    accumulator.setValue(value);
  }

  @Override
  public String name() {
    return accumulator.name().get();
  }
}// END OF SparkMAccumulator