57 lines
2.3 KiB
Java
57 lines
2.3 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.common.ErrorCode;
|
||
|
|
import cn.wildfirechat.proto.ProtoConstants;
|
||
|
|
import cn.wildfirechat.proto.WFCMessage;
|
||
|
|
import com.hazelcast.util.StringUtil;
|
||
|
|
import io.moquette.BrokerConstants;
|
||
|
|
import io.moquette.spi.impl.Qos1PublishHandler;
|
||
|
|
import io.netty.buffer.ByteBuf;
|
||
|
|
import win.liyufan.im.IMTopic;
|
||
|
|
import win.liyufan.im.MessageShardingUtil;
|
||
|
|
|
||
|
|
import java.util.Set;
|
||
|
|
|
||
|
|
import static cn.wildfirechat.proto.ProtoConstants.ContentType.Text;
|
||
|
|
|
||
|
|
@Handler(value = IMTopic.BroadcastMessageTopic)
|
||
|
|
public class BroadcastMessageHandler extends IMHandler<WFCMessage.Message> {
|
||
|
|
@Override
|
||
|
|
public ErrorCode action(ByteBuf ackPayload, String clientID, String fromUser, ProtoConstants.RequestSourceType requestSourceType, WFCMessage.Message message, Qos1PublishHandler.IMCallback callback) {
|
||
|
|
ErrorCode errorCode = ErrorCode.ERROR_CODE_SUCCESS;
|
||
|
|
boolean isAdmin = requestSourceType == ProtoConstants.RequestSourceType.Request_From_Admin;
|
||
|
|
if (message != null) {
|
||
|
|
if (!isAdmin) { //only admin can broadcast message
|
||
|
|
return ErrorCode.ERROR_CODE_NOT_RIGHT;
|
||
|
|
}
|
||
|
|
|
||
|
|
long timestamp = System.currentTimeMillis();
|
||
|
|
long messageId = 0;
|
||
|
|
try {
|
||
|
|
messageId = MessageShardingUtil.generateId();
|
||
|
|
} catch (Exception e) {
|
||
|
|
e.printStackTrace();
|
||
|
|
return ErrorCode.ERROR_CODE_SERVER_ERROR;
|
||
|
|
}
|
||
|
|
message = message.toBuilder().setFromUser(fromUser).setMessageId(messageId).setServerTimestamp(timestamp).setConversation(WFCMessage.Conversation.newBuilder().setTarget(fromUser).setLine(message.getConversation().getLine()).setType(ProtoConstants.ConversationType.ConversationType_Private).build()).build();
|
||
|
|
|
||
|
|
long count = saveAndBroadcast(fromUser, clientID, message);
|
||
|
|
ackPayload = ackPayload.capacity(20);
|
||
|
|
ackPayload.writeLong(messageId);
|
||
|
|
ackPayload.writeLong(count);
|
||
|
|
} else {
|
||
|
|
errorCode = ErrorCode.ERROR_CODE_INVALID_MESSAGE;
|
||
|
|
}
|
||
|
|
return errorCode;
|
||
|
|
}
|
||
|
|
|
||
|
|
}
|