com.uber.cadence.internal.sync.WorkflowQueueImpl Maven / Gradle / Ivy

Go to download
/*
 *  Copyright 2012-2016 Amazon.com, Inc. or its affiliates. All Rights Reserved.
 *
 *  Modifications copyright (C) 2017 Uber Technologies, Inc.
 *
 *  Licensed under the Apache License, Version 2.0 (the "License"). You may not
 *  use this file except in compliance with the License. A copy of the License is
 *  located at
 *
 *  http://aws.amazon.com/apache2.0
 *
 *  or in the "license" file accompanying this file. This file 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.uber.cadence.internal.sync;

import com.uber.cadence.workflow.Functions;
import com.uber.cadence.workflow.QueueConsumer;
import com.uber.cadence.workflow.WorkflowQueue;
import java.util.ArrayDeque;
import java.util.Deque;
import java.util.concurrent.TimeUnit;

final class WorkflowQueueImpl implements WorkflowQueue {

  private final Deque queue = new ArrayDeque<>();
  private final int capacity;

  public WorkflowQueueImpl(int capacity) {
    if (capacity < 1) {
      throw new IllegalArgumentException("Capacity less than 1: " + capacity);
    }
    this.capacity = capacity;
  }

  @Override
  public E take() throws InterruptedException {
    WorkflowThread.await("WorkflowQueue.take", () -> !queue.isEmpty());
    return queue.pollLast();
  }

  @Override
  public E poll(long timeout, TimeUnit unit) throws InterruptedException {
    WorkflowThread.await(unit.toMillis(timeout), "WorkflowQueue.poll", () -> !queue.isEmpty());
    if (queue.isEmpty()) {
      return null;
    }
    return queue.remove();
  }

  @Override
  public boolean offer(E e) {
    if (queue.size() == capacity) {
      return false;
    }
    queue.addLast(e);
    return true;
  }

  @Override
  public void put(E e) throws InterruptedException {
    // This condition is excessive as await already checks it.
    // But await can be called only from the sync owned thread.
    // This condition allows puts outside the sync thread which
    // is used by signal handling logic.
    if (queue.size() >= capacity) {
      WorkflowThread.await("WorkflowQueue.put", () -> queue.size() < capacity);
    }
    queue.addLast(e);
  }

  @Override
  public boolean offer(E e, long timeout, TimeUnit unit) throws InterruptedException {
    boolean timedOut =
        WorkflowThread.await(
            unit.toMillis(timeout), "WorkflowQueue.offer", () -> queue.size() < capacity);
    if (timedOut) {
      return false;
    }
    queue.addLast(e);
    return true;
  }

  @Override
  public  QueueConsumer map(Functions.Func1 mapper) {
    return new MappedQueueConsumer(this, mapper);
  }

  private static class MappedQueueConsumer implements QueueConsumer {

    private QueueConsumer source;
    private final Functions.Func1 mapper;

    public MappedQueueConsumer(
        QueueConsumer source, Functions.Func1 mapper) {
      this.source = source;
      this.mapper = mapper;
    }

    @Override
    public R take() throws InterruptedException {
      E element = source.take();
      try {
        return mapper.apply(element);
      } catch (Exception e) {
        throw new RuntimeException("Failure mapping an element", e);
      }
    }

    @Override
    public R poll(long timeout, TimeUnit unit) throws InterruptedException {
      E element = source.poll(timeout, unit);
      try {
        if (element == null) {
          return null;
        }
        return mapper.apply(element);
      } catch (Exception e) {
        throw new RuntimeException("Failure mapping an element", e);
      }
    }

    @Override
    public  QueueConsumer map(Functions.Func1 mapper) {
      return new MappedQueueConsumer<>(this, mapper);
    }
  }
}