55 lines
1.1 KiB
PowerShell
55 lines
1.1 KiB
PowerShell
param([string]$RequestID)
|
|
|
|
if (-not $RequestID) { throw "RequestID is required" }
|
|
|
|
. .\tests\scripts\load-env.ps1 | Out-Null
|
|
|
|
$env:DUMP_RID = $RequestID
|
|
@'
|
|
package main
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"log"
|
|
"os"
|
|
"time"
|
|
|
|
"github.com/ClickHouse/clickhouse-go/v2"
|
|
|
|
"git.misaka.ren/M1saka/token_thief/db"
|
|
)
|
|
|
|
func main() {
|
|
log.SetFlags(0)
|
|
dsn := os.Getenv("CLICKHOUSE_URL")
|
|
if dsn == "" {
|
|
log.Fatal("CLICKHOUSE_URL not set")
|
|
}
|
|
rid := os.Getenv("DUMP_RID")
|
|
ctx, cancel := context.WithTimeout(context.Background(), 10*time.Second)
|
|
defer cancel()
|
|
opts, err := db.ClickHouseOptions(dsn)
|
|
if err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
conn, err := clickhouse.Open(opts)
|
|
if err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
defer conn.Close()
|
|
var body string
|
|
if err := conn.QueryRow(ctx,
|
|
`SELECT response_body FROM proxy_logs WHERE request_id = ? ORDER BY started_at DESC LIMIT 1`,
|
|
rid,
|
|
).Scan(&body); err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
fmt.Println(body)
|
|
}
|
|
'@ | Set-Content -Path tmp_dump.go -Encoding UTF8 -NoNewline
|
|
|
|
go run tmp_dump.go
|
|
Remove-Item tmp_dump.go -Force
|
|
Remove-Item Env:DUMP_RID -ErrorAction SilentlyContinue
|