Azure Function连接PostgreSQL异常：Excel更新数据库代码求助

核心问题梳理 #

你的代码存在几个关键问题导致崩溃：

• Blob流获取方式错误，无法正确读取Excel文件
• PostgreSQL连接池创建位置不当，引发资源泄漏与未定义错误
• Excel表头被误当作数据行处理，导致SQL更新异常
• 未处理工作表不存在、单元格值为空的边界情况

分步修复说明 #

1. 修正Blob流获取逻辑
   Azure Functions的Blob触发器中，Blob内容通过context.bindings[触发器名称]获取，而非context.bindingData.blobTrigger。替换流获取代码：

// 错误写法
const blobStream = context.bindingData.blobTrigger.openReadStream();
// 正确写法
const blobStream = context.bindings.storageBlobTrigger.openReadStream();

2. 将PostgreSQL连接池改为全局单例
   每次触发都创建新连接池会导致连接耗尽，将Pool初始化移到函数外部，复用连接资源：

// 移到handler外部，全局初始化一次
const pool = new Pool({
    user: '****',
    host: '***',
    database: '***',
    password: '****',
    port: 5432,
    ssl: {
        rejectUnauthorized: false
    }
});

3. 跳过Excel表头行
   默认eachRow会遍历所有行，包括表头，需从第二行开始处理数据：

worksheet.eachRow({ includeEmpty: false }, (row, rowNumber) => {
    if (rowNumber === 1) return; // 跳过表头
    // 后续处理逻辑
});

4. 增加边界情况处理

• 检查工作表是否存在，避免getWorksheet(1)返回undefined
• 处理单元格值为空的情况，给默认值或跳过无效行

5. 修复finally块的未定义风险
   原代码中如果Pool创建失败，finally块调用pool.end()会报错，现在Pool是全局的，无需每次触发关闭，只需在函数退出时（可选）处理，或者移除多余的pool.end()。

完整修复代码 #

const { app } = require('@azure/functions');
const ExcelJS = require('exceljs');
const { Pool } = require('pg');

// 全局初始化PostgreSQL连接池，复用连接
const pool = new Pool({
    user: '****',
    host: '***',
    database: '***',
    password: '****',
    port: 5432,
    ssl: {
        rejectUnauthorized: false // 注意：生产环境建议配置合法证书，关闭此选项存在安全风险
    }
});

app.storageBlob('storageBlobTrigger', {
    path: 'attachments/{name}',
    connection: 'MyStorageConnectionString',
    handler: async (context) => {
        let client;
        try {
            context.log(`Processing blob "${context.bindingData.name}" with size ${context.bindingData.length} bytes`);

            // 正确获取Blob流
            const blobStream = context.bindings.storageBlobTrigger.openReadStream();

            const workbook = new ExcelJS.Workbook();
            await workbook.xlsx.read(blobStream);

            // 检查工作表是否存在
            const worksheet = workbook.getWorksheet(1);
            if (!worksheet) {
                throw new Error('Excel文件中不存在工作表');
            }

            // 从连接池获取客户端
            client = await pool.connect();
            await client.query('BEGIN');

            const updates = [];
            // 遍历行，跳过表头，忽略空行
            worksheet.eachRow({ includeEmpty: false }, (row, rowNumber) => {
                if (rowNumber === 1) return; // 跳过表头行

                // 提取单元格值，处理空值
                const values = Array.from({ length: 8 }, (_, idx) => {
                    const cell = row.getCell(idx + 1);
                    return cell.value ?? null; // 空值替换为null
                });

                // 检查主键字段是否为空，避免无效更新
                if (!values[0]) {
                    context.log.warn(`Row ${rowNumber} 的product_name为空，跳过更新`);
                    return;
                }

                const updatePromise = client.query(`
                    UPDATE stock_overview SET
                    material_code = $2, producer_item_code = $3, description_id = $4, serial_number = $5,
                    total_stock = $6, unit = $7, warehouse = $8
                    WHERE product_name = $1;
                `, values);

                updates.push(updatePromise);
                updatePromise.then(result => {
                    context.log(`Row ${rowNumber} 更新完成，影响行数: ${result.rowCount}`);
                }).catch(err => {
                    context.log.error(`Row ${rowNumber} 更新失败: ${err.message}`);
                });
            });

            await Promise.all(updates);
            await client.query('COMMIT');
            context.log('所有数据更新完成，事务已提交');

        } catch (error) {
            context.log.error(`处理Excel文件失败: ${error.message}`);
            if (client) {
                await client.query('ROLLBACK').catch(rollbackErr => {
                    context.log.error(`回滚事务失败: ${rollbackErr.message}`);
                });
            }
            throw error;
        } finally {
            if (client) {
                client.release(); // 释放客户端回连接池
            }
            // 不要每次触发都关闭连接池，全局池保持复用
        }
    }
});

额外建议 #

• 生产环境中，建议将数据库凭据存储在Azure Key Vault中，通过环境变量或Functions配置读取，避免硬编码
• 开启PostgreSQL的SSL证书验证，不要长期设置rejectUnauthorized: false，可下载Azure提供的根证书配置
• 增加对Excel文件格式的校验（如扩展名、MIME类型），避免处理非Excel文件

内容的提问来源于stack exchange，提问作者Hossei Asifi