PSFramework

0.9.0.0

internal/scripts/async-logging.TempPoint.ps1

$scriptBlock = {
    try
    {
        $script:ScriptName = 'PSFramework.Logging'
        
        #region Helper Functions
        function Clean-ErrorXml
        {
            [CmdletBinding()]
            Param (
                $Path
            )
            
            $totalLength = $Null
            $files = Get-ChildItem -Path $Path.FullName -Filter "$($env:ComputerName)_$($pid)_error_*.xml" | Sort-Object LastWriteTime
            $totalLength = $files | Measure-Object Length -Sum | Select-Object -ExpandProperty Sum
            if (([PSFramework.Message.LogHost]::MaxErrorFileBytes) -gt $totalLength) { return }
            
            $removed = 0
            foreach ($file in $files)
            {
                $removed += $file.Length
                Remove-Item -Path $file.FullName -Force -Confirm:$false
                
                if (($totalLength - $removed) -lt ([PSFramework.Message.LogHost]::MaxErrorFileBytes)) { break }
            }
        }
        
        function Clean-MessageLog
        {
            [CmdletBinding()]
            Param (
                $Path
            )
            
            if ([PSFramework.Message.LogHost]::MaxMessagefileCount -eq 0) { return }
            
            $files = Get-ChildItem -Path $Path.FullName -Filter "$($env:ComputerName)_$($pid)_message_*.log" | Sort-Object LastWriteTime
            if (([PSFramework.Message.LogHost]::MaxMessagefileCount) -ge $files.Count) { return }
            
            $removed = 0
            foreach ($file in $files)
            {
                $removed++
                Remove-Item -Path $file.FullName -Force -Confirm:$false
                
                if (($files.Count - $removed) -le ([PSFramework.Message.LogHost]::MaxMessagefileCount)) { break }
            }
        }
        
        function Clean-GlobalLog
        {
            [CmdletBinding()]
            Param (
                $Path
            )
            
            # Kill too old files
            Get-ChildItem -Path "$($Path.FullName)\*" | Where-Object Name -Match "^$([regex]::Escape($env:ComputerName))_.+" | Where-Object LastWriteTime -LT ((Get-Date) - ([PSFramework.Message.LogHost]::MaxLogFileAge)) | Remove-Item -Force -Confirm:$false
            
            # Handle the global overcrowding
            $files = Get-ChildItem -Path "$($Path.FullName)\*" | Where-Object Name -Match "^$([regex]::Escape($env:ComputerName))_.+" | Sort-Object LastWriteTime
            if (-not ($files)) { return }
            $totalLength = $files | Measure-Object Length -Sum | Select-Object -ExpandProperty Sum
            
            if (([PSFramework.Message.LogHost]::MaxTotalFolderSize) -gt $totalLength) { return }
            
            $removed = 0
            foreach ($file in $files)
            {
                $removed += $file.Length
                Remove-Item -Path $file.FullName -Force -Confirm:$false
                
                if (($totalLength - $removed) -lt ([PSFramework.Message.LogHost]::MaxTotalFolderSize)) { break }
            }
        }
        #endregion Helper Functions
        
        
        while ($true)
        {
            # This portion is critical to gracefully closing the script
            if ([PSFramework.Runspace.RunspaceHost]::Runspaces[$ScriptName.ToLower()].State -notlike "Running")
            {
                break
            }
            
            $path = [PSFramework.Message.LogHost]::LoggingPath
            if (-not (Test-Path $path))
            {
                $root = New-Item $path -ItemType Directory -Force -ErrorAction Stop
            }
            else { $root = Get-Item -Path $path }
            
            [int]$num_Error = (Get-ChildItem -Path $Path.FullName -Filter "$($env:ComputerName)_$($pid)_error_*.xml" | Sort-Object LastWriteTime -Descending | Select-Object -First 1 -ExpandProperty Name | Select-String -Pattern "(\d+)" -AllMatches).Matches[1].Value
            [int]$num_Message = (Get-ChildItem -Path $Path.FullName -Filter "$($env:ComputerName)_$($pid)_message_*.log" | Sort-Object LastWriteTime -Descending | Select-Object -First 1 -ExpandProperty Name | Select-String -Pattern "(\d+)" -AllMatches).Matches[1].Value
            if (-not ($num_Error)) { $num_Error = 0 }
            if (-not ($num_Message)) { $num_Message = 0 }
            
            #region Process Errors
            while ([PSFramework.Message.LogHost]::OutQueueError.Count -gt 0)
            {
                $num_Error++
                
                $Record = $null
                [PSFramework.Message.LogHost]::OutQueueError.TryDequeue([ref]$Record)
                
                if ($Record)
                {
                    $Record | Export-Clixml -Path "$($root.FullName)\$($env:ComputerName)_$($pid)_error_$($num_Error).xml" -Depth 3
                }
                
                Clean-ErrorXml -Path $root
            }
            #endregion Process Errors
            
            #region Process Logs
            while ([PSFramework.Message.LogHost]::OutQueueLog.Count -gt 0)
            {
                $CurrentFile = "$($root.FullName)\$($env:ComputerName)_$($pid)_message_$($num_Message).log"
                if (Test-Path $CurrentFile)
                {
                    $item = Get-Item $CurrentFile
                    if ($item.Length -gt ([PSFramework.Message.LogHost]::MaxMessagefileBytes))
                    {
                        $num_Message++
                        $CurrentFile = "$($root.FullName)\$($env:ComputerName)_$($pid)_message_$($num_Message).log"
                    }
                }
                
                $Entry = $null
                [PSFramework.Message.LogHost]::OutQueueLog.TryDequeue([ref]$Entry)
                if ($Entry)
                {
                    Add-Content -Path $CurrentFile -Value (ConvertTo-Csv -InputObject $Entry -NoTypeInformation)[1]
                }
            }
            #endregion Process Logs
            
            Clean-MessageLog -Path $root
            Clean-GlobalLog -Path $root
            
            Start-Sleep -Seconds 5
        }
    }
    catch
    {
        [PSFramework.Runspace.RunspaceHost]::Runspaces[$ScriptName.ToLower()].Kill()
    }
}

Register-PSFRunspace -ScriptBlock $scriptBlock -Name 'PSFramework.Logging'
Start-PSFRunspace -Name 'PSFramework.Logging'