WindChat/windchat-connector/src/main/java/com/windchat/im/connector/handler/ImMessageHandler.java

/** 
 * 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);
	}

}