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 committed by Niko Storni
parent 848fce5afa
commit ed3622d0a6
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.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) {
vars := mux.Vars(r)

6
peer/http3/worker.go
View file

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