snowflake_sample_data.rb

# frozen_string_literal: true

require_relative 'common_sample_interface.rb' # Creates/uses test_data table in the db you point to
# Assumes you have access to snowflake_sample_data https://docs.snowflake.net/manuals/user-guide/sample-data.html
# Set env vars: SNOWFLAKE_TEST_ACCOUNT, SNOWFLAKE_TEST_USER, SNOWFLAKE_TEST_PASSWORD, SNOWFLAKE_TEST_WAREHOUSE
# optionally set SNOWFLAKE_TEST_SCHEMA, SNOWFLAKE_TEST_ROLE

class SnowflakeSampleData < CommonSampleInterface
  def initialize
    super('SNOWFLAKE_SAMPLE_DATA')
  end

  def get_customer_names(where = "c_last_name = 'Flowers'")
    raise('db not connected') unless @db_pointer

    query = 'select c_first_name, c_last_name from "CUSTOMER"'
    query += " where #{where}" if where

    GoSnowflakeClient.select(@db_pointer, query) { |row| @logger.info("#{row[0]} #{row[1]}") }
  end

  # @example process_unshipped_web_sales {|row| check_shipping_queue(row)}
  def process_unshipped_web_sales(limit = 1_000, &block)
    raise('db not connected') unless @db_pointer

    query = <<~QUERY
      select c_first_name, c_last_name, ws_sold_date_sk, ws_list_price
      from "CUSTOMER"
      inner join "WEB_SALES"
      ON c_customer_sk = ws_bill_customer_sk
      where ws_ship_date_sk is null
      #{"limit #{limit}" if limit}
    QUERY

    GoSnowflakeClient.select(@db_pointer, query, &block)
  end
end