Java DocIdPusher.FeedType方法代码示例

import com.google.enterprise.adaptor.DocIdPusher; //导入方法依赖的package包/类
@Override
public void getDocIds(DocIdPusher pusher) throws InterruptedException {
  Authentication authentication = this.soapFactory.newAuthentication();
  String authenticationToken =
      getAuthenticationToken(this.username, this.password);
  DocumentManagement documentManagement =
      soapFactory.newDocumentManagement(authenticationToken);
  ArrayList<DocId> docIds = new ArrayList<DocId>();
  for (StartPoint startPoint : this.startPoints) {
    if (isValidStartPoint(startPoint, documentManagement)) {
      docIds.add(
          new DocId(startPoint.getName() + ":" + startPoint.getNodeId()));
    }
  }
  log.log(Level.FINER, "Sending doc ids: {0}", docIds);
  pusher.pushDocIds(docIds);

  if (!this.markAllDocsAsPublic) {
    MemberService memberService =
        this.soapFactory.newMemberService(documentManagement);
    SOAPFaultException groupsException = null;
    Groups groups = new Groups(memberService);
    try {
      groups.addGroups();
      if (this.publicAccessGroupEnabled) {
        groups.addPublicAccessGroup("[Public Access]", this.localNamespace);
      }
    } catch (SOAPFaultException e) {
      groupsException = e;
      log.log(Level.WARNING, "Error while reading groups", e);
    }

    // If some groups were found, push them; otherwise, if an
    // exception was thrown, rethrow it as before.
    Map<GroupPrincipal, List<Principal>> groupDefinitions =
        groups.getGroupDefinitions();
    if (groupDefinitions.size() > 0) {
      DocIdPusher.FeedType feedType = DocIdPusher.FeedType.REPLACE;
      if (groups.isPartialResults() || groupsException != null) {
        feedType = DocIdPusher.FeedType.INCREMENTAL;
        log.log(Level.FINE, "Sending partial groups results");
      }
      pusher.pushGroupDefinitions(groupDefinitions,
          /* case sensitive */ true, feedType,
          /* group source */ null, /* exception handler */ null);
    } else if (groupsException != null) {
      throw groupsException;
    }
  }
}
 

import com.google.enterprise.adaptor.DocIdPusher; //导入方法依赖的package包/类
@Test
public void testGroupsError() throws InterruptedException {
  class MemberServiceMockError extends MemberServiceMock {
    boolean called;
    public MemberSearchResults getSearchResults(PageHandle pageHandle) {
      if (!called)  {
        called = true;
        return new MemberSearchResultsMock((PageHandleMock) pageHandle);
      }
      throw getSoapFaultException("MemberService SOAPFaultException",
          "urn:opentextadaptortest", "Test.MemberServiceException", "ns0");
    }
  }
  SoapFactoryMock soapFactory = new SoapFactoryMock();
  soapFactory.memberServiceMock = new MemberServiceMockError();
  for (int i = 0; i < 20; i++) {
    soapFactory.memberServiceMock.addMember(
        getMember(1000 + i, "user" + i, "User"));
  }
  // The search uses a page size of 50, and we want to trigger
  // an exception on the second page in order to get partial
  // results, so create a bunch of groups.
  for (int i = 0; i < 64; i++) {
    soapFactory.memberServiceMock.addMember(
        getMember(2000 + i, "group" + i, "Group"));
    soapFactory.memberServiceMock.addMemberToGroup(
        2000 + i,
        soapFactory.memberServiceMock.getMemberById(1000 + (i % 20)));
  }
  OpentextAdaptor adaptor = new OpentextAdaptor(soapFactory);
  AdaptorContext context = ProxyAdaptorContext.getInstance();
  Config config = initConfig(adaptor, context);
  adaptor.init(context);

  class MyRecordingDocIdPusher extends RecordingDocIdPusher {
    DocIdPusher.FeedType latestFeedType;

    @Override
    public GroupPrincipal pushGroupDefinitions(
        Map<GroupPrincipal, ? extends Collection<Principal>> defs,
        boolean caseSensitive, FeedType feedType, String sourceName,
        ExceptionHandler exceptionHandler) throws InterruptedException {
      this.latestFeedType = feedType;
      return super.pushGroupDefinitions(
          defs, caseSensitive, feedType, sourceName, exceptionHandler);
    }
  };

  MyRecordingDocIdPusher pusher = new MyRecordingDocIdPusher();
  adaptor.getDocIds(pusher);
  Map<GroupPrincipal, Collection<Principal>> groupDefinitions =
      pusher.getGroupDefinitions();
  assertEquals(50, groupDefinitions.size());
  assertEquals(DocIdPusher.FeedType.INCREMENTAL, pusher.latestFeedType);
}