package com.alibaba.ons.api.impl.rocketmq;
import java.util.List;
import java.util.Properties;
import java.util.concurrent.ConcurrentHashMap;
import com.alibaba.ons.api.Message;
import com.alibaba.ons.api.PropertyKeyConst;
import com.alibaba.ons.api.exception.ONSClientException;
import com.alibaba.ons.api.order.ConsumeOrderContext;
import com.alibaba.ons.api.order.MessageOrderListener;
import com.alibaba.ons.api.order.OrderAction;
import com.alibaba.ons.api.order.OrderConsumer;
import com.alibaba.rocketmq.client.consumer.DefaultMQPushConsumer;
import com.alibaba.rocketmq.client.consumer.listener.ConsumeOrderlyContext;
import com.alibaba.rocketmq.client.consumer.listener.ConsumeOrderlyStatus;
import com.alibaba.rocketmq.client.consumer.listener.MessageListenerOrderly;
import com.alibaba.rocketmq.client.exception.MQClientException;
import com.alibaba.rocketmq.common.message.MessageExt;
public class OrderConsumerImpl extends ONSClientAbstract implements OrderConsumer {
private final DefaultMQPushConsumer defaultMQPushConsumer;
private final ConcurrentHashMap<String/* Topic */, MessageOrderListener> subscribeTable =
new ConcurrentHashMap<String, MessageOrderListener>();
public OrderConsumerImpl(final Properties properties) {
super(properties);
this.defaultMQPushConsumer = new DefaultMQPushConsumer();
String consumerGroup = properties.getProperty(PropertyKeyConst.ConsumerId);
if (null == consumerGroup) {
throw new ONSClientException("\'ConsumerGroup\' property is null");
}
this.defaultMQPushConsumer.setConsumerGroup(consumerGroup);
this.defaultMQPushConsumer.setInstanceName(this.buildIntanceName());
}
@Override
public Properties getProperties() {
return this.properties;
}
@Override
public void start() {
this.defaultMQPushConsumer.registerMessageListener(new MessageListenerOrderlyImpl());
try {
this.defaultMQPushConsumer.start();
}
catch (Exception e) {
throw new ONSClientException("defaultMQPushConsumer start exception", e);
}
}
@Override
public void shutdown() {
this.defaultMQPushConsumer.shutdown();
}
class MessageListenerOrderlyImpl implements MessageListenerOrderly {
@Override
public ConsumeOrderlyStatus consumeMessage(List<MessageExt> arg0, ConsumeOrderlyContext arg1) {
MessageExt msgRMQ = arg0.get(0);
Message msg = ONSUtil.msgConvert(msgRMQ);
MessageOrderListener listener = OrderConsumerImpl.this.subscribeTable.get(msg.getTopic());
if (null == listener) {
throw new ONSClientException("MessageOrderListener is null");
}
final ConsumeOrderContext context = new ConsumeOrderContext();
OrderAction action = listener.consume(msg, context);
if (action != null) {
switch (action) {
case ConsumeSuccessfully:
return ConsumeOrderlyStatus.SUCCESS;
case SuspendMoment:
return ConsumeOrderlyStatus.SUSPEND_CURRENT_QUEUE_A_MOMENT;
default:
break;
}
}
return ConsumeOrderlyStatus.SUSPEND_CURRENT_QUEUE_A_MOMENT;
}
}
@Override
public void subscribe(String topic, String subExpression, MessageOrderListener listener) {
if (null == topic) {
throw new ONSClientException("topic is null");
}
if (null == listener) {
throw new ONSClientException("listener is null");
}
try {
this.subscribeTable.put(topic, listener);
this.defaultMQPushConsumer.subscribe(topic, subExpression);
}
catch (MQClientException e) {
throw new ONSClientException("defaultMQPushConsumer subscribe exception", e);
}
}
}