io.atomix.core.map.impl.TranscodingAsyncDistributedMap 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 2016-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.map.impl;

import com.google.common.collect.Maps;
import io.atomix.core.collection.AsyncDistributedCollection;
import io.atomix.core.collection.impl.TranscodingAsyncDistributedCollection;
import io.atomix.core.map.AsyncDistributedMap;
import io.atomix.core.map.DistributedMap;
import io.atomix.core.map.MapEvent;
import io.atomix.core.map.MapEventListener;
import io.atomix.core.set.AsyncDistributedSet;
import io.atomix.core.set.impl.TranscodingAsyncDistributedSet;
import io.atomix.primitive.impl.DelegatingAsyncPrimitive;
import io.atomix.utils.concurrent.Futures;

import java.time.Duration;
import java.util.Map;
import java.util.Map.Entry;
import java.util.concurrent.CompletableFuture;
import java.util.concurrent.Executor;
import java.util.function.BiFunction;
import java.util.function.Function;
import java.util.stream.Collectors;

/**
 * An {@code AsyncDistributedMap} that maps its operations to operations on a
 * differently typed {@code AsyncDistributedMap} by transcoding operation inputs and outputs.
 *
 * @param  key type of other map
 * @param  value type of other map
 * @param  key type of this map
 * @param  value type of this map
 */
public class TranscodingAsyncDistributedMap extends DelegatingAsyncPrimitive implements AsyncDistributedMap {

  private final AsyncDistributedMap backingMap;
  private final Function keyEncoder;
  private final Function keyDecoder;
  private final Function valueDecoder;
  private final Function valueEncoder;
  private final Function, Entry> entryDecoder;
  private final Function, Entry> entryEncoder;
  private final Map, InternalBackingMapEventListener> listeners =
      Maps.newIdentityHashMap();

  public TranscodingAsyncDistributedMap(
      AsyncDistributedMap backingMap,
      Function keyEncoder,
      Function keyDecoder,
      Function valueEncoder,
      Function valueDecoder) {
    super(backingMap);
    this.backingMap = backingMap;
    this.keyEncoder = k -> k == null ? null : keyEncoder.apply(k);
    this.keyDecoder = k -> k == null ? null : keyDecoder.apply(k);
    this.valueEncoder = v -> v == null ? null : valueEncoder.apply(v);
    this.valueDecoder = v -> v == null ? null : valueDecoder.apply(v);
    this.entryDecoder = e -> e == null ? null : Maps.immutableEntry(keyDecoder.apply(e.getKey()), valueDecoder.apply(e.getValue()));
    this.entryEncoder = e -> e == null ? null : Maps.immutableEntry(keyEncoder.apply(e.getKey()), valueEncoder.apply(e.getValue()));
  }

  @Override
  public CompletableFuture size() {
    return backingMap.size();
  }

  @Override
  public CompletableFuture isEmpty() {
    return backingMap.isEmpty();
  }

  @Override
  public CompletableFuture containsKey(K1 key) {
    try {
      return backingMap.containsKey(keyEncoder.apply(key));
    } catch (Exception e) {
      return Futures.exceptionalFuture(e);
    }
  }

  @Override
  public CompletableFuture containsValue(V1 value) {
    try {
      return backingMap.containsValue(valueEncoder.apply(value));
    } catch (Exception e) {
      return Futures.exceptionalFuture(e);
    }
  }

  @Override
  public CompletableFuture get(K1 key) {
    try {
      return backingMap.get(keyEncoder.apply(key)).thenApply(valueDecoder);
    } catch (Exception e) {
      return Futures.exceptionalFuture(e);
    }
  }

  @Override
  public CompletableFuture getOrDefault(K1 key, V1 defaultValue) {
    try {
      return backingMap.getOrDefault(keyEncoder.apply(key), valueEncoder.apply(defaultValue))
          .thenApply(valueDecoder);
    } catch (Exception e) {
      return Futures.exceptionalFuture(e);
    }
  }

  @Override
  public CompletableFuture put(K1 key, V1 value) {
    try {
      return backingMap.put(keyEncoder.apply(key), valueEncoder.apply(value))
          .thenApply(valueDecoder);
    } catch (Exception e) {
      return Futures.exceptionalFuture(e);
    }
  }

  @Override
  public CompletableFuture putAll(Map map) {
    return backingMap.putAll(map.entrySet().stream()
        .map(e -> entryEncoder.apply((Map.Entry) e))
        .collect(Collectors.toMap(e -> e.getKey(), e -> e.getValue())));
  }

  @Override
  public CompletableFuture remove(K1 key) {
    try {
      return backingMap.remove(keyEncoder.apply(key)).thenApply(valueDecoder);
    } catch (Exception e) {
      return Futures.exceptionalFuture(e);
    }
  }

  @Override
  public CompletableFuture computeIfAbsent(K1 key, Function mappingFunction) {
    return backingMap.computeIfAbsent(keyEncoder.apply(key), (k) ->
        valueEncoder.apply(mappingFunction.apply(keyDecoder.apply(k))))
        .thenApply(valueDecoder);
  }

  @Override
  public CompletableFuture computeIfPresent(K1 key, BiFunction remappingFunction) {
    return backingMap.computeIfPresent(keyEncoder.apply(key), (k, v) ->
        valueEncoder.apply(remappingFunction.apply(keyDecoder.apply(k), valueDecoder.apply(v))))
        .thenApply(valueDecoder);
  }

  @Override
  public CompletableFuture compute(K1 key, BiFunction remappingFunction) {
    return backingMap.compute(keyEncoder.apply(key), (k, v) ->
        valueEncoder.apply(remappingFunction.apply(keyDecoder.apply(k), valueDecoder.apply(v))))
        .thenApply(valueDecoder);
  }

  @Override
  public CompletableFuture clear() {
    return backingMap.clear();
  }

  @Override
  public AsyncDistributedSet keySet() {
    return new TranscodingAsyncDistributedSet<>(backingMap.keySet(), keyEncoder, keyDecoder);
  }

  @Override
  public AsyncDistributedCollection values() {
    return new TranscodingAsyncDistributedCollection<>(backingMap.values(), valueEncoder, valueDecoder);
  }

  @Override
  public AsyncDistributedSet> entrySet() {
    return new TranscodingAsyncDistributedSet<>(backingMap.entrySet(), entryEncoder, entryDecoder);
  }

  @Override
  public CompletableFuture putIfAbsent(K1 key, V1 value) {
    try {
      return backingMap.putIfAbsent(keyEncoder.apply(key), valueEncoder.apply(value))
          .thenApply(valueDecoder);
    } catch (Exception e) {
      return Futures.exceptionalFuture(e);
    }
  }

  @Override
  public CompletableFuture remove(K1 key, V1 value) {
    try {
      return backingMap.remove(keyEncoder.apply(key), valueEncoder.apply(value));
    } catch (Exception e) {
      return Futures.exceptionalFuture(e);
    }
  }

  @Override
  public CompletableFuture replace(K1 key, V1 value) {
    try {
      return backingMap.replace(keyEncoder.apply(key), valueEncoder.apply(value))
          .thenApply(valueDecoder);
    } catch (Exception e) {
      return Futures.exceptionalFuture(e);
    }
  }

  @Override
  public CompletableFuture replace(K1 key, V1 oldValue, V1 newValue) {
    try {
      return backingMap.replace(keyEncoder.apply(key),
          valueEncoder.apply(oldValue),
          valueEncoder.apply(newValue));
    } catch (Exception e) {
      return Futures.exceptionalFuture(e);
    }
  }

  @Override
  public CompletableFuture addListener(MapEventListener listener, Executor executor) {
    synchronized (listeners) {
      InternalBackingMapEventListener backingMapListener =
          listeners.computeIfAbsent(listener, k -> new InternalBackingMapEventListener(listener));
      return backingMap.addListener(backingMapListener, executor);
    }
  }

  @Override
  public CompletableFuture removeListener(MapEventListener listener) {
    synchronized (listeners) {
      InternalBackingMapEventListener backingMapListener = listeners.remove(listener);
      if (backingMapListener != null) {
        return backingMap.removeListener(backingMapListener);
      } else {
        return CompletableFuture.completedFuture(null);
      }
    }
  }

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

  private class InternalBackingMapEventListener implements MapEventListener {

    private final MapEventListener listener;

    InternalBackingMapEventListener(MapEventListener listener) {
      this.listener = listener;
    }

    @Override
    public void event(MapEvent event) {
      listener.event(new MapEvent<>(
          event.type(),
          keyDecoder.apply(event.key()),
          event.newValue() != null ? valueDecoder.apply(event.newValue()) : null,
          event.oldValue() != null ? valueDecoder.apply(event.oldValue()) : null));
    }
  }
}