zishuo/apinto
1
0
Fork 0
mirror of https://github.com/eolinker/apinto synced 2025-10-05 08:47:04 +08:00
Code Issues Packages Projects Releases Wiki Activity

新增AI provider和ai prompt

Browse Source
This commit is contained in:
Liujian
2024-09-30 10:03:01 +08:00
parent 5fded26881
commit 0af5bcc5e6
62 changed files with 2314 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
node/http-context/context.go
View File

@@ -143,9 +143,11 @@ func (ctx *HttpContext) SendTo(scheme string, node eoscContext.INode, timeout ti
case eoscContext.NodeHost:
rewriteHost = host
request.URI().SetHost(host)
//ctx.proxyRequest.Header().SetHost(targetHost)
case eoscContext.ReWriteHost:
rewriteHost = targetHost
request.URI().SetHost(targetHost)
//ctx.proxyRequest.Header().SetHost(targetHost)
}
beginTime := time.Now()
ctx.response.responseError = fasthttp_client.ProxyTimeout(scheme, rewriteHost, node, request, &ctx.fastHttpRequestCtx.Response, timeout)
@@ -153,17 +155,23 @@ func (ctx *HttpContext) SendTo(scheme string, node eoscContext.INode, timeout ti
if ctx.response.Response != nil {
responseHeader = ctx.response.Response.Header
}
agent := newRequestAgent(&ctx.proxyRequest, host, scheme, responseHeader, beginTime, time.Now())
if ctx.response.responseError != nil {
agent.setStatusCode(504)
} else {
ctx.response.ResponseHeader.refresh()
agent.setStatusCode(ctx.fastHttpRequestCtx.Response.StatusCode())
}
if ctx.fastHttpRequestCtx.Response.RemoteAddr() != nil {
ip, port := parseAddr(ctx.fastHttpRequestCtx.Response.RemoteAddr().String())
agent.setRemoteIP(ip)
agent.setRemotePort(port)
ctx.response.remoteIP = ip
ctx.response.remotePort = port
agent.setStatusCode(ctx.fastHttpRequestCtx.Response.StatusCode())
}
agent.responseBody = string(ctx.response.Response.Body())