feat: 优化代理服务
This commit is contained in:
@@ -10,17 +10,14 @@ import (
|
||||
)
|
||||
|
||||
// ChannelProxy defines the interface for different API channel proxies.
|
||||
// It's responsible for channel-specific logic like URL building and request modification.
|
||||
type ChannelProxy interface {
|
||||
// BuildUpstreamURL constructs the target URL for the upstream service.
|
||||
BuildUpstreamURL(originalURL *url.URL, group *models.Group) (string, error)
|
||||
|
||||
// ModifyRequest allows the channel to add specific headers or modify the request
|
||||
// before it's sent to the upstream service.
|
||||
ModifyRequest(req *http.Request, apiKey *models.APIKey, group *models.Group)
|
||||
|
||||
// IsStreamRequest checks if the request is for a streaming response,
|
||||
// now using the cached request body to avoid re-reading the stream.
|
||||
IsStreamRequest(c *gin.Context, bodyBytes []byte) bool
|
||||
|
||||
// ValidateKey checks if the given API key is valid.
|
||||
|
Reference in New Issue
Block a user