io.atomix.core.queue.impl.CachingAsyncDistributedQueue Maven / Gradle / Ivy

Go to download

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

There is a newer version: 3.2.0-alpha12

/*
 * Copyright 2018-present Open Networking Foundation
 *
 * 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 io.atomix.core.queue.impl;

import io.atomix.core.cache.CacheConfig;
import io.atomix.core.collection.impl.CachingAsyncDistributedCollection;
import io.atomix.core.queue.AsyncDistributedQueue;
import io.atomix.core.queue.DistributedQueue;

import java.time.Duration;
import java.util.concurrent.CompletableFuture;

/**
 * Caching asynchronous distributed queue.
 */
public class CachingAsyncDistributedQueue extends CachingAsyncDistributedCollection implements AsyncDistributedQueue {
  private final AsyncDistributedQueue backingQueue;

  public CachingAsyncDistributedQueue(AsyncDistributedQueue backingCollection, CacheConfig cacheConfig) {
    super(backingCollection, cacheConfig);
    this.backingQueue = backingCollection;
  }

  @Override
  public CompletableFuture offer(E e) {
    return backingQueue.offer(e);
  }

  @Override
  public CompletableFuture remove() {
    return backingQueue.remove().thenApply(result -> {
      cache.invalidate(result);
      return result;
    });
  }

  @Override
  public CompletableFuture poll() {
    return backingQueue.poll().thenApply(result -> {
      cache.invalidate(result);
      return result;
    });
  }

  @Override
  public CompletableFuture element() {
    return backingQueue.element().thenApply(result -> {
      cache.invalidate(result);
      return result;
    });
  }

  @Override
  public CompletableFuture peek() {
    return backingQueue.peek();
  }

  @Override
  public DistributedQueue sync(Duration operationTimeout) {
    return new BlockingDistributedQueue<>(this, operationTimeout.toMillis());
  }
}