Management Api执行近4小时后停止上报数据的技术排查请求

问题背景 #

使用PowerShell脚本导出Microsoft 365 DLP日志时，前700MB数据（耗时3-4小时）导出正常，之后数据上报停止。怀疑是Access Token过期导致，但作为新手不会实现定时刷新逻辑，尝试过每小时刷新令牌的代码但无效，需要解决方法，同时排查其他可能原因。

当前令牌获取代码 #

$body = @{grant_type="client_credentials";resource=$APIResource;client_id=$AppClientID;client_secret=$ClientSecretValue}
Write-Host -ForegroundColor Blue -BackgroundColor white "Obtaining authentication token..." -NoNewline
try{
    $oauth = Invoke-RestMethod -Method Post -Uri "$loginURL/$tenantdomain/oauth2/token?api-version=1.0" -Body $body -ErrorAction Stop
    $OfficeToken = @{'Authorization'="$($oauth.token_type) $($oauth.access_token)"}
    Write-Host -ForegroundColor Green "Authentication token obtained"
} catch {
    write-host -ForegroundColor Red "FAILED"
    write-host -ForegroundColor Red "Invoke-RestMethod failed."
    Write-host -ForegroundColor Red $error[0]
    exit
}

无效的刷新尝试 #

曾用以下代码尝试每小时刷新令牌，但未解决问题：

$StopWatch = [System.Diagnostics.Stopwatch]::StartNew() if($StopWatch.Elapsed.TotalSeconds -ge 3599){$OfficeToken = Get-Token; $StopWatch.Restart()}

正确的令牌刷新实现 #

你的问题出在：秒表是从脚本启动开始计时，没有结合令牌实际的过期时间，且未在每次API请求前检查令牌状态。正确的做法是利用OAuth返回的expires_in字段（通常为3600秒），提前刷新令牌：

1. 封装令牌获取函数（保存过期时间） #

function Get-AuthToken {
    param(
        [string]$APIResource,
        [string]$AppClientID,
        [string]$ClientSecretValue,
        [string]$loginURL,
        [string]$tenantdomain
    )
    $body = @{
        grant_type    = "client_credentials"
        resource      = $APIResource
        client_id     = $AppClientID
        client_secret = $ClientSecretValue
    }
    Write-Host -ForegroundColor Blue -BackgroundColor white "Obtaining authentication token..." -NoNewline
    try{
        $oauth = Invoke-RestMethod -Method Post -Uri "$loginURL/$tenantdomain/oauth2/token?api-version=1.0" -Body $body -ErrorAction Stop
        # 提前5分钟刷新令牌，避免临界时间点失效
        return @{
            Token     = @{'Authorization' = "$($oauth.token_type) $($oauth.access_token)"}
            ExpiresAt = (Get-Date).AddSeconds($oauth.expires_in - 300)
        }
    } catch {
        write-host -ForegroundColor Red "FAILED"
        write-host -ForegroundColor Red "Invoke-RestMethod failed."
        Write-host -ForegroundColor Red $error[0]
        exit
    }
}

2. 初始化令牌信息 #

# 替换为你的实际变量
$authInfo = Get-AuthToken -APIResource $APIResource -AppClientID $AppClientID -ClientSecretValue $ClientSecretValue -loginURL $loginURL -tenantdomain $tenantdomain
$OfficeToken = $authInfo.Token

3. 每次API请求前检查并刷新令牌 #

在你调用DLP API的循环/代码前，加入以下检查：

# 检查令牌是否即将过期，是则刷新
if((Get-Date) -ge $authInfo.ExpiresAt){
    $authInfo = Get-AuthToken -APIResource $APIResource -AppClientID $AppClientID -ClientSecretValue $ClientSecretValue -loginURL $loginURL -tenantdomain $tenantdomain
    $OfficeToken = $authInfo.Token
}

# 执行DLP API请求示例
$dlpLogs = Invoke-RestMethod -Uri "你的DLP日志API地址" -Headers $OfficeToken -Method Get -ErrorAction Stop

其他可能的故障原因 #

如果不是令牌过期问题，排查以下方向：

• API限流：Microsoft 365 Compliance API有调用频率限制，超过阈值会被限流，导致请求失败。可查看API返回的错误信息（如429状态码）确认。
• 网络中断：长时间运行中出现网络波动、超时，导致请求中断且脚本未做重试处理。
• 内存不足：处理大体积日志时，PowerShell内存占用过高，导致脚本卡顿或崩溃。可尝试分批导出数据。
• API响应异常：脚本未处理API返回的错误状态码（如500、403），导致后续流程直接终止。

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