package bit.com.spring.websock;

import java.util.Date;
import java.util.Map;
import java.util.concurrent.ConcurrentHashMap;

import org.springframework.web.socket.CloseStatus;
import org.springframework.web.socket.TextMessage;
import org.springframework.web.socket.WebSocketSession;
import org.springframework.web.socket.handler.TextWebSocketHandler;

//import org.springframework.stereotype.Component;

//@Component 이 클래스를 자동 생성해서 동작을 하게한다 . 이방법이 있고 context.xml에 생성을 하게 하는 방법이 있다.

// web socket server
public class WebSocket extends TextWebSocketHandler{
	
	//클라이언트들을 저장하고 관리를 할 수 있도록 해주어야 한다.
	private Map<String, WebSocketSession> users = new ConcurrentHashMap<String, WebSocketSession>();
	
	public WebSocket() {
		System.out.println("EchoHandler 생성됨" + new Date());
	}

	
	//연결된 다음 실행되는 메소드 accept 부분에 해당 (오버라이드)
	@Override
	public void afterConnectionEstablished(WebSocketSession session) throws Exception {
		//연결 확인(연결 후 실행)
		System.out.println("연결됨 " + session.getId() + " " + new Date());
		users.put(session.getId(), session); //session에 들어오는 값이 계속 달라진다
	}


	//연결이 종료 되었을때 실행
	@Override
	public void afterConnectionClosed(WebSocketSession session, CloseStatus status) throws Exception {
		//연결 종료
		System.out.println("연결종료" + session.getId());
		users.remove(session.getId());	//연결 종류 된 후 저장되어있던 세션을 지워 준다.
	}


	//수신되는 부분을 체크하여 다시 보내주는역할
	@Override
	protected void handleTextMessage(WebSocketSession session, TextMessage message) throws Exception {
		//메시지 수신
		
		System.out.println("메세지 수신: "+ message.getPayload() + " "+ new Date());
		
		//송신 (send)
		for(WebSocketSession s: users.values()) {		//하나하나의 소켓 세션을 꺼낼 수 있다
			s.sendMessage(message);
		}
	}

	
	//예외가 발생되었을때 실행되는 메소드
	@Override
	public void handleTransportError(WebSocketSession session, Throwable exception) throws Exception {
	//예외 발생
		System.out.println(session.getId() + "Exception 발생" + new Date());
	}
}

package bit.com.spring.controller;

import org.springframework.stereotype.Controller;
import org.springframework.ui.Model;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RequestMethod;

/*
 AOP와 다른 Package에 사용 권장
 */
@Controller
public class bWebSocket {

	@RequestMapping(value = "chatting.do", method = {RequestMethod.GET,RequestMethod.POST})
	public String chatting(Model model) {
		model.addAttribute("doc_title", "채팅");
		return "chatting.tiles";
	}
}

@Override
public void afterConnectionEstablished(WebSocketSession session) throws Exception {
	//연결 확인(연결 후 실행)
	System.out.println("연결됨 " + session.getId() + " " + new Date());
	String senderId = getid(session);	//아래 세션으로 저장된 멤버dto의 id를 가져옴
	System.out.println(senderId);
	users.put(senderId, session); //session에 들어오는 값이 계속 달라진다
}

private String getid(WebSocketSession session) {
	Map<String, Object> httpSession = session.getAttributes();	// 세션을 가져온다
	MemberDto loginUser = (MemberDto)httpSession.get("login");	// 세션에 저장되어있는 객체를 가져온다

	if(loginUser == null) {
		return session.getId();
	} else {
		return loginUser.getId();	// id를 return 해준다
	}
}

@Override
protected void handleTextMessage(WebSocketSession session, TextMessage message) throws Exception {
	//메시지 수신

	String msg = message.getPayload();
	System.out.println(msg);
	String str[] = msg.split(":");	// 뷰에서 보내준 id값을 추출하기 위해 스플릿한다
	System.out.println(str[1]);
	String id = str[1];

	//송신 (send)
	WebSocketSession s = users.get(id); // map에 저장되어있는 아이디를 불러온다 (value에 지정 session 저장되어있다)
	s.sendMessage(message);
}