graphql.kickstart.spring.webflux.ReactiveWebSocketSubscriptionSession Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of graphql-kickstart-spring-webflux-javax Show documentation

GraphQL Spring Framework Boot

The newest version!

package graphql.kickstart.spring.webflux;

import graphql.kickstart.execution.subscriptions.DefaultSubscriptionSession;
import graphql.kickstart.execution.subscriptions.GraphQLSubscriptionMapper;
import java.util.Map;
import org.springframework.web.reactive.socket.WebSocketSession;

public class ReactiveWebSocketSubscriptionSession extends DefaultSubscriptionSession {

  private final WebSocketSession webSocketSession;
  private boolean opened = true;

  public ReactiveWebSocketSubscriptionSession(
      GraphQLSubscriptionMapper mapper, WebSocketSession webSocketSession) {
    super(mapper);
    this.webSocketSession = webSocketSession;
  }

  @Override
  public boolean isOpen() {
    return opened;
  }

  @Override
  public Map getUserProperties() {
    return webSocketSession.getAttributes();
  }

  @Override
  public String getId() {
    return webSocketSession.getId();
  }

  @Override
  public WebSocketSession unwrap() {
    return webSocketSession;
  }

  @Override
  public void close(final String reason) {
    super.close(reason);
    this.opened = false;
  }
}