class Google::Cloud::Bigquery::Data

def self.from_gapi_json gapi_json, table_gapi, job_gapi, service, format_options_use_int64_timestamp
  rows = gapi_json[:rows] || []
  rows = Convert.format_rows rows, table_gapi.schema.fields unless rows.empty?
  data = new rows
  data.table_gapi = table_gapi
  data.gapi_json = gapi_json
  data.job_gapi = job_gapi
  data.service = service
  data.format_options_use_int64_timestamp = format_options_use_int64_timestamp
  data
end

def all request_limit: nil, &block
  request_limit = request_limit.to_i if request_limit
  return enum_for :all, request_limit: request_limit unless block_given?
  results = self
  loop do
    results.each(&block)
    if request_limit
      request_limit -= 1
      break if request_limit.negative?
    end
    break unless results.next?
    results = results.next
  end
end

def ddl?
  [
    "ALTER_TABLE",
    "CREATE_MODEL",
    "CREATE_TABLE",
    "CREATE_TABLE_AS_SELECT",
    "CREATE_VIEW",
    "DROP_MODEL",
    "DROP_TABLE",
    "DROP_VIEW"
  ].include? statement_type
end

def ddl_operation_performed
  job_gapi&.statistics&.query&.ddl_operation_performed
end

def ddl_target_routine
  ensure_service!
  routine = job_gapi&.statistics&.query&.ddl_target_routine
  return nil if routine.nil?
  Google::Cloud::Bigquery::Routine.new_reference_from_gapi routine, service
end

def ddl_target_table
  ensure_service!
  table = job_gapi&.statistics&.query&.ddl_target_table
  return nil if table.nil?
  Google::Cloud::Bigquery::Table.new_reference_from_gapi table, service
end

def deleted_row_count
  job_gapi&.statistics&.query&.dml_stats&.deleted_row_count
end

def dml?
  [
    "INSERT",
    "UPDATE",
    "MERGE",
    "DELETE"
  ].include? statement_type
end

def ensure_service!
  raise "Must have active connection" unless service
end

def etag
  @gapi_json[:etag]
end

def fields
  schema.fields
end

def headers
  schema.headers
end

def initialize arr = []
  @service = nil
  @table_gapi = nil
  @gapi_json = nil
  super arr
end

def inserted_row_count
  job_gapi&.statistics&.query&.dml_stats&.inserted_row_count
end

def kind
  @gapi_json[:kind]
end

def next
  return nil unless next?
  ensure_service!
  data_json = service.list_tabledata \
    @table_gapi.table_reference.dataset_id,
    @table_gapi.table_reference.table_id,
    token: token,
    format_options_use_int64_timestamp: @format_options_use_int64_timestamp
  self.class.from_gapi_json data_json, @table_gapi, job_gapi, @service, @format_options_use_int64_timestamp
end

def next?
  !token.nil?
end

def num_dml_affected_rows
  job_gapi&.statistics&.query&.num_dml_affected_rows
end

def param_types
  schema.param_types
end

def schema
  return nil unless @table_gapi
  Schema.from_gapi(@table_gapi.schema).freeze
end

def statement_type
  job_gapi&.statistics&.query&.statement_type
end

def token
  @gapi_json[:pageToken]
end

def total
  Integer @gapi_json[:totalRows]
rescue StandardError
  nil
end

def updated_row_count
  job_gapi&.statistics&.query&.dml_stats&.updated_row_count
end