深入解析：解决DynamoDB从CSV文件批量写入数据不完整的问题

params.Item = {
    "sc": {
        "S": item_str.split(",")[0] // 错误：文档客户端不需要 { "S": "..." }
    },
    "t": {
        "N": String(item_str.split(",")[1]) // 错误：文档客户端不需要 { "N": "..." }
    }
};

params.Item = {
    "sc": item_str.split(",")[0], // 正确：直接使用字符串
    "t": parseInt(item_str.split(",")[1]) // 正确：直接使用数字，确保类型为Number
};

const fileContents = fs.readFileSync("cities.csv", "utf8").split('\n').map(async (item_str) => {
    // ... DynamoDB PutCommand 逻辑 ...
    const response = await docClient.send(new PutCommand(params));
    // ...
});
// 此时 fileContents 只是一个 Promise 数组，而不是最终结果
// 如果没有 await，函数可能会在此处或之后立即退出

const fileContents = fs.readFileSync("cities.csv", "utf8").split('\n');
const putPromises = fileContents.map(async (item_str) => {
    // ...
    // 在这里构建 params.Item，确保使用正确的文档客户端格式
    params.Item = {
        "sc": item_str.split(",")[0],
        "t": parseInt(item_str.split(",")[1])
    };
    try {
        const response = await docClient.send(new PutCommand(params));
        console.log("写入成功:", response);
        return response; // 返回每个 PutCommand 的结果
    } catch (err) {
        console.error("写入错误:", err);
        throw err; // 抛出错误以便 Promise.all 可以捕获
    }
});

// 等待所有 Promise 完成
await Promise.all(putPromises);
console.log("所有数据写入完成。");

import * as fs from 'fs';
import { DynamoDBClient } from "@aws-sdk/client-dynamodb";
import { PutCommand, DynamoDBDocumentClient } from "@aws-sdk/lib-dynamodb";

export const handler = async (event, context) => { // 建议使用 context 而非 callback
    console.log("进入处理函数");

    const DDB = new DynamoDBClient({ region: 'us-east-1' });
    const docClient = DynamoDBDocumentClient.from(DDB);

    const tableName = "weather"; // 定义表名

    let fileContents;
    try {
        // 假设 cities.csv 文件在 Lambda 部署包的根目录
        fileContents = fs.readFileSync("/var/task/cities.csv", "utf8"); 
    } catch (readErr) {
        console.error("读取CSV文件失败:", readErr);
        return { statusCode: 500, body: JSON.stringify({ message: "无法读取CSV文件" }) };
    }

    const itemsToProcess = fileContents.split('\n').filter(line => line.trim() !== ''); // 过滤空行

    const putPromises = itemsToProcess.map(async (item_str) => {
        const parts = item_str.split(",");
        if (parts.length < 2) {
            console.warn(`跳过无效行: ${item_str}`);
            return null; // 返回 null 或抛出错误，根据需求处理
        }

        const params = {
            TableName: tableName,
            Item: {
                "sc": parts[0].trim(), // 城市代码，直接字符串
                "t": parseInt(parts[1].trim()) // 温度，直接数字
            },
            ReturnConsumedCapacity: "TOTAL"
        };

        try {
            const response = await docClient.send(new PutCommand(params));
            console.log(`成功写入项目: ${item_str}`, response.ConsumedCapacity);
            return response;
        } catch (err) {
            console.error(`写入项目失败: ${item_str}`, err);
            // 可以选择在这里重新抛出错误，或返回一个表示失败的对象
            throw new Error(`Failed to write item ${item_str}: ${err.message}`);
        }
    });

    try {
        // 等待所有写入操作完成
        await Promise.all(putPromises);
        console.log("所有数据已成功写入DynamoDB。");
        return { statusCode: 200, body: JSON.stringify({ message: "数据导入成功" }) };
    } catch (allErr) {
        console.error("部分或全部数据写入失败:", allErr);
        return { statusCode: 500, body: JSON.stringify({ message: "数据导入失败", error: allErr.message }) };
    }
};