easyai
e07a997aa9
兼容接口现在以入口协议作为最终响应协议，同协议保留官方 Wire 响应，跨协议统一转换成功、任务状态与错误结构。 同时修正异步提交状态边界，持久化兼容公开任务标识和官方提交响应，并新增迁移、流式响应及协议契约测试。 验证：go vet ./...；go test ./...；govulncheck ./...；pnpm lint；pnpm test；pnpm build；pnpm audit --audit-level high；pnpm openapi；全部 CI 脚本。
59 lines
1.6 KiB
Go
59 lines
1.6 KiB
Go
package store
|
|
|
|
import (
|
|
"context"
|
|
"encoding/json"
|
|
"strings"
|
|
)
|
|
|
|
type CompatibilitySubmission struct {
|
|
TargetProtocol string
|
|
PublicID string
|
|
SourceProtocol string
|
|
HTTPStatus int
|
|
Headers map[string]any
|
|
Body map[string]any
|
|
}
|
|
|
|
func (s *Store) SetTaskCompatibilitySubmission(ctx context.Context, taskID string, submission CompatibilitySubmission) error {
|
|
headers, _ := json.Marshal(emptyObjectIfNil(submission.Headers))
|
|
body, _ := json.Marshal(emptyObjectIfNil(submission.Body))
|
|
_, err := s.pool.Exec(ctx, `
|
|
UPDATE gateway_tasks
|
|
SET compatibility_protocol = NULLIF($2, ''),
|
|
compatibility_public_id = NULLIF($3, ''),
|
|
compatibility_source_protocol = NULLIF($4, ''),
|
|
compatibility_submit_http_status = NULLIF($5, 0),
|
|
compatibility_submit_headers = $6::jsonb,
|
|
compatibility_submit_body = $7::jsonb,
|
|
updated_at = now()
|
|
WHERE id = $1::uuid`,
|
|
taskID,
|
|
strings.TrimSpace(submission.TargetProtocol),
|
|
strings.TrimSpace(submission.PublicID),
|
|
strings.TrimSpace(submission.SourceProtocol),
|
|
submission.HTTPStatus,
|
|
string(headers),
|
|
string(body),
|
|
)
|
|
return err
|
|
}
|
|
|
|
func (s *Store) GetCompatibilityTask(ctx context.Context, protocol string, publicID string) (GatewayTask, error) {
|
|
task, err := scanGatewayTask(s.pool.QueryRow(ctx, `
|
|
SELECT `+gatewayTaskColumns+`
|
|
FROM gateway_tasks
|
|
WHERE compatibility_protocol = $1
|
|
AND compatibility_public_id = $2
|
|
LIMIT 1`, strings.TrimSpace(protocol), strings.TrimSpace(publicID)))
|
|
if err != nil {
|
|
return GatewayTask{}, err
|
|
}
|
|
attempts, err := s.ListTaskAttempts(ctx, task.ID)
|
|
if err != nil {
|
|
return GatewayTask{}, err
|
|
}
|
|
task.Attempts = attempts
|
|
return task, nil
|
|
}
|