package com.qianwen.core.websocket.handler;

import com.qianwen.core.websocket.config.WebSocketMessageSender;
import com.qianwen.core.websocket.message.JsonWebSocketMessage;
import com.qianwen.core.websocket.message.PingJsonWebSocketMessage;
import com.qianwen.core.websocket.message.PongJsonWebSocketMessage;
import com.qianwen.core.websocket.message.WebSocketMessageTypeEnum;
import org.springframework.web.socket.WebSocketSession;

public class PingJsonMessageHandler implements JsonMessageHandler<PingJsonWebSocketMessage> {
    @Override 
    public void handle(WebSocketSession session, PingJsonWebSocketMessage message) {
        JsonWebSocketMessage pongJsonWebSocketMessage = new PongJsonWebSocketMessage();
        WebSocketMessageSender.send(session, pongJsonWebSocketMessage);
    }

    @Override
    public String type() {
        return WebSocketMessageTypeEnum.PING.getValue();
    }

    @Override
    public Class<PingJsonWebSocketMessage> getMessageClass() {
        return PingJsonWebSocketMessage.class;
    }
}