48 lines
2.1 KiB
Java
48 lines
2.1 KiB
Java
|
/*
|
||
|
|
* This file is part of the Wildfire Chat package.
|
||
|
|
* (c) Heavyrain2012 <[email protected]>
|
||
|
|
*
|
||
|
|
* For the full copyright and license information, please view the LICENSE
|
||
|
|
* file that was distributed with this source code.
|
||
|
|
*/
|
||
|
|
|
||
|
|
package io.moquette.imhandler;
|
||
|
|
|
||
|
|
import cn.wildfirechat.proto.ProtoConstants;
|
||
|
|
import cn.wildfirechat.proto.WFCMessage;
|
||
|
|
import io.moquette.spi.impl.Qos1PublishHandler;
|
||
|
|
import io.netty.buffer.ByteBuf;
|
||
|
|
import cn.wildfirechat.common.ErrorCode;
|
||
|
|
import win.liyufan.im.IMTopic;
|
||
|
|
|
||
|
|
@Handler(value = IMTopic.LoadRemoteMessagesTopic)
|
||
|
|
public class LoadRemoteMessagesHandler extends IMHandler<WFCMessage.LoadRemoteMessages> {
|
||
|
|
@Override
|
||
|
|
public ErrorCode action(ByteBuf ackPayload, String clientID, String fromUser, ProtoConstants.RequestSourceType requestSourceType, WFCMessage.LoadRemoteMessages request, Qos1PublishHandler.IMCallback callback) {
|
||
|
|
ErrorCode errorCode = ErrorCode.ERROR_CODE_SUCCESS;
|
||
|
|
|
||
|
|
long beforeUid = request.getBeforeUid();
|
||
|
|
if (beforeUid == 0) {
|
||
|
|
beforeUid = Long.MAX_VALUE;
|
||
|
|
}
|
||
|
|
|
||
|
|
if (request.getConversation().getType() == ProtoConstants.ConversationType.ConversationType_Group) {
|
||
|
|
if (!m_messagesStore.isMemberInGroup(fromUser, request.getConversation().getTarget())) {
|
||
|
|
return ErrorCode.ERROR_CODE_NOT_IN_GROUP;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
if (request.getConversation().getType() == ProtoConstants.ConversationType.ConversationType_Channel) {
|
||
|
|
if (!m_messagesStore.canSendMessageInChannel(fromUser, request.getConversation().getTarget())) {
|
||
|
|
return ErrorCode.ERROR_CODE_NOT_IN_CHANNEL;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
WFCMessage.PullMessageResult result = m_messagesStore.loadRemoteMessages(fromUser, request.getConversation(), beforeUid, request.getCount(), request.getContentTypeList());
|
||
|
|
byte[] data = result.toByteArray();
|
||
|
|
LOG.info("User {} load message with count({}), payload size({})", fromUser, result.getMessageCount(), data.length);
|
||
|
|
ackPayload.ensureWritable(data.length).writeBytes(data);
|
||
|
|
return errorCode;
|
||
|
|
}
|
||
|
|
}
|