org.nkjmlab.util.jakarta.websocket.WebsocketClientSession Maven / Gradle / Ivy

Go to download
package org.nkjmlab.util.jakarta.websocket;

import java.io.IOException;
import java.net.URI;
import java.util.function.BiConsumer;
import org.nkjmlab.util.java.function.Try;
import org.nkjmlab.util.java.lang.StringFormatter;
import org.nkjmlab.util.java.logging.LogManager;
import org.nkjmlab.util.java.net.UriUtils;
import io.netty.util.internal.shaded.org.jctools.queues.MessagePassingQueue.Consumer;
import jakarta.websocket.CloseReason;
import jakarta.websocket.ContainerProvider;
import jakarta.websocket.DeploymentException;
import jakarta.websocket.Session;

public class WebsocketClientSession {
  private static final org.nkjmlab.util.java.logging.SimpleLogger log = LogManager.createLogger();

  private final Session session;

  /**
   *    * WebsocketClientSession session =
   *     WebsocketClientSession.builder("wss://foo.nkjmlab.org/websocket/connection").open();
   * 
   *
   * @param uri destination uri
   * @return
   */
  public static WebsocketClientSession.Builder builder(String uri) {
    return new Builder(uri);
  }

  public void close() {
    try {
      session.close();
    } catch (IOException e) {
      log.error(e.getMessage());
    }
  }

  public void sendMessage(String message) {
    try {
      session.getBasicRemote().sendText(message);
    } catch (IOException e) {
      throw Try.rethrow(e);
    }
  }

  @Override
  public String toString() {
    return "WebsocketClient [session=" + session + "]";
  }


  public Session getSession() {
    return session;
  }

  private WebsocketClientSession(Session session) {
    this.session = session;
  }

  public static class Builder {
    private Consumer onOpenHandler = session -> {
      log.debug("onOpen={}", session);
    };
    private BiConsumer onMessageHandler = (message, session) -> {
      log.debug("[{}] onMessage {}", session.getId(), message);
    };

    private BiConsumer onCloseHandler = (closeReason, session) -> {
      log.debug("[{}] onClose {}", session.getId(), closeReason);
      try {
        session.close();
      } catch (IOException e) {
        log.error(e, e);
      }
    };

    private BiConsumer onErrorHandler = (th, session) -> {
      log.debug("[{}] onError {}", session.getId(), th);
    };
    private URI uri;



    private Builder(String uri) {
      this.uri = UriUtils.of(uri);

    }

    public void setOnOpenHandler(Consumer onOpenHandler) {
      this.onOpenHandler = onOpenHandler;
    }



    public void setOnMessageHandler(BiConsumer onMessageHandler) {
      this.onMessageHandler = onMessageHandler;
    }



    public void setOnCloseHandler(BiConsumer onCloseHandler) {
      this.onCloseHandler = onCloseHandler;
    }



    public void setOnErrorHandler(BiConsumer onErrorHandler) {
      this.onErrorHandler = onErrorHandler;
    }

    public WebsocketClientSession open() {
      try {
        WebsocketClientEndpoint endpoint = new WebsocketClientEndpoint(onOpenHandler,
            onMessageHandler, onCloseHandler, onErrorHandler);
        Session session = ContainerProvider.getWebSocketContainer().connectToServer(endpoint, uri);
        if (session.isOpen()) {
          log.debug("[{}] session is open to {}", session.getId(), uri);
        }
        return new WebsocketClientSession(session);
      } catch (DeploymentException | IOException e) {
        log.error(StringFormatter.format("error occurs {}", uri), e);
        throw Try.rethrow(e);
      }
    }

  }



}