com.tinkerpop.gremlin.tinkergraph.process.computer.TinkerMapEmitter Maven / Gradle / Ivy
package com.tinkerpop.gremlin.tinkergraph.process.computer;
import com.tinkerpop.gremlin.process.computer.MapReduce;
import com.tinkerpop.gremlin.process.util.MapHelper;
import org.javatuples.Pair;
import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import java.util.Queue;
import java.util.concurrent.ConcurrentHashMap;
import java.util.concurrent.ConcurrentLinkedQueue;
import java.util.concurrent.CopyOnWriteArrayList;
/**
* @author Marko A. Rodriguez (http://markorodriguez.com)
*/
class TinkerMapEmitter implements MapReduce.MapEmitter {
public Map> reduceMap;
public Queue> mapQueue;
private final boolean doReduce;
public TinkerMapEmitter(final boolean doReduce) {
this.doReduce = doReduce;
if(this.doReduce)
this.reduceMap = new ConcurrentHashMap<>();
else
this.mapQueue = new ConcurrentLinkedQueue<>();
}
@Override
public void emit(K key, V value) {
if (this.doReduce)
MapHelper.concurrentIncr(this.reduceMap, key, value);
else
this.mapQueue.add(new Pair<>(key, value));
}
}