LBRYCommunity/reflector.go
1
0
Fork 0
Code Issues 7 Pull requests Projects Releases 10 Packages Wiki Activity

Wait for request to be handled before returning

Browse source
This commit is contained in:
Mark Beamer Jr 2021-04-06 14:21:05 -04:00
parent bd13836897
commit b97595311f
No known key found for this signature in database
GPG key ID: 1C314FB89AD76973
2 changed files with 9 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

6
peer/http3/server.go
View file

@ -70,7 +70,11 @@ func (s *Server) Start(address string) error {
} }
r := mux.NewRouter() r := mux.NewRouter()
r.HandleFunc("/get/{hash}", func(w http.ResponseWriter, r *http.Request) { r.HandleFunc("/get/{hash}", func(w http.ResponseWriter, r *http.Request) {
enqueue(&blobRequest{request: r, reply: w}) waiter := stop.New()
enqueue(&blobRequest{request: r, reply: w, finished: waiter})
select {
case <-waiter.Ch():
}
}) })
r.HandleFunc("/has/{hash}", func(w http.ResponseWriter, r *http.Request) { r.HandleFunc("/has/{hash}", func(w http.ResponseWriter, r *http.Request) {
vars := mux.Vars(r) vars := mux.Vars(r)

6
peer/http3/worker.go
View file

@ -9,8 +9,9 @@ import (
) )
type blobRequest struct { type blobRequest struct {
request *http.Request request *http.Request
reply http.ResponseWriter reply http.ResponseWriter
finished *stop.Group
} }
var getReqCh = make(chan *blobRequest) var getReqCh = make(chan *blobRequest)
@ -37,4 +38,5 @@ func enqueue(b *blobRequest) {
func process(server *Server, r *blobRequest) { func process(server *Server, r *blobRequest) {
server.HandleGetBlob(r.reply, r.request) server.HandleGetBlob(r.reply, r.request)
r.finished.Done()
} }