/**
* Copyright 2012 Comcast Corporation
*
* Licensed 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 com.comcast.cqs.controller;
import java.util.Arrays;
import java.util.HashMap;
import java.util.Random;
import javax.servlet.AsyncContext;
import javax.servlet.http.HttpServletResponse;
import org.apache.log4j.Logger;
import com.comcast.cmb.common.model.User;
import com.comcast.cmb.common.persistence.PersistenceFactory;
import com.comcast.cmb.common.util.CMBErrorCodes;
import com.comcast.cmb.common.util.CMBException;
import com.comcast.cmb.common.util.CMBProperties;
import com.comcast.cmb.common.util.Util;
import com.comcast.cqs.io.CQSMessagePopulator;
import com.comcast.cqs.model.CQSMessage;
import com.comcast.cqs.model.CQSQueue;
import com.comcast.cqs.util.CQSConstants;
/**
* Send message
* @author aseem, bwolf, vvenkatraman, baosen
*
*/
public class CQSSendMessageAction extends CQSAction {
protected static Logger logger = Logger.getLogger(CQSSendMessageAction.class);
private static Random rand = new Random();
public CQSSendMessageAction() {
super("SendMessage");
}
@Override
public boolean doAction(User user, AsyncContext asyncContext) throws Exception {
CQSHttpServletRequest request = (CQSHttpServletRequest)asyncContext.getRequest();
HttpServletResponse response = (HttpServletResponse)asyncContext.getResponse();
String messageBody = request.getParameter(CQSConstants.MESSAGE_BODY);
if (messageBody == null) {
throw new CMBException(CMBErrorCodes.MissingParameter, "MessageBody not found");
}
if (messageBody.length() > CMBProperties.getInstance().getCQSMaxMessageSize()) {
throw new CMBException(CMBErrorCodes.InvalidParameterValue, "Value for parameter MessageBody is invalid. Reason: Message body must be shorter than " + CMBProperties.getInstance().getCQSMaxMessageSize() + " bytes");
}
if (!Util.isValidUnicode(messageBody)) {
throw new CMBException(CMBErrorCodes.InvalidMessageContents, "The message contains characters outside the allowed set.");
}
HashMap<String, String> attributes = new HashMap<String, String>();
String delaySecondsReq = request.getParameter(CQSConstants.DELAY_SECONDS);
if (delaySecondsReq != null) {
try {
int delaySeconds = Integer.parseInt(delaySecondsReq);
if (delaySeconds < 0 || delaySeconds > CMBProperties.getInstance().getCQSMaxMessageDelaySeconds()) {
throw new CMBException(CMBErrorCodes.InvalidParameterValue, "DelaySeconds should be from 0 to " + CMBProperties.getInstance().getCQSMaxMessageDelaySeconds());
} else {
attributes.put(CQSConstants.DELAY_SECONDS, "" + delaySeconds);
}
} catch (NumberFormatException e) {
throw new CMBException(CMBErrorCodes.InvalidParameterValue, "DelaySeconds must be integer value");
}
}
attributes.put(CQSConstants.SENDER_ID, user.getUserId());
attributes.put(CQSConstants.SENT_TIMESTAMP, "" + System.currentTimeMillis());
attributes.put(CQSConstants.APPROXIMATE_RECEIVE_COUNT, "0");
attributes.put(CQSConstants.APPROXIMATE_FIRST_RECEIVE_TIMESTAMP, "");
CQSMessage message = new CQSMessage(messageBody, attributes);
CQSQueue queue = CQSCache.getCachedQueue(user, request);
int shard = 0;
String shardNumber = request.getParameter("Shard");
if (shardNumber != null) {
shard = Integer.parseInt(shardNumber);
} else if (queue.getNumberOfShards() > 1) {
shard = rand.nextInt(queue.getNumberOfShards());
}
if (shard < 0 || shard >= queue.getNumberOfShards()) {
throw new CMBException(CMBErrorCodes.InvalidParameterValue, "Shard number " + shard +" exceeds number of available shards in queue (" + queue.getNumberOfShards() + ").");
}
String receiptHandle = PersistenceFactory.getCQSMessagePersistence().sendMessage(queue, shard, message);
if (receiptHandle == null || receiptHandle.isEmpty()) {
throw new CMBException(CMBErrorCodes.InternalError, "Failed to add message to queue");
}
request.setReceiptHandles(Arrays.asList(new String[] {receiptHandle}));
message.setReceiptHandle(receiptHandle);
message.setMessageId(receiptHandle);
String out = CQSMessagePopulator.getSendMessageResponse(message);
writeResponse(out, response);
CQSMonitor.getInstance().addNumberOfMessagesReceived(queue.getRelativeUrl(), 1);
try {
CQSLongPollSender.send(queue.getArn());
} catch (Exception ex) {
logger.warn("event=failed_to_send_longpoll_notification", ex);
}
return true;
}
}