/*
* Licensed to the Apache Software Foundation (ASF) under one or more
* contributor license agreements. See the NOTICE file distributed with
* this work for additional information regarding copyright ownership.
* The ASF licenses this file to You under the Apache License, Version 2.0
* (the "License"); you may not use this file except in compliance with
* the License. You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
package org.apache.usergrid.rest.applications.queues;
import com.fasterxml.jackson.jaxrs.json.annotation.JSONP;
import org.apache.commons.lang.StringUtils;
import org.apache.usergrid.mq.QueueManager;
import org.apache.usergrid.mq.QueueSet;
import org.apache.usergrid.rest.AbstractContextResource;
import org.apache.usergrid.rest.security.annotations.CheckPermissionsForPath;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.context.annotation.Scope;
import org.springframework.stereotype.Component;
import javax.ws.rs.*;
import javax.ws.rs.core.Context;
import javax.ws.rs.core.MediaType;
import javax.ws.rs.core.UriInfo;
import java.util.List;
import java.util.Map;
@Component
@Scope("prototype")
@Produces({
MediaType.APPLICATION_JSON, "application/javascript", "application/x-javascript", "text/ecmascript",
"application/ecmascript", "text/jscript"
})
public class QueueSubscriptionResource extends AbstractContextResource {
static final Logger logger = LoggerFactory.getLogger( QueueSubscriptionResource.class );
QueueManager mq;
String queuePath = "";
String subscriptionPath = "";
public QueueSubscriptionResource() {
}
public QueueSubscriptionResource init( QueueManager mq, String queuePath ) {
this.mq = mq;
this.queuePath = queuePath;
return this;
}
public QueueSubscriptionResource init( QueueManager mq, String queuePath, String subscriptionPath )
throws Exception {
this.mq = mq;
this.queuePath = queuePath;
this.subscriptionPath = subscriptionPath;
return this;
}
@Path("{subPath}")
public QueueSubscriptionResource getSubPath( @Context UriInfo ui, @PathParam("subPath") String subPath )
throws Exception {
if (logger.isTraceEnabled()) {
logger.info("QueueSubscriptionResource.getSubPath");
}
return getSubResource( QueueSubscriptionResource.class )
.init( mq, queuePath, subscriptionPath + "/" + subPath );
}
@CheckPermissionsForPath
@GET
@JSONP
@Produces({MediaType.APPLICATION_JSON, "application/javascript"})
public QueueSet executeGet( @Context UriInfo ui, @QueryParam("start") String firstSubscriptionQueuePath,
@QueryParam("limit") @DefaultValue("10") int limit,
@QueryParam("callback") @DefaultValue("callback") String callback )
throws Exception {
if (logger.isTraceEnabled()) {
logger.trace("QueueSubscriptionResource.executeGet: {}", queuePath);
}
QueueSet results = mq.getSubscriptions( queuePath, firstSubscriptionQueuePath, limit );
return results;
}
@CheckPermissionsForPath
@POST
@Consumes(MediaType.APPLICATION_JSON)
@JSONP
@Produces({MediaType.APPLICATION_JSON, "application/javascript"})
public QueueSet executePost( @Context UriInfo ui, Map<String, Object> body,
@QueryParam("callback") @DefaultValue("callback") String callback )
throws Exception {
if (logger.isTraceEnabled()) {
logger.trace("QueueSubscriptionResource.executePost: {}", queuePath);
}
return executePut( ui, body, callback );
}
@CheckPermissionsForPath
@PUT
@Consumes(MediaType.APPLICATION_JSON)
@JSONP
@Produces({MediaType.APPLICATION_JSON, "application/javascript"})
public QueueSet executePut( @Context UriInfo ui, Map<String, Object> body,
@QueryParam("callback") @DefaultValue("callback") String callback )
throws Exception {
if (logger.isTraceEnabled()) {
logger.trace("QueueSubscriptionResource.executePut: {}", queuePath);
}
Map<String, Object> json = body;
if ( StringUtils.isNotBlank( subscriptionPath ) ) {
return mq.subscribeToQueue( subscriptionPath, queuePath );
}
else if ( ( json != null ) && ( json.containsKey( "subscriber" ) ) ) {
String supscription = ( String ) json.get( "supscription" );
return mq.subscribeToQueue( supscription, queuePath );
}
else if ( ( json != null ) && ( json.containsKey( "subscribers" ) ) ) {
@SuppressWarnings("unchecked") List<String> supscriptions = ( List<String> ) json.get( "supscriptions" );
return mq.unsubscribeFromQueues( queuePath, supscriptions );
}
return null;
}
@CheckPermissionsForPath
@DELETE
@JSONP
@Produces({MediaType.APPLICATION_JSON, "application/javascript"})
public QueueSet executeDelete( @Context UriInfo ui,
@QueryParam("callback") @DefaultValue("callback") String callback )
throws Exception {
if (logger.isTraceEnabled()) {
logger.trace("QueueSubscriptionResource.executeDelete: {}", queuePath);
}
if ( StringUtils.isNotBlank( subscriptionPath ) ) {
return mq.unsubscribeFromQueue( subscriptionPath, queuePath );
}
return null;
}
}