7efa8b54c4
新增 DB/Redis 连接池配置与校验，并补充单测 网关请求体大小限制与 413 处理 HTTP/req 客户端池化并调整上游连接池默认值 并发槽位改为 ZSET+Lua 与指数退避 用量统计改 SQL 聚合并新增索引迁移 计费缓存写入改工作池并补测试/基准 测试: 在 backend/ 下运行 go test ./...
105 lines
3.4 KiB
Go
105 lines
3.4 KiB
Go
package repository
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"time"
|
|
|
|
"github.com/Wei-Shaw/sub2api/internal/pkg/geminicli"
|
|
"github.com/Wei-Shaw/sub2api/internal/service"
|
|
|
|
"github.com/imroc/req/v3"
|
|
)
|
|
|
|
type geminiCliCodeAssistClient struct {
|
|
baseURL string
|
|
}
|
|
|
|
func NewGeminiCliCodeAssistClient() service.GeminiCliCodeAssistClient {
|
|
return &geminiCliCodeAssistClient{baseURL: geminicli.GeminiCliBaseURL}
|
|
}
|
|
|
|
func (c *geminiCliCodeAssistClient) LoadCodeAssist(ctx context.Context, accessToken, proxyURL string, reqBody *geminicli.LoadCodeAssistRequest) (*geminicli.LoadCodeAssistResponse, error) {
|
|
if reqBody == nil {
|
|
reqBody = defaultLoadCodeAssistRequest()
|
|
}
|
|
|
|
var out geminicli.LoadCodeAssistResponse
|
|
resp, err := createGeminiCliReqClient(proxyURL).R().
|
|
SetContext(ctx).
|
|
SetHeader("Authorization", "Bearer "+accessToken).
|
|
SetHeader("Content-Type", "application/json").
|
|
SetHeader("User-Agent", geminicli.GeminiCLIUserAgent).
|
|
SetBody(reqBody).
|
|
SetSuccessResult(&out).
|
|
Post(c.baseURL + "/v1internal:loadCodeAssist")
|
|
if err != nil {
|
|
fmt.Printf("[CodeAssist] LoadCodeAssist request error: %v\n", err)
|
|
return nil, fmt.Errorf("request failed: %w", err)
|
|
}
|
|
if !resp.IsSuccessState() {
|
|
body := geminicli.SanitizeBodyForLogs(resp.String())
|
|
fmt.Printf("[CodeAssist] LoadCodeAssist failed: status %d, body: %s\n", resp.StatusCode, body)
|
|
return nil, fmt.Errorf("loadCodeAssist failed: status %d, body: %s", resp.StatusCode, body)
|
|
}
|
|
fmt.Printf("[CodeAssist] LoadCodeAssist success: status %d, response: %+v\n", resp.StatusCode, out)
|
|
return &out, nil
|
|
}
|
|
|
|
func (c *geminiCliCodeAssistClient) OnboardUser(ctx context.Context, accessToken, proxyURL string, reqBody *geminicli.OnboardUserRequest) (*geminicli.OnboardUserResponse, error) {
|
|
if reqBody == nil {
|
|
reqBody = defaultOnboardUserRequest()
|
|
}
|
|
|
|
fmt.Printf("[CodeAssist] OnboardUser request body: %+v\n", reqBody)
|
|
|
|
var out geminicli.OnboardUserResponse
|
|
resp, err := createGeminiCliReqClient(proxyURL).R().
|
|
SetContext(ctx).
|
|
SetHeader("Authorization", "Bearer "+accessToken).
|
|
SetHeader("Content-Type", "application/json").
|
|
SetHeader("User-Agent", geminicli.GeminiCLIUserAgent).
|
|
SetBody(reqBody).
|
|
SetSuccessResult(&out).
|
|
Post(c.baseURL + "/v1internal:onboardUser")
|
|
if err != nil {
|
|
fmt.Printf("[CodeAssist] OnboardUser request error: %v\n", err)
|
|
return nil, fmt.Errorf("request failed: %w", err)
|
|
}
|
|
if !resp.IsSuccessState() {
|
|
body := geminicli.SanitizeBodyForLogs(resp.String())
|
|
fmt.Printf("[CodeAssist] OnboardUser failed: status %d, body: %s\n", resp.StatusCode, body)
|
|
return nil, fmt.Errorf("onboardUser failed: status %d, body: %s", resp.StatusCode, body)
|
|
}
|
|
fmt.Printf("[CodeAssist] OnboardUser success: status %d, response: %+v\n", resp.StatusCode, out)
|
|
return &out, nil
|
|
}
|
|
|
|
func createGeminiCliReqClient(proxyURL string) *req.Client {
|
|
return getSharedReqClient(reqClientOptions{
|
|
ProxyURL: proxyURL,
|
|
Timeout: 30 * time.Second,
|
|
})
|
|
}
|
|
|
|
func defaultLoadCodeAssistRequest() *geminicli.LoadCodeAssistRequest {
|
|
return &geminicli.LoadCodeAssistRequest{
|
|
Metadata: geminicli.LoadCodeAssistMetadata{
|
|
IDEType: "ANTIGRAVITY",
|
|
Platform: "PLATFORM_UNSPECIFIED",
|
|
PluginType: "GEMINI",
|
|
},
|
|
}
|
|
}
|
|
|
|
func defaultOnboardUserRequest() *geminicli.OnboardUserRequest {
|
|
return &geminicli.OnboardUserRequest{
|
|
TierID: "LEGACY",
|
|
Metadata: geminicli.LoadCodeAssistMetadata{
|
|
IDEType: "ANTIGRAVITY",
|
|
Platform: "PLATFORM_UNSPECIFIED",
|
|
PluginType: "GEMINI",
|
|
},
|
|
}
|
|
}
|