net.snowflake.ingest.streaming.internal.ChannelsStatusRequest Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of snowflake-ingest-sdk Show documentation
Snowflake Ingest SDK
The newest version!
/*
 * Copyright (c) 2021-2024 Snowflake Computing Inc. All rights reserved.
 */

package net.snowflake.ingest.streaming.internal;

import com.fasterxml.jackson.annotation.JsonProperty;
import java.util.List;
import java.util.stream.Collectors;
import net.snowflake.ingest.utils.Utils;

/** Class to deserialize a request from a channel status request */
class ChannelsStatusRequest implements IStreamingIngestRequest {

  // Used to deserialize a channel request
  static class ChannelStatusRequestDTO {
    // Database name
    private final String databaseName;

    // Schema name
    private final String schemaName;

    // Table Name
    private final String tableName;

    // Channel Name
    private final String channelName;

    // Client Sequencer
    private final Long clientSequencer;

    ChannelStatusRequestDTO(SnowflakeStreamingIngestChannelInternal channel) {
      this.channelName = channel.getName();
      this.databaseName = channel.getDBName();
      this.schemaName = channel.getSchemaName();
      this.tableName = channel.getTableName();
      this.clientSequencer = channel.getChannelSequencer();
    }

    @JsonProperty("table")
    String getTableName() {
      return tableName;
    }

    @JsonProperty("database")
    String getDatabaseName() {
      return databaseName;
    }

    @JsonProperty("schema")
    String getSchemaName() {
      return schemaName;
    }

    @JsonProperty("channel_name")
    String getChannelName() {
      return channelName;
    }

    @JsonProperty("client_sequencer")
    Long getClientSequencer() {
      return clientSequencer;
    }
  }

  // Channels in request
  private List channels;

  // Snowflake role used by client
  private String role;

  @JsonProperty("role")
  public String getRole() {
    return role;
  }

  @JsonProperty("role")
  public void setRole(String role) {
    this.role = role;
  }

  @JsonProperty("channels")
  void setChannels(List channels) {
    this.channels = channels;
  }

  @JsonProperty("channels")
  List getChannels() {
    return channels;
  }

  @Override
  public String getStringForLogging() {
    return String.format(
        "ChannelsStatusRequest(role=%s, channels=[%s])",
        role,
        channels.stream()
            .map(
                r ->
                    Utils.getFullyQualifiedChannelName(
                        r.getDatabaseName(),
                        r.getSchemaName(),
                        r.getTableName(),
                        r.getChannelName()))
            .collect(Collectors.joining(", ")));
  }
}