spring boot websocket推送kafka消费消息

1、websocket配置类

package com.example.equipkafka.websocketService;

import org.nfunk.jep.function.Str;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.stereotype.Service;

import javax.websocket.*;
import javax.websocket.server.PathParam;
import javax.websocket.server.ServerEndpoint;
import java.io.IOException;
import java.util.concurrent.CopyOnWriteArraySet;
import java.util.concurrent.atomic.AtomicInteger;

@ServerEndpoint("/ws/{id}")
@Service
public class TestWebsocket {
    private Session session;
    public static CopyOnWriteArraySet<TestWebsocket> wbSockets = new CopyOnWriteArraySet<TestWebsocket>(); //此处定义静态变量，以在其他方法中获取到所有连接


    @OnOpen
    public void onOpen(Session session, @PathParam("id") String id){
        System.out.println("接收参数id");
        System.out.println(id);
        this.session = session;
        wbSockets.add(this); //将此对象存入集合中以在之后广播用，如果要实现一对一订阅，则类型对应为Map。由于这里广播就可以了随意用Set
    }

    @OnClose
    public void onClose(){
        wbSockets.remove(this);//将socket对象从集合中移除，以便广播时不发送次连接。如果不移除会报错(需要测试)
        System.out.println("A session insert,sessionId is "+ session.getId());
    }

    @OnError
    public void onError(Session session, Throwable error) {
        wbSockets.remove(this);//将socket对象从集合中移除，以便广播时不发送次连接。如果不移除会报错(需要测试)
        error.printStackTrace();
    }

    @OnMessage
    public void onMessage(String message ,Session session){
        System.out.println(message + "from " + session.getId());
    }

    public void sendMessage(String message) throws IOException {
        this.session.getBasicRemote().sendText(message);
    }

}

2、在kafka消费者中，执行如下

for (TestWebsocket webSocket : wbSockets){
                        System.out.println("消费消息");
                        System.out.println(record.value());
                        webSocket.sendMessage(record.value());
                    }