use of org.apache.directory.api.ldap.model.message.SearchResultDone in project directory-ldap-api by apache.
the class SearchCursorImpl method next.
/**
* {@inheritDoc}
*/
@Override
public boolean next() throws LdapException, CursorException {
if (done) {
return false;
}
try {
if (future.isCancelled()) {
response = null;
done = true;
return false;
}
response = future.get(timeout, timeUnit);
} catch (Exception e) {
LdapException ldapException = new LdapException(LdapNetworkConnection.NO_RESPONSE_ERROR, e);
// Send an abandon request
if (!future.isCancelled()) {
future.cancel(true);
}
// close the cursor
try {
close(ldapException);
} catch (IOException ioe) {
throw new LdapException(ioe.getMessage(), ioe);
}
throw ldapException;
}
if (response == null) {
future.cancel(true);
throw new LdapConnectionTimeOutException(LdapNetworkConnection.TIME_OUT_ERROR);
}
done = response instanceof SearchResultDone;
if (done) {
searchDoneResp = (SearchResultDone) response;
response = null;
}
return !done;
}
use of org.apache.directory.api.ldap.model.message.SearchResultDone in project directory-ldap-api by apache.
the class MessageDecorator method getDecorator.
/**
* Gets the decorator associated with a given message
*
* @param codec The LdapApiService to use
* @param decoratedMessage The message to decorate
* @return The decorator instance
*/
public static MessageDecorator<? extends Message> getDecorator(LdapApiService codec, Message decoratedMessage) {
if (decoratedMessage instanceof MessageDecorator) {
return (MessageDecorator<?>) decoratedMessage;
}
MessageDecorator<?> decorator;
switch(decoratedMessage.getType()) {
case ABANDON_REQUEST:
decorator = new AbandonRequestDecorator(codec, (AbandonRequest) decoratedMessage);
break;
case ADD_REQUEST:
decorator = new AddRequestDecorator(codec, (AddRequest) decoratedMessage);
break;
case ADD_RESPONSE:
decorator = new AddResponseDecorator(codec, (AddResponse) decoratedMessage);
break;
case BIND_REQUEST:
decorator = new BindRequestDecorator(codec, (BindRequest) decoratedMessage);
break;
case BIND_RESPONSE:
decorator = new BindResponseDecorator(codec, (BindResponse) decoratedMessage);
break;
case COMPARE_REQUEST:
decorator = new CompareRequestDecorator(codec, (CompareRequest) decoratedMessage);
break;
case COMPARE_RESPONSE:
decorator = new CompareResponseDecorator(codec, (CompareResponse) decoratedMessage);
break;
case DEL_REQUEST:
decorator = new DeleteRequestDecorator(codec, (DeleteRequest) decoratedMessage);
break;
case DEL_RESPONSE:
decorator = new DeleteResponseDecorator(codec, (DeleteResponse) decoratedMessage);
break;
case EXTENDED_REQUEST:
decorator = codec.decorate((ExtendedRequest) decoratedMessage);
break;
case EXTENDED_RESPONSE:
decorator = codec.decorate((ExtendedResponse) decoratedMessage);
break;
case INTERMEDIATE_RESPONSE:
decorator = new IntermediateResponseDecorator(codec, (IntermediateResponse) decoratedMessage);
break;
case MODIFY_REQUEST:
decorator = new ModifyRequestDecorator(codec, (ModifyRequest) decoratedMessage);
break;
case MODIFY_RESPONSE:
decorator = new ModifyResponseDecorator(codec, (ModifyResponse) decoratedMessage);
break;
case MODIFYDN_REQUEST:
decorator = new ModifyDnRequestDecorator(codec, (ModifyDnRequest) decoratedMessage);
break;
case MODIFYDN_RESPONSE:
decorator = new ModifyDnResponseDecorator(codec, (ModifyDnResponse) decoratedMessage);
break;
case SEARCH_REQUEST:
decorator = new SearchRequestDecorator(codec, (SearchRequest) decoratedMessage);
break;
case SEARCH_RESULT_DONE:
decorator = new SearchResultDoneDecorator(codec, (SearchResultDone) decoratedMessage);
break;
case SEARCH_RESULT_ENTRY:
decorator = new SearchResultEntryDecorator(codec, (SearchResultEntry) decoratedMessage);
break;
case SEARCH_RESULT_REFERENCE:
decorator = new SearchResultReferenceDecorator(codec, (SearchResultReference) decoratedMessage);
break;
case UNBIND_REQUEST:
decorator = new UnbindRequestDecorator(codec, (UnbindRequest) decoratedMessage);
break;
default:
return null;
}
Map<String, Control> controls = decoratedMessage.getControls();
if (controls != null) {
for (Control control : controls.values()) {
decorator.addControl(control);
}
}
return decorator;
}
use of org.apache.directory.api.ldap.model.message.SearchResultDone in project directory-ldap-api by apache.
the class EntryCursorImpl method next.
/**
* {@inheritDoc}
*/
@Override
public boolean next() throws LdapException, CursorException {
if (!searchCursor.next()) {
return false;
}
try {
do {
response = searchCursor.get();
if (response == null) {
throw new LdapException(LdapNetworkConnection.TIME_OUT_ERROR);
}
messageId = response.getMessageId();
if (response instanceof SearchResultEntry) {
return true;
}
if (response instanceof SearchResultReference) {
return true;
}
} while (!(response instanceof SearchResultDone));
return false;
} catch (Exception e) {
LdapException ldapException = new LdapException(LdapNetworkConnection.NO_RESPONSE_ERROR);
ldapException.initCause(e);
// close the cursor
try {
close(ldapException);
} catch (IOException ioe) {
throw new LdapException(ioe.getMessage(), ioe);
}
throw ldapException;
}
}
use of org.apache.directory.api.ldap.model.message.SearchResultDone in project directory-ldap-api by apache.
the class SearchResultDoneTest method testResponseWithMatchedDNAttribute.
/**
* Test parsing of a response with MatchedDN attribute
*/
@Test
public void testResponseWithMatchedDNAttribute() {
Dsmlv2ResponseParser parser = null;
try {
parser = new Dsmlv2ResponseParser(getCodec());
parser.setInput(SearchResultDoneTest.class.getResource("response_with_matchedDN_attribute.xml").openStream(), "UTF-8");
parser.parse();
} catch (Exception e) {
fail(e.getMessage());
}
SearchResultDone searchResultDone = ((SearchResponse) parser.getBatchResponse().getCurrentResponse().getDecorated()).getSearchResultDone();
LdapResult ldapResult = searchResultDone.getLdapResult();
assertTrue(ldapResult.getMatchedDn().equals("cn=Bob Rush,ou=Dev,dc=Example,dc=COM"));
}
use of org.apache.directory.api.ldap.model.message.SearchResultDone in project directory-ldap-api by apache.
the class SearchResultDoneTest method testResponseWithResultCode.
/**
* Test parsing of a response with Result Code
*/
@Test
public void testResponseWithResultCode() {
Dsmlv2ResponseParser parser = null;
try {
parser = new Dsmlv2ResponseParser(getCodec());
parser.setInput(SearchResultDoneTest.class.getResource("response_with_result_code.xml").openStream(), "UTF-8");
parser.parse();
} catch (Exception e) {
fail(e.getMessage());
}
SearchResultDone searchResultDone = ((SearchResponse) parser.getBatchResponse().getCurrentResponse().getDecorated()).getSearchResultDone();
LdapResult ldapResult = searchResultDone.getLdapResult();
assertEquals(ResultCodeEnum.PROTOCOL_ERROR, ldapResult.getResultCode());
}
Aggregations