FriendOfFriendResult process(int partition, List<Integer> friends) throws IOException {
    List<DocIdSet> docSets = new ArrayList<DocIdSet>(friends.size());
    List<int[]> fids = _core.getFriendIds(partition, friends);
    System.out.println("find finds " + fids.size());

    for (int[] ids : fids) {
      IntArrayDocIdSet intSet = new IntArrayDocIdSet();
      intSet.addDocs(ids, 0, ids.length);
      docSets.add(intSet);
    }
    OrDocIdSet orSet = new OrDocIdSet(docSets);
    DocIdSetIterator iter = orSet.iterator();
    List<Integer> res = new ArrayList<Integer>();

    int docId = iter.nextDoc();
    while (docId != DocIdSetIterator.NO_MORE_DOCS) {
      res.add(docId);
      docId = iter.nextDoc();
    }

    int[] arr = new int[res.size()];
    for (int i = 0; i < res.size(); ++i) {
      arr[i] = res.get(i);
    }
    FriendOfFriendResult result = new FriendOfFriendResult();
    result.setRes(arr);
    return result;
  }
Esempio n. 2
0
 @Test
 public void testOrDocIdSetWithIntArrayPForDelta() throws Exception {
   // test OrDocIdSet with IntArrayDocIdSet and PForDeltaDocIdSet
   System.out.println("Running testOrDocIdSetWithIntArrayPForDelta() Test case...");
   DocIdSet[] DocList;
   DocList = new DocIdSet[5];
   int maxdoc = 100000;
   for (int i = 0; i < DocList.length; ++i) {
     IntArrayDocIdSet docset = new IntArrayDocIdSet(maxdoc);
     for (int k = 0; k < maxdoc; k++) {
       docset.addDoc(k);
     }
     DocList[i] = docset;
   }
   // PForDeltaOrDocIdSet orset = new PForDeltaOrDocIdSet(Arrays.asList(DocList));
   OrDocIdSet orset = new OrDocIdSet(Arrays.asList(DocList));
   DocIdSetIterator iter = orset.iterator();
   int doc;
   while ((doc = iter.nextDoc()) != DocIdSetIterator.NO_MORE_DOCS) {}
   System.out.println("--------completed-----------");
 }