Relay done but not working

This commit is contained in:
JustSong
2023-04-23 18:24:11 +08:00
parent 9fc375c604
commit 852af57c03
12 changed files with 225 additions and 70 deletions

View File

@@ -56,7 +56,7 @@ func GetChannel(c *gin.Context) {
})
return
}
channel, err := model.GetChannelById(id)
channel, err := model.GetChannelById(id, false)
if err != nil {
c.JSON(http.StatusOK, gin.H{
"success": false,

52
controller/relay.go Normal file
View File

@@ -0,0 +1,52 @@
package controller
import (
"fmt"
"github.com/gin-gonic/gin"
"io"
"net/http"
"one-api/common"
)
func Relay(c *gin.Context) {
channelType := c.GetInt("channel")
host := common.ChannelHosts[channelType]
req, err := http.NewRequest(c.Request.Method, fmt.Sprintf("%s/%s", host, c.Request.URL.String()), c.Request.Body)
if err != nil {
c.JSON(http.StatusOK, gin.H{
"error": gin.H{
"message": err.Error(),
"type": "one_api_error",
},
})
return
}
req.Header = c.Request.Header.Clone()
client := &http.Client{}
resp, err := client.Do(req)
if err != nil {
c.JSON(http.StatusOK, gin.H{
"error": gin.H{
"message": err.Error(),
"type": "one_api_error",
},
})
return
}
for k, v := range resp.Header {
c.Writer.Header().Set(k, v[0])
}
_, err = io.Copy(c.Writer, resp.Body)
//body, err := io.ReadAll(resp.Body)
//_, err = c.Writer.Write(body)
if err != nil {
c.JSON(http.StatusOK, gin.H{
"error": gin.H{
"message": err.Error(),
"type": "one_api_error",
},
})
return
}
}