diff options
author | 2018-04-12 14:32:44 +0300 | |
---|---|---|
committer | 2018-04-12 14:32:44 +0300 | |
commit | c5649915e6259e921cfd087042525b69533d53b0 (patch) | |
tree | 54e2d7bdba8a21a640b3e2a0c427b72d33c445c2 /juick-notifications/src/main/java/com/juick/components/service/JuickServerComponent.java | |
parent | 9e5a6b3f2f6907f18f22718bcb6f0ac7f79c95cd (diff) |
notifications: pushserver is the only websocket component for now
Diffstat (limited to 'juick-notifications/src/main/java/com/juick/components/service/JuickServerComponent.java')
-rw-r--r-- | juick-notifications/src/main/java/com/juick/components/service/JuickServerComponent.java | 78 |
1 files changed, 0 insertions, 78 deletions
diff --git a/juick-notifications/src/main/java/com/juick/components/service/JuickServerComponent.java b/juick-notifications/src/main/java/com/juick/components/service/JuickServerComponent.java deleted file mode 100644 index 0f62e702..00000000 --- a/juick-notifications/src/main/java/com/juick/components/service/JuickServerComponent.java +++ /dev/null @@ -1,78 +0,0 @@ -package com.juick.components.service; - -import com.fasterxml.jackson.databind.ObjectMapper; -import com.juick.components.Notifications; -import com.juick.server.component.DisconnectedEvent; -import com.juick.server.component.MessageEvent; -import org.slf4j.Logger; -import org.slf4j.LoggerFactory; -import org.springframework.context.ApplicationEventPublisher; -import org.springframework.scheduling.annotation.Scheduled; -import org.springframework.stereotype.Component; -import org.springframework.web.socket.CloseStatus; -import org.springframework.web.socket.PingMessage; -import org.springframework.web.socket.TextMessage; -import org.springframework.web.socket.WebSocketSession; -import org.springframework.web.socket.handler.TextWebSocketHandler; - -import javax.annotation.PostConstruct; -import javax.annotation.PreDestroy; -import javax.inject.Inject; -import java.io.IOException; -import java.util.concurrent.atomic.AtomicBoolean; - -@Component -public class JuickServerComponent extends TextWebSocketHandler { - private static Logger logger = LoggerFactory.getLogger(JuickServerComponent.class); - @Inject - private ApplicationEventPublisher applicationEventPublisher; - @Inject - private Notifications notifications; - @Inject - private ObjectMapper jsonMapper; - - private WebSocketSession session; - private final AtomicBoolean closeFlag = new AtomicBoolean(false); - - @PostConstruct - public void init() { - closeFlag.set(false); - } - @Override - public void afterConnectionEstablished(WebSocketSession session) throws Exception { - logger.info("WebSocket connected"); - this.session = session; - } - - @Override - public void afterConnectionClosed(WebSocketSession session, CloseStatus status) throws Exception { - logger.info("WebSocket disconnected with code {}: {}", status.getCode(), status.getReason()); - applicationEventPublisher.publishEvent(new DisconnectedEvent(this)); - } - - @Override - protected void handleTextMessage(WebSocketSession session, TextMessage text) throws Exception { - com.juick.Message jmsg = jsonMapper.readValue(text.asBytes(), com.juick.Message.class); - - if (logger.isInfoEnabled()) // prevent writeValueAsString execution if logger disabled - logger.info("got jmsg: {}", jsonMapper.writeValueAsString(jmsg)); - if (!closeFlag.get()) { - notifications.messageReceived(jmsg); - } - } - - @Scheduled(fixedRate = 30000, initialDelay = 30000) - public void ping() throws IOException { - if (session != null && session.isOpen()) { - logger.debug("Sending WebSocket ping"); - session.sendMessage(new PingMessage()); - } else if (!closeFlag.get()) { - applicationEventPublisher.publishEvent(new DisconnectedEvent(this)); - } - } - @PreDestroy - public void close() { - closeFlag.set(true); - } - -} |