使用Ruby删除DynamoDB表中的所有项目

时间:2016-12-20 20:23:00

标签: ruby amazon-web-services amazon-dynamodb aws-sdk-ruby

我正在尝试编写一个简单的ruby脚本来删除DynamoDB表中的所有项目,但是我无法理解传递给“delete_items”的哪个参数,这是我到目前为止所做的:

dynamoDB = Aws::DynamoDB::Resource.new(region: 'us-west-2')

dynamoDB.tables.each do |table|
  puts "Table #{table.name}"
  scan_output = table.scan({
    select: "ALL_ATTRIBUTES"
    })

  scan_output.items.each do |item|
    keys = item.keys
    table.delete_item({
      key: ???
    })
  end
end 

我尝试传递项目或item.keys - 两者都不起作用。

谢谢!

2 个答案:

答案 0 :(得分:1)

以下是扫描和删除DynamoDB表中所有项目的代码,但我不确定为什么不能删除表格并重新创建,如果您要删除表格中的所有项目。

请注意,这是不推荐的方法,除非您有一些非常具体的用例。当代码从表中读取项目然后删除项目时,这将花费您。

<强>代码: -

您可能需要更改以下代码中的表名和键值。在下面的代码中,使用的表名为files,其键值为fileName

如果您同时拥有分区键和排序键,则需要同时设置这两个值。 files表只有分区键。

#! /usr/bin/ruby

require "aws-sdk-core"

# Configure SDK

# use credentials file at .aws/credentials
Aws.config[:credentials] = Aws::SharedCredentials.new
Aws.config[:region] = "us-west-2"

# point to DynamoDB Local, comment out this line to use real DynamoDB
Aws.config[:dynamodb] = { endpoint: "http://localhost:8000" }

dynamodb = Aws::DynamoDB::Client.new

tableName = "files"

scanParams = {
  table_name: tableName
}

puts "Scanning files table."

begin
  loop do
    result = dynamodb.scan(scanParams)

    result.items.each{|files|
      puts "Item :" + "#{files}"
      puts "Going to delete item :" + "#{files["fileName"]}"

      deleteParams = {
        table_name: tableName,
        key: {
          fileName: files["fileName"]

        }
      }
      begin
        deleteResult = dynamodb.delete_item(deleteParams)
        puts "Deleted item." + files["fileName"]            

      rescue  Aws::DynamoDB::Errors::ServiceError => error
        puts "Unable to delete item:"
        puts "#{error.message}"
      end

    }


    break if result.last_evaluated_key.nil?
    puts "Scanning for more..."
    scanParams[:exclusive_start_key] = result.last_evaluated_key

  end

rescue  Aws::DynamoDB::Errors::ServiceError => error
  puts "Unable to scan:"
  puts "#{error.message}"
end

答案 1 :(得分:1)

我最终编写了这个脚本,删除了所有表中的所有记录(对于大多数情况来说不是很有用,但对我来说这正是我所需要的,因为我在专用的测试帐户中使用它):

#!/usr/bin/env ruby

require 'aws-sdk'

dynamoDB = Aws::DynamoDB::Resource.new(region: 'us-west-2')

dynamoDB.tables.each do |table|
  puts "Table #{table.name}"
  scan_output = table.scan({
    select: "ALL_ATTRIBUTES"
    })

  scan_output.items.each do |item|
    item_key = Hash.new
    table.key_schema.each do |k|
      item_key[k.attribute_name] = item[k.attribute_name]
    end
    table.delete_item({
      key: item_key
    })
  end
end