package org.springblade.modules.sse.controller;

import lombok.AllArgsConstructor;
import lombok.extern.slf4j.Slf4j;
import org.springblade.modules.sse.server.SSEServer;
import org.springblade.modules.sse.vo.SseVO;
import org.springframework.web.bind.annotation.CrossOrigin;
import org.springframework.web.bind.annotation.GetMapping;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RestController;
import org.springframework.web.servlet.mvc.method.annotation.SseEmitter;

@Slf4j
@RestController
@CrossOrigin
@RequestMapping("/sse/sse")
@AllArgsConstructor
public class SSEController {

	/**
	 * 建立连接
	 * @param sse
	 * @return
	 */
	@GetMapping("/connect")
	public SseEmitter connect(SseVO sse){
		String userId = sse.getType() + ":" + sse.getUserId();
		return SSEServer.connect(userId);
	}

	/**
	 * 断开连接
	 * @param sse
	 * @return
	 */
	@GetMapping("/disconnect")
	public void disconnect(SseVO sse){
		String userId = sse.getType() + ":" + sse.getUserId();
		SSEServer.removeUser(userId);
	}
}