日本综合一区二区|亚洲中文天堂综合|日韩欧美自拍一区|男女精品天堂一区|欧美自拍第6页亚洲成人精品一区|亚洲黄色天堂一区二区成人|超碰91偷拍第一页|日韩av夜夜嗨中文字幕|久久蜜综合视频官网|精美人妻一区二区三区

 io.netty
 netty-all 
 4.1.33.Final



 cn.hutool
 hutool-all
 5.2.3

package com.sixj.nettypush.config;

import io.netty.channel.Channel;
import io.netty.channel.group.ChannelGroup;
import io.netty.channel.group.DefaultChannelGroup;
import io.netty.util.concurrent.GlobalEventExecutor;

import java.util.concurrent.ConcurrentHashMap;

/**
 * @author sixiaojie
 * @date 2020-03-28-15:07
 */
public class NettyConfig {
  /**
   * 定義一個(gè)channel組，管理所有的channel
   * GlobalEventExecutor.INSTANCE 是全局的事件執(zhí)行器，是一個(gè)單例
   */
  private static ChannelGroup channelGroup = new DefaultChannelGroup(GlobalEventExecutor.INSTANCE);

  /**
   * 存放用戶與Chanel的對(duì)應(yīng)信息，用于給指定用戶發(fā)送消息
   */
  private static ConcurrentHashMap userChannelMap = new ConcurrentHashMap<>();

  private NettyConfig() {}
  
  /**
   * 獲取channel組
   * @return
   */
  public static ChannelGroup getChannelGroup() {
    return channelGroup;
  }

  /**
   * 獲取用戶channel map
   * @return
   */
  public static ConcurrentHashMap getUserChannelMap(){
    return userChannelMap;
  }
}

package com.sixj.nettypush.websocket;

import io.netty.bootstrap.ServerBootstrap;
import io.netty.channel.ChannelFuture;
import io.netty.channel.ChannelInitializer;
import io.netty.channel.EventLoopGroup;
import io.netty.channel.nio.NioEventLoopGroup;
import io.netty.channel.socket.SocketChannel;
import io.netty.channel.socket.nio.NioServerSocketChannel;
import io.netty.handler.codec.http.HttpObjectAggregator;
import io.netty.handler.codec.http.HttpServerCodec;
import io.netty.handler.codec.http.websocketx.WebSocketServerProtocolHandler;
import io.netty.handler.codec.serialization.ObjectEncoder;
import io.netty.handler.stream.ChunkedWriteHandler;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.stereotype.Component;

import javax.annotation.PostConstruct;
import javax.annotation.PreDestroy;
import java.net.InetSocketAddress;

/**
 * @author sixiaojie
 * @date 2020-03-28-13:44
 */

@Component
public class NettyServer{
  private static final Logger log = LoggerFactory.getLogger(NettyServer.class);
  /**
   * webSocket協(xié)議名
   */
  private static final String WEBSOCKET_PROTOCOL = "WebSocket";

  /**
   * 端口號(hào)
   */
  @Value("${webSocket.netty.port:58080}")
  private int port;

  /**
   * webSocket路徑
   */
  @Value("${webSocket.netty.path:/webSocket}")
  private String webSocketPath;

  @Autowired
  private WebSocketHandler webSocketHandler;

  private EventLoopGroup bossGroup;
  private EventLoopGroup workGroup;

  /**
   * 啟動(dòng)
   * @throws InterruptedException
   */
  private void start() throws InterruptedException {
    bossGroup = new NioEventLoopGroup();
    workGroup = new NioEventLoopGroup();
    ServerBootstrap bootstrap = new ServerBootstrap();
    // bossGroup輔助客戶端的tcp連接請(qǐng)求, workGroup負(fù)責(zé)與客戶端之前的讀寫操作
    bootstrap.group(bossGroup,workGroup);
    // 設(shè)置NIO類型的channel
    bootstrap.channel(NioServerSocketChannel.class);
    // 設(shè)置監(jiān)聽端口
    bootstrap.localAddress(new InetSocketAddress(port));
    // 連接到達(dá)時(shí)會(huì)創(chuàng)建一個(gè)通道
    bootstrap.childHandler(new ChannelInitializer() {

      @Override
      protected void initChannel(SocketChannel ch) throws Exception {
        // 流水線管理通道中的處理程序（Handler），用來處理業(yè)務(wù)
        // webSocket協(xié)議本身是基于http協(xié)議的，所以這邊也要使用http編解碼器
        ch.pipeline().addLast(new HttpServerCodec());
        ch.pipeline().addLast(new ObjectEncoder());
        // 以塊的方式來寫的處理器
        ch.pipeline().addLast(new ChunkedWriteHandler());
        /*
        說明：
        1、http數(shù)據(jù)在傳輸過程中是分段的，HttpObjectAggregator可以將多個(gè)段聚合
        2、這就是為什么，當(dāng)瀏覽器發(fā)送大量數(shù)據(jù)時(shí)，就會(huì)發(fā)送多次http請(qǐng)求
         */
        ch.pipeline().addLast(new HttpObjectAggregator(8192));
        /*
        說明：
        1、對(duì)應(yīng)webSocket，它的數(shù)據(jù)是以幀（frame）的形式傳遞
        2、瀏覽器請(qǐng)求時(shí) ws://localhost:58080/xxx 表示請(qǐng)求的uri
        3、核心功能是將http協(xié)議升級(jí)為ws協(xié)議，保持長(zhǎng)連接
        */
        ch.pipeline().addLast(new WebSocketServerProtocolHandler(webSocketPath, WEBSOCKET_PROTOCOL, true, 65536 * 10));
        // 自定義的handler，處理業(yè)務(wù)邏輯
        ch.pipeline().addLast(webSocketHandler);

      }
    });
    // 配置完成，開始綁定server，通過調(diào)用sync同步方法阻塞直到綁定成功
    ChannelFuture channelFuture = bootstrap.bind().sync();
    log.info("Server started and listen on:{}",channelFuture.channel().localAddress());
    // 對(duì)關(guān)閉通道進(jìn)行監(jiān)聽
    channelFuture.channel().closeFuture().sync();
  }

  /**
   * 釋放資源
   * @throws InterruptedException
   */
  @PreDestroy
  public void destroy() throws InterruptedException {
    if(bossGroup != null){
      bossGroup.shutdownGracefully().sync();
    }
    if(workGroup != null){
      workGroup.shutdownGracefully().sync();
    }
  }
  @PostConstruct()
  public void init() {
    //需要開啟一個(gè)新的線程來執(zhí)行netty server 服務(wù)器
    new Thread(() -> {
      try {
        start();
      } catch (InterruptedException e) {
        e.printStackTrace();
      }
    }).start();
  }
}

package com.sixj.nettypush.websocket;

import cn.hutool.json.JSONObject;
import cn.hutool.json.JSONUtil;
import com.sixj.nettypush.config.NettyConfig;
import io.netty.channel.ChannelHandler;
import io.netty.channel.ChannelHandlerContext;
import io.netty.channel.SimpleChannelInboundHandler;
import io.netty.handler.codec.http.websocketx.TextWebSocketFrame;
import io.netty.util.AttributeKey;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.stereotype.Component;


/**
 * TextWebSocketFrame類型， 表示一個(gè)文本幀
 * @author sixiaojie
 * @date 2020-03-28-13:47
 */
@Component
@ChannelHandler.Sharable
public class WebSocketHandler extends SimpleChannelInboundHandler {
  private static final Logger log = LoggerFactory.getLogger(NettyServer.class);

  /**
   * 一旦連接，第一個(gè)被執(zhí)行
   * @param ctx
   * @throws Exception
   */
  @Override
  public void handlerAdded(ChannelHandlerContext ctx) throws Exception {
    log.info("handlerAdded 被調(diào)用"+ctx.channel().id().asLongText());
    // 添加到channelGroup 通道組
    NettyConfig.getChannelGroup().add(ctx.channel());
  }

  /**
   * 讀取數(shù)據(jù)
   */
  @Override
  protected void channelRead0(ChannelHandlerContext ctx, TextWebSocketFrame msg) throws Exception {
    log.info("服務(wù)器收到消息：{}",msg.text());

    // 獲取用戶ID,關(guān)聯(lián)channel
    JSONObject jsonObject = JSONUtil.parseObj(msg.text());
    String uid = jsonObject.getStr("uid");
    NettyConfig.getUserChannelMap().put(uid,ctx.channel());

    // 將用戶ID作為自定義屬性加入到channel中，方便隨時(shí)channel中獲取用戶ID
    AttributeKey key = AttributeKey.valueOf("userId");
    ctx.channel().attr(key).setIfAbsent(uid);

    // 回復(fù)消息
    ctx.channel().writeAndFlush(new TextWebSocketFrame("服務(wù)器連接成功！"));
  }

  @Override
  public void handlerRemoved(ChannelHandlerContext ctx) throws Exception {
    log.info("handlerRemoved 被調(diào)用"+ctx.channel().id().asLongText());
    // 刪除通道
    NettyConfig.getChannelGroup().remove(ctx.channel());
    removeUserId(ctx);
  }

  @Override
  public void exceptionCaught(ChannelHandlerContext ctx, Throwable cause) throws Exception {
    log.info("異常：{}",cause.getMessage());
    // 刪除通道
    NettyConfig.getChannelGroup().remove(ctx.channel());
    removeUserId(ctx);
    ctx.close();
  }

  /**
   * 刪除用戶與channel的對(duì)應(yīng)關(guān)系
   * @param ctx
   */
  private void removeUserId(ChannelHandlerContext ctx){
    AttributeKey key = AttributeKey.valueOf("userId");
    String userId = ctx.channel().attr(key).get();
    NettyConfig.getUserChannelMap().remove(userId);
  }
}

**第四步：**具體消息推送的接口
public interface PushService {
  /**
   * 推送給指定用戶
   * @param userId
   * @param msg
   */
  void pushMsgToOne(String userId,String msg);

  /**
   * 推送給所有用戶
   * @param msg
   */
  void pushMsgToAll(String msg);
}

import java.util.concurrent.ConcurrentHashMap;

/**
 * @author sixiaojie
 * @date 2020-03-30-20:10
 */
@Service
public class PushServiceImpl implements PushService {

  @Override
  public void pushMsgToOne(String userId, String msg){
    ConcurrentHashMap userChannelMap = NettyConfig.getUserChannelMap();
    Channel channel = userChannelMap.get(userId);
    channel.writeAndFlush(new TextWebSocketFrame(msg));
  }
  @Override
  public void pushMsgToAll(String msg){
    NettyConfig.getChannelGroup().writeAndFlush(new TextWebSocketFrame(msg));
  }
}

package com.sixj.nettypush.controller;

import com.sixj.nettypush.service.PushService;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.web.bind.annotation.PostMapping;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RequestParam;
import org.springframework.web.bind.annotation.RestController;

/**
 * @author sixiaojie
 * @date 2020-03-30-20:08
 */
@RestController
@RequestMapping("/push")
public class PushController {

  @Autowired
  private PushService pushService;

  /**
   * 推送給所有用戶
   * @param msg
   */
  @PostMapping("/pushAll")
  public void pushToAll(@RequestParam("msg") String msg){
    pushService.pushMsgToAll(msg);
  }
  /**
   * 推送給指定用戶
   * @param userId
   * @param msg
   */
  @PostMapping("/pushOne")
  public void pushMsgToOne(@RequestParam("userId") String userId,@RequestParam("msg") String msg){
    pushService.pushMsgToOne(userId,msg);
  }
}