diff --git a/build.gradle b/build.gradle index c92a1c7..b82f0f1 100644 --- a/build.gradle +++ b/build.gradle @@ -18,6 +18,8 @@ implementation 'org.springframework.boot:spring-boot-starter-web' providedRuntime 'org.springframework.boot:spring-boot-starter-tomcat' testImplementation 'org.springframework.boot:spring-boot-starter-test' + + implementation 'org.springframework.boot:spring-boot-starter-websocket' } tasks.named('test') { diff --git a/src/main/java/com/ntlab/irisserver/WebSocketConfig.java b/src/main/java/com/ntlab/irisserver/WebSocketConfig.java new file mode 100644 index 0000000..b4ad29e --- /dev/null +++ b/src/main/java/com/ntlab/irisserver/WebSocketConfig.java @@ -0,0 +1,27 @@ +package com.ntlab.irisserver; +import com.ntlab.irisserver.resources.HelloWorldWebSocket; +import org.springframework.context.annotation.Bean; +import org.springframework.context.annotation.Configuration; +import org.springframework.web.socket.config.annotation.*; +import org.springframework.web.socket.server.standard.ServerEndpointExporter; + +@Configuration +@EnableWebSocket +public class WebSocketConfig implements WebSocketConfigurer { + + @Override + public void registerWebSocketHandlers(WebSocketHandlerRegistry registry) { + //registry.addHandler(unitWebSocketHandler, "/unit").setAllowedOrigins("*"); + } + + @Bean + public HelloWorldWebSocket helloWorldWebSocket() { + return new HelloWorldWebSocket(); + } + + @Bean + public ServerEndpointExporter serverEndpointExporter() { + return new ServerEndpointExporter(); + } + +} diff --git a/src/main/java/com/ntlab/irisserver/resources/HelloWorld.java b/src/main/java/com/ntlab/irisserver/resources/HelloWorld.java index de0e2ed..d38541e 100644 --- a/src/main/java/com/ntlab/irisserver/resources/HelloWorld.java +++ b/src/main/java/com/ntlab/irisserver/resources/HelloWorld.java @@ -12,7 +12,7 @@ public class HelloWorld { @GET public String getHello() { - return "Hello World!(WebSocketバージョン)。12/11-10:48"; + return "WebSocketのデモの追加。12/11-10:54"; } } diff --git a/src/main/java/com/ntlab/irisserver/resources/HelloWorldWebSocket.java b/src/main/java/com/ntlab/irisserver/resources/HelloWorldWebSocket.java new file mode 100644 index 0000000..92a70ef --- /dev/null +++ b/src/main/java/com/ntlab/irisserver/resources/HelloWorldWebSocket.java @@ -0,0 +1,56 @@ +package com.ntlab.irisserver.resources; + +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("/HelloWorldWebSocket") +public class HelloWorldWebSocket { + private static final Queue sessions = new ConcurrentLinkedQueue<>(); + + static { + ScheduledExecutorService service = Executors.newSingleThreadScheduledExecutor(); + service.scheduleWithFixedDelay(HelloWorldWebSocket::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