mirror of
https://github.com/seaweedfs/seaweedfs.git
synced 2024-01-19 02:48:24 +00:00
filer: add "proxyToFileId" to reverse proxy to a volume server
This commit is contained in:
parent
3ecd989500
commit
937cfacc01
|
@ -17,9 +17,16 @@ func (fs *FilerServer) filerHandler(w http.ResponseWriter, r *http.Request) {
|
||||||
start := time.Now()
|
start := time.Now()
|
||||||
switch r.Method {
|
switch r.Method {
|
||||||
case "GET":
|
case "GET":
|
||||||
stats.FilerRequestCounter.WithLabelValues("get").Inc()
|
fileId := r.FormValue("proxyToFileId")
|
||||||
fs.GetOrHeadHandler(w, r, true)
|
if fileId != "" {
|
||||||
stats.FilerRequestHistogram.WithLabelValues("get").Observe(time.Since(start).Seconds())
|
stats.FilerRequestCounter.WithLabelValues("proxy").Inc()
|
||||||
|
fs.proxyToVolumeServer(w,r,fileId)
|
||||||
|
stats.FilerRequestHistogram.WithLabelValues("proxy").Observe(time.Since(start).Seconds())
|
||||||
|
} else {
|
||||||
|
stats.FilerRequestCounter.WithLabelValues("get").Inc()
|
||||||
|
fs.GetOrHeadHandler(w, r, true)
|
||||||
|
stats.FilerRequestHistogram.WithLabelValues("get").Observe(time.Since(start).Seconds())
|
||||||
|
}
|
||||||
case "HEAD":
|
case "HEAD":
|
||||||
stats.FilerRequestCounter.WithLabelValues("head").Inc()
|
stats.FilerRequestCounter.WithLabelValues("head").Inc()
|
||||||
fs.GetOrHeadHandler(w, r, false)
|
fs.GetOrHeadHandler(w, r, false)
|
||||||
|
|
66
weed/server/filer_server_handlers_proxy.go
Normal file
66
weed/server/filer_server_handlers_proxy.go
Normal file
|
@ -0,0 +1,66 @@
|
||||||
|
package weed_server
|
||||||
|
|
||||||
|
import (
|
||||||
|
"github.com/chrislusf/seaweedfs/weed/glog"
|
||||||
|
"github.com/chrislusf/seaweedfs/weed/util"
|
||||||
|
"io"
|
||||||
|
"math/rand"
|
||||||
|
"net/http"
|
||||||
|
)
|
||||||
|
|
||||||
|
var (
|
||||||
|
client *http.Client
|
||||||
|
)
|
||||||
|
|
||||||
|
func init() {
|
||||||
|
client = &http.Client{Transport: &http.Transport{
|
||||||
|
MaxIdleConnsPerHost: 1024,
|
||||||
|
}}
|
||||||
|
}
|
||||||
|
|
||||||
|
func (fs *FilerServer) proxyToVolumeServer(w http.ResponseWriter, r *http.Request, fileId string) {
|
||||||
|
|
||||||
|
urlStrings, err := fs.filer.MasterClient.GetLookupFileIdFunction()(fileId)
|
||||||
|
if err != nil {
|
||||||
|
glog.Errorf("locate %s: %v", fileId, err)
|
||||||
|
w.WriteHeader(http.StatusInternalServerError)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
if len(urlStrings) == 0 {
|
||||||
|
w.WriteHeader(http.StatusNotFound)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
proxyReq, err := http.NewRequest("GET", urlStrings[rand.Intn(len(urlStrings))], r.Body)
|
||||||
|
if err != nil {
|
||||||
|
glog.Errorf("NewRequest %s: %v", urlStrings[0], err)
|
||||||
|
w.WriteHeader(http.StatusInternalServerError)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
proxyReq.Header.Set("Host", r.Host)
|
||||||
|
proxyReq.Header.Set("X-Forwarded-For", r.RemoteAddr)
|
||||||
|
|
||||||
|
for header, values := range r.Header {
|
||||||
|
for _, value := range values {
|
||||||
|
proxyReq.Header.Add(header, value)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
proxyResponse, postErr := client.Do(proxyReq)
|
||||||
|
|
||||||
|
if postErr != nil {
|
||||||
|
glog.Errorf("post to filer: %v", postErr)
|
||||||
|
w.WriteHeader(http.StatusInternalServerError)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
defer util.CloseResponse(proxyResponse)
|
||||||
|
|
||||||
|
for k, v := range proxyResponse.Header {
|
||||||
|
w.Header()[k] = v
|
||||||
|
}
|
||||||
|
w.WriteHeader(proxyResponse.StatusCode)
|
||||||
|
io.Copy(w, proxyResponse.Body)
|
||||||
|
|
||||||
|
}
|
Loading…
Reference in a new issue