项目：dropwizard-sqs-bundle
文件：SqsBundleHealthCheck.java
@Override
protected Result check() throws Exception {
try {
ListQueuesResult listQueuesResult = sqs.listQueues();
if (listQueuesResult != null) {
return Result.healthy("OK");
}
else {
return Result.unhealthy("Could not fetch queues list from AWS");
}
} catch (AmazonClientException e) {
LOGGER.error(e.getMessage(),e);
return Result.unhealthy("Could not reach AWS to list queues");
}
}
项目：queue-slayer
文件：AmazonSQSPrioritizedMessageProviderTest.java
@Test
public void oneQueue() {
AmazonSQS amazonSQS = mock(AmazonSQS.class);
// return one queue
when(amazonSQS.listQueues(any(ListQueuesRequest.class)))
.thenReturn(new ListQueuesResult().withQueueUrls("test-foo"));
// return 3 messages from the queue
when(amazonSQS.receiveMessage(any(ReceiveMessageRequest.class)))
.thenReturn(new ReceiveMessageResult().withMessages(newMessage("foo"),newMessage("foo"),newMessage("foo")));
AmazonSQSPrioritizedMessageProvider provider = new AmazonSQSPrioritizedMessageProvider(amazonSQS,"test",1,60 * 1000);
List<Message> messages = provider.next();
assertMessages(messages,3,"foo");
verify(amazonSQS).listQueues(any(ListQueuesRequest.class));
verify(amazonSQS).receiveMessage(any(ReceiveMessageRequest.class));
}
项目：unitstack
文件：MockSqsTest.java
@Test
public void testCreateGetUrlListQueue_shouldCreateReturnUrlAndListQueue() {
// create first queue
CreateQueueResult createdQueue = sqs.createQueue(new CreateQueueRequest().withQueueName("tea-earl-grey-queue"));
assertNotNull("verify that,on creation,queue url was returned",createdQueue.getQueueUrl());
// create other queues
CreateQueueResult secondTeaQueue = sqs.createQueue(new CreateQueueRequest().withQueueName("tea-mate-queue"));
CreateQueueResult anotherQueue = sqs.createQueue(new CreateQueueRequest().withQueueName("coffee-queue"));
// get queue url
GetQueueUrlResult queueUrlResult = sqs.getQueueUrl(new GetQueueUrlRequest()
.withQueueName("tea-earl-grey-queue").withQueueOwnerAWSAccountId("some owner"));
assertNotNull("verify that,on fetch,queueUrlResult.getQueueUrl());
// get all queues
ListQueuesResult allQueues = sqs.listQueues();
assertEquals("verify all queues are returned",allQueues.getQueueUrls().size());
assertTrue("verify that all queues contain first queue",allQueues.getQueueUrls().contains(createdQueue.getQueueUrl()));
assertTrue("verify that all queues contain second tea queue",allQueues.getQueueUrls().contains(secondTeaQueue.getQueueUrl()));
assertTrue("verify that all queues contain coffee queue",allQueues.getQueueUrls().contains(anotherQueue.getQueueUrl()));
// get only queues that start with 'tea'
ListQueuesResult teaQueues = sqs.listQueues(new ListQueuesRequest("tea"));
assertEquals("verify only tea queues are returned",2,teaQueues.getQueueUrls().size());
assertTrue("verify that tea queues contain first queue",teaQueues.getQueueUrls().contains(createdQueue.getQueueUrl()));
assertTrue("verify that tea queues contain second tea queue",teaQueues.getQueueUrls().contains(secondTeaQueue.getQueueUrl()));
assertNotNull("verify that delete queue returned ok",sqs.deleteQueue(new DeleteQueueRequest().withQueueUrl(queueUrlResult.getQueueUrl())));
assertFalse("verify that the queue was removed",sqs.listQueues().getQueueUrls().stream()
.anyMatch( queueUrl -> StringUtils.equals(queueUrl,queueUrlResult.getQueueUrl()) ));
// cleanup
getQueues().remove("tea-earl-grey-queue");
getQueues().remove("tea-mate-queue");
getQueues().remove("coffee-queue");
}
项目：Camel
文件：SqsEndpointTest.java
@Test
public void doStartShouldNotCallUpdateQueueAttributesIfQueueExistAndNoOptionIsSpecified() throws Exception {
EasyMock.expect(amazonSQSClient.listQueues())
.andReturn(new ListQueuesResult().withQueueUrls("https://sqs.us-east-1.amazonaws.com/ID/dummy-queue","https://sqs.us-east-1.amazonaws.com/ID/test-queue"));
EasyMock.replay(amazonSQSClient);
endpoint.doStart();
EasyMock.verify(amazonSQSClient);
}
项目：Camel
文件：SqsEndpointUseExistingQueueTest.java
@Override
public ListQueuesResult listQueues() throws AmazonServiceException,AmazonClientException {
ListQueuesResult result = new ListQueuesResult();
result.getQueueUrls().add("http://queue.amazonaws.com/0815/Foo");
result.getQueueUrls().add("http://queue.amazonaws.com/0815/MyQueue");
result.getQueueUrls().add("http://queue.amazonaws.com/0815/Bar");
return result;
}
项目：apex-malhar
文件：SQSTestBase.java
/**
* Each test creates its own uniquely named queue in SQS and then deletes it afterwards.
* We try to scrub any leftover queues from the previous runs just in case tests were
* aborted
*
* @param currentQueueNamePrefix
*/
public void generateCurrentQueueName(String currentQueueNamePrefix)
{
if (validateTestCreds()) {
ListQueuesResult list = sqs.listQueues(currentQueueNamePrefix);
for (String url : list.getQueueUrls()) {
sqs.deleteQueue(url);
}
}
this.currentQueueName = currentQueueNamePrefix + System.currentTimeMillis();
}
项目：dropwizard-sqs-bundle
文件：SqsBundleHealthCheckTest.java
@Test
public void shouldBeHealthy() throws Exception {
//GIVEN
String queueUrl1 = "https://eu-central-1/queue.amazonaws.com/123456/test-queue";
when(sqs.listQueues()).thenReturn(new ListQueuesResult().withQueueUrls(queueUrl1));
//WHEN
HealthCheck.Result result = sqsBundleHealthCheck.check();
//THEN
assertThat(result.isHealthy()).isTrue();
}
项目：aws-java-sdk-stubs
文件：AmazonSQSStubTest.java
@Test
public void createQueue() {
final String queueName = "bizo";
final CreateQueueRequest createQueueRequest = new CreateQueueRequest().withQueueName(queueName);
sqs.createQueue(createQueueRequest);
final ListQueuesResult listQueuesResult = sqs.listQueues();
assertThat(listQueuesResult.getQueueUrls(),hasItem(containsString(queueName)));
}
项目：s3-bucket-loader
文件：TOCQueue.java
/**
* Note here we attempt to the TOCQueue which may take some time to be shown as available
* @param isConsumer
* @param maxAttempts
* @throws Exception
*/
public void connectToQueue(boolean isConsumer,int maxAttempts) throws Exception{
for (int i=0; i<maxAttempts; i++) {
logger.debug("connectToQueue() attempt: " + (i+1));
ListQueuesResult queuesResult = sqsClient.listQueues();
if (queuesResult != null) {
for (String queueUrl : queuesResult.getQueueUrls()) {
if (queueUrl.indexOf(sqsQueueName) != -1) {
tocQueueUrl = queueUrl;
break;
}
}
}
// if consumer,retry,otherwise is master,so just exit quick to create...
if (tocQueueUrl == null && isConsumer) {
Thread.currentThread().sleep(1000);
continue;
} else {
break; // exit;
}
}
if (tocQueueUrl == null && !isConsumer) {
CreateQueueResult createQueueResult = sqsClient.createQueue(sqsQueueName);
this.tocQueueUrl = createQueueResult.getQueueUrl();
} else if (tocQueueUrl == null) {
throw new Exception("TOCQueue() isConsumer:"+ isConsumer+ " cannot start,sqsQueueName has yet to be created by master?: " + sqsQueueName);
}
}
项目：awslocal
文件：DirectorySQS.java
@Override
public ListQueuesResult listQueues(ListQueuesRequest listQueuesRequest) throws AmazonClientException {
List<String> queueUrls = Lists.newArrayListWithCapacity(_queuesByUrl.size());
try (DirectoryStream<Path> queuePaths = Files.newDirectoryStream(_rootDirectory.toPath())) {
for (Path queuePath : queuePaths) {
if (listQueuesRequest.getQueueNamePrefix() == null || queuePath.getFileName().toString().startsWith(listQueuesRequest.getQueueNamePrefix())) {
queueUrls.add(queuePath.toUri().toString());
}
}
} catch (IOException e) {
throw new AmazonServiceException("could not get queue list",e);
}
return new ListQueuesResult().withQueueUrls(queueUrls);
}
项目：queue-slayer
文件：AmazonSQSPrioritizedMessageProvider.java
private void updateAvailableQueues() {
try {
ListQueuesResult result = sqs.listQueues(new ListQueuesRequest(queuePrefix));
List<String> availableQueues = Lists.newArrayList(Iterables.filter(result.getQueueUrls(),include));
Collections.sort(availableQueues,queueComparator);
messageProviders.clear();
for (String queueUrl : availableQueues) {
messageProviders.add(new AmazonSQSMessageProvider(sqs,queueUrl,waitTimeSeconds));
}
} catch (AmazonClientException e) {
LOG.error("An error occurred while listing SQS queues: {}",e);
}
}
项目：queue-slayer
文件：AmazonSQSPrioritizedMessageProviderTest.java
@Test
public void noQueues() {
AmazonSQS amazonSQS = mock(AmazonSQS.class);
// return an empty list
when(amazonSQS.listQueues(any(ListQueuesRequest.class))).thenReturn(new ListQueuesResult());
AmazonSQSPrioritizedMessageProvider provider = new AmazonSQSPrioritizedMessageProvider(amazonSQS,10 * 1000);
List<Message> empty = provider.next();
Assert.assertEquals(0,empty.size());
verify(amazonSQS).listQueues(any(ListQueuesRequest.class));
}
项目：kolich-aws
文件：KolichSQSClient.java
@Override
public Either<HttpFailure,ListQueuesResult> listQueues() {
return new AwsSQSHttpClosure<ListQueuesResult>(client_,SC_OK,new ListQueuesResultStaxUnmarshaller()) {
@Override
public void prepare(final AwsHttpRequest request) throws Exception {
request.addParameter(SQS_ACTION_PARAM,SQS_ACTION_LIST_QUEUES);
}
}.post();
}
项目：conductor
文件：SQSObservableQueue.java
private List<String> listQueues(String queueName) {
ListQueuesRequest listQueuesRequest = new ListQueuesRequest().withQueueNamePrefix(queueName);
ListQueuesResult resultList = client.listQueues(listQueuesRequest);
List<String> queueUrls = resultList.getQueueUrls().stream().filter(queueUrl -> queueUrl.contains(queueName)).collect(Collectors.toList());
return queueUrls;
}
项目：reactive-sqs-client
文件：ReactiveSqsClient.java
public Observable<ListQueuesResult> listQueuesAsync(ListQueuesRequest request) {
return Observable.from(sqsClient.listQueuesAsync(request));
}
项目：reactive-sqs-client
文件：ReactiveSqsClient.java
public Observable<ListQueuesResult> listQueuesAsync() {
return Observable.from(sqsClient.listQueuesAsync());
}
项目：reactive-sqs-client
文件：ReactiveSqsClient.java
public Observable<ListQueuesResult> listQueuesAsync(String queueNamePrefix) {
return Observable.from(sqsClient.listQueuesAsync(queueNamePrefix));
}
项目：Camel
文件：SqsEndpoint.java
@Override
protected void doStart() throws Exception {
client = getConfiguration().getAmazonSQSClient() != null
? getConfiguration().getAmazonSQSClient() : getClient();
// Override the endpoint location
if (ObjectHelper.isNotEmpty(getConfiguration().getAmazonSQSEndpoint())) {
client.setEndpoint(getConfiguration().getAmazonSQSEndpoint());
}
// check the setting the headerFilterStrategy
if (headerFilterStrategy == null) {
headerFilterStrategy = new SqsHeaderFilterStrategy();
}
// If both region and Account ID is provided the queue URL can be built manually.
// This allows accessing queues where you don't have permission to list queues or query queues
if (configuration.getRegion() != null && configuration.getQueueOwnerAWSAccountId() != null) {
queueUrl = "https://sqs." + configuration.getRegion() + ".amazonaws.com/"
+ configuration.getQueueOwnerAWSAccountId() + "/" + configuration.getQueueName();
} else if (configuration.getQueueOwnerAWSAccountId() != null) {
GetQueueUrlRequest getQueueUrlRequest = new GetQueueUrlRequest();
getQueueUrlRequest.setQueueName(configuration.getQueueName());
getQueueUrlRequest.setQueueOwnerAWSAccountId(configuration.getQueueOwnerAWSAccountId());
GetQueueUrlResult getQueueUrlResult = client.getQueueUrl(getQueueUrlRequest);
queueUrl = getQueueUrlResult.getQueueUrl();
} else {
// check whether the queue already exists
ListQueuesResult listQueuesResult = client.listQueues();
for (String url : listQueuesResult.getQueueUrls()) {
if (url.endsWith("/" + configuration.getQueueName())) {
queueUrl = url;
LOG.trace("Queue available at '{}'.",queueUrl);
break;
}
}
}
if (queueUrl == null) {
createQueue(client);
} else {
updateQueueAttributes(client);
}
}
项目：Camel
文件：AmazonSQSClientMock.java
@Override
public ListQueuesResult listQueues() throws AmazonServiceException,AmazonClientException {
ListQueuesResult result = new ListQueuesResult();
return result;
}
项目：cloudml
文件：BeanstalkConnector.java
public List<String> listQueues(){
ListQueuesResult result= sqsClient.listQueues();
return result.getQueueUrls();
}
项目：queue-slayer
文件：AmazonSQSPrioritizedMessageProviderTest.java
@Test
public void drainThreeQueues() {
AmazonSQS amazonSQS = mock(AmazonSQS.class);
// return three queues
when(amazonSQS.listQueues(any(ListQueuesRequest.class)))
.thenReturn(new ListQueuesResult().withQueueUrls("test-A","test-C","test-B","test-D"));
// each queue has N messages to return and then 0 messages
when(amazonSQS.receiveMessage(any(ReceiveMessageRequest.class)))
.thenAnswer(new Answer<Object>() {
// fake that a queue has no messages left with flags
boolean aDone = false;
boolean bDone = false;
boolean cDone = false;
@Override
public Object answer(InvocationOnMock invocation) throws Throwable {
ReceiveMessageRequest receiveMessageRequest = (ReceiveMessageRequest) invocation.getArguments()[0];
if (receiveMessageRequest.getQueueUrl().equals("test-A")) {
// return 3 messages for A,then no more messages
if (!aDone) {
aDone = true;
return new ReceiveMessageResult().withMessages(newMessage("A"),newMessage("A"),newMessage("A"));
}
} else if (receiveMessageRequest.getQueueUrl().equals("test-B")) {
// return 4 messages for B,then no more messages
if (!bDone) {
bDone = true;
return new ReceiveMessageResult().withMessages(newMessage("B"),newMessage("B"),newMessage("B"));
}
} else if (receiveMessageRequest.getQueueUrl().equals("test-C")) {
// return 1 message for C,then no more messages
if (!cDone) {
cDone = true;
return new ReceiveMessageResult().withMessages(newMessage("C"));
}
}
// fall through to return 0 messages
return new ReceiveMessageResult().withMessages();
}
});
// verify the order of next objects by counting the number of messages we return
AmazonSQSPrioritizedMessageProvider provider = new AmazonSQSPrioritizedMessageProvider(amazonSQS,TimeUnit.MINUTES.toMillis(30));
// for queues with messages,1 receive returns messages,1 receive returns none and removes the provider
assertMessages(provider.next(),"A");
assertMessages(provider.next(),4,"B");
assertMessages(provider.next(),"C");
// for queues with no messages,"D-but-nothing-really");
// when we have no queues left,there are no messages
assertMessages(provider.next(),"nothing");
assertMessages(provider.next(),"nothing");
// we only checked for queues once here
verify(amazonSQS,times(1)).listQueues(any(ListQueuesRequest.class));
// the total number of receives sent to AWS
verify(amazonSQS,times(7)).receiveMessage(any(ReceiveMessageRequest.class));
}
项目：queue-slayer
文件：AmazonSQSPrioritizedMessageProviderTest.java
@Test
public void updatingQueues() {
AmazonSQS amazonSQS = mock(AmazonSQS.class);
// return queues
when(amazonSQS.listQueues(any(ListQueuesRequest.class)))
.thenReturn(
new ListQueuesResult().withQueueUrls("test-A","test-D"),new ListQueuesResult().withQueueUrls("test-C","test-D"));
// always return messages from these queues
when(amazonSQS.receiveMessage(any(ReceiveMessageRequest.class)))
.thenAnswer(new Answer<Object>() {
@Override
public Object answer(InvocationOnMock invocation) throws Throwable {
ReceiveMessageRequest receiveMessageRequest = (ReceiveMessageRequest) invocation.getArguments()[0];
return new ReceiveMessageResult().withMessages(newMessage(receiveMessageRequest.getQueueUrl()),newMessage(receiveMessageRequest.getQueueUrl()));
}
});
// check for new queues every 30 minutes,should be way longer than this test runs so we can simulate elapsed time
final long intervalNs = TimeUnit.MINUTES.toNanos(30);
// simulate the passage of time
final Iterator<Long> time = Lists.newArrayList(
0L,// start at 0,always an initial update
1 * intervalNs / 2,// update
1 * intervalNs,// no update
1 * intervalNs + (intervalNs / 2),// update
2 * intervalNs,// no update
3 * intervalNs + 1,// should update
3 * intervalNs + 2 // no update
).iterator();
AmazonSQSPrioritizedMessageProvider provider = new AmazonSQSPrioritizedMessageProvider(amazonSQS,TimeUnit.NANOSECONDS.toMillis(intervalNs)) {
@Override
public long currentNanoTime() {
return time.next();
}
};
// no update on constructor
verify(amazonSQS,times(0)).listQueues(any(ListQueuesRequest.class));
// update on first next
provider.next();
verify(amazonSQS,times(1)).listQueues(any(ListQueuesRequest.class));
// no update
provider.next();
verify(amazonSQS,times(1)).listQueues(any(ListQueuesRequest.class));
// update
provider.next();
verify(amazonSQS,times(2)).listQueues(any(ListQueuesRequest.class));
// no update
provider.next();
verify(amazonSQS,times(2)).listQueues(any(ListQueuesRequest.class));
// update
provider.next();
verify(amazonSQS,times(3)).listQueues(any(ListQueuesRequest.class));
// no update
provider.next();
verify(amazonSQS,times(3)).listQueues(any(ListQueuesRequest.class));
}
项目：queue-slayer
文件：AmazonSQSPrioritizedMessageProviderTest.java
@Test
public void filteredQueues() {
AmazonSQS amazonSQS = mock(AmazonSQS.class);
// return queues
when(amazonSQS.listQueues(any(ListQueuesRequest.class)))
.thenReturn(
new ListQueuesResult().withQueueUrls("test-A",newMessage(receiveMessageRequest.getQueueUrl()));
}
});
AmazonSQSPrioritizedMessageProvider provider = new AmazonSQSPrioritizedMessageProvider(amazonSQS,0)
.withInclude(new Predicate<String>() {
@Override
public boolean apply(String input) {
return input.equals("test-B");
}
});
// no update on constructor
verify(amazonSQS,times(0)).listQueues(any(ListQueuesRequest.class));
// update on every next,only test-B messages
assertMessages(provider.next(),"test-B");
verify(amazonSQS,times(1)).listQueues(any(ListQueuesRequest.class));
// update on every next,times(2)).listQueues(any(ListQueuesRequest.class));
// update on every next,should be no messages when there is no B
assertMessages(provider.next(),"nothing");
verify(amazonSQS,times(3)).listQueues(any(ListQueuesRequest.class));
}
项目：queue-slayer
文件：AmazonSQSPrioritizedMessageProviderTest.java
@Test
public void sortedQueuesReverse() {
AmazonSQS amazonSQS = mock(AmazonSQS.class);
// return queues
when(amazonSQS.listQueues(any(ListQueuesRequest.class)))
.thenReturn(
new ListQueuesResult().withQueueUrls("test-A",new ListQueuesResult().withQueueUrls("test-A","test-B"),"test-e","test-D","test-F","test-A"));
// always return messages from these queues
when(amazonSQS.receiveMessage(any(ReceiveMessageRequest.class)))
.thenAnswer(new Answer<Object>() {
@Override
public Object answer(InvocationOnMock invocation) throws Throwable {
ReceiveMessageRequest receiveMessageRequest = (ReceiveMessageRequest) invocation.getArguments()[0];
return new ReceiveMessageResult().withMessages(newMessage(receiveMessageRequest.getQueueUrl()),0)
.withQueueComparator(new Comparator<String>() {
@Override
public int compare(String s1,String s2) {
// reverse sorting
return s2.compareTo(s1);
}
});
// no update on constructor
verify(amazonSQS,only test-D messages
assertMessages(provider.next(),"test-D");
verify(amazonSQS,only test-C messages
assertMessages(provider.next(),"test-C");
verify(amazonSQS,only test-e messages
assertMessages(provider.next(),"test-e");
verify(amazonSQS,times(3)).listQueues(any(ListQueuesRequest.class));
}
项目：queue-slayer
文件：AmazonSQSPrioritizedMessageProviderTest.java
@Test
public void sortedQueuesReverseIgnoreCase() {
AmazonSQS amazonSQS = mock(AmazonSQS.class);
// return queues
when(amazonSQS.listQueues(any(ListQueuesRequest.class)))
.thenReturn(
new ListQueuesResult().withQueueUrls("test-A",String s2) {
// reverse sorting,ignore case
return s2.compareToIgnoreCase(s1);
}
});
// no update on constructor
verify(amazonSQS,only test-F messages
assertMessages(provider.next(),"test-F");
verify(amazonSQS,times(3)).listQueues(any(ListQueuesRequest.class));
}
项目：amazon-sqs-java-extended-client-lib
文件：AmazonSQSExtendedClientBase.java
/**
* <p>
* Returns a list of your queues. The maximum number of queues that can be
* returned is 1000. If you specify a value for the optional
* <code>QueueNamePrefix</code> parameter,only queues with a name beginning
* with the specified value are returned.
* </p>
*
* @param listQueuesRequest
* Container for the necessary parameters to execute the
* ListQueues service method on AmazonSQS.
*
* @return The response from the ListQueues service method,as returned by
* AmazonSQS.
*
*
* @throws AmazonClientException
* If any internal errors are encountered inside the client
* while attempting to make the request or handle the response.
* For example if a network connection is not available.
* @throws AmazonServiceException
* If an error response is returned by AmazonSQS indicating
* either a problem with the data in the request,or a server
* side issue.
*/
public ListQueuesResult listQueues(ListQueuesRequest listQueuesRequest) throws AmazonServiceException,AmazonClientException {
return amazonSqsToBeExtended.listQueues(listQueuesRequest);
}
项目：amazon-sqs-java-extended-client-lib
文件：AmazonSQSExtendedClientBase.java
/**
* <p>
* Returns a list of your queues. The maximum number of queues that can be
* returned is 1000. If you specify a value for the optional
* <code>QueueNamePrefix</code> parameter,only queues with a name beginning
* with the specified value are returned.
* </p>
*
* @return The response from the ListQueues service method,or a server
* side issue.
*/
public ListQueuesResult listQueues() throws AmazonServiceException,AmazonClientException {
return amazonSqsToBeExtended.listQueues();
}
项目：amazon-sqs-java-extended-client-lib
文件：AmazonSQSExtendedClientBase.java
/**
* <p>
* Returns a list of your queues. The maximum number of queues that can be
* returned is 1000. If you specify a value for the optional
* <code>QueueNamePrefix</code> parameter,only queues with a name beginning
* with the specified value are returned.
* </p>
*
* @param queueNamePrefix
* A string to use for filtering the list results. Only those
* queues whose name begins with the specified string are
* returned.
*
* @return The response from the ListQueues service method,or a server
* side issue.
*/
public ListQueuesResult listQueues(String queueNamePrefix) throws AmazonServiceException,AmazonClientException {
return amazonSqsToBeExtended.listQueues(queueNamePrefix);
}
项目：kolich-aws
文件：SQSClient.java
/**
* List all queues.
* @return a {@link ListQueuesResult} representing all queues
*/
public Either<HttpFailure,ListQueuesResult> listQueues();