package com.qianwen.smartman.common.websocket.fms;

import java.util.HashMap;
import java.util.List;
import java.util.Map;
import com.qianwen.core.websocket.custom.SubscriptionSessionFilter;
import org.springframework.stereotype.Component;
import org.springframework.web.socket.WebSocketSession;

@Component
public class FmsVisualDataFilter implements SubscriptionSessionFilter<FmsVisualDataRequestMessage> {
    public Map<String, FmsVisualDataRequestMessage> dataMap = new HashMap();

    public String type() {
        return "fmsVisualData";
    }

    public void onMessage(WebSocketSession session, FmsVisualDataRequestMessage message) {
        this.dataMap.put(session.getId(), message);
    }

    public List<String> onResponse(String message) {
        return null;
    }

    public void onClose(WebSocketSession session) {
        this.dataMap.remove(session.getId());
    }
}