Ejemplo n.º 1
0
  public void expandVertex(
      Node target,
      TimeEntry removed,
      HashMap<Long, Integer> wasTraversed,
      PriorityQueue<TimeEntry> queue,
      HashMap<Long, RouteEntry> parents) {

    Long2IntMap neig = graph.accessNeighborhood(graph.getNode(removed.getId()));

    for (long vid : neig.keySet()) {

      int arrivalTime = graph.getArrival(removed.getArrivalTime(), neig.get(vid));
      int travelTime = removed.getTravelTime() + neig.get(vid);
      TimeEntry newEntry = new TimeEntry(vid, travelTime, arrivalTime, removed.getId());

      Edge edge = null;
      int distance = -1;

      if (!wasTraversed.containsKey(vid)) {

        queue.offer(newEntry);
        wasTraversed.put(newEntry.getId(), newEntry.getTravelTime());

        distance = neig.get(vid);
        edge = getEdge(removed.getId(), vid, distance);
        parents.put(
            vid, new RouteEntry(removed.getId(), distance / 17, edge.getId(), edge.getLabel()));
      } else {

        int cost = wasTraversed.get(vid);

        if (cost != wasRemoved && cost > newEntry.getTravelTime()) {
          queue.remove(newEntry);
          queue.offer(newEntry);
          wasTraversed.remove(newEntry.getId());
          wasTraversed.put(newEntry.getId(), newEntry.getTravelTime());

          parents.remove(vid);
          distance = neig.get(vid);
          edge = getEdge(removed.getId(), vid, distance);
          parents.put(
              vid, new RouteEntry(removed.getId(), distance / 17, edge.getId(), edge.getLabel()));
        }
      }
    }
  }
 public int getIndexByUID(long uid) {
   Lock lock = globalLock.readLock();
   try {
     lock.lock();
     if (!uidToArrayIndex.containsKey(uid)) {
       return -1;
     }
     return uidToArrayIndex.get(uid);
   } finally {
     lock.unlock();
   }
 }
 public int getIntValueByUID(long uid, String column) {
   Lock lock = globalLock.readLock();
   try {
     lock.lock();
     if (!uidToArrayIndex.containsKey(uid)) {
       return Integer.MIN_VALUE;
     }
     return ((ActivityIntValues) getActivityValues(column)).getIntValue(uidToArrayIndex.get(uid));
   } finally {
     lock.unlock();
   }
 }
  public int update(long uid, final String version, Map<String, Object> map) {
    if (valuesMap.isEmpty()) {
      return -1;
    }
    if (versionComparator.compare(lastVersion, version) > 0) {
      versionRejectionCounter.inc();
      return -1;
    }
    if (map.isEmpty()) {
      lastVersion = version;
      return -1;
    }
    int index = -1;

    Lock writeLock = globalLock.writeLock();
    boolean needToFlush = false;
    try {
      writeLock.lock();
      totalUpdatesCounter.inc();
      if (uidToArrayIndex.containsKey(uid)) {
        index = uidToArrayIndex.get(uid);
      } else {
        insertedDocumentsCounter.inc();
        synchronized (deletedIndexes) {
          if (deletedIndexes.size() > 0) {
            index = deletedIndexes.removeInt(deletedIndexes.size() - 1);
          } else {
            index = indexSize.getAndIncrement();
          }
        }
        uidToArrayIndex.put(uid, index);
        recentlyAddedUids.add(uid);
        needToFlush = updateBatch.addFieldUpdate(new Update(index, uid));
      }
      boolean currentUpdate = updateActivities(map, index);
      needToFlush = needToFlush || currentUpdate;
      lastVersion = version;
    } finally {
      writeLock.unlock();
    }
    if (needToFlush) {
      flush();
    }
    return index;
  }
 public int[] precomputeArrayIndexes(long[] uids) {
   int[] ret = new int[uids.length];
   for (int i = 0; i < uids.length; i++) {
     long uid = uids[i];
     if (uid == ZoieSegmentReader.DELETED_UID) {
       ret[i] = -1;
       continue;
     }
     Lock lock = globalLock.readLock();
     try {
       lock.lock();
       if (!uidToArrayIndex.containsKey(uid)) {
         ret[i] = -1;
       } else {
         ret[i] = uidToArrayIndex.get(uid);
       }
     } finally {
       lock.unlock();
     }
   }
   return ret;
 }