72 lines
1.6 KiB
PowerShell
72 lines
1.6 KiB
PowerShell
|
param(
|
||
|
|
[string]$BaseUrl = "http://localhost:8880",
|
||
|
|
[string]$WorkerId = "local-worker-01",
|
||
|
|
[string]$WorkerToken = "local-worker-dev-token"
|
||
|
|
)
|
||
|
|
|
||
|
|
$headers = @{
|
||
|
|
"X-Worker-Token" = $WorkerToken
|
||
|
|
}
|
||
|
|
|
||
|
|
$pollBody = @{
|
||
|
|
workerId = $WorkerId
|
||
|
|
platforms = @("douyin", "xhs", "wxvideo", "kuaishou", "bilibili", "weibo")
|
||
|
|
maxTasks = 5
|
||
|
|
} | ConvertTo-Json
|
||
|
|
|
||
|
|
$tasks = Invoke-RestMethod `
|
||
|
|
-Uri "$BaseUrl/v1/media/workers/poll" `
|
||
|
|
-Method Post `
|
||
|
|
-Headers $headers `
|
||
|
|
-ContentType "application/json" `
|
||
|
|
-Body $pollBody
|
||
|
|
|
||
|
|
foreach ($task in $tasks) {
|
||
|
|
if ($task.taskType -eq "SYNC_COMMENTS") {
|
||
|
|
$result = @{
|
||
|
|
status = "SUCCESS"
|
||
|
|
result = @{
|
||
|
|
comments = @(
|
||
|
|
@{
|
||
|
|
platform = $task.platform
|
||
|
|
accountId = $task.accountId
|
||
|
|
userName = "测试用户"
|
||
|
|
content = "这个设备多少钱？"
|
||
|
|
likedCount = 12
|
||
|
|
intent = "sales_lead"
|
||
|
|
}
|
||
|
|
)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
} elseif ($task.taskType -eq "SYNC_METRICS") {
|
||
|
|
$result = @{
|
||
|
|
status = "SUCCESS"
|
||
|
|
result = @{
|
||
|
|
metrics = @{
|
||
|
|
views = 12000
|
||
|
|
likes = 860
|
||
|
|
comments = 56
|
||
|
|
collects = 128
|
||
|
|
shares = 43
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
} else {
|
||
|
|
$result = @{
|
||
|
|
status = "SUCCESS"
|
||
|
|
result = @{
|
||
|
|
message = "local worker mock completed"
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
Invoke-RestMethod `
|
||
|
|
-Uri "$BaseUrl/v1/media/workers/tasks/$($task.id)/result" `
|
||
|
|
-Method Post `
|
||
|
|
-Headers $headers `
|
||
|
|
-ContentType "application/json" `
|
||
|
|
-Body ($result | ConvertTo-Json -Depth 10) | Out-Null
|
||
|
|
|
||
|
|
Write-Host "completed $($task.id) $($task.taskType) $($task.platform)"
|
||
|
|
}
|