如何判断PowerShell日志函数的调用场景并适配日志命名？

下午好，感恩节快乐！针对你的问题，完全可以通过添加判断逻辑来解决，以下是具体实现方案：

当直接在控制台调用Write-Log时，$MyInvocation.PSCommandPath为空值，此时Get-ChildItem $MyInvocation.PSCommandPath等价于不带参数执行Get-ChildItem，会获取当前目录下所有文件的名称，最终拼接出的文件名过长导致报错。

修改日志文件名的生成逻辑，区分脚本调用和控制台直接调用两种场景：

• 脚本内调用：沿用原逻辑，使用脚本名称作为日志文件名
• 控制台直接调用：提示用户输入自定义日志名，或使用默认命名（如ConsoleSession.log）

function Write-Log {
    param(
        [Parameter(Mandatory=$true)]
        [string]$Message
    )

    # 生成日志文件名
    $logFileName = if (-not [string]::IsNullOrEmpty($MyInvocation.PSCommandPath)) {
        # 脚本内调用：基于脚本名称生成日志文件
        (Get-Item -Path $MyInvocation.PSCommandPath).Name.Replace(".ps1", ".log")
    }
    else {
        # 控制台直接调用：获取用户输入或使用默认名
        $inputName = Read-Host "请输入日志文件名（直接回车使用默认名：ConsoleSession.log）"
        if ([string]::IsNullOrEmpty($inputName)) {
            "ConsoleSession.log"
        }
        else {
            # 确保文件名以.log后缀结尾
            if ($inputName -notlike "*.log") {
                "$inputName.log"
            }
            else {
                $inputName
            }
        }
    }

    # 日志写入逻辑（可根据你的原有逻辑调整）
    $Message | Out-File -FilePath $logFileName -Append -Encoding utf8
}

• 用[string]::IsNullOrEmpty($MyInvocation.PSCommandPath)准确判断场景：脚本内调用时该变量会包含脚本的完整路径，控制台直接调用时为空
• 控制台场景下通过Read-Host交互获取用户输入，兼顾灵活性和易用性
• 将原代码中的Get-ChildItem替换为Get-Item，更高效（已知具体路径时无需遍历目录）
• 添加了后缀检查，避免用户输入无.log后缀的文件名

1. 脚本内调用（原有正常场景）：

PS> . .\test.ps1
PS> Write-Log "Hello World"

仍会生成test.log日志文件，逻辑保持一致。

2. 控制台直接调用：

PS> Write-Log "Hello World"
请输入日志文件名（直接回车使用默认名：ConsoleSession.log）:

• 输入自定义名称（如myConsoleLog），会生成myConsoleLog.log
• 直接回车，会生成ConsoleSession.log

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