Пример #1
0
 private List<String> getBaseDNs() {
   List<String> result;
   QualifiedUid container = options.getContainer();
   if (container != null) {
     result =
         singletonList(
             LdapSearches.findEntryDN(conn, container.getObjectClass(), container.getUid()));
   } else {
     result = Arrays.asList(baseDNs);
   }
   assert result != null;
   return result;
 }
Пример #2
0
  public final void executeADQuery(final ResultsHandler handler) {
    final String[] attrsToGetOption = options.getAttributesToGet();
    final Set<String> attrsToGet = utils.getAttributesToGet(attrsToGetOption, oclass);

    final LdapInternalSearch search = getInternalSearch(attrsToGet);

    search.execute(
        new SearchResultsHandler() {

          @Override
          public boolean handle(String baseDN, SearchResult result) throws NamingException {
            return handler.handle(
                utils.createConnectorObject(
                    result.getNameInNamespace(), result, attrsToGet, oclass));
          }
        });
  }
Пример #3
0
  private int getLdapSearchScope() {
    String scope = options.getScope();

    if (scope == null) {
      if (oclass.is(ObjectClass.ACCOUNT_NAME)) {
        scope = ((ADConfiguration) conn.getConfiguration()).getUserSearchScope();
      } else {
        scope = ((ADConfiguration) conn.getConfiguration()).getGroupSearchScope();
      }
    }

    if (OperationOptions.SCOPE_OBJECT.equals(scope)) {
      return SearchControls.OBJECT_SCOPE;
    } else if (OperationOptions.SCOPE_ONE_LEVEL.equals(scope)) {
      return SearchControls.ONELEVEL_SCOPE;
    } else if (OperationOptions.SCOPE_SUBTREE.equals(scope) || scope == null) {
      return SearchControls.SUBTREE_SCOPE;
    } else {
      throw new IllegalArgumentException("Invalid search scope " + scope);
    }
  }
Пример #4
0
 public static String getSearchFilter(OperationOptions options) {
   return (String) options.getOptions().get(SEARCH_FILTER_NAME);
 }
Пример #5
0
 public static String[] getLdapUidAttributes(OperationOptions options) {
   return (String[]) options.getOptions().get(LDAP_UID_ATTRS_NAME);
 }
Пример #6
0
  public void executeQuery(
      ObjectClass objectClass, Filter query, ResultsHandler handler, OperationOptions options) {

    SortKey[] sortKeys = options.getSortKeys();
    if (null == sortKeys) {
      sortKeys = new SortKey[] {new SortKey(Name.NAME, true)};
    }

    // Rebuild the full result set.
    TreeSet<ConnectorObject> resultSet =
        new TreeSet<ConnectorObject>(new ResourceComparator(sortKeys));

    if (null != query) {
      for (ConnectorObject co : collection.values()) {
        if (query.accept(co)) {
          resultSet.add(co);
        }
      }
    } else {
      resultSet.addAll(collection.values());
    }

    // Handle the results
    if (null != options.getPageSize()) {
      // Paged Search
      final String pagedResultsCookie = options.getPagedResultsCookie();
      String currentPagedResultsCookie = options.getPagedResultsCookie();
      final Integer pagedResultsOffset =
          null != options.getPagedResultsOffset()
              ? Math.max(0, options.getPagedResultsOffset())
              : 0;
      final Integer pageSize = options.getPageSize();

      int index = 0;
      int pageStartIndex = null == pagedResultsCookie ? 0 : -1;
      int handled = 0;

      for (ConnectorObject entry : resultSet) {
        if (pageStartIndex < 0 && pagedResultsCookie.equals(entry.getName().getNameValue())) {
          pageStartIndex = index + 1;
        }

        if (pageStartIndex < 0 || index < pageStartIndex) {
          index++;
          continue;
        }

        if (handled >= pageSize) {
          break;
        }

        if (index >= pagedResultsOffset + pageStartIndex) {
          if (handler.handle(entry)) {
            handled++;
            currentPagedResultsCookie = entry.getName().getNameValue();
          } else {
            break;
          }
        }
        index++;
      }

      if (index == resultSet.size()) {
        currentPagedResultsCookie = null;
      }

      if (handler instanceof SearchResultsHandler) {
        ((SearchResultsHandler) handler)
            .handleResult(new SearchResult(currentPagedResultsCookie, resultSet.size() - index));
      }
    } else {
      // Normal Search
      for (ConnectorObject entry : resultSet) {
        if (!handler.handle(entry)) {
          break;
        }
      }
      if (handler instanceof SearchResultsHandler) {
        ((SearchResultsHandler) handler).handleResult(new SearchResult());
      }
    }
  }