Java DeliveryState类代码示例

import org.apache.qpid.proton.amqp.transport.DeliveryState; //导入依赖的package包/类
/**
 * Verifies that the endpoint rejects malformed request messages.
 */
@Test
public void testHandleMessageRejectsMalformedMessage() {

    Message msg = ProtonHelper.message();
    ProtonConnection con = mock(ProtonConnection.class);
    ProtonDelivery delivery = mock(ProtonDelivery.class);
    RequestResponseEndpoint<ServiceConfigProperties> endpoint = getEndpoint(false);

    // WHEN a malformed message is received
    endpoint.handleMessage(con, receiver, resource, delivery, msg);

    // THEN the link is closed and the message is rejected
    ArgumentCaptor<DeliveryState> deliveryState = ArgumentCaptor.forClass(DeliveryState.class);
    verify(delivery).disposition(deliveryState.capture(), booleanThat(is(Boolean.TRUE)));
    assertThat(deliveryState.getValue(), instanceOf(Rejected.class));
    verify(receiver, never()).close();
}
 

import org.apache.qpid.proton.amqp.transport.DeliveryState; //导入依赖的package包/类
/**
 * Verifies that the endpoint rejects request messages for operations the client
 * is not authorized to invoke.
 */
@Test
public void testHandleMessageRejectsUnauthorizedRequests() {

    Message msg = ProtonHelper.message();
    msg.setSubject("unauthorized");
    ProtonConnection con = mock(ProtonConnection.class);
    ProtonDelivery delivery = mock(ProtonDelivery.class);
    AuthorizationService authService = mock(AuthorizationService.class);
    when(authService.isAuthorized(any(HonoUser.class), any(ResourceIdentifier.class), anyString())).thenReturn(Future.succeededFuture(Boolean.FALSE));
    Future<Void> processingTracker = Future.future();
    RequestResponseEndpoint<ServiceConfigProperties> endpoint = getEndpoint(true, processingTracker);
    endpoint.setAuthorizationService(authService);

    // WHEN a request for an operation is received that the client is not authorized to invoke
    endpoint.handleMessage(con, receiver, resource, delivery, msg);

    // THEN the the message is rejected
    ArgumentCaptor<DeliveryState> deliveryState = ArgumentCaptor.forClass(DeliveryState.class);
    verify(delivery).disposition(deliveryState.capture(), booleanThat(is(Boolean.TRUE)));
    assertThat(deliveryState.getValue(), instanceOf(Rejected.class));
    verify(receiver, never()).close();
    verify(authService).isAuthorized(Constants.PRINCIPAL_ANONYMOUS, resource, "unauthorized");
    assertFalse(processingTracker.isComplete());
}
 

import org.apache.qpid.proton.amqp.transport.DeliveryState; //导入依赖的package包/类
/**
 * Verifies that the endpoint processes request messages for operations the client
 * is authorized to invoke.
 */
@Test
public void testHandleMessageProcessesAuthorizedRequests() {

    Message msg = ProtonHelper.message();
    msg.setSubject("get");
    ProtonConnection con = mock(ProtonConnection.class);
    ProtonDelivery delivery = mock(ProtonDelivery.class);
    AuthorizationService authService = mock(AuthorizationService.class);
    when(authService.isAuthorized(any(HonoUser.class), any(ResourceIdentifier.class), anyString())).thenReturn(Future.succeededFuture(Boolean.TRUE));

    Future<Void> processingTracker = Future.future();
    RequestResponseEndpoint<ServiceConfigProperties> endpoint = getEndpoint(true, processingTracker);
    endpoint.setAuthorizationService(authService);

    // WHEN a request for an operation is received that the client is authorized to invoke
    endpoint.handleMessage(con, receiver, resource, delivery, msg);

    // THEN then the message gets processed
    ArgumentCaptor<DeliveryState> deliveryState = ArgumentCaptor.forClass(DeliveryState.class);
    verify(delivery).disposition(deliveryState.capture(), booleanThat(is(Boolean.TRUE)));
    assertThat(deliveryState.getValue(), instanceOf(Accepted.class));
    verify(receiver, never()).close();
    verify(authService).isAuthorized(Constants.PRINCIPAL_ANONYMOUS, resource, "get");
    assertTrue(processingTracker.isComplete());
}
 

import org.apache.qpid.proton.amqp.transport.DeliveryState; //导入依赖的package包/类
void disposition(long sequence, DeliveryState state, int... flags) {
  int flag = flags.length == 1 ? flags[0] : 0;
  boolean cumilative = (flag & CUMULATIVE) != 0;
  boolean settle = (flag & SETTLE) != 0;

  long count = cumilative ? _lastDispositionMark.get() : sequence;
  long end = sequence;

  while (count <= end) {
    Delivery d = _unsettled.get(count);
    if (d != null) {
      d.disposition(state);
    }
    count++;
  }
  _lastDispositionMark.set(end);
  if (settle) {
    settle(sequence, cumilative, false);
  }
  _conn.write();
}
 

import org.apache.qpid.proton.amqp.transport.DeliveryState; //导入依赖的package包/类
void disposition(InboundMessage msg, DeliveryState state, int... flags) throws NetworkException
{
    int flag = flags.length == 1 ? flags[0] : 0;
    boolean cumilative = (flag & CUMULATIVE) != 0;
    boolean settle = (flag & SETTLE) != 0;

    long count = cumilative ? _lastDispositionMark.get() : msg.getSequence();
    long end = msg.getSequence();

    while (count <= end)
    {
        Delivery d = _unsettled.get(count);
        if (d != null)
        {
            d.disposition(state);
        }
        count++;
    }
    _lastDispositionMark.set(end);
    if (settle)
    {
        settle(msg, cumilative, false);
    }
    _conn.write();
}
 

import org.apache.qpid.proton.amqp.transport.DeliveryState; //导入依赖的package包/类
private boolean processTransportWorkReceiver(DeliveryImpl delivery,
                                             ReceiverImpl rcv)
{
    TransportDelivery tpDelivery = delivery.getTransportDelivery();
    SessionImpl session = rcv.getSession();
    TransportSession tpSession = session.getTransportSession();

    if (tpSession.isLocalChannelSet())
    {
        boolean settled = delivery.isSettled();
        DeliveryState localState = delivery.getLocalState();

        Disposition disposition = new Disposition();
        disposition.setFirst(tpDelivery.getDeliveryId());
        disposition.setLast(tpDelivery.getDeliveryId());
        disposition.setRole(Role.RECEIVER);
        disposition.setSettled(settled);
        disposition.setState(localState);

        if(localState == null && settled) {
            disposition.setState(delivery.getDefaultDeliveryState());
        }

        writeFrame(tpSession.getLocalChannel(), disposition, null, null);
        if (settled)
        {
            tpDelivery.settled();
        }
        return true;
    }

    return false;
}
 

import org.apache.qpid.proton.amqp.transport.DeliveryState; //导入依赖的package包/类
@Override
public void disposition(final DeliveryState state)
{
    _deliveryState = state;
    if(!_remoteSettled)
    {
        addToTransportWorkList();
    }
}
 

import org.apache.qpid.proton.amqp.transport.DeliveryState; //导入依赖的package包/类
public Disposition newInstance(Object described)
{
    List l = (List) described;

    Disposition o = new Disposition();

    if(l.isEmpty())
    {
        throw new DecodeException("The first field cannot be omitted");
    }

    switch(6 - l.size())
    {

        case 0:
            Boolean batchable = (Boolean) l.get(5);
            o.setBatchable(batchable == null ? false : batchable);
        case 1:
            o.setState( (DeliveryState) l.get( 4 ) );
        case 2:
            Boolean settled = (Boolean) l.get(3);
            o.setSettled(settled == null ? false : settled);
        case 3:
            o.setLast( (UnsignedInteger) l.get( 2 ) );
        case 4:
            o.setFirst( (UnsignedInteger) l.get( 1 ) );
        case 5:
            o.setRole( Boolean.TRUE.equals(l.get( 0 )) ? Role.RECEIVER : Role.SENDER );
    }


    return o;
}
 

import org.apache.qpid.proton.amqp.transport.DeliveryState; //导入依赖的package包/类
@Override
public ProtonDelivery disposition(DeliveryState state, boolean settle) {
  delivery.disposition(state);
  if (settle) {
    settle();
  } else {
    flushConnection();
  }

  return this;
}
 

import org.apache.qpid.proton.amqp.transport.DeliveryState; //导入依赖的package包/类
protected void settleDelivery(Delivery incoming, DeliveryState state) {
    incoming.disposition(state);
    incoming.settle();
    // TODO: this flows credit, which we might not want, e.g if
    // a drain was issued to stop the link.
    sendFlowIfNeeded();
}
 

import org.apache.qpid.proton.amqp.transport.DeliveryState; //导入依赖的package包/类
@Override
public void processDeliveryUpdates() {
    List<Delivery> toRemove = new ArrayList<Delivery>();

    for (Delivery delivery : pending) {
        DeliveryState state = delivery.getRemoteState();
        if (state == null) {
            continue;
        }

        @SuppressWarnings("unchecked")
        AsyncResult<Void> request = (AsyncResult<Void>) delivery.getContext();

        if (state instanceof Accepted) {
            toRemove.add(delivery);
            LOG.trace("State of delivery accepted: {}", delivery);
            tagGenerator.returnTag(delivery.getTag());
            if (request != null && !request.isComplete()) {
                request.onSuccess();
            }
        } else if (state instanceof Rejected) {
            Exception remoteError = getRemoteError();
            toRemove.add(delivery);
            tagGenerator.returnTag(delivery.getTag());
            if (request != null && !request.isComplete()) {
                request.onFailure(remoteError);
            } else {
                connection.getProvider().fireProviderException(remoteError);
            }
        } else if (state instanceof TransactionalState) {
            LOG.info("State of delivery is Transacted: {}", state);
        } else {
            LOG.warn("Message send updated with unsupported state: {}", state);
        }
    }

    pending.removeAll(toRemove);
}
 

import org.apache.qpid.proton.amqp.transport.DeliveryState; //导入依赖的package包/类
@Override
public DeliveryState getLocalState()
{
    return _deliveryState;
}
 

import org.apache.qpid.proton.amqp.transport.DeliveryState; //导入依赖的package包/类
@Override
public DeliveryState getRemoteState()
{
    return _remoteDeliveryState;
}
 

import org.apache.qpid.proton.amqp.transport.DeliveryState; //导入依赖的package包/类
void setRemoteDeliveryState(DeliveryState remoteDeliveryState)
{
    _remoteDeliveryState = remoteDeliveryState;
    _updated = true;
}
 

import org.apache.qpid.proton.amqp.transport.DeliveryState; //导入依赖的package包/类
@Override
public void setDefaultDeliveryState(DeliveryState state)
{
    _defaultDeliveryState = state;
}