diff --git a/src/main/java/com/ntlab/irisserver/resources/HelloWorld.java b/src/main/java/com/ntlab/irisserver/resources/HelloWorld.java index 8f1b994..9a1982d 100644 --- a/src/main/java/com/ntlab/irisserver/resources/HelloWorld.java +++ b/src/main/java/com/ntlab/irisserver/resources/HelloWorld.java @@ -10,7 +10,7 @@ public class HelloWorld { @GET public String getHello() { - return "WebSocketのデモの追加。12/15-16:00"; + return "WebSocketのデモの追加。12/15-16:05"; } } diff --git a/src/main/java/com/ntlab/irisserver/websockets/HelloWebSocket.java b/src/main/java/com/ntlab/irisserver/websockets/HelloWebSocket.java new file mode 100644 index 0000000..2eceb16 --- /dev/null +++ b/src/main/java/com/ntlab/irisserver/websockets/HelloWebSocket.java @@ -0,0 +1,59 @@ +package com.ntlab.irisserver.websockets; + +import org.springframework.web.socket.WebSocketHandler; +import org.springframework.web.socket.handler.TextWebSocketHandler; + +import javax.websocket.*; +import javax.websocket.server.ServerEndpoint; +import java.text.SimpleDateFormat; +import java.util.Date; +import java.util.Queue; +import java.util.concurrent.ConcurrentLinkedQueue; +import java.util.concurrent.Executors; +import java.util.concurrent.ScheduledExecutorService; +import java.util.concurrent.TimeUnit; + + +// 5秒に1回HelloWorldを送るデモ +@ServerEndpoint("/hello-websocket") +public class HelloWebSocket { + private static final Queue sessions = new ConcurrentLinkedQueue<>(); + + static { + ScheduledExecutorService service = Executors.newSingleThreadScheduledExecutor(); + service.scheduleWithFixedDelay(HelloWebSocket::broadcast, 5, 5, TimeUnit.SECONDS); + } + + @OnMessage + public String onMessage(String message) { + System.out.println("WebSocketで受信したメッセージ/ " + message); + return "WebSocketでメッセージを正常に受信しました!"; + } + + @OnError + public void onError(Throwable th) { + System.out.println("WebSocketエラーが発生/ " + th.getMessage()); + } + + @OnOpen + public void onOpen(Session session) { + System.out.println("WebSocketセッション確立"); + sessions.add(session); + } + + @OnClose + public void onClose(Session session) { + System.out.println("WebSocketセッション終了"); + sessions.remove(session); + } + + public static void broadcast() { + Date now = new Date(); + SimpleDateFormat formatter = new SimpleDateFormat("yyyy/MM/dd HH:mm:ss"); + +// sessions.forEach(session -> { +// session.getAsyncRemote().sendText("HelloWorld! : " + formatter.format(now)); +// }); + } + +} \ No newline at end of file