114 lines
4.4 KiB
Java
Executable File
114 lines
4.4 KiB
Java
Executable File
/**
|
||
* Copyright 2018-2028 WindChat Group
|
||
*
|
||
* 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.windchat.im.connector.handler;
|
||
|
||
import java.util.HashMap;
|
||
import java.util.Map;
|
||
|
||
import org.apache.commons.lang3.StringUtils;
|
||
import org.slf4j.Logger;
|
||
import org.slf4j.LoggerFactory;
|
||
|
||
import com.windchat.common.channel.ChannelManager;
|
||
import com.windchat.common.channel.ChannelSession;
|
||
import com.windchat.common.command.Command;
|
||
import com.windchat.common.command.CommandResponse;
|
||
import com.windchat.common.command.RedisCommand;
|
||
import com.windchat.common.constant.CommandConst;
|
||
import com.windchat.common.constant.ErrorCode2;
|
||
import com.windchat.common.constant.RequestAction;
|
||
import com.windchat.common.utils.StringHelper;
|
||
import com.akaxin.proto.core.CoreProto;
|
||
import com.windchat.im.connector.constant.AkxProject;
|
||
import com.windchat.im.message.service.ImMessageService;
|
||
|
||
/**
|
||
* 这里负责下发消息至message模块进行处理
|
||
*
|
||
* @author Sam{@link an.guoyue254@gmail.com}
|
||
* @since 2017-11-15 11:11:42
|
||
*/
|
||
public class ImMessageHandler extends AbstractCommonHandler<Command, CommandResponse> {
|
||
private static final Logger logger = LoggerFactory.getLogger(ImMessageHandler.class);
|
||
|
||
public CommandResponse handle(Command command) {
|
||
try {
|
||
ChannelSession channelSession = command.getChannelSession();
|
||
String deviceId = channelSession.getDeviceId();
|
||
if (StringUtils.isEmpty(deviceId)) {
|
||
channelSession.getChannel().close();
|
||
logger.error("{} client={} im request error with deviceId={}.", AkxProject.PLN, command.getClientIp(),
|
||
deviceId);
|
||
return customResponse(ErrorCode2.ERROR);
|
||
}
|
||
|
||
ChannelSession acsession = ChannelManager.getChannelSession(deviceId);
|
||
if (acsession == null) {
|
||
channelSession.getChannel().close();
|
||
logger.error("{} client={} im request error with channelSession={}", AkxProject.PLN,
|
||
command.getClientIp(), acsession);
|
||
return customResponse(ErrorCode2.ERROR);
|
||
}
|
||
|
||
if (!checkSiteUserId(command.getSiteUserId(), acsession.getUserId())) {
|
||
channelSession.getChannel().close();
|
||
logger.error("{} client={} im request fail siteUserId={},sessionUserId={}", AkxProject.PLN,
|
||
command.getClientIp(), command.getSiteUserId(), acsession.getUserId());
|
||
return customResponse(ErrorCode2.ERROR);
|
||
}
|
||
|
||
if (RequestAction.IM_CTS_PING.getName().equalsIgnoreCase(command.getAction())) {
|
||
Map<Integer, String> header = new HashMap<Integer, String>();
|
||
header.put(CoreProto.HeaderKey.SITE_SERVER_VERSION_VALUE, CommandConst.SITE_VERSION);
|
||
CoreProto.TransportPackageData.Builder packBuilder = CoreProto.TransportPackageData.newBuilder();
|
||
packBuilder.putAllHeader(header);
|
||
channelSession.getChannel().writeAndFlush(new RedisCommand().add(CommandConst.PROTOCOL_VERSION)
|
||
.add(RequestAction.IM_STC_PONG.getName()).add(packBuilder.build().toByteArray()));
|
||
// 检测是否需要给用户发送PSN
|
||
if (channelSession.detectPsn()) {
|
||
channelSession.getChannel().writeAndFlush(new RedisCommand().add(CommandConst.PROTOCOL_VERSION)
|
||
.add(RequestAction.IM_STC_PSN.getName()).add(packBuilder.build().toByteArray()));
|
||
logger.debug("{} client={} siteUserId={} deviceId={} push psn {} {}", AkxProject.PLN,
|
||
command.getClientIp(), command.getSiteUserId(), command.getDeviceId(),
|
||
channelSession.getPsnTime(), channelSession.getSynFinTime());
|
||
}
|
||
|
||
} else {
|
||
// 排除ping请求,其他请求走im服务
|
||
new ImMessageService().execute(command);
|
||
}
|
||
|
||
return customResponse(ErrorCode2.SUCCESS);
|
||
} catch (Exception e) {
|
||
logger.error(StringHelper.format("{} client={} im request error.", AkxProject.PLN, command.getClientIp()),
|
||
e);
|
||
}
|
||
return defaultErrorResponse();
|
||
}
|
||
|
||
/**
|
||
* 比较
|
||
*
|
||
* @param cmdUserId
|
||
* @param sessionUserId
|
||
* @return
|
||
*/
|
||
private boolean checkSiteUserId(String cmdUserId, String sessionUserId) {
|
||
return StringUtils.isBlank(sessionUserId) ? false : sessionUserId.equals(cmdUserId);
|
||
}
|
||
|
||
}
|