io.lightflame.websocket.FlameWsContext Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of lightflame-core Show documentation

Full web framework

The newest version!

package io.lightflame.websocket;


import io.netty.channel.ChannelHandlerContext;
import io.netty.handler.codec.http.websocketx.TextWebSocketFrame;

import java.util.Collection;

/**
 * FlameWebSocketCtx
 */
public class FlameWsContext {

    private ChannelHandlerContext ctx;
    private Session session;


    FlameWsContext(ChannelHandlerContext ctx, Session session){
        this.ctx = ctx;
        this.session = session;
    }

    public String message() {
        return ctx.channel().attr(WsAttributes.requestAttKey).get();
    }

    public void writeToChannel(String msg) {
        ctx.channel().writeAndFlush(new TextWebSocketFrame(msg));
    }

    public void writeToAllChannels(String msg) {
        for (ChannelHandlerContext ctx : session.getAllSessions().values()){
            ctx.channel().writeAndFlush(new TextWebSocketFrame(msg));
        }
    }

}