Method execute

backend/app/executors/database.py:252–307

(self, config: dict, context: ExecutionContext)

        return "db_update"
    
    async def execute(self, config: dict, context: ExecutionContext) -> ModuleResult:
        connection_name = context.resolve_value(config.get('connectionName', 'default'))
        table = context.resolve_value(config.get('table', ''))
        data = config.get('data', {})
        where = context.resolve_value(config.get('where', ''))
        variable_name = context.resolve_value(config.get('variableName', ''))
        
        connections = get_db_connections(context)
        conn = connections.get(connection_name)
        
        if not conn:
            return ModuleResult(
                success=False, 
                error=f"数据库连接 '{connection_name}' 不存在，请先使用「连接数据库」模块"
            )
        
        if not table:
            return ModuleResult(success=False, error="请指定表名")
        
        try:
            # 解析数据
            if isinstance(data, str):
                import json
                data = json.loads(context.resolve_value(data))
            
            # 解析数据中的变量
            resolved_data = {}
            for key, value in data.items():
                if isinstance(value, str):
                    resolved_data[key] = context.resolve_value(value)
                else:
                    resolved_data[key] = value
            
            if not resolved_data:
                return ModuleResult(success=False, error="更新数据不能为空")
            
            # 构建UPDATE语句
            set_clause = ', '.join(f'`{k}` = %s' for k in resolved_data.keys())
            sql = f"UPDATE `{table}` SET {set_clause}"
            if where:
                sql += f" WHERE {where}"
            
            with conn.cursor() as cursor:
                affected_rows = cursor.execute(sql, list(resolved_data.values()))
            
            # 保存影响行数到变量
            if variable_name:
                context.variables[variable_name] = affected_rows
            
            return ModuleResult(
                success=True,
                message=f"更新成功，影响 {affected_rows} 行",
                data={"affectedRows": affected_rows}
            )
        except Exception as e:
            return ModuleResult(success=False, error=f"更新失败: {str(e)}")