Java ReceiveMessageRequest.setMaxNumberOfMessages方法代碼示例

import com.amazonaws.services.sqs.model.ReceiveMessageRequest; //導入方法依賴的package包/類
@SuppressWarnings("unchecked")
public T peekMessage(int waitFor) throws Exception {
       // Receive messages
	logger.info("Trying to recieve message from: " + _queueName);
       ReceiveMessageRequest receiveMessageRequest = new ReceiveMessageRequest(_queueURL);
       receiveMessageRequest.setMaxNumberOfMessages(1);
       receiveMessageRequest.setWaitTimeSeconds(waitFor);
       
       List<Message> messages = _sqs.receiveMessage(receiveMessageRequest).getMessages();
       
       for (Message message : messages) {
       	logger.info("  Got Message");
       	logger.info("    Body:          " + message.getBody());
       	logger.info("    Handle:        " + message.getReceiptHandle());
           
           _lastMessage = message;
           GenericMessage msg =  GenericMessage.fromXML(message.getBody());
   		if (!msg.type.equals(_msgClass.getName()))
   			throw new Exception("Invalid message type recieved.");
   		
           return (T) msg.body;
       }
	return null;
}
 

import com.amazonaws.services.sqs.model.ReceiveMessageRequest; //導入方法依賴的package包/類
public static String popFrom(String name) {
	try {
		String queueUrl = getConnection().createQueue(
				new CreateQueueRequest(name)).getQueueUrl();
		ReceiveMessageRequest receiveMessageRequest = new ReceiveMessageRequest(
				queueUrl);
		receiveMessageRequest.setMaxNumberOfMessages(1);
		if (null != receiveMessageRequest) {
			List<Message> messages = getConnection().receiveMessage(
					receiveMessageRequest).getMessages();
			if (messages.size() > 0) {
				String messageRecieptHandle = messages.get(0)
						.getReceiptHandle();
				getConnection().deleteMessage(
						new DeleteMessageRequest(receiveMessageRequest
								.getQueueUrl(), messageRecieptHandle));
				return messages.get(0).getBody();
			}
		}
	} catch (Exception e) {
		e.printStackTrace();
	}
	return null;
}
 

import com.amazonaws.services.sqs.model.ReceiveMessageRequest; //導入方法依賴的package包/類
@Override
public ReceiveMessageRequest createReceiveMessageRequest(String queueUrl, final int maxNumberMessages, final int waitTimeSeconds) {
    final ReceiveMessageRequest request = new ReceiveMessageRequest(queueUrl);
    request.setMaxNumberOfMessages(maxNumberMessages);
    request.setWaitTimeSeconds(waitTimeSeconds);
    return request;
}
 

import com.amazonaws.services.sqs.model.ReceiveMessageRequest; //導入方法依賴的package包/類
@Override
public void run() {
    String queueUrl = client.getQueueUrl(queue).getQueueUrl();

    ReceiveMessageRequest receiveRequest = new ReceiveMessageRequest(queueUrl);
    receiveRequest.setMaxNumberOfMessages(10);

    while (true) {
        ReceiveMessageResult receiveResult= client.receiveMessage(receiveRequest);
        List<Message> messages = receiveResult.getMessages();

        messages.forEach(System.out::println);
    }
}
 

import com.amazonaws.services.sqs.model.ReceiveMessageRequest; //導入方法依賴的package包/類
@Override
protected int poll() throws Exception {
    // must reset for each poll
    shutdownRunningTask = null;
    pendingExchanges = 0;
    
    ReceiveMessageRequest request = new ReceiveMessageRequest(getQueueUrl());
    request.setMaxNumberOfMessages(getMaxMessagesPerPoll() > 0 ? getMaxMessagesPerPoll() : null);
    request.setVisibilityTimeout(getConfiguration().getVisibilityTimeout() != null ? getConfiguration().getVisibilityTimeout() : null);
    request.setWaitTimeSeconds(getConfiguration().getWaitTimeSeconds() != null ? getConfiguration().getWaitTimeSeconds() : null);

    if (attributeNames != null) {
        request.setAttributeNames(attributeNames);
    }
    if (messageAttributeNames != null) {
        request.setMessageAttributeNames(messageAttributeNames);
    }

    LOG.trace("Receiving messages with request [{}]...", request);
    
    ReceiveMessageResult messageResult = null;
    try {
        messageResult = getClient().receiveMessage(request);
    } catch (QueueDoesNotExistException e) {
        LOG.info("Queue does not exist....recreating now...");
        reConnectToQueue();
        messageResult = getClient().receiveMessage(request);
    }

    if (LOG.isTraceEnabled()) {
        LOG.trace("Received {} messages", messageResult.getMessages().size());
    }
    
    Queue<Exchange> exchanges = createExchanges(messageResult.getMessages());
    return processBatch(CastUtils.cast(exchanges));
}
 

import com.amazonaws.services.sqs.model.ReceiveMessageRequest; //導入方法依賴的package包/類
@Test
public void shouldGetMessages10per10WithAttributes() throws Exception {
    //GIVEN
    when(sqs.receiveMessage((ReceiveMessageRequest) anyObject())).thenReturn(new ReceiveMessageResult());

    //WHEN
    receiverHandler.start();

    //THEN
    Thread.sleep(100);
    ReceiveMessageRequest expected = new ReceiveMessageRequest(queueUrl).withMessageAttributeNames("All");
    expected.setMaxNumberOfMessages(10);
    verify(sqs, atLeastOnce()).receiveMessage(expected);
    verify(sqs, never()).receiveMessage(new ReceiveMessageRequest(queueUrl));
}
 

import com.amazonaws.services.sqs.model.ReceiveMessageRequest; //導入方法依賴的package包/類
/**
 * Poll SQS queue for incoming messages, filter them, and return a list of SQS Messages.
 *
 * @return a list of SQS messages.
 */
public List<Message> pollQueue() {
    boolean success = false;
    ProgressStatus pollQueueStatus = new ProgressStatus(ProgressState.pollQueue, new BasicPollQueueInfo(0, success));
    final Object reportObject = progressReporter.reportStart(pollQueueStatus);

    ReceiveMessageRequest request = new ReceiveMessageRequest().withAttributeNames(ALL_ATTRIBUTES);
    request.setQueueUrl(config.getSqsUrl());
    request.setVisibilityTimeout(config.getVisibilityTimeout());
    request.setMaxNumberOfMessages(DEFAULT_SQS_MESSAGE_SIZE_LIMIT);
    request.setWaitTimeSeconds(DEFAULT_WAIT_TIME_SECONDS);

    List<Message> sqsMessages = new ArrayList<Message>();
    try {

        ReceiveMessageResult result = sqsClient.receiveMessage(request);
        sqsMessages = result.getMessages();
        logger.info("Polled " + sqsMessages.size() + " sqs messages from " + config.getSqsUrl());

        success = true;
    } catch (AmazonServiceException e) {
        LibraryUtils.handleException(exceptionHandler, pollQueueStatus, e, "Failed to poll sqs message.");

    } finally {
        LibraryUtils.endToProcess(progressReporter, success, pollQueueStatus, reportObject);
    }
    return sqsMessages;

}
 

import com.amazonaws.services.sqs.model.ReceiveMessageRequest; //導入方法依賴的package包/類
public List<CloudtrailSNSNotification> getNotifications() {
    LOG.debug("Fetching SQS CloudTrail notifications.");

    List<CloudtrailSNSNotification> notifications = Lists.newArrayList();

    ReceiveMessageRequest request = new ReceiveMessageRequest(queueName);
    request.setMaxNumberOfMessages(10);
    ReceiveMessageResult result = sqs.receiveMessage(request);

    LOG.debug("Received [{}] SQS CloudTrail notifications.", result.getMessages().size());

    CloudtrailSNSNotificationParser parser = new CloudtrailSNSNotificationParser(objectMapper);

    for (Message message : result.getMessages()) {
        notifications.addAll(parser.parse(message));
    }

    return notifications;
}