|
|
@ -1,6 +1,7 @@ |
|
|
|
package weed_server |
|
|
|
|
|
|
|
import ( |
|
|
|
"fmt" |
|
|
|
"net/http" |
|
|
|
"strconv" |
|
|
|
"strings" |
|
|
@ -51,12 +52,12 @@ func (vs *VolumeServer) privateStoreHandler(w http.ResponseWriter, r *http.Reque |
|
|
|
|
|
|
|
// wait until in flight data is less than the limit
|
|
|
|
contentLength := getContentLength(r) |
|
|
|
vs.inFlightUploadDataLimitCond.L.Lock() |
|
|
|
for vs.concurrentUploadLimit != 0 && atomic.LoadInt64(&vs.inFlightUploadDataSize) > vs.concurrentUploadLimit { |
|
|
|
glog.V(4).Infof("wait because inflight upload data %d > %d", vs.inFlightUploadDataSize, vs.concurrentUploadLimit) |
|
|
|
vs.inFlightUploadDataLimitCond.Wait() |
|
|
|
err := fmt.Errorf("reject because inflight upload data %d > %d", vs.inFlightUploadDataSize, vs.concurrentUploadLimit) |
|
|
|
glog.V(1).Infof("too many requests: %v", err) |
|
|
|
writeJsonError(w, r, http.StatusTooManyRequests, err) |
|
|
|
return |
|
|
|
} |
|
|
|
vs.inFlightUploadDataLimitCond.L.Unlock() |
|
|
|
atomic.AddInt64(&vs.inFlightUploadDataSize, contentLength) |
|
|
|
defer func() { |
|
|
|
atomic.AddInt64(&vs.inFlightUploadDataSize, -contentLength) |
|
|
|