Method execute

backend/app/executors/database.py:187–241

(self, config: dict, context: ExecutionContext)

        return "db_insert"
    
    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', {})
        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="插入数据不能为空")
            
            # 构建INSERT语句
            columns = ', '.join(f'`{k}`' for k in resolved_data.keys())
            placeholders = ', '.join(['%s'] * len(resolved_data))
            sql = f"INSERT INTO `{table}` ({columns}) VALUES ({placeholders})"
            
            with conn.cursor() as cursor:
                cursor.execute(sql, list(resolved_data.values()))
                last_id = cursor.lastrowid
            
            # 保存插入ID到变量
            if variable_name:
                context.variables[variable_name] = last_id
            
            return ModuleResult(
                success=True,
                message=f"插入成功，ID: {last_id}",
                data={"lastInsertId": last_id}
            )
        except Exception as e:
            return ModuleResult(success=False, error=f"插入失败: {str(e)}")


@register_executor