package com.fzzy.igds.websocket;

import com.alibaba.fastjson.JSONObject;
import com.ruoyi.common.utils.StringUtils;
import lombok.extern.slf4j.Slf4j;
import org.springframework.stereotype.Component;
import javax.websocket.*;
import javax.websocket.server.PathParam;
import javax.websocket.server.ServerEndpoint;
import java.util.HashMap;
import java.util.Map;
import java.util.concurrent.ConcurrentHashMap;

/**
 * @Description WEB-Socket服务，大屏使用：请求路径定义："/websocket/{组织编码}"
 * @Author CZT
 * @Date 2025/12/18 19:19
 */
@Slf4j
@Component
@ServerEndpoint(value = "/websocket/{deptId}/{bizType}/{bizTag}/{userId}")
public class WebSocketServer {

	private static Map<String, Session> sessionPool = new ConcurrentHashMap<>();
	private static Map<String, String> sessionIds = new ConcurrentHashMap<>();
	
	/**
	 * 配置模块在线状态
	 */
	public static Map<String,Boolean> contextOnLineMap = new HashMap<String, Boolean>();

	// 与某个客户端的连接会话，需要通过它来给客户端发送数据
	private Session session;

	@OnOpen
	public void onOpen(Session session,
					   @PathParam("deptId") String deptId,
					   @PathParam("bizType") String bizType,
					   @PathParam("bizTag") String bizTag,
					   @PathParam("userId") String userId) throws Exception {

		this.session = session;

		String key = deptId + "-" + bizType + "-" + bizTag + "-" + userId;

		sessionPool.put(key, session);
		sessionIds.put(session.getId(), key);

		log.info("new webSocket,key={}", key);
	}

	@OnClose
	public void onClose() {
		String key = sessionIds.get(session.getId());

		sessionPool.remove(key);
		sessionIds.remove(session.getId());

		log.info("WebSocket连接关闭={}", key);
		
		String bizType = key.split("-")[2];
		
		contextOnLineMap.put(bizType, false);
	}

	/**
	 * 收到前端发送的信息
	 * 
	 * @param message
	 * @param session
	 */
	@OnMessage
	public void onMessage(String message, Session session) {

		contextOnLineMap.put(message,true);

		log.info("来自客户端信息:\n" + message);
	}

	@OnError
	public void onError(Session session, Throwable error) {
		log.error("发生错误");

		String key = sessionIds.get(session.getId());

		sessionPool.remove(key);
		sessionIds.remove(session.getId());
		error.printStackTrace();
	}

	/**
	 * 后端向前端推送信息，接受者为同一个业务类型操作所有人员
	 *
	 * @param packet
	 */
	public static void sendByPocket(WebSocketPacket packet) {
		if (StringUtils.isEmpty(packet.getBizType())) {
			log.error("WebSocket信息推送失败，业务类型不可为空。");
			return;
		}

		String tag = packet.getDeptId() + "-" + packet.getBizType() + "-" + packet.getBizTag();

		// 遍历推送，只要是bizType一致的均推送
		Session session;
		for (String key : sessionPool.keySet()) {
			if (key.indexOf(tag) != -1) {
				session = sessionPool.get(key);
				session.getAsyncRemote().sendText(
						JSONObject.toJSONString(packet));
			}
		}
	}

	/**
	 * 发送给指定人员
	 *
	 * @param packet
	 */
	public static void sendByUser(WebSocketPacket packet) {
		if (StringUtils.isEmpty(packet.getBizType())
				|| StringUtils.isEmpty(packet.getDeptId())
				|| StringUtils.isEmpty(packet.getUserId())) {
			log.error("WebSocket信息推送失败，组织编码和业务类型或者指定人信息参数没有获取到！");

			return;
		}

		String tag = packet.getDeptId() + "-" + packet.getBizType() + "-"
				+ packet.getBizTag() + "-" + packet.getUserId();

		Session session;
		for (String key : sessionPool.keySet()) {
			if (tag.equals(key)) {
				session = sessionPool.get(key);
				session.getAsyncRemote().sendText(
						JSONObject.toJSONString(packet));
			}
		}
	}

	/**
	 * 根据指定的BizId标签推送
	 *
	 * @param packet
	 */
	public static void sendByBizTag(WebSocketPacket packet) {
		if (StringUtils.isEmpty(packet.getBizType())
				|| StringUtils.isEmpty(packet.getDeptId())
				|| StringUtils.isEmpty(packet.getBizTag())) {
			log.error("WebSocket信息推送失败，组织编码和业务类型或者指定人信息参数没有获取到！");

			return;
		}

		String tag = packet.getDeptId() + "-" + packet.getBizType() + "-" + packet.getBizTag();

		// 遍历推送，只要是bizType一致的均推送
		Session session;
		for (String key : sessionPool.keySet()) {
			if (key.indexOf(tag) != -1) {
				session = sessionPool.get(key);
				session.getAsyncRemote().sendText(
						JSONObject.toJSONString(packet));
			}
		}

	}
}