com.github.dm.jrt.operator.CountInvocation Maven / Gradle / Ivy

Go to download
/*
 * Copyright 2016 Davide Maestroni
 *
 * 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 com.github.dm.jrt.operator;

import com.github.dm.jrt.core.channel.Channel;
import com.github.dm.jrt.core.invocation.Invocation;
import com.github.dm.jrt.core.invocation.InvocationFactory;
import com.github.dm.jrt.core.invocation.TemplateInvocation;

import org.jetbrains.annotations.NotNull;

/**
 * Invocation computing the number of outputs.
 * 
 * Created by davide-maestroni on 04/26/2016.
 */
class CountInvocation extends TemplateInvocation {

  private static final InvocationFactory sFactory =
      new InvocationFactory(null) {

        @NotNull
        @Override
        public Invocation newInvocation() {
          return new CountInvocation();
        }
      };

  private long mCount;

  /**
   * Constructor.
   */
  private CountInvocation() {
  }

  /**
   * Returns the factory of counting invocations.
   *
   * @return the factory instance.
   */
  @NotNull
  static InvocationFactory factoryOf() {
    return sFactory;
  }

  @Override
  public void onComplete(@NotNull final Channel result) {
    result.pass(mCount);
  }

  @Override
  public void onInput(final Object input, @NotNull final Channel result) {
    ++mCount;
  }

  @Override
  public void onRestart() {
    mCount = 0;
  }
}