服务器端实时推送技术之 SseEmitter

直接上代码

后端:

首先建一个 Result 类

package com.ljv.chat.util;

import org.springframework.web.servlet.mvc.method.annotation.SseEmitter;

public class Result {
    public String clientId;
    public long timestamp;
    public SseEmitter sseEmitter;

    public Result(String clientId, long timestamp, SseEmitter sseEmitter) {
        this.clientId = clientId;
        this.timestamp = timestamp;
        this.sseEmitter = sseEmitter;
    }
}

然后创建一个 SseMap 类

package com.ljv.chat.util;

import java.util.Map;
import java.util.concurrent.ConcurrentHashMap;

public class SseMap {
    public static final Map<String, Result> sseEmitterMap = new ConcurrentHashMap<>();
}

创建 SaticScheduleTask 类 定时任务

package com.ljv.chat.util;

import org.springframework.scheduling.annotation.Scheduled;
import org.springframework.stereotype.Component;

@Component
public class SaticScheduleTask {
    //3.添加定时任务
//    @Scheduled(cron = "0 0 3 * * ? ")


    @Scheduled(cron = "0 0 3 * * ? ")
    private void configureTasks() {
        SseMap.sseEmitterMap.clear();
    }
}

创建 SseEmitterController

package com.ljv.chat.test.controller;

import com.ljv.chat.util.Result;
import com.ljv.chat.util.SseMap;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.web.bind.annotation.CrossOrigin;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RestController;
import org.springframework.web.servlet.mvc.method.annotation.SseEmitter;

import java.io.IOException;
import java.util.ArrayList;
import java.util.List;

/**
 * 服务器端实时推送技术之 SseEmitter 的用法测试
 * <p>
 * 测试步骤:
 * 1.请求http://localhost:8888/sse/start?clientId=111接口,浏览器会阻塞,等待服务器返回结果;
 * 2.请求http://localhost:8888/sse/send?clientId=111接口,可以请求多次,并观察第1步的浏览器返回结果;
 * 3.请求http://localhost:8888/sse/end?clientId=111接口结束某个请求,第1步的浏览器将结束阻塞;
 * 其中clientId代表请求的唯一标志;
 *
 * @author
 */

@CrossOrigin
@RestController
@RequestMapping("/web/sse")
public class SseEmitterController {
    private static final Logger logger = LoggerFactory.getLogger(SseEmitterController.class);

    // 用于保存每个请求对应的 SseEmitter


    /**
     * 返回SseEmitter对象
     *
     * @param clientId
     * @return
     */
    @RequestMapping("/start")
    public SseEmitter testSseEmitter(String clientId) {
        // 默认30秒超时,设置为0L则永不超时
        SseEmitter sseEmitter = new SseEmitter(0L);
        SseMap.sseEmitterMap.put(clientId, new Result(clientId, System.currentTimeMillis(), sseEmitter));
        return sseEmitter;
    }

    /**
     * 向SseEmitter对象发送数据
     *
     * @param clientId
     * @return
     */
    @RequestMapping("/send")
    public String setSseEmitter(String clientId) {
        try {
            Result result =  SseMap.sseEmitterMap.get(clientId);
            if (result != null && result.sseEmitter != null) {
                long timestamp = System.currentTimeMillis();
                String re = "{\"success\":true,\"message\":\"操作成功!\",\"code\":200,\"result\":{\"id\":1018},\"timestamp\":1629254216358}";
                result.sseEmitter.send(re);
            }
        } catch (IOException e) {
            logger.error("IOException!", e);
            return "error";
        }

        return "Succeed!";
    }

    /**
     * 将SseEmitter对象设置成完成
     *
     * @param clientId
     * @return
     */
    @RequestMapping("/end")
    public String completeSseEmitter(String clientId) {
        Result result =  SseMap.sseEmitterMap.get(clientId);
        if (result != null) {
            SseMap.sseEmitterMap.remove(clientId);
            result.sseEmitter.complete();
        }

        return "Succeed!";
    }

}

创建 ChatController

@CrossOrigin   //解决跨域
@RestController
@RequestMapping("chat")
public class ChatController {

   /**
    * 发送消息
    * @param homeId  房间id
    * @param msg   发送的消息
    * @param userId  用户id
    * @return
    */
   @GetMapping("send")
   public Result2<?> send(String homeId,String msg,String userId){

       for (String key :  SseMap.sseEmitterMap.keySet()) {
           if(key.indexOf(String.valueOf(homeId))!=-1){
               try{
                   Result result = SseMap.sseEmitterMap.get(key);
                   Map map=new HashMap();
                   map.put("msg",msg);
                   map.put("userId",userId);
                   String string = JSON.toJSONString(map);
                   result.sseEmitter.send(string);
               }catch (Exception e){
                   SseMap.sseEmitterMap.remove(key);
               }

           }
       }

       return Result2.OK("发送成功");
   }
}

前端(uniapp写的) 建立连接和关闭都是一样的,样式就是简单的测试使用

<template>
    <view class="content">
        <image class="logo" src="/static/logo.png"></image>
        <view class="text-area">
            <!-- <navigator :url="'/pages/index/chat?id='+homeId" hover-class="navigator-hover"> -->
                <button @click="join()"><text class="title">加入{{homeId}}房间</text></button>
            <!-- </navigator> -->
            <button @click="exit()"><text class="title">退出房间</text></button>
            <input v-if="send" type="text" v-model="msg" placeholder="请输入信息" style="border: 1rpx #4CD964 solid; border-radius: 10rpx;" />
            <button @click="sendMsg()">{{title}}</button>
        </view>
        <view v-if="getMsg!=null && getMsg!='' && !isMe">收到消息:<text style="color: red;">{{getMsg}}</text></view>
        <view v-if="getMsg!=null && getMsg!='' && isMe">发送消息:<text style="color: red;">{{getMsg}}</text> 成功</view>
    </view>
</template>

<script>
    export default {
        data() {
            return {
                title: '发送',
                randNum: '',
                msg:'',
                homeId: '1000',
                send: true,
                getMsg: '',
                isMe: false,
                                source:null,
            }
        },
        onLoad() {

        },
        onShow() {
            
        },
        onHide() {
            
        },
        methods: {
            sendMsg(){
                // alert("1")
                let self = this
                self.$http.get('/chat/send', {
                    homeId: self.homeId,
                    msg: self.msg,
                    userId: self.randNum
                }).then((res) => {
                    console.log(res.data)
                    
                });
            },
            exit(){
                // alert("2")
                this.close1();
            },
            join(){
                this.auto();
                // this.send=true
            },
            /**
             * 关闭连接
             */
            close1() {
                let self = this
                self.$http.get('/chat/end', {
                    clientId: self.randNum
                }).then((
                    res) => {
                        
                                                self.source.close();
                        uni.showToast({
                            title: '已退出房间',
                            duration: 2000
                        });
                    });
            },
            /**
             * 与服务器端建立连接
             */
            auto() {
                let self = this
                if (!!window.EventSource) {
                    this.randNum = this.homeId + this.uuid()
                    
                    self.source = new EventSource('http://localhost:8989/chat/start?clientId=' + this
                        .randNum);
                    uni.showToast({
                        title: '加入房间成功',
                        duration: 2000
                    });
                    self.source.onopen = function(event) {
                        console.log("链接建立")
                        
                    };
                    self.source.addEventListener('message', function(e) {
                        let data = JSON.parse(e.data)
                        console.log(data)
                        self.getMsg=data.msg
                        self.isMe=data.userId==self.randNum?true:false
                        // alert(data.message)
                    }, false);
                    self.source.addEventListener('error', function(e) {
                        console.log(e)
                    });
            
            
                } else {
                    alert("不支持sse")
                }
            },
            uuid() {
                var s = [];
                var hexDigits = "0123456789abcdef";
                for (var i = 0; i < 36; i++) {
                    s[i] = hexDigits.substr(Math.floor(Math.random() * 0x10), 1);
                }
                s[14] = "4"; // bits 12-15 of the time_hi_and_version field to 0010
                s[19] = hexDigits.substr((s[19] & 0x3) | 0x8, 1); // bits 6-7 of the clock_seq_hi_and_reserved to 01
                s[8] = s[13] = s[18] = s[23] = "-";
            
                var uuid = s.join("");
                return uuid;
            },
        }
    }
</script>

<style>
    .content {
        display: flex;
        flex-direction: column;
        align-items: center;
        justify-content: center;
    }

    .logo {
        height: 200rpx;
        width: 200rpx;
        margin-top: 200rpx;
        margin-left: auto;
        margin-right: auto;
        margin-bottom: 50rpx;
    }

    .text-area {
        display: flex;
        justify-content: center;
    }

    .title {
        font-size: 36rpx;
        color: #8f8f94;
    }
</style>

最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 160,026评论 4 364
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 67,655评论 1 296
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 109,726评论 0 244
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 44,204评论 0 213
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 52,558评论 3 287
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 40,731评论 1 222
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 31,944评论 2 314
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 30,698评论 0 203
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 34,438评论 1 246
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 30,633评论 2 247
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 32,125评论 1 260
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 28,444评论 3 255
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 33,137评论 3 238
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 26,103评论 0 8
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 26,888评论 0 197
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 35,772评论 2 276
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 35,669评论 2 271

推荐阅读更多精彩内容