xie.bx
3 years ago
39 changed files with 544 additions and 198 deletions
@ -0,0 +1,9 @@ |
|||
package com.lx.common.model.im; |
|||
|
|||
import lombok.Data; |
|||
|
|||
@Data |
|||
public class LoginInfo { |
|||
|
|||
private long userId; |
|||
} |
|||
@ -0,0 +1,64 @@ |
|||
package com.lx.implatform.imserver.websocket.processor; |
|||
|
|||
import cn.hutool.core.bean.BeanUtil; |
|||
import com.lx.common.contant.RedisKey; |
|||
import com.lx.common.enums.WSCmdEnum; |
|||
import com.lx.common.model.im.HeartbeatInfo; |
|||
import com.lx.common.model.im.LoginInfo; |
|||
import com.lx.common.model.im.SendInfo; |
|||
import com.lx.implatform.imserver.websocket.WebsocketChannelCtxHloder; |
|||
import com.lx.implatform.imserver.websocket.WebsocketServer; |
|||
import io.netty.channel.ChannelHandlerContext; |
|||
import io.netty.util.AttributeKey; |
|||
import lombok.extern.slf4j.Slf4j; |
|||
import org.springframework.beans.factory.annotation.Autowired; |
|||
import org.springframework.data.redis.core.RedisTemplate; |
|||
import org.springframework.stereotype.Component; |
|||
|
|||
import java.util.HashMap; |
|||
import java.util.concurrent.TimeUnit; |
|||
|
|||
@Slf4j |
|||
@Component |
|||
public class LoginProcessor extends MessageProcessor<LoginInfo> { |
|||
|
|||
|
|||
@Autowired |
|||
private WebsocketServer WSServer; |
|||
|
|||
@Autowired |
|||
RedisTemplate<String,Object> redisTemplate; |
|||
|
|||
@Override |
|||
synchronized public void process(ChannelHandlerContext ctx, LoginInfo loginInfo) { |
|||
log.info("用户登录，userId:{}",loginInfo.getUserId()); |
|||
ChannelHandlerContext context = WebsocketChannelCtxHloder.getChannelCtx(loginInfo.getUserId()); |
|||
if(context != null){ |
|||
// 不允许多地登录,强制下线
|
|||
SendInfo sendInfo = new SendInfo(); |
|||
sendInfo.setCmd(WSCmdEnum.FORCE_LOGUT.getCode()); |
|||
context.channel().writeAndFlush(sendInfo); |
|||
} |
|||
|
|||
// 绑定用户和channel
|
|||
WebsocketChannelCtxHloder.addChannelCtx(loginInfo.getUserId(),ctx); |
|||
// 设置属性
|
|||
AttributeKey<Long> attr = AttributeKey.valueOf("USER_ID"); |
|||
ctx.channel().attr(attr).set(loginInfo.getUserId()); |
|||
// 在redis上记录每个user的channelId，15秒没有心跳，则自动过期
|
|||
String key = RedisKey.IM_USER_SERVER_ID+loginInfo.getUserId(); |
|||
redisTemplate.opsForValue().set(key, WSServer.getServerId()); |
|||
// 响应ws
|
|||
SendInfo sendInfo = new SendInfo(); |
|||
sendInfo.setCmd(WSCmdEnum.LOGIN.getCode()); |
|||
ctx.channel().writeAndFlush(sendInfo); |
|||
} |
|||
|
|||
|
|||
@Override |
|||
public LoginInfo transForm(Object o) { |
|||
HashMap map = (HashMap)o; |
|||
LoginInfo loginInfo = BeanUtil.fillBeanWithMap(map, new LoginInfo(), false); |
|||
return loginInfo; |
|||
} |
|||
} |
|||
@ -1,9 +1,17 @@ |
|||
package com.lx.implatform.imserver.websocket.processor; |
|||
|
|||
|
|||
import io.netty.channel.ChannelHandlerContext; |
|||
|
|||
public interface MessageProcessor<T> { |
|||
public abstract class MessageProcessor<T> { |
|||
|
|||
public void process(ChannelHandlerContext ctx,T data){} |
|||
|
|||
public void process(T data){} |
|||
|
|||
public T transForm(Object o){ |
|||
return (T)o; |
|||
} |
|||
|
|||
void process(ChannelHandlerContext ctx,T data); |
|||
|
|||
} |
|||
|
|||
Loading…
Reference in new issue