要在DynamoDB中通过单个主键获取所有范围键，可以使用query操作。以下是一个在AWS Step Functions中使用DynamoDB查询的示例代码：

{
  "Comment": "Get all range keys for a given partition key from DynamoDB",
  "StartAt": "Query DynamoDB",
  "States": {
    "Query DynamoDB": {
      "Type": "Task",
      "Resource": "arn:aws:states:::dynamodb:query",
      "Parameters": {
        "TableName": "your-dynamodb-table-name",
        "KeyConditionExpression": "partitionKey = :pk",
        "ExpressionAttributeValues": {
          ":pk": { "S.$": "$.partitionKeyValue" }
        },
        "ProjectionExpression": "rangeKey",
        "IndexName": "your-index-name", // If using a secondary index
        "Limit": 100 // Adjust the limit as per your requirement
      },
      "ResultPath": "$.dynamodbResult",
      "Next": "Process Results"
    },
    "Process Results": {
      "Type": "Pass",
      "Result": "$.dynamodbResult.Items",
      "End": true
    }
  }
}

在上面的示例中，需要替换以下值：

• "your-dynamodb-table-name"：替换为你的DynamoDB表名。
• "your-index-name"：如果使用了二级索引，则替换为你的索引名。
• "partitionKeyValue"：替换为你的要查询的分区键值。

这个示例使用了AWS Step Functions的Query操作来查询DynamoDB表并获取所有范围键。查询结果存储在"dynamodbResult.Items"字段中，可以在下一步中进行处理。

注意：上述代码仅为示例，实际使用时请根据自己的需求进行相应的修改和配置。